Combined device serial id

Author: YuvrajRaghuvanshiS

helpers/adb_device_serial_id.py

@@ -0,0 +1,128 @@
+import os
+import platform
+import subprocess as sp
+
+from custom_ci import custom_input, custom_print
+
+
+def init(mode, tcp_ip='', tcp_port=''):
+    custom_print(
+        f'>>> I am in device_serial_id.init({mode=!s}, {tcp_ip=!s}, {tcp_port=!s})', is_print=False)
+    # Detect OS
+    is_windows = False
+    is_linux = False
+    if platform.system() == 'Windows':
+        is_windows = True
+    if platform.system() == 'Linux':
+        is_linux = True
+
+    # Global command line helpers
+    curr_dir = os.path.dirname(os.path.realpath(__file__))
+    root_dir = os.path.abspath(os.path.join(curr_dir, '..'))
+
+    if(is_windows):
+        adb = f'{root_dir}\\bin\\adb.exe'
+    else:
+        adb = 'adb'
+
+    # Kill server before getting list to avoid daemon texts.
+    os.system(f'{adb} kill-server')
+    os.system(f'{adb} start-server')
+
+    combo = f'{tcp_ip}:{tcp_port}'
+    cmd = ''
+    if mode == 'USB':
+        cmd = f'{adb} devices'
+    elif mode == 'TCP':
+        cmd = f'{adb} connect {combo}'
+    else:
+        pass
+        # FIXME: Wrong choice.
+    proc = sp.Popen(cmd.split(), stdin=sp.PIPE, stdout=sp.PIPE,
+                    stderr=sp.PIPE, shell=False)
+    output, error = proc.communicate()
+    output = output.decode('utf-8')
+    error = error.decode('utf-8')
+
+    if len(output) == 0 or error:
+        output = None
+        custom_print(error, 'red')
+        kill_me()
+
+    if mode == 'USB':
+        output = [x.strip() for x in output.split('\n') if len(x.strip()) > 0]
+
+        if(len(output) == 1):
+            custom_print(
+                'Could not find any connected device. Is USB Debugging on?', 'red')
+            return ''
+
+        device_to_connect = None
+        i = 1
+        if(len(output) == 2):
+            if(output[1].split()[1] == 'offline'):
+                custom_print(
+                    'Device is offline, try turning off USB debugging and turn on again.', 'yellow')
+                kill_me()
+            if(output[1].split()[1] == 'unauthorized'):
+                custom_print(
+                    'Device unauthorized. Please check the confirmation dialog on your device.', 'red')
+                kill_me()
+            return output[1].split()[0]
+
+        custom_print(output[0])
+        custom_print('\n', is_get_time=False)
+        if device_to_connect is None:
+            for index, device in enumerate(output[1:]):
+                name = f'{adb} -s {device.split()[0]} shell getprop ro.product.model'
+                custom_print(
+                    f'{index}. {device.split()[0]} {device.split()[1]} {sp.getoutput(name).strip()}')
+
+        while device_to_connect is None:
+            device_index = int(custom_input(
+                'Enter device number (for ex: 2): '))
+            if device_index <= 0 or device_index + 1 > len(output):
+                continue
+            device_to_connect = output[device_index]
+
+        if(device_to_connect.split()[1] == 'offline'):
+            custom_print(
+                'Device is offline, try turning off USB debugging and turn on again.', 'yellow')
+            kill_me()
+        if(device_to_connect.split()[1] == 'unauthorized'):
+            custom_print(
+                'Device unauthorized. Please check the confirmation dialog on your device.', 'red')
+            kill_me()
+        return device_to_connect.split()[0]
+
+    elif mode == 'TCP':
+        output = [x.strip() for x in output.split() if len(x.strip()) > 0]
+        if('connected' in (x.lower() for x in output)):
+            return combo
+        if('authenticate' in (x.lower() for x in output)):
+            custom_print(
+                'Device unauthorized. Please check the confirmation dialog on your device.', 'red')
+            kill_me()
+        if('refused' in (x.lower() for x in output)):
+            custom_print(
+                'Could not find any connected device. Either USB Debugging is off or device is not running ADB over TCP', 'red')
+            return ''
+        ''' Possible outputs
+        ['connected', 'to', '192.168.43.130:5555']
+        ['failed', 'to', 'authenticate', 'to', '192.168.43.130:5555'] 
+        ['cannot', 'connect', 'to', '192.168.43.130:5555:', 'No', 'connection', 'could', 'be', 'made', 'because', 'the', 'target', 'machine', 'actively', 'refused', 'it.', '(10061)']
+        '''
+    else:
+        pass
+        # FIXME: Wrong choice.
+
+
+def kill_me():
+    custom_print(
+        '>>> I am in device_serial_id.kill_me()', is_print=False)
+    custom_print('\n', is_get_time=False)
+    custom_print('Exiting...')
+    custom_print(
+        'Turn off USB debugging [and USB debugging (Security Settings)] if you\'re done.', 'cyan')
+    custom_input('Hit \"Enter\" key to continue....', 'cyan')
+    quit()

helpers/device_serial_id.py

@@ -1,4 +1,3 @@
-from curses.ascii import isprint
 import os
 import re
 import shutil

helpers/handler.py

@@ -4,8 +4,7 @@
 import platform
 import subprocess
 
-import helpers.adb_device_serial_id as adb_device_id
-import helpers.tcp_device_serial_id as tcp_device_id
+import helpers.device_serial_id as device_id
 from helpers.custom_ci import custom_input, custom_print
 
 # Detect OS
@@ -83,19 +82,17 @@ def show_banner():
     parser = argparse.ArgumentParser()
     parser.add_argument('-tip',
                         '--tcp-ip', help='Connects to a remote device via TCP mode.')
-    parser.add_argument('-tp',
-                        '--tcp-port', help='Port number to connect to. Default: 5555')
+    parser.add_argument('-tp', '--tcp-port', default='5555',
+                        help='Port number to connect to. Default: 5555')
     args = parser.parse_args()
-    # args = parser.parse_args('--tcp-ip 192.168.43.130'.split())
+    # args = parser.parse_args('--tcp-ip 192.168.168.117'.split())
 
     tcp_ip = args.tcp_ip
     tcp_port = args.tcp_port
     if(tcp_ip):
-        if(not tcp_port):
-            tcp_port = '5555'
-        adb_device_serial_id = tcp_device_id.init(tcp_ip, tcp_port)
+        adb_device_serial_id = device_id.init('TCP', tcp_ip, tcp_port)
     else:
-        adb_device_serial_id = adb_device_id.init()
+        adb_device_serial_id = device_id.init('USB')
     if(not adb_device_serial_id):
         kill_me()
 

helpers/tcp_device_serial_id.py

@@ -27,8 +27,7 @@
 import subprocess
 import time
 
-import helpers.adb_device_serial_id as adb_device_id
-import helpers.tcp_device_serial_id as tcp_device_id
+import helpers.device_serial_id as device_id
 from helpers.custom_ci import custom_input, custom_print
 from helpers.handler import handler
 from view_extract import extract_ab
@@ -411,7 +410,7 @@ def usb_mode():
                         help='Allow reboot of device before installation of LegacyWhatsApp.apk to prevent some issues like [INSTALL_FAILED_VERSION_DOWNGRADE]')
     parser.add_argument('-tip', '--tcp-ip',
                         help='Connects to a remote device via TCP mode.')
-    parser.add_argument('-tp', '--tcp-port',
+    parser.add_argument('-tp', '--tcp-port', default='5555',
                         help='Port number to connect to. Default: 5555')
     parser.add_argument('-s', '--scrcpy', action='store_true',
                         help='Run ScrCpy to see and control Android device.')
@@ -426,11 +425,9 @@ def usb_mode():
     is_scrcpy = args.scrcpy
     is_tar_only = args.tar_only
     if(tcp_ip):
-        if(not tcp_port):
-            tcp_port = '5555'
-        adb_device_serial_id = tcp_device_id.init(tcp_ip, tcp_port)
+        adb_device_serial_id = device_id.init('TCP', tcp_ip, tcp_port)
     else:
-        adb_device_serial_id = adb_device_id.init()
+        adb_device_serial_id = device_id.init('USB')
     if(not adb_device_serial_id):
         kill_me()