iOSAI/Module/Main.py

import os
import sys
from pathlib import Path

import wda

import tidevice
from Module.DeviceInfo import DeviceInfo
from Module.FlaskSubprocessManager import FlaskSubprocessManager
from Utils.DevDiskImageDeployer import DevDiskImageDeployer

# 确定 exe 或 py 文件所在目录
BASE = Path(getattr(sys, 'frozen', False) and sys.executable or __file__).resolve().parent
LOG_DIR = BASE / "log"
LOG_DIR.mkdir(exist_ok=True)  # 确保 log 目录存在

print(f"日志目录: {LOG_DIR}")

def _run_flask_role():
    from Module import FlaskService
    port = int(os.getenv("FLASK_COMM_PORT", "34567"))  # 固定端口的兜底仍是 34567
    app_factory = getattr(FlaskService, "create_app", None)
    app = app_factory() if callable(app_factory) else FlaskService.app
    app.run(host="0.0.0.0", port=port + 1, debug=False, use_reloader=False)

if "--role=flask" in sys.argv:
    _run_flask_role()
    sys.exit(0)


# 项目入口
if __name__ == "__main__":

    wda.debug = True

    # 1. 打包环境
    if hasattr(sys, '_MEIPASS'):
        base_dir = sys._MEIPASS
    # 2. 源码调试环境
    else:
        base_dir = os.path.dirname(
            os.path.dirname(os.path.abspath(__file__))  # 向上跳一级
        )

    # 3. 拼 iproxy 目录
    iproxy_dir = os.path.join(base_dir, 'resources', 'iproxy')

    # 4. 剩余逻辑不变
    if os.path.isdir(iproxy_dir):
        sep = os.pathsep
        old_path = os.environ.get('PATH', '')
        if iproxy_dir not in old_path:
            os.environ['PATH'] = old_path + sep + iproxy_dir
    else:
        print(f'warning: {iproxy_dir} not found', file=sys.stderr)

    # 添加iOS开发包到电脑上
    deployer = DevDiskImageDeployer(verbose=True)
    deployer.deploy_all()

    # 启动 Flask 子进程
    manager = FlaskSubprocessManager.get_instance()
    manager.start()

    # 设备监听（即使失败/很快返回，也不会导致主进程退出）
    try:
        info = DeviceInfo()
        info.listen()
    except Exception as e:
        print("[WARN] Device listener not running:", e)


    # === 保活：阻塞主线程，直到收到 Ctrl+C/关闭 ===
    import threading, time, signal

    stop = threading.Event()

    def _handle(_sig, _frm):
        stop.set()
    
    # Windows 上 SIGINT/SIGTERM 都可以拦到
    try:
        signal.signal(signal.SIGINT, _handle)
        signal.signal(signal.SIGTERM, _handle)
    except Exception:
        pass  # 某些环境可能不支持，忽略

    try:
        while not stop.is_set():
            time.sleep(1)
    finally:
        # 进程退出前记得把子进程关掉
        manager.stop()