kotonebot 0.5.0__py3-none-any.whl → 0.6.0__py3-none-any.whl

kotonebot/client/host/leidian_host.py

@@ -1,197 +1,197 @@
-import os
-import subprocess
-from typing import Literal
-from functools import lru_cache
-from typing_extensions import override
-
-from kotonebot import logging
-from kotonebot.client import Device
-from kotonebot.util import Countdown, Interval
-from .protocol import HostProtocol, Instance, copy_type, AdbHostConfig
-from .adb_common import AdbRecipes, CommonAdbCreateDeviceMixin
-from ...interop.win.reg import read_reg
-
-logger = logging.getLogger(__name__)
-LeidianRecipes = AdbRecipes
-
-
-class LeidianInstance(CommonAdbCreateDeviceMixin, Instance[AdbHostConfig]):
-    @copy_type(Instance.__init__)
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-        self._args = args
-        self.index: int | None = None
-        self.is_running: bool = False
-
-    @override
-    def refresh(self):
-        ins = LeidianHost.query(id=self.id)
-        assert isinstance(ins, LeidianInstance), f'Expected LeidianInstance, got {type(ins)}'
-        if ins is not None:
-            self.adb_port = ins.adb_port
-            self.adb_ip = ins.adb_ip
-            self.adb_name = ins.adb_name
-            self.is_running = ins.is_running
-            logger.debug('Refreshed Leidian instance: %s', repr(ins))
-
-    @override
-    def start(self):
-        if self.running():
-            logger.warning('Instance is already running.')
-            return
-        logger.info('Starting Leidian instance %s', self)
-        LeidianHost._invoke_manager(['launch', '--index', str(self.index)])
-        self.refresh()
-
-    @override
-    def stop(self):
-        if not self.running():
-            logger.warning('Instance is not running.')
-            return
-        logger.info('Stopping Leidian instance id=%s name=%s...', self.id, self.name)
-        LeidianHost._invoke_manager(['quit', '--index', str(self.index)])
-        self.refresh()
-
-    @override
-    def wait_available(self, timeout: float = 180):
-        cd = Countdown(timeout)
-        it = Interval(5)
-        while not cd.expired() and not self.running():
-            it.wait()
-            self.refresh()
-        if not self.running():
-            raise TimeoutError(f'Leidian instance "{self.name}" is not available.')
-
-    @override
-    def running(self) -> bool:
-        return self.is_running
-
-    @override
-    def create_device(self, impl: LeidianRecipes, host_config: AdbHostConfig) -> Device:
-        """为雷电模拟器实例创建 Device。"""
-        if self.adb_port is None:
-            raise ValueError("ADB port is not set and is required.")
-
-        return super().create_device(impl, host_config, connect=False, disconnect=False)
-
-class LeidianHost(HostProtocol[LeidianRecipes]):
-    @staticmethod
-    @lru_cache(maxsize=1)
-    def _read_install_path() -> str | None:
-        """
-        从注册表中读取雷电模拟器的安装路径。
-
-        :return: 安装路径，如果未找到则返回 None。
-        """
-        if os.name != 'nt':
-            return None
-
-        try:
-            icon_path = read_reg('HKCU', r'Software\leidian\LDPlayer9', 'DisplayIcon', default=None)
-            if icon_path and isinstance(icon_path, str):
-                icon_path = icon_path.replace('"', '')
-                path = os.path.dirname(icon_path)
-                logger.debug('Leidian installation path (from DisplayIcon): %s', path)
-                return path
-            install_dir = read_reg('HKCU', r'Software\leidian\LDPlayer9', 'InstallDir', default=None)
-            if install_dir and isinstance(install_dir, str):
-                install_dir = install_dir.replace('"', '')
-                logger.debug('Leidian installation path (from InstallDir): %s', install_dir)
-                return install_dir
-        except Exception as e:
-            logger.error(f'Failed to read Leidian installation path from registry: {e}')
-
-        return None
-
-    @staticmethod
-    def _invoke_manager(args: list[str]) -> str:
-        """
-        调用 ldconsole.exe。
-        
-        参考文档：https://www.ldmnq.com/forum/30.html，以及命令行帮助。
-        另外还有个 ld.exe，封装了 adb.exe，可以直接执行 adb 命令。（https://www.ldmnq.com/forum/9178.html）
-
-        :param args: 命令行参数列表。
-        :return: 命令执行的输出。
-        """
-        install_path = LeidianHost._read_install_path()
-        if install_path is None:
-            raise RuntimeError('Leidian is not installed.')
-        manager_path = os.path.join(install_path, 'ldconsole.exe')
-        logger.debug('ldconsole execute: %s', repr(args))
-        output = subprocess.run(
-            [manager_path] + args,
-            capture_output=True,
-            text=True,
-            # encoding='utf-8', # 居然不是 utf-8 编码
-            # https://stackoverflow.com/questions/6011235/run-a-program-from-python-and-have-it-continue-to-run-after-the-script-is-kille
-            creationflags=subprocess.DETACHED_PROCESS | subprocess.CREATE_NEW_PROCESS_GROUP
-        )
-        if output.returncode != 0:
-            raise RuntimeError(f'Failed to invoke ldconsole: {output.stderr}')
-        return output.stdout
-
-    @staticmethod
-    def installed() -> bool:
-        return LeidianHost._read_install_path() is not None
-
-    @staticmethod
-    def list() -> list[Instance]:
-        output = LeidianHost._invoke_manager(['list2'])
-        instances = []
-
-        # 解析 list2 命令的输出
-        # 格式: 索引,标题,顶层窗口句柄,绑定窗口句柄,是否进入android,进程PID,VBox进程PID
-        for line in output.strip().split('\n'):
-            if not line:
-                continue
-
-            parts = line.split(',')
-            if len(parts) < 5:
-                logger.warning(f'Invalid list2 output line: {line}')
-                continue
-
-            index = parts[0]
-            name = parts[1]
-            is_android_started = parts[4] == '1'
-            # 端口号规则 https://help.ldmnq.com/docs/LD9adbserver#a67730c2e7e2e0400d40bcab37d0e0cf
-            adb_port = 5554 + (int(index) * 2)
-
-            instance = LeidianInstance(
-                id=index,
-                name=name,
-                adb_port=adb_port,
-                adb_ip='127.0.0.1',
-                adb_name=f'emulator-{adb_port}'
-            )
-            instance.index = int(index)
-            instance.is_running = is_android_started
-            logger.debug('Leidian instance: %s', repr(instance))
-            instances.append(instance)
-
-        return instances
-
-    @staticmethod
-    def query(*, id: str) -> Instance | None:
-        instances = LeidianHost.list()
-        for instance in instances:
-            if instance.id == id:
-                return instance
-        return None
-
-    @staticmethod
-    def recipes() -> 'list[LeidianRecipes]':
-        return ['adb', 'adb_raw', 'uiautomator2']
-
-if __name__ == '__main__':
-    logging.basicConfig(level=logging.DEBUG, format='[%(asctime)s] [%(levelname)s] [%(name)s] [%(funcName)s] [%(lineno)d] %(message)s')
-    print(LeidianHost._read_install_path())
-    print(LeidianHost.installed())
-    print(LeidianHost.list())
-    print(ins:=LeidianHost.query(id='0'))
-    assert isinstance(ins, LeidianInstance)
-    ins.start()
-    ins.wait_available()
-    print('status', ins.running(), ins.adb_port, ins.adb_ip)
-    # ins.stop()
+import os
+import subprocess
+from typing import Literal
+from functools import lru_cache
+from typing_extensions import override
+
+from kotonebot import logging
+from kotonebot.client import Device
+from kotonebot.util import Countdown, Interval
+from .protocol import HostProtocol, Instance, copy_type, AdbHostConfig
+from .adb_common import AdbRecipes, CommonAdbCreateDeviceMixin
+from ...interop.win.reg import read_reg
+
+logger = logging.getLogger(__name__)
+LeidianRecipes = AdbRecipes
+
+
+class LeidianInstance(CommonAdbCreateDeviceMixin, Instance[AdbHostConfig]):
+    @copy_type(Instance.__init__)
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self._args = args
+        self.index: int | None = None
+        self.is_running: bool = False
+
+    @override
+    def refresh(self):
+        ins = LeidianHost.query(id=self.id)
+        assert isinstance(ins, LeidianInstance), f'Expected LeidianInstance, got {type(ins)}'
+        if ins is not None:
+            self.adb_port = ins.adb_port
+            self.adb_ip = ins.adb_ip
+            self.adb_name = ins.adb_name
+            self.is_running = ins.is_running
+            logger.debug('Refreshed Leidian instance: %s', repr(ins))
+
+    @override
+    def start(self):
+        if self.running():
+            logger.warning('Instance is already running.')
+            return
+        logger.info('Starting Leidian instance %s', self)
+        LeidianHost._invoke_manager(['launch', '--index', str(self.index)])
+        self.refresh()
+
+    @override
+    def stop(self):
+        if not self.running():
+            logger.warning('Instance is not running.')
+            return
+        logger.info('Stopping Leidian instance id=%s name=%s...', self.id, self.name)
+        LeidianHost._invoke_manager(['quit', '--index', str(self.index)])
+        self.refresh()
+
+    @override
+    def wait_available(self, timeout: float = 180):
+        cd = Countdown(timeout)
+        it = Interval(5)
+        while not cd.expired() and not self.running():
+            it.wait()
+            self.refresh()
+        if not self.running():
+            raise TimeoutError(f'Leidian instance "{self.name}" is not available.')
+
+    @override
+    def running(self) -> bool:
+        return self.is_running
+
+    @override
+    def create_device(self, impl: LeidianRecipes, host_config: AdbHostConfig) -> Device:
+        """为雷电模拟器实例创建 Device。"""
+        if self.adb_port is None:
+            raise ValueError("ADB port is not set and is required.")
+
+        return super().create_device(impl, host_config, connect=False, disconnect=False)
+
+class LeidianHost(HostProtocol[LeidianRecipes]):
+    @staticmethod
+    @lru_cache(maxsize=1)
+    def _read_install_path() -> str | None:
+        """
+        从注册表中读取雷电模拟器的安装路径。
+
+        :return: 安装路径，如果未找到则返回 None。
+        """
+        if os.name != 'nt':
+            return None
+
+        try:
+            icon_path = read_reg('HKCU', r'Software\leidian\LDPlayer9', 'DisplayIcon', default=None)
+            if icon_path and isinstance(icon_path, str):
+                icon_path = icon_path.replace('"', '')
+                path = os.path.dirname(icon_path)
+                logger.debug('Leidian installation path (from DisplayIcon): %s', path)
+                return path
+            install_dir = read_reg('HKCU', r'Software\leidian\LDPlayer9', 'InstallDir', default=None)
+            if install_dir and isinstance(install_dir, str):
+                install_dir = install_dir.replace('"', '')
+                logger.debug('Leidian installation path (from InstallDir): %s', install_dir)
+                return install_dir
+        except Exception as e:
+            logger.error(f'Failed to read Leidian installation path from registry: {e}')
+
+        return None
+
+    @staticmethod
+    def _invoke_manager(args: list[str]) -> str:
+        """
+        调用 ldconsole.exe。
+        
+        参考文档：https://www.ldmnq.com/forum/30.html，以及命令行帮助。
+        另外还有个 ld.exe，封装了 adb.exe，可以直接执行 adb 命令。（https://www.ldmnq.com/forum/9178.html）
+
+        :param args: 命令行参数列表。
+        :return: 命令执行的输出。
+        """
+        install_path = LeidianHost._read_install_path()
+        if install_path is None:
+            raise RuntimeError('Leidian is not installed.')
+        manager_path = os.path.join(install_path, 'ldconsole.exe')
+        logger.debug('ldconsole execute: %s', repr(args))
+        output = subprocess.run(
+            [manager_path] + args,
+            capture_output=True,
+            text=True,
+            # encoding='utf-8', # 居然不是 utf-8 编码
+            # https://stackoverflow.com/questions/6011235/run-a-program-from-python-and-have-it-continue-to-run-after-the-script-is-kille
+            creationflags=subprocess.DETACHED_PROCESS | subprocess.CREATE_NEW_PROCESS_GROUP
+        )
+        if output.returncode != 0:
+            raise RuntimeError(f'Failed to invoke ldconsole: {output.stderr}')
+        return output.stdout
+
+    @staticmethod
+    def installed() -> bool:
+        return LeidianHost._read_install_path() is not None
+
+    @staticmethod
+    def list() -> list[Instance]:
+        output = LeidianHost._invoke_manager(['list2'])
+        instances = []
+
+        # 解析 list2 命令的输出
+        # 格式: 索引,标题,顶层窗口句柄,绑定窗口句柄,是否进入android,进程PID,VBox进程PID
+        for line in output.strip().split('\n'):
+            if not line:
+                continue
+
+            parts = line.split(',')
+            if len(parts) < 5:
+                logger.warning(f'Invalid list2 output line: {line}')
+                continue
+
+            index = parts[0]
+            name = parts[1]
+            is_android_started = parts[4] == '1'
+            # 端口号规则 https://help.ldmnq.com/docs/LD9adbserver#a67730c2e7e2e0400d40bcab37d0e0cf
+            adb_port = 5554 + (int(index) * 2)
+
+            instance = LeidianInstance(
+                id=index,
+                name=name,
+                adb_port=adb_port,
+                adb_ip='127.0.0.1',
+                adb_name=f'emulator-{adb_port}'
+            )
+            instance.index = int(index)
+            instance.is_running = is_android_started
+            logger.debug('Leidian instance: %s', repr(instance))
+            instances.append(instance)
+
+        return instances
+
+    @staticmethod
+    def query(*, id: str) -> Instance | None:
+        instances = LeidianHost.list()
+        for instance in instances:
+            if instance.id == id:
+                return instance
+        return None
+
+    @staticmethod
+    def recipes() -> 'list[LeidianRecipes]':
+        return ['adb', 'uiautomator2']
+
+if __name__ == '__main__':
+    logging.basicConfig(level=logging.DEBUG, format='[%(asctime)s] [%(levelname)s] [%(name)s] [%(funcName)s] [%(lineno)d] %(message)s')
+    print(LeidianHost._read_install_path())
+    print(LeidianHost.installed())
+    print(LeidianHost.list())
+    print(ins:=LeidianHost.query(id='0'))
+    assert isinstance(ins, LeidianInstance)
+    ins.start()
+    ins.wait_available()
+    print('status', ins.running(), ins.adb_port, ins.adb_ip)
+    # ins.stop()
     # print('status', ins.running(), ins.adb_port, ins.adb_ip)