show_device_options.py

import subprocess
import os
import sys
import re
from workflow import Workflow
import commands
import time

adb_path = os.getenv('adb_path')
serial = os.getenv('serial')
api = os.getenv('device_api')
ip = os.getenv("ip")


def wordMatch(arg, sentence):
    words = arg.lower().split(" ")
    sentenceComponents = sentence.lower().split(" ")
    for word in words:
        included = False
        for sentenceComponent in sentenceComponents:
            if word in sentenceComponent:
                included = True
                break
        if not included:
            return False
    return True


def main(wf):

    isWifiDevice = re.match(".*:5555$", serial)
    isEmulator = re.match("emulator-55[5-8][02468]", serial)
    arg = wf.args[0].strip()
    log.debug(arg)
    addAll = False
    if arg == '':
        addAll = True

    itemCount = 0

    # SHOW INSTALLED PACKAGES
    title = "Show apps list"

    if addAll or (wordMatch(arg, title + " start launch uninstall force stop clear info") and not arg.startswith("ins")):
        func = ""
        if "start" in arg.lower() or "launch" in arg.lower():
            title = "Select app to launch"
            func = "start_app"
        elif "uninstall" in arg.lower():
            title = "Select app to uninstall"
            func = "uninstall_app"
        elif "force" in arg.lower() or "stop" in arg.lower():
            title = "Select app to force stop"
            func = "force_stop"
        elif "clear" in arg.lower():
            title = "Select app to clear data"
            func = "clear_app_data"
        elif "info" in arg.lower():
            title = "Select app to show app info"
            func = "app_info"
        it = wf.add_item(title=title,
                         uid="list_app",
                         arg="list_app",
                         copytext=commands.CMD_LIST_APPS,
                         valid=True)
        it.setvar("func", func)
        if func == "":
            m = it.add_modifier('cmd', "Select app to launch")
            m.setvar("func", "start_app")

            m = it.add_modifier('alt', "Select app to uninstall")
            m.setvar("func", "uninstall_app")

            m = it.add_modifier('ctrl', "Select app to force stop")
            m.setvar("func", "force_stop")

            m = it.add_modifier('fn', "Select app to clear data")
            m.setvar("func", "clear_app_data")

            m = it.add_modifier('shift', "Select app to show app info")
            m.setvar("func", "app_info")
        itemCount += 1

    # INSTALL APK
    title = "Install apk"

    if addAll or wordMatch(arg, title):
        wf.add_item(title=title,
                    uid="install_apk",
                    arg="install_apk",
                    subtitle="install apk or apks",
                    copytext=commands.CMD_INSTALL_APP.format("", ""),
                    valid=True)
        itemCount += 1

    # SCREENSHOT
    title = "Take screenshot"

    if addAll or wordMatch(arg, title):
        timestamp = int(time.time())
        it = wf.add_item(title=title,
                         uid="screenshot",
                         arg="screenshot:to_clipboard",
                         subtitle="take a screenshot and copy to clipboard, `cmd` to copy to desktop",
                         copytext=commands.CMD_SCREENCAP.format(timestamp),
                         valid=True)
        it.add_modifier(
            'cmd', 'take a screenshot and copy to desktop', arg="screenshot:to_desktop")
        itemCount += 1

    # OPEN SETTINGS
    title = "Open settings"

    if addAll or wordMatch(arg, title):
        it = wf.add_item(title=title,
                         uid="open_settings",
                         arg="open_settings",
                         subtitle="'cmd' - Dev Tool, 'alt' - WiFi, 'ctrl' - App, 'fn' - Date, 'shift' - Accessibility",
                         copytext=commands.CMD_OPEN_SETTINGS.format("android.settings.SETTINGS"),
                         valid=True)
        it.add_modifier('cmd', 'Open Developer Settings',
                        arg="open_settings:developer_options")
        it.add_modifier('alt', 'Open WiFi Settings', arg="open_settings:wifi")
        it.add_modifier('fn', 'Open Date Settings', arg="open_settings:date")
        it.add_modifier('shift', 'Open Accessibility Settings',
                        arg="open_settings:accessibility")
        it.add_modifier('ctrl', 'Open Application Settings',
                        arg="open_settings:application")
        itemCount += 1

    # TOGGLE DEBUG LAYOUT
    title = "Toggle debug layout"

    if addAll or wordMatch(arg, title):
        it = wf.add_item(title=title,
                         uid="debug_layout",
                         arg="debug_layout",
                         copytext=commands.CMD_SET_DEBUG_LAYOUT.format('on'),
                         valid=True)
        itemCount += 1
        it.add_modifier('cmd', 'Toggle pointer location',
                        arg="pointer_location")
        it.add_modifier('alt', 'Toggle show taps', arg="show_taps")
        it.add_modifier('ctrl', 'Toggle GPU profile', arg="gpu_profile")
        it.add_modifier('fn', 'Toggle GPU overdraw', arg="gpu_overdraw")
        it.add_modifier('shift', 'Turn off everything', arg="debug_off")

    # DEMO MODE
    if api and int(api) >= 23:
        title = "Toggle demo mode"

        if addAll or wordMatch(arg, title):
            wf.add_item(title=title,
                        uid="demo_mode",
                        arg="demo_mode",
                        copytext=commands.CMD_DEMO_MODE_STATUS,
                        valid=True)
            itemCount += 1

    # REBOOT SYSTEM
    title = "Reboot system"

    if addAll or wordMatch(arg, title) or wordMatch(arg, "restart"):
        it = wf.add_item(title=title,
                         uid="adb_reboot",
                         arg="adb_reboot",
                         subtitle="'cmd' - Bootloader, 'alt' - Recovery, 'ctrl' - Sideload",
                         copytext=commands.CMD_REBOOT.format(''),
                         valid=True)
        itemCount += 1
        it.add_modifier('cmd', 'Bootloader', arg="adb_reboot:bootloader")
        it.add_modifier('alt', 'Recovery', arg="adb_reboot:recovery")
        it.add_modifier('ctrl', 'Sideload', arg="adb_reboot:sideload")

    # CONNECT OVER WIFI
    if not isWifiDevice and not isEmulator and ip:
        title = "Connect over Wi-Fi"

        if addAll or wordMatch(arg, title) or wordMatch(arg, "Connect over WiFi"):
            wf.add_item(title=title,
                        subtitle=ip,
                        uid="debug_wifi",
                        arg="debug_wifi",
                        copytext=commands.CMD_TCPIP + ";" + commands.CMD_ADB_CONNECT.format(ip.split("/")[0]),
                        valid=True)
            itemCount += 1

    # KEY INPUT
    title = "Keyevent input"

    if addAll or wordMatch(arg, title):
        it = wf.add_item(title=title,
                         uid="keyevent_input",
                         arg="keyevent_input",
                         copytext=commands.CMD_INPUT.format("keyevent", "", "KEYCODE_HOME"),
                         valid=True)
        it.add_modifier("cmd", "Back", arg='keyevent_input_KEYCODE_BACK')
        it.add_modifier("alt", "Home", arg='keyevent_input_KEYCODE_HOME')
        it.add_modifier("ctrl", "App Switch",
                        arg='keyevent_input_KEYCODE_APP_SWITCH')
        it.add_modifier("fn", "Escape", arg='keyevent_input_KEYCODE_ESCAPE')
        it.add_modifier("shift", "Power", arg='keyevent_input_KEYCODE_POWER')
        itemCount += 1

    # DUMP STACK
    title = "Dump task stacks"

    if addAll or wordMatch(arg, title):
        it = wf.add_item(title=title,
                         uid="dump_stack",
                         arg="dump_stack",
                         copytext=commands.CMD_DUMP_STACK,
                         valid=True)
        itemCount += 1
        it.add_modifier('cmd', 'Dump the first application',
                        arg="dump_stack:first_app")
        it.add_modifier('alt', 'Dump the first stack',
                        arg="dump_stack:first_stack")

    # DUMP NOTIFICATIONS
    if api and int(api) >= 23:
        title = "Dump notifications"

        if addAll or wordMatch(arg, title):
            it = wf.add_item(title=title,
                        uid="dump_notification",
                        arg="dump_notification",
                        copytext=commands.CMD_DUMP_NOTIFICATION,
                        valid=True)
            itemCount += 1
            it.add_modifier('alt', 'Exclude empty notification', arg="dump_notification:exclude_empty")


    # Screen Copy with scrcpy
    title = "Screen Copy with scrcpy"
    log.debug("Screen Copy with scrcpy")
    try:
        rc = subprocess.check_output(['/usr/local/bin/scrcpy', '-v'])
    except OSError:
        log.error("OS error 1")
        try:
            rc = subprocess.check_output(['/opt/homebrew/bin/scrcpy', '-v'])
        except OSError:
            log.error("OS error 2")
            rc = -1
    if rc != -1 and (addAll or wordMatch(arg, title)) and not isEmulator:
        it = wf.add_item(title=title,
                         uid="scr_cpy",
                         arg="scr_cpy",
                         copytext="scrcpy -s {}".format(serial),
                         valid=True)

        it.setvar("dimension", 0)
        mod = it.add_modifier(
            "cmd", subtitle="Run with maximum resolution restriction 1024")
        mod.setvar("dimension", 1024)
        mod = it.add_modifier(
            "alt", subtitle="Record screen and save to User folder")
        mod.setvar("record", 1)
        log.debug("Manufacturer {0}".format(
            os.getenv("ro.product.manufacturer")))
        if os.getenv("ro.product.manufacturer") == "Oculus" and os.getenv("name") == "Quest 2":
            mod = it.add_modifier("ctrl", subtitle="Quest 2 - Mirror one eye")
            mod.setvar("crop", "1832:1920:0:0")
        itemCount += 1
    log.debug(rc)

    # COMMAND HISTORY
    title = "Command history"
    lastFuncs = wf.cached_data('last_func:' + os.getenv("his_tag"), max_age=0)

    if (addAll or wordMatch(arg, title)) and lastFuncs and len(lastFuncs) > 0:
        it = wf.add_item(title=title,
                         subtitle="show command history",
                         arg="cmd_history",
                         valid=True)
        itemCount += 1
        it.add_modifier('cmd', 'clear command history',
                        arg="cmd_history:clear")

    idx = 1
    while idx > 0:
        config = os.getenv('self_script_device_%d' % idx)
        if config:
            title = config.split("|")[0]
            if addAll or wordMatch(arg, title) or wordMatch(arg, "Self script"):
                path = config.split("|")[1]
                it = wf.add_item(title=title,
                                 subtitle="with script: %s" % path,
                                 arg="self_script_device_%d" % idx,
                                 valid=True)
                it.setvar("self_script_device", config)
                mod = it.add_modifier("cmd", subtitle="apply cmd modifier")
                mod.setvar("mod", "cmd")
                mod = it.add_modifier("alt", subtitle="apply alt modifier")
                mod.setvar("mod", "alt")
                mod = it.add_modifier("ctrl", subtitle="apply ctrl modifier")
                mod.setvar("mod", "ctrl")
                mod = it.add_modifier("fn", subtitle="apply fn modifier")
                mod.setvar("mod", "fn")
                mod = it.add_modifier("shift", subtitle="apply shift modifier")
                mod.setvar("mod", "shift")
            idx = idx + 1
        else:
            idx = -1

    # CUSTOM ACTION
    if itemCount == 0:
        it = wf.add_item(title="Execute custom command for " + serial,
                         subtitle="adb " + arg,
                         arg="adb_cmd:in_terminal:" + arg,
                         valid=True)
        m = it.add_modifier('cmd', 'Run without opening terminal',
                            arg="adb_cmd:in_background:" + arg)

    wf.send_feedback()


if __name__ == '__main__':
    wf = Workflow()
    log = wf.logger
    sys.exit(wf.run(main))