dsh-desktop-windows 0.4.0 → 0.4.1

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "dsh-desktop-windows",
   "productName": "DeepSeek Harness Desktop",
-  "version": "0.4.0",
+  "version": "0.4.1",
   "description": "DeepSeek Harness 桌面版（Windows）· Electron desktop shell for DeepSeek Harness — 双击即用 / double-click to run · 支持选区截图提问、系统托盘、会话管理 / region screenshot, system tray, session manager",
   "main": "src/main.js",
   "license": "MIT",
@@ -67,3 +67,4 @@
 
 
 
+

package/src/main.js

@@ -351,24 +351,81 @@ let isShuttingDownDsh = false
 // 已自动重启过一次的标志（同一次会话最多自愈一次，避免崩溃循环）
 let didAutoRestartDsh = false
 
+/* ------------------------------------------------------------------ *
+ * dsh 0.1.5+ 启动令牌（launch token）
+ *
+ * 从 0.1.5 起 dsh web 的 Web UI 需要启动时打印的带 token URL 才能访问：
+ *   首次带 ?token=... 访问 → 服务端种下会话 cookie → 之后访问干净的 / 即可。
+ * 本壳捕获那行 URL 并用它加载窗口，cookie 建好后 F5 刷新照常工作。
+ * 若服务不是本壳启动的（外部/WSL），拿不到 token，只能提示用户。
+ * ------------------------------------------------------------------ */
+
+let localLaunchUrl = null
+let launchUrlWaiters = []
+const LAUNCH_URL_RE = /dsh web:\s*(https?:\/\/[^\s()]+)/
+const LAUNCH_URL_TIMEOUT_MS = 15_000
+
+/** 从 dsh 输出流里捕获带 token 的 URL（只认第一行）。 */
+function captureLaunchUrl(text) {
+  if (localLaunchUrl) return
+  const m = LAUNCH_URL_RE.exec(text)
+  if (!m) return
+  localLaunchUrl = m[1]
+  // 只记「捕获到了」，不把 token 写进日志
+  log('captured authenticated dsh web url')
+  const waiters = launchUrlWaiters
+  launchUrlWaiters = []
+  for (const resolve of waiters) resolve(localLaunchUrl)
+}
+
+/** 等带 token 的 URL 出现；超时返回 null（不阻塞启动流程）。 */
+function waitForLaunchUrl(timeoutMs = LAUNCH_URL_TIMEOUT_MS) {
+  if (localLaunchUrl) return Promise.resolve(localLaunchUrl)
+  return new Promise((resolve) => {
+    const onUrl = (url) => {
+      clearTimeout(timer)
+      resolve(url)
+    }
+    const timer = setTimeout(() => {
+      launchUrlWaiters = launchUrlWaiters.filter((w) => w !== onUrl)
+      resolve(null)
+    }, timeoutMs)
+    launchUrlWaiters.push(onUrl)
+  })
+}
+
+/** 重启 dsh 前重置，避免用上一次的旧 token。 */
+function resetLaunchUrl() {
+  localLaunchUrl = null
+}
+
 /** 启动 dsh web，日志追加到 userData/dsh.log。返回子进程。 */
 function startDsh(port) {
   const bin = resolveDshBin()
   const logFile = path.join(app.getPath('userData'), 'dsh.log')
-  const out = fs.openSync(logFile, 'a')
+  // 用管道而不是直接写 fd：既要落日志，又要捕获启动时打印的带 token URL
+  const out = fs.createWriteStream(logFile, { flags: 'a' })
   // shell:true 时把参数拼进命令字符串（参数均为常量），避免 DEP0190 警告
   // --no-open 关闭 dsh 默认的自动打开浏览器行为（桌面壳自己用 BrowserWindow 加载，不弹浏览器）
   const cmd = `"${bin}" web --port ${port} --no-open`
   const child = spawn(cmd, {
     shell: true,
     windowsHide: true,
-    stdio: ['ignore', out, out],
+    stdio: ['ignore', 'pipe', 'pipe'],
     env: { ...process.env },
   })
+  const onData = (chunk) => {
+    const text = chunk.toString()
+    out.write(text)
+    captureLaunchUrl(text)
+  }
+  child.stdout.on('data', onData)
+  child.stderr.on('data', onData)
   log(`spawned dsh: ${cmd} (pid ${child.pid})`)
   log(`dsh log: ${logFile}`)
   child.on('exit', (code, signal) => {
     log(`dsh exited: code=${code} signal=${signal}`)
+    out.end()
     if (ownedDsh !== child) return
     ownedDsh = null
     // 崩溃自愈：本壳启动的 dsh 意外退出（非主动关闭/非退出中）时自动重启一次
@@ -409,12 +466,16 @@ function restartOwnedDsh(port, exitCode) {
         log('port already serving after crash; not restarting')
         return
       }
+      resetLaunchUrl()
       const child = startDsh(port)
       ownedDsh = child
       // 等它就绪；若起不来则等 exit 回调里的二次提示
-      waitForServer(port, START_TIMEOUT_MS).then((ready) => {
+      waitForServer(port, START_TIMEOUT_MS).then(async (ready) => {
         if (ready) {
           log('dsh auto-restarted successfully')
+          // 新进程有新 token：等捕获后重新加载页面
+          await waitForLaunchUrl()
+          loadApp()
           dialog.showMessageBox(mainWindow || undefined, {
             type: 'info',
             title: 'dsh 已自动重启',
@@ -428,6 +489,22 @@ function restartOwnedDsh(port, exitCode) {
   }, 3000)
 }
 
+/** 查监听指定端口的进程 PID（netstat 解析，找不到返回 null）。 */
+function portOwnerPid(port) {
+  return new Promise((resolve) => {
+    execFile('netstat', ['-ano', '-p', 'tcp'], { windowsHide: true }, (err, stdout) => {
+      if (err || !stdout) return resolve(null)
+      const line = String(stdout)
+        .split('\n')
+        .find((l) => l.includes(`:${port} `) && /LISTENING/i.test(l))
+      if (!line) return resolve(null)
+      const parts = line.trim().split(/\s+/)
+      const pid = Number(parts[parts.length - 1])
+      resolve(Number.isInteger(pid) && pid > 0 ? pid : null)
+    })
+  })
+}
+
 /** 杀掉进程树（Windows 用 taskkill /T 覆盖子孙进程），带超时兜底。 */
 function killProcessTree(pid) {
   return new Promise((resolve) => {
@@ -458,23 +535,50 @@ async function shutdownOwnedDsh() {
 async function restartDshService() {
   if (!ownedDsh) {
     log('restart: no owned dsh to restart')
-    // 没有自管的 dsh（可能复用了外部服务）→ 无法安全重启，提示用户
-    dialog.showMessageBox(mainWindow || undefined, {
-      type: 'info',
-      title: '无法重启',
-      message: '当前使用的是外部已有的 dsh 服务，不是本应用启动的，无法从这里重启。',
-      detail: '请在外部终端手动重启该服务，然后刷新页面（F5）。',
-      buttons: ['好'],
+    if (!targetIsLocal) {
+      // 远程模式：本壳无权重启对方服务器
+      dialog.showMessageBox(mainWindow || undefined, {
+        type: 'info',
+        title: '无法重启',
+        message: '当前连接的是远程服务器上的 dsh。',
+        detail: '请在服务器上重启该服务（如 docker compose 重建容器），然后刷新页面（F5）。',
+        buttons: ['好'],
+        noLink: true,
+      })
+      return
+    }
+    // 本机、但服务不是本壳启动的（残留/手动启动）：dsh 0.1.5+ 需要启动令牌，
+    // 本壳拿不到，页面会 401。经用户确认后接管：结束占用端口的进程，再由本壳启动。
+    const port = resolvePort()
+    const { response } = await dialog.showMessageBox(mainWindow || undefined, {
+      type: 'question',
+      title: '接管本机 dsh 服务',
+      message: `端口 ${port} 上有一个不是本应用启动的 dsh 服务。`,
+      detail:
+        'dsh 0.1.5 起，Web UI 需要启动时打印的令牌才能访问，外部启动的服务本应用无法自动连接。\n\n' +
+        '是否结束该进程，并由本应用重新启动 dsh？（当前页面会短暂中断）',
+      buttons: ['接管并重启', '取消'],
+      defaultId: 0,
+      cancelId: 1,
       noLink: true,
     })
-    return
+    if (response !== 0) return
+    const pid = await portOwnerPid(port)
+    if (pid) {
+      log(`restart: taking over external dsh on port ${port} (pid ${pid})`)
+      await killProcessTree(pid)
+      await new Promise((r) => setTimeout(r, 800))
+    } else {
+      log(`restart: no listener found on port ${port}; starting a fresh service`)
+    }
   }
   const port = resolvePort()
   log(`restart: restarting dsh service on port ${port}…`)
   // 先停（标记主动关停，避免触发崩溃自愈逻辑）
   await shutdownOwnedDsh()
   isShuttingDownDsh = false // 重置，让后续退出能触发自愈
-  // 重新启动
+  // 重新启动（换新 token，先清掉旧的）
+  resetLaunchUrl()
   ownedDsh = startDsh(port)
   const ready = await waitForServer(port, START_TIMEOUT_MS)
   if (!ready) {
@@ -489,10 +593,11 @@ async function restartDshService() {
     })
     return
   }
+  // 等新的带 token URL（dsh 0.1.5+）
+  await waitForLaunchUrl()
   log('restart: dsh back up; reloading window')
-  // 等页面也能加载后刷新
   if (mainWindow && !mainWindow.isDestroyed()) {
-    mainWindow.loadURL(`http://127.0.0.1:${port}`)
+    loadApp()
   }
   dialog.showMessageBox(mainWindow || undefined, {
     type: 'info',
@@ -588,6 +693,7 @@ async function performDshUpdate(targetVersion) {
   const port = resolvePort()
   await shutdownOwnedDsh()
   isShuttingDownDsh = false // 重置，让后续崩溃自愈逻辑仍生效
+  resetLaunchUrl()
   ownedDsh = startDsh(port)
 
   // 4) 轮询直到服务恢复，然后重载页面
@@ -600,6 +706,8 @@ async function performDshUpdate(targetVersion) {
   }
   setUpdateProgress(-1)
   if (back) {
+    // 新版 dsh 同样需要带 token 的 URL（0.1.5+）
+    await waitForLaunchUrl()
     loadApp()
   } else {
     log('dsh update: service did not come back within timeout')
@@ -1225,7 +1333,46 @@ function showMainWindow() {
 
 /** 回到应用主页面（不能用 reload()——它会把临时页如更新进度页再刷一遍，把人困住）。 */
 function loadApp() {
-  if (mainWindow && !mainWindow.isDestroyed() && target) mainWindow.loadURL(target).catch(() => {})
+  if (!mainWindow || mainWindow.isDestroyed()) return
+  // 本机模式优先用捕获到的带 token URL（dsh 0.1.5+ 必需；cookie 建立后普通刷新即可）
+  const url = targetIsLocal && localLaunchUrl ? localLaunchUrl : target
+  if (!url) return
+  guardAuthRequired()
+  mainWindow.loadURL(url).catch(() => {})
+}
+
+let authNoticeShown = false
+
+/**
+ * 若页面落到「authentication required」401（常见于复用了外部启动的 dsh，
+ * 本壳拿不到启动令牌），给出可执行的指引而不是让用户看一屏报错。
+ */
+function guardAuthRequired() {
+  if (!mainWindow || mainWindow.isDestroyed()) return
+  mainWindow.webContents.once('did-finish-load', async () => {
+    if (authNoticeShown) return
+    try {
+      const text = await mainWindow.webContents.executeJavaScript(
+        'document.body ? document.body.innerText.slice(0, 300) : ""'
+      )
+      if (!/authentication required/i.test(String(text))) return
+      authNoticeShown = true
+      dialog.showMessageBox(mainWindow, {
+        type: 'warning',
+        title: 'dsh 服务需要启动令牌',
+        message: '当前运行中的 dsh 服务需要启动令牌才能访问（dsh 0.1.5+ 的新认证机制）。',
+        detail:
+          '该服务不是本桌面壳启动的，因此拿不到令牌。\n\n' +
+          '解决办法：托盘右键（或 Alt → 文件）→「重启 dsh 服务」，选择「接管并重启」，' +
+          '由桌面壳结束该进程并重新启动 dsh 即可。\n\n' +
+          '也可以完全退出 DeepSeek Harness Desktop 并结束已有 dsh web 进程后，重新打开桌面壳。',
+        buttons: ['好'],
+        noLink: true,
+      }).catch(() => {})
+    } catch {
+      /* 忽略检测失败 */
+    }
+  })
 }
 
 /** 显示连接页。mode: first=首次使用 / offline=连不上 / switch=从应用主动换服务器 */
@@ -1297,6 +1444,7 @@ async function startLocalMode({ persist = true, fromStartup = false } = {}) {
   let external = await probe(port)
   if (!external) {
     // 2) 没有 → 自己启动，等就绪
+    resetLaunchUrl()
     ownedDsh = startDsh(port)
     const ready = await waitForServer(port, START_TIMEOUT_MS)
     if (!ready) {
@@ -1318,6 +1466,9 @@ async function startLocalMode({ persist = true, fromStartup = false } = {}) {
     log('reusing external dsh service on port ' + port)
   }
 
+  // 2.5) 本壳启动的 dsh：等带 token 的 URL 出现（dsh 0.1.5+ 必需）
+  if (!external) await waitForLaunchUrl()
+
   // 3) 就绪：首次创建窗口（连接页触发时窗口已存在，直接进入）
   if (!mainWindow || mainWindow.isDestroyed()) {
     createWindow({ mode: 'app' })