fix(installer+tray): 开始菜单卸载指向uninstall.exe + 停止后启动/重启 + 启动后浏览器时序

Author: quantclaws

installer/create-shortcuts.ps1

@@ -169,15 +169,19 @@ if ($Actions -contains 'Start') {
 }
 
 # 2. 卸载：托盘菜单接管了"停止"/"重启"，开始菜单不再列停止入口
+# NSIS 的 WriteUninstaller 生成的是 uninstall.exe（见 installer.nsi -Post
+# 段），而不是 uninstall.bat。早期版本误指向 uninstall.bat，导致开始菜单
+# 的 Uninstall.lnk 点击后提示找不到文件（控制面板走的是注册表
+# UninstallString，指向 uninstall.exe，所以控制面板卸载正常）。
 if ($Actions -contains 'Uninstall') {
     New-Shortcut `
         -Path (Join-Path $smAppDir '卸载.lnk') `
-        -Target (Join-Path $InstallDir 'uninstall.bat') `
+        -Target (Join-Path $InstallDir 'uninstall.exe') `
         -WorkingDirectory $InstallDir `
         -Description '卸载 QMT Gateway'
     New-Shortcut `
         -Path (Join-Path $smAppDir 'Uninstall.lnk') `
-        -Target (Join-Path $InstallDir 'uninstall.bat') `
+        -Target (Join-Path $InstallDir 'uninstall.exe') `
         -WorkingDirectory $InstallDir `
         -Description 'Uninstall QMT Gateway'
 }

installer/start.bat

@@ -21,11 +21,30 @@ if not errorlevel 1 (
     exit /b 0
 )
 
-start "" "http://localhost:%GATEWAY_PORT%"
+:: 端口没响应——可能是 gateway 没启动（用户刚从托盘"停止"过，.port 已被
+:: atexit 删掉），也可能是上一个实例崩了。先在后台启动一个"等待就绪并打开
+:: 浏览器"的任务，再在前台启动 gateway——这样 gateway 的日志留在当前控制
+:: 台窗口，浏览器在 gateway 监听端口后自动弹出。旧实现是在启动 gateway
+:: 之前就 start 浏览器，结果浏览器先于 gateway 就绪弹出，显示"无法连接"。
+::
+:: .port 可能因为 8130 被占用而指向 8131-8139，所以等待结束后重新读取，
+:: 而不是沿用启动前的值。
+start "" /B powershell.exe -NoProfile -ExecutionPolicy Bypass -Command ^
+  "$portFile='%QMT_GATEWAY_HOME%\.port';" ^
+  "$w=0;" ^
+  "while($w -lt 15){" ^
+    "if(Test-Path -LiteralPath $portFile){$r=Get-Content -LiteralPath $portFile -ErrorAction SilentlyContinue; if($r -match '^\d+$'){break}}" ^
+    "Start-Sleep -Seconds 1; $w++" ^
+  "};" ^
+  "$port=8130; if(Test-Path -LiteralPath $portFile){$r=Get-Content -LiteralPath $portFile -ErrorAction SilentlyContinue; if($r -match '^\d+$'){$port=[int]$r}};" ^
+  "$w=0;" ^
+  "while($w -lt 30){" ^
+    "try{$c=New-Object Net.Sockets.TcpClient; $c.Connect('127.0.0.1',$port); $c.Close(); start \"\" \"http://localhost:$port\"; exit 0}catch{}; Start-Sleep -Seconds 1; $w++" ^
+  "}"
 python\python.exe -m qmt_gateway
 set "RC=%ERRORLEVEL%"
-:: 退出码 2 = 另一个实例已在跑（被单实例锁拒绝）。浏览器已经打开，
-:: 不需要提示用户出错。
+:: 退出码 2 = 另一个实例已在跑（被单实例锁拒绝）。浏览器由上面的等待逻辑
+:: 打开，不需要提示用户出错。
 if not "%RC%"=="2" (
     if not "%RC%"=="0" (
         echo qmt-gateway 启动失败，退出码 %RC%

qmt_gateway/tray.py

@@ -4,8 +4,9 @@
 右键菜单：
 
   - 打开管理界面  →  浏览器打开 http://localhost:<port>（.port 文件）
+  - 启动 Gateway  →  停止后重新拉起 gateway（若已在跑则打开浏览器）
   - 重启 Gateway  →  杀掉旧 PID，等端口空闲，再启动新的 gateway
-  - 停止 Gateway  →  杀掉 gateway 进程（托盘自身保留，可手动重启）
+  - 停止 Gateway  →  杀掉 gateway 进程（托盘自身保留，可手动启动/重启）
   - 退出托盘      →  关闭托盘（gateway 继续运行）
 
 入口由 ``__main__`` 在 lifespan startup 阶段以独立子进程拉起，确保
@@ -234,10 +235,21 @@ def _open_browser(icon: pystray.Icon, item: pystray.MenuItem) -> None:
 
 
 def _restart_gateway(icon: pystray.Icon, item: pystray.MenuItem) -> None:
-    port = _read_port()
     logger.info("用户触发重启")
     # 先清过期锁（gateway 已被外部 kill -9 / 任务管理器结束的场景）
     _clear_stale_lock()
+    # "停止"之后 gateway 已死、.lock 已被 atexit 删掉，_read_pid() 返回 None。
+    # 此时 _kill_gateway() 返回 False，旧实现会短路掉 _spawn_gateway()——
+    # 结果用户点"重启"什么也没发生。这里改成：没有正在跑的进程就直接拉起。
+    # 注意不调 _wait_for_port_free：停止后 .port 已被删，_read_port() 退回
+    # 8130，但 8130 可能被别的应用占用——此时等它空闲会误判为"端口占用"。
+    # gateway 内部的 find_available_port 会自己探测可用端口，托盘不需要干预。
+    pid = _read_pid()
+    if pid is None:
+        logger.info("未发现正在运行的 gateway，直接启动新实例")
+        _spawn_gateway()
+        return
+    port = _read_port()
     if _kill_gateway() and _wait_for_port_free(port, timeout=10.0):
         _spawn_gateway()
     else:
@@ -255,9 +267,29 @@ def _quit_tray(icon: pystray.Icon, item: pystray.MenuItem) -> None:
     icon.stop()
 
 
+def _start_gateway(icon: pystray.Icon, item: pystray.MenuItem) -> None:
+    """启动 gateway（停止后用来重新拉起服务）。
+
+    与 _restart_gateway 的区别：这里假定当前没有正在运行的 gateway，
+    不做 kill，直接 spawn。gateway 内部的 find_available_port 会自己
+    探测可用端口，托盘不需要预先等端口空闲（否则 8130 被别的应用占用时
+    会误判为"无法启动"）。
+    """
+    logger.info("用户触发启动")
+    _clear_stale_lock()
+    pid = _read_pid()
+    if pid is not None and _is_pid_alive_windows(pid):
+        # 已经在跑——直接打开浏览器，不重复拉起
+        logger.info("gateway 已在运行，打开浏览器")
+        _open_browser(icon, item)
+        return
+    _spawn_gateway()
+
+
 def _build_menu() -> pystray.Menu:
     return pystray.Menu(
         pystray.MenuItem("打开管理界面", _open_browser, default=True),
+        pystray.MenuItem("启动 QMT Gateway", _start_gateway),
         pystray.MenuItem("重启 QMT Gateway", _restart_gateway),
         pystray.MenuItem("停止 QMT Gateway", _stop_gateway),
         pystray.Menu.SEPARATOR,