703 lines
28 KiB
Python
703 lines
28 KiB
Python
"""启动时的「环境准备」窗口（Windows）。
|
||
|
||
需求：每次打开 exe 都弹出一个窗口，列出所有需要的依赖：
|
||
- 没装的在右侧有「安装」按钮
|
||
- 装好的显示 ✓
|
||
- 必需项全部就绪后，「进入程序」按钮才可点击
|
||
|
||
设计要点：
|
||
- 用标准库 **tkinter**（已随 Python 打包进 exe），因此不依赖机器上有浏览器/网络，
|
||
也不依赖我们已经启动的 HTTP 服务 —— 它必须能作为"第一个界面"独立工作。
|
||
- 检测与安装都放在后台线程，通过队列回传 UI 线程（tkinter 不是线程安全的）。
|
||
- 任何异常都不能让程序打不开：tkinter 不可用时返回 None，由调用方降级处理。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import queue
|
||
import threading
|
||
from pathlib import Path
|
||
from typing import Any, Callable
|
||
|
||
from . import apple_driver
|
||
from . import platform as plat
|
||
from . import protocol
|
||
|
||
# 颜色（与 web 端保持同一套语义色）
|
||
COLOR_OK = "#1a7f37"
|
||
COLOR_BAD = "#c0392b"
|
||
COLOR_BUSY = "#b7791f"
|
||
COLOR_IDLE = "#6b7280"
|
||
COLOR_BG = "#ffffff"
|
||
COLOR_MUTED = "#57606a"
|
||
|
||
ICON_OK = "✓"
|
||
ICON_BAD = "✗"
|
||
ICON_BUSY = "…"
|
||
ICON_OPT = "·"
|
||
ICON_IDLE = "○"
|
||
|
||
|
||
class EnvItem:
|
||
"""一个待检查/待安装的依赖项。"""
|
||
|
||
def __init__(
|
||
self,
|
||
key: str,
|
||
label: str,
|
||
description: str,
|
||
check: Callable[[], tuple[bool, str]],
|
||
install: Callable[[Callable[[str, str], None]], tuple[bool, str]] | None = None,
|
||
required: bool = True,
|
||
install_label: str = "安装",
|
||
size_hint: str = "",
|
||
) -> None:
|
||
self.key = key
|
||
self.label = label
|
||
self.description = description
|
||
self.check = check
|
||
self.install = install
|
||
self.required = required
|
||
self.install_label = install_label
|
||
self.size_hint = size_hint
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 各个依赖项的检测 / 安装实现
|
||
# ---------------------------------------------------------------------------
|
||
def _check_python() -> tuple[bool, str]:
|
||
import sys
|
||
|
||
return True, f"已内置 Python {sys.version.split()[0]}"
|
||
|
||
|
||
def _check_pymobiledevice3() -> tuple[bool, str]:
|
||
try:
|
||
import pymobiledevice3
|
||
|
||
version = getattr(pymobiledevice3, "__version__", None) or "已安装"
|
||
return True, f"已内置 pymobiledevice3 {version}"
|
||
except Exception as exc:
|
||
return False, f"导入失败：{exc}"
|
||
|
||
|
||
def _check_tunnel_deps() -> tuple[bool, str]:
|
||
"""userspace 隧道依赖（纯 Python 协议栈，**不需要 Wintun 驱动**）。"""
|
||
missing = []
|
||
for module in ("pytun_pmd3", "pmd_pytcp", "pmd_net_addr"):
|
||
try:
|
||
__import__(module)
|
||
except Exception:
|
||
missing.append(module)
|
||
if missing:
|
||
return False, "缺少：" + "、".join(missing)
|
||
try:
|
||
from pymobiledevice3.remote.userspace_tunnel import UserspaceRsdTunnel # noqa: F401
|
||
|
||
return True, "已内置（纯 Python 用户态协议栈，无需 Wintun 驱动、无需管理员权限）"
|
||
except Exception as exc:
|
||
return False, f"隧道模块导入失败：{exc}"
|
||
|
||
|
||
def _check_apple_driver() -> tuple[bool, str]:
|
||
status = apple_driver.status(deep=True)
|
||
if not status.get("applicable", True):
|
||
return True, "非 Windows 平台，系统自带 usbmuxd"
|
||
if status.get("ready"):
|
||
product = status.get("product") or {}
|
||
version = product.get("displayversion")
|
||
devices = (status.get("devices") or {}).get("count", 0)
|
||
detail = "usbmuxd 可通信"
|
||
if version:
|
||
detail = f"v{version}，{detail}"
|
||
if devices:
|
||
detail += f"，已发现 {devices} 台设备"
|
||
return True, detail
|
||
return False, status.get("summary") or "不可用"
|
||
|
||
|
||
def _install_apple_driver(log: Callable[[str, str], None]) -> tuple[bool, str]:
|
||
if not plat.is_admin():
|
||
return False, "需要管理员权限（本程序启动时应已提权）"
|
||
result = apple_driver.ensure(log=log, allow_install=True)
|
||
if result.get("ok"):
|
||
if result.get("reboot_required"):
|
||
return True, "安装完成，但需要重启 Windows 才生效"
|
||
return True, "安装完成"
|
||
return False, result.get("error") or "安装失败"
|
||
|
||
|
||
def _check_developer_image() -> tuple[bool, str]:
|
||
from . import apple_driver as ad
|
||
|
||
status = ad.ddi_status()
|
||
if status["ok"]:
|
||
return True, f"已就绪（{status['total_mb']} MB，iOS 17 及以上全系共用）"
|
||
return False, "缺少：" + "、".join(status["missing"])
|
||
|
||
|
||
def _install_developer_image(log: Callable[[str, str], None]) -> tuple[bool, str]:
|
||
result = apple_driver.download_developer_image(log=log)
|
||
if result.get("ok"):
|
||
return True, result.get("detail") or "下载完成"
|
||
return False, result.get("error") or "下载失败"
|
||
|
||
|
||
def _check_permission() -> tuple[bool, str]:
|
||
if plat.is_admin():
|
||
return True, "已以管理员身份运行"
|
||
return False, "未提权（安装 Apple 驱动会失败）"
|
||
|
||
|
||
def _install_permission(log: Callable[[str, str], None]) -> tuple[bool, str]:
|
||
"""「安装」管理员权限 = 以管理员身份重新启动本程序。
|
||
|
||
之前这一项没有安装动作，导致**非管理员用户永远无法点「进入程序」**（死锁）。
|
||
现在点这个按钮会通过 UAC 拉起一个提权的新实例，本实例随即退出。
|
||
"""
|
||
if plat.is_admin():
|
||
return True, "已经是管理员身份"
|
||
log("info", "正在通过 UAC 以管理员身份重新启动程序…")
|
||
launched = plat.relaunch_elevated()
|
||
if launched:
|
||
log("info", "已发起提权重启；请在系统弹窗里点「是」")
|
||
return True, "已发起提权重启，本窗口即将关闭"
|
||
return False, "无法自动提权，请关闭程序后右键 →「以管理员身份运行」"
|
||
|
||
|
||
def _check_device() -> tuple[bool, str]:
|
||
if not plat.IS_WINDOWS:
|
||
return True, "非 Windows 平台"
|
||
result = apple_driver.list_devices()
|
||
if not result.get("ok"):
|
||
return False, str(result.get("error") or "usbmuxd 不可用")
|
||
count = result.get("count", 0)
|
||
if count == 0:
|
||
return False, "未检测到 iPhone（用数据线连接、解锁并点「信任」）"
|
||
first = result["devices"][0]
|
||
return True, f"{count} 台（{first.get('serial', '')}，{first.get('connection', '?')}）"
|
||
|
||
|
||
def _check_network() -> tuple[bool, str]:
|
||
"""检查两个**用途不同**的网络依赖，分开报，不要混为一谈。
|
||
|
||
- ``gs.apple.com``：挂载开发者镜像时向 Apple 申请 TSS 签名，**必须**能连。
|
||
- GitHub / jsDelivr：只在「开发者镜像缺失、需要现下」时用；镜像已内置则无所谓。
|
||
国内经常只有 GitHub 被墙而 Apple 正常，以前那种"二选一"的判断会给出误导性结论。
|
||
"""
|
||
import socket
|
||
|
||
def _reachable(host: str, port: int = 443) -> bool:
|
||
try:
|
||
with socket.create_connection((host, port), timeout=6):
|
||
return True
|
||
except OSError:
|
||
return False
|
||
|
||
apple_ok = _reachable("gs.apple.com")
|
||
mirror_ok = _reachable("raw.githubusercontent.com") or _reachable("cdn.jsdelivr.net")
|
||
|
||
if apple_ok and mirror_ok:
|
||
return True, "Apple 签名服务与镜像下载源均可连通"
|
||
if apple_ok:
|
||
return True, "Apple 签名服务可连通（镜像源不可达，但镜像已内置，通常无影响）"
|
||
return False, "无法连接 Apple 签名服务器 gs.apple.com —— 挂载开发者镜像时会失败"
|
||
|
||
|
||
def build_items() -> list[EnvItem]:
|
||
items = [
|
||
EnvItem("python", "Python 运行时", "程序自带的解释器",
|
||
_check_python, None, required=True),
|
||
EnvItem("pymobiledevice3", "pymobiledevice3", "与 iOS 设备通信的核心库",
|
||
_check_pymobiledevice3, None, required=True),
|
||
]
|
||
if plat.IS_WINDOWS:
|
||
items.append(EnvItem(
|
||
"apple_driver", "Apple Mobile Device Support",
|
||
"Windows 上的 usbmuxd（127.0.0.1:27015），用来发现 iPhone",
|
||
_check_apple_driver, _install_apple_driver, required=True,
|
||
install_label="安装", size_hint="约 40~200 MB",
|
||
))
|
||
items.append(EnvItem(
|
||
"tunnel_deps", "RSD 隧道依赖",
|
||
"iOS 17+ 的用户态隧道（PyTCP）",
|
||
_check_tunnel_deps, None, required=True,
|
||
))
|
||
items.append(EnvItem(
|
||
"developer_image", "开发者镜像",
|
||
"iOS 17 及以上全系共用的个性化镜像素材",
|
||
_check_developer_image, _install_developer_image, required=True,
|
||
install_label="下载", size_hint="约 15 MB",
|
||
))
|
||
items.append(EnvItem(
|
||
"permission", "管理员权限",
|
||
"安装 Apple 驱动需要；虚拟定位本身不需要",
|
||
_check_permission, _install_permission, required=True,
|
||
install_label="重启提权",
|
||
))
|
||
items.append(EnvItem(
|
||
"device", "iPhone 连接",
|
||
"可选项：现在没插也可以，之后接上再点「重新检测」",
|
||
_check_device, None, required=False,
|
||
))
|
||
items.append(EnvItem(
|
||
"network", "网络连通",
|
||
"可选项：挂载开发者镜像时 Apple 需要在线签名",
|
||
_check_network, None, required=False,
|
||
))
|
||
return items
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 窗口
|
||
# ---------------------------------------------------------------------------
|
||
class SetupWindow:
|
||
"""环境准备窗口。所有耗时操作都在后台线程里跑。"""
|
||
|
||
def __init__(self, root: Any, items: list[EnvItem] | None = None,
|
||
title: str = "环境准备") -> None:
|
||
import tkinter as tk
|
||
from tkinter import ttk
|
||
|
||
self.tk = tk
|
||
self.ttk = ttk
|
||
self.root = root
|
||
self.items = items if items is not None else build_items()
|
||
self.state: dict[str, dict[str, Any]] = {
|
||
item.key: {"ok": None, "detail": "等待检测…", "busy": False}
|
||
for item in self.items
|
||
}
|
||
self.events: queue.Queue[tuple[str, Any]] = queue.Queue()
|
||
self.installing = False
|
||
self.proceeded = False
|
||
self.auto_install = False
|
||
self._rows: dict[str, dict[str, Any]] = {}
|
||
|
||
root.title(f"{title} — iOS 虚拟定位轨迹工具")
|
||
root.configure(bg=COLOR_BG)
|
||
# 先建内容再定尺寸：否则 Tk 会按控件的"要求尺寸"把窗口撑高，
|
||
# geometry() 设的高度会被覆盖，底部的「进入程序」按钮就被挤出屏幕。
|
||
self._build(title)
|
||
root.resizable(False, True)
|
||
self._fit_to_screen(root)
|
||
root.protocol("WM_DELETE_WINDOW", self._on_close)
|
||
self._pump_after_id: str | None = None
|
||
self._schedule_pump()
|
||
self.refresh_all()
|
||
|
||
def _schedule_pump(self) -> None:
|
||
"""排下一次 UI 泵，并记住 after id 以便销毁前取消。"""
|
||
try:
|
||
self._pump_after_id = self.root.after(80, self._pump)
|
||
except Exception:
|
||
self._pump_after_id = None
|
||
|
||
def _cancel_pump(self) -> None:
|
||
"""销毁窗口前取消待执行的 after，否则 Tk 会打印
|
||
``invalid command name "..._pump"``（窗口程序看不到，命令行会刷屏）。"""
|
||
if self._pump_after_id is not None:
|
||
try:
|
||
self.root.after_cancel(self._pump_after_id)
|
||
except Exception:
|
||
pass
|
||
self._pump_after_id = None
|
||
|
||
def _destroy(self) -> None:
|
||
self._cancel_pump()
|
||
try:
|
||
self.root.destroy()
|
||
except Exception:
|
||
pass
|
||
|
||
# ---------------- 布局 ----------------
|
||
@staticmethod
|
||
def _fit_to_screen(window: Any, min_width: int = 820, min_height: int = 560) -> None:
|
||
"""按内容实际需要的高度定尺寸，并保证窗口完整落在屏幕内。"""
|
||
window.update_idletasks()
|
||
try:
|
||
screen_w = window.winfo_screenwidth()
|
||
screen_h = window.winfo_screenheight()
|
||
# 控件"要求"的尺寸 + 一点余量
|
||
need_w = max(min_width, window.winfo_reqwidth() + 24)
|
||
need_h = window.winfo_reqheight() + 24
|
||
width = min(need_w, max(640, screen_w - 60))
|
||
height = min(need_h, max(min_height, screen_h - 100))
|
||
x = max(0, (screen_w - width) // 2)
|
||
y = max(0, (screen_h - height) // 3)
|
||
window.geometry(f"{width}x{height}+{x}+{y}")
|
||
window.minsize(min(760, width), min(520, height))
|
||
except Exception:
|
||
window.geometry("960x760")
|
||
|
||
def _build(self, title: str) -> None:
|
||
tk, ttk = self.tk, self.ttk
|
||
style = ttk.Style()
|
||
try:
|
||
style.theme_use("vista")
|
||
except Exception:
|
||
pass
|
||
|
||
header = tk.Frame(self.root, bg=COLOR_BG)
|
||
header.pack(fill="x", padx=20, pady=(16, 6))
|
||
title_row = tk.Frame(header, bg=COLOR_BG)
|
||
title_row.pack(fill="x", anchor="w")
|
||
tk.Label(title_row, text=title, font=("Microsoft YaHei UI", 15, "bold"),
|
||
bg=COLOR_BG).pack(side="left")
|
||
tk.Label(title_row, text=f"v{protocol.APP_VERSION} · {plat.platform_label()}",
|
||
font=("Microsoft YaHei UI", 9), fg=COLOR_MUTED,
|
||
bg=COLOR_BG).pack(side="left", padx=(10, 0), pady=(4, 0))
|
||
tk.Label(
|
||
header,
|
||
text="首次使用需要准备以下组件。缺失的项目点右侧按钮即可自动安装；"
|
||
"全部就绪后点「进入程序」。",
|
||
font=("Microsoft YaHei UI", 9), fg=COLOR_MUTED, bg=COLOR_BG,
|
||
justify="left",
|
||
).pack(anchor="w", pady=(4, 0))
|
||
|
||
ttk.Separator(self.root).pack(fill="x", padx=20, pady=(10, 4))
|
||
|
||
body = tk.Frame(self.root, bg=COLOR_BG)
|
||
body.pack(fill="both", expand=False, padx=20)
|
||
body.columnconfigure(1, weight=1)
|
||
|
||
for row, item in enumerate(self.items):
|
||
icon = tk.Label(body, text=ICON_IDLE, font=("Segoe UI Symbol", 13, "bold"),
|
||
fg=COLOR_IDLE, bg=COLOR_BG, width=2)
|
||
icon.grid(row=row, column=0, sticky="nw", pady=(6, 0))
|
||
|
||
text = tk.Frame(body, bg=COLOR_BG)
|
||
text.grid(row=row, column=1, sticky="ew", padx=(2, 10), pady=(6, 0))
|
||
name = tk.Label(text, text=item.label, font=("Microsoft YaHei UI", 10, "bold"),
|
||
bg=COLOR_BG, anchor="w")
|
||
name.pack(anchor="w")
|
||
detail = tk.Label(text, text=item.description, font=("Microsoft YaHei UI", 8),
|
||
fg=COLOR_MUTED, bg=COLOR_BG, anchor="w", justify="left",
|
||
wraplength=560)
|
||
detail.pack(anchor="w")
|
||
|
||
button = ttk.Button(body, text=item.install_label, width=10,
|
||
command=lambda i=item: self.install_one(i))
|
||
button.grid(row=row, column=2, sticky="ne", pady=(6, 0))
|
||
if item.install is None:
|
||
button.state(["disabled"])
|
||
|
||
tag = tk.Label(body, text="必需" if item.required else "可选",
|
||
font=("Microsoft YaHei UI", 8),
|
||
fg=COLOR_BAD if item.required else COLOR_MUTED,
|
||
bg=COLOR_BG, width=5)
|
||
tag.grid(row=row, column=3, sticky="ne", pady=(6, 0))
|
||
|
||
self._rows[item.key] = {
|
||
"icon": icon, "detail": detail, "button": button, "item": item
|
||
}
|
||
|
||
ttk.Separator(self.root).pack(fill="x", padx=20, pady=(12, 6))
|
||
|
||
log_frame = tk.Frame(self.root, bg=COLOR_BG)
|
||
log_frame.pack(fill="both", expand=True, padx=20)
|
||
tk.Label(log_frame, text="安装日志", font=("Microsoft YaHei UI", 9, "bold"),
|
||
bg=COLOR_BG).pack(anchor="w")
|
||
self.log = tk.Text(log_frame, height=6, wrap="word", font=("Consolas", 8),
|
||
bg="#f6f8fa", fg="#24292f", relief="flat",
|
||
highlightthickness=1, highlightbackground="#d0d7de")
|
||
scroll = ttk.Scrollbar(log_frame, command=self.log.yview)
|
||
self.log.configure(yscrollcommand=scroll.set)
|
||
scroll.pack(side="right", fill="y")
|
||
self.log.pack(side="left", fill="both", expand=True, pady=(3, 0))
|
||
self.log.configure(state="disabled")
|
||
|
||
footer = tk.Frame(self.root, bg=COLOR_BG)
|
||
footer.pack(fill="x", padx=20, pady=(10, 16))
|
||
|
||
self.status_label = tk.Label(footer, text="正在检测…",
|
||
font=("Microsoft YaHei UI", 9), fg=COLOR_MUTED,
|
||
bg=COLOR_BG)
|
||
self.status_label.pack(side="left")
|
||
|
||
self.confirm_button = ttk.Button(footer, text="进入程序", width=16,
|
||
command=self._on_confirm)
|
||
self.confirm_button.pack(side="right", padx=(8, 0))
|
||
self.confirm_button.state(["disabled"])
|
||
|
||
self.install_all_button = ttk.Button(footer, text="全部安装", width=12,
|
||
command=self.install_all)
|
||
self.install_all_button.pack(side="right", padx=(8, 0))
|
||
|
||
self.recheck_button = ttk.Button(footer, text="重新检测", width=12,
|
||
command=self.refresh_all)
|
||
self.recheck_button.pack(side="right")
|
||
|
||
self.open_logs_button = ttk.Button(footer, text="打开日志目录", width=13,
|
||
command=self._open_logs_dir)
|
||
self.open_logs_button.pack(side="right", padx=(0, 8))
|
||
|
||
def _open_logs_dir(self) -> None:
|
||
"""打开 %LOCALAPPDATA%\\iOSLocationTrack\\logs —— 装驱动失败时让用户能自己取证。"""
|
||
try:
|
||
target = plat.data_dir() / "logs"
|
||
target.mkdir(parents=True, exist_ok=True)
|
||
if not plat.open_in_file_manager(target):
|
||
self.append_log(f"日志目录：{target}")
|
||
except Exception as exc:
|
||
self.append_log(f"无法打开日志目录：{exc}")
|
||
|
||
# ---------------- 日志 ----------------
|
||
def append_log(self, text: str) -> None:
|
||
self.log.configure(state="normal")
|
||
self.log.insert("end", text.rstrip() + "\n")
|
||
self.log.see("end")
|
||
self.log.configure(state="disabled")
|
||
|
||
# ---------------- 检测 ----------------
|
||
def refresh_all(self) -> None:
|
||
if self.installing:
|
||
return
|
||
for item in self.items:
|
||
self.state[item.key] = {"ok": None, "detail": "检测中…", "busy": False}
|
||
self._render(item.key)
|
||
self._update_footer()
|
||
threading.Thread(target=self._check_worker, name="env-check", daemon=True).start()
|
||
|
||
def _check_worker(self) -> None:
|
||
for item in self.items:
|
||
try:
|
||
ok, detail = item.check()
|
||
except Exception as exc:
|
||
ok, detail = False, f"检测异常：{exc}"
|
||
self.events.put(("check", (item.key, ok, detail)))
|
||
self.events.put(("checks_done", None))
|
||
|
||
# ---------------- 安装 ----------------
|
||
def install_one(self, item: EnvItem) -> None:
|
||
if self.installing or item.install is None:
|
||
return
|
||
state = self.state[item.key]
|
||
if state.get("ok") is True:
|
||
return
|
||
self.installing = True
|
||
self.install_all_button.state(["disabled"])
|
||
self.recheck_button.state(["disabled"])
|
||
self._set_busy(item.key, "正在安装…")
|
||
self.append_log(f"===== 开始安装：{item.label} =====")
|
||
threading.Thread(target=self._install_worker, args=([item],),
|
||
name=f"install-{item.key}", daemon=True).start()
|
||
|
||
def install_all(self) -> None:
|
||
if self.installing:
|
||
return
|
||
pending = [i for i in self.items
|
||
if i.install is not None and self.state[i.key].get("ok") is not True]
|
||
if not pending:
|
||
self.append_log("没有需要安装的项目。")
|
||
return
|
||
self.installing = True
|
||
self.install_all_button.state(["disabled"])
|
||
self.recheck_button.state(["disabled"])
|
||
for item in pending:
|
||
self._set_busy(item.key, "排队中…")
|
||
threading.Thread(target=self._install_worker, args=(pending,),
|
||
name="install-all", daemon=True).start()
|
||
|
||
def _install_worker(self, items: list[EnvItem]) -> None:
|
||
for item in items:
|
||
self.events.put(("busy", (item.key, "正在安装…")))
|
||
|
||
def log(level: str, message: str, _label: str = item.label) -> None:
|
||
self.events.put(("log", f"[{_label}] {message}"))
|
||
|
||
try:
|
||
assert item.install is not None
|
||
ok, detail = item.install(log)
|
||
except Exception as exc:
|
||
ok, detail = False, f"安装异常：{exc}"
|
||
self.events.put(("install", (item.key, ok, detail)))
|
||
if item.key == "permission" and ok:
|
||
# 提权重启已发起：关掉本窗口，本进程退出，由新实例接管
|
||
self.events.put(("close", None))
|
||
return
|
||
self.events.put(("installs_done", None))
|
||
|
||
# ---------------- UI 线程泵 ----------------
|
||
def _pump(self) -> None:
|
||
# 窗口一旦销毁（点「进入程序」/关闭窗口/提权重启），就不要再碰控件：
|
||
# 否则队列里排着的 after 回调会在 Tk 里报 invalid command name ..._pump。
|
||
try:
|
||
if not self.root.winfo_exists():
|
||
return
|
||
except Exception:
|
||
return
|
||
try:
|
||
while True:
|
||
kind, payload = self.events.get_nowait()
|
||
if kind == "check":
|
||
key, ok, detail = payload
|
||
self.state[key] = {"ok": ok, "detail": detail, "busy": False}
|
||
self._render(key)
|
||
# 每完成一项就刷新底部：否则必需项都绿了，
|
||
# 还要等可选项（网络探测可能 3×6s / PnP 45s）跑完按钮才亮。
|
||
self._update_footer()
|
||
elif kind == "checks_done":
|
||
self._update_footer()
|
||
elif kind == "busy":
|
||
key, detail = payload
|
||
self.state[key]["busy"] = True
|
||
self.state[key]["detail"] = detail
|
||
self._render(key)
|
||
elif kind == "install":
|
||
key, ok, detail = payload
|
||
self.state[key] = {"ok": ok, "detail": detail, "busy": False}
|
||
self._render(key)
|
||
self.append_log(("安装成功：" if ok else "安装失败：") + detail)
|
||
elif kind == "installs_done":
|
||
self.installing = False
|
||
self.install_all_button.state(["!disabled"])
|
||
self.recheck_button.state(["!disabled"])
|
||
self._update_footer()
|
||
# 安装后可能影响其它项（例如驱动装好后设备就能发现了），重新检测一遍
|
||
self.refresh_all()
|
||
elif kind == "log":
|
||
self.append_log(payload)
|
||
elif kind == "close":
|
||
# 提权重启：关掉本窗口。main() 看到 proceeded=False 会正常退出，
|
||
# 由刚拉起的提权实例继续跑（它自己会再弹一次本窗口，那时已是管理员）。
|
||
self.proceeded = False
|
||
self._destroy()
|
||
return
|
||
except queue.Empty:
|
||
pass
|
||
if not self.proceeded:
|
||
self._schedule_pump()
|
||
|
||
# ---------------- 渲染 ----------------
|
||
def _render(self, key: str) -> None:
|
||
row = self._rows.get(key)
|
||
if row is None:
|
||
return
|
||
state = self.state[key]
|
||
item: EnvItem = row["item"]
|
||
ok = state.get("ok")
|
||
busy = state.get("busy")
|
||
|
||
if busy:
|
||
icon, color = ICON_BUSY, COLOR_BUSY
|
||
elif ok is True:
|
||
icon, color = ICON_OK, COLOR_OK
|
||
elif ok is False:
|
||
icon = ICON_BAD if item.required else ICON_OPT
|
||
color = COLOR_BAD if item.required else COLOR_MUTED
|
||
else:
|
||
icon, color = ICON_IDLE, COLOR_IDLE
|
||
|
||
row["icon"].configure(text=icon, fg=color)
|
||
suffix = f" （{item.size_hint}）" if item.size_hint and ok is not True else ""
|
||
row["detail"].configure(text=f"{state.get('detail', '')}{suffix}",
|
||
fg=COLOR_OK if ok is True else COLOR_MUTED)
|
||
|
||
button = row["button"]
|
||
if ok is True:
|
||
# 已就绪：按钮文字改成"已就绪"并禁用，避免用户误以为还要再装一次
|
||
button.configure(text="已就绪")
|
||
button.state(["disabled"])
|
||
elif item.install is None:
|
||
button.configure(text="—")
|
||
button.state(["disabled"])
|
||
elif busy:
|
||
button.configure(text="安装中…")
|
||
button.state(["disabled"])
|
||
else:
|
||
button.configure(text=item.install_label)
|
||
button.state(["!disabled"])
|
||
|
||
def _set_busy(self, key: str, detail: str) -> None:
|
||
self.state[key]["busy"] = True
|
||
self.state[key]["detail"] = detail
|
||
self._render(key)
|
||
|
||
def _needs_item(self, item: EnvItem) -> bool:
|
||
"""判断某一项此刻是否真的"必需"。
|
||
|
||
「管理员权限」只在**需要安装 Apple 驱动**时才是必需的：驱动已就绪时，
|
||
建隧道（userspace 免管理员）和下发定位都不需要提权，不该因为它把
|
||
「进入程序」锁死 —— 否则源码模式 / 非管理员启动就永远进不去。
|
||
"""
|
||
if item.key == "permission":
|
||
driver = self.state.get("apple_driver") or {}
|
||
return driver.get("ok") is not True
|
||
return item.required
|
||
|
||
def _update_footer(self) -> None:
|
||
required = [i for i in self.items if self._needs_item(i)]
|
||
pending = [i for i in required if self.state[i.key].get("ok") is not True]
|
||
optional_bad = [i for i in self.items
|
||
if not i.required and self.state[i.key].get("ok") is False]
|
||
|
||
if self.installing:
|
||
self.status_label.configure(text="正在安装，请稍候…", fg=COLOR_BUSY)
|
||
elif pending:
|
||
names = "、".join(i.label for i in pending)
|
||
self.status_label.configure(text=f"还差 {len(pending)} 项：{names}", fg=COLOR_BAD)
|
||
else:
|
||
text = "全部就绪，可以开始使用"
|
||
if optional_bad:
|
||
text += "（可选：" + "、".join(i.label for i in optional_bad) + " 未就绪）"
|
||
self.status_label.configure(text=text, fg=COLOR_OK)
|
||
|
||
if not pending and not self.installing:
|
||
self.confirm_button.state(["!disabled"])
|
||
else:
|
||
self.confirm_button.state(["disabled"])
|
||
|
||
any_missing = any(i.install is not None and self.state[i.key].get("ok") is not True
|
||
for i in self.items)
|
||
if any_missing and not self.installing:
|
||
self.install_all_button.state(["!disabled"])
|
||
else:
|
||
self.install_all_button.state(["disabled"])
|
||
|
||
# ---------------- 结束 ----------------
|
||
def _on_confirm(self) -> None:
|
||
self.proceeded = True
|
||
self._destroy()
|
||
|
||
def _on_close(self) -> None:
|
||
self.proceeded = False
|
||
self._destroy()
|
||
|
||
|
||
def run_setup_window(title: str = "环境准备", auto_install_missing: bool = False) -> bool | None:
|
||
"""显示环境准备窗口。
|
||
|
||
返回：
|
||
True —— 全部必需项就绪，用户点了「进入程序」
|
||
False —— 用户关闭了窗口（调用方应退出）
|
||
None —— 无法显示窗口（tkinter 不可用/构建失败），调用方自行降级
|
||
"""
|
||
import traceback
|
||
|
||
try:
|
||
import tkinter as tk
|
||
except Exception as exc:
|
||
print(f" [提示] tkinter 不可用：{exc}")
|
||
return None
|
||
|
||
try:
|
||
root = tk.Tk()
|
||
except Exception as exc:
|
||
print(f" [提示] 无法创建窗口（可能没有图形界面）：{exc}")
|
||
return None
|
||
|
||
try:
|
||
window = SetupWindow(root, title=title)
|
||
if auto_install_missing:
|
||
root.after(600, window.install_all)
|
||
root.mainloop()
|
||
return bool(window.proceeded)
|
||
except Exception:
|
||
# 这里必须把真实原因打出来，否则窗口不显示时无从排查
|
||
print(" [错误] 「环境准备」窗口构建失败：")
|
||
traceback.print_exc()
|
||
try:
|
||
root.destroy()
|
||
except Exception:
|
||
pass
|
||
return None
|