deskbuddy 0.2.0-beta.2 → 0.2.0-beta.4

package/hook/hook.cjs

@@ -26,7 +26,7 @@ const MAX_SESSIONS = 20
  * 快照整包带的能力声明（前置 spec 2026-09-29 §4.4，服务端写 deskbuddy:caps:{clientId}）。
  * 与 src/core/snapshot.ts 的 CAPS 一致
  */
-const CAPS = ['cmd_result', 'window_inject']
+const CAPS = ['cmd_result', 'window_inject', 'resume_ended']
 /** 状态文件里保留的最近事件条数，与 src/core/state.ts 的 RECENT_MAX 一致 */
 const RECENT_MAX = 8
 const REPORT_TIMEOUT_MS = 2000
@@ -556,6 +556,40 @@ function writeState(s) {
   fs.renameSync(tmp, statePath(s.sessionId))
 }
 
+// ---------- 已结束记录（与 src/core/ended.ts 同构，续跑方案 spec 2026-09-29 §5.1） ----------
+
+const ENDED_TTL_MS = 7 * 24 * 60 * 60 * 1000
+
+function writeEnded(s, now) {
+  const sid = String((s && s.sessionId) || '')
+  if (!/^[A-Za-z0-9_-]{1,64}$/.test(sid)) return
+  try {
+    const dir = path.join(deskbuddyHome(), 'ended')
+    fs.mkdirSync(dir, { recursive: true, mode: 0o700 })
+    const t = now || Date.now()
+    const rec = { sessionId: sid, tool: s.tool || '', cwd: s.cwd || '', interactive: s.interactive === true, endedAt: t }
+    const file = path.join(dir, sid + '.json')
+    fs.writeFileSync(file + '.tmp', JSON.stringify(rec), { encoding: 'utf8', mode: 0o600 })
+    fs.renameSync(file + '.tmp', file)
+    for (const n of fs.readdirSync(dir)) {
+      const f = path.join(dir, n)
+      try {
+        if (t - fs.statSync(f).mtimeMs > ENDED_TTL_MS) fs.rmSync(f, { force: true })
+      } catch (e) {
+        /* ignore */
+      }
+    }
+  } catch (e) {
+    /* 记不下：只是这个会话结束后不能从设备续跑 */
+  }
+}
+
+/** 会话结束：先写已结束记录，再删状态（删状态的地方都走这里，/clear 除外） */
+function retireState(s, now) {
+  writeEnded(s, now)
+  removeState(s.sessionId)
+}
+
 function removeState(sessionId) {
   try {
     fs.rmSync(statePath(sessionId), { force: true })
@@ -591,7 +625,7 @@ function readAliveStates(tool) {
     }
     if (!s || s.tool !== tool) continue
     if (s.pid && !isAlive(s.pid, s.pidName)) {
-      removeState(s.sessionId) // 窗口已关：剔除并自我清理，state 目录不再无限增长
+      retireState(s) // 窗口已关：记下已结束、剔除并自我清理，state 目录不再无限增长
       continue
     }
     alive.push(s)
@@ -764,8 +798,35 @@ async function reportIfAuthorized(tool, nowMs) {
   // clientId 为空说明还没 setup 过（服务端靠 clientId ↔ userId 绑定关系认人）：
   // 绝不能在这里现编一个，那等于凭空造出一台没人认领的机器
   if (!cfg.clientId || cfg.enabledTools.indexOf(tool) === -1) return
+  // Codex：以 daemon 连着的 app-server 清单判 remoteWindow（切片 B）；Claude Code：状态文件 + 窗口唤醒标记
   const remote = tool === 'codex' ? readCodexRemote() || new Set() : null
-  await postReport(cfg, buildSnapshot(readAliveStates(tool), tool, cfg, nowMs, remote))
+  await postReport(cfg, buildSnapshot(withWake(readAliveStates(tool)), tool, cfg, nowMs, remote))
+}
+
+// ---------- 窗口唤醒（asyncRewake，与 src/core/wake.ts 同构） ----------
+
+/**
+ * daemon 写的 `wake/marks/<sessionId>.json`：这个会话开着的窗口里有等待者，设备指令能直接唤醒窗口执行。
+ * 标记里的智能体进程还在才算（kill 0，不查进程名：每次上报都要调）
+ */
+function wakeMarked(sessionId) {
+  const safe = String(sessionId).replace(/[^A-Za-z0-9_-]/g, '')
+  if (!safe) return false
+  try {
+    const m = JSON.parse(fs.readFileSync(path.join(deskbuddyHome(), 'wake', 'marks', safe + '.json'), 'utf8'))
+    // 写标记的 daemon 死了就不算（daemon 退出来不及清标记时，不能一直误报 remoteWindow）
+    const daemonPid = Number(m && m.daemonPid) || 0
+    if (!daemonPid || !pidExists(daemonPid)) return false
+    const pid = Number(m && m.agentPid) || 0
+    return pid > 0 ? pidExists(pid) : true
+  } catch (e) {
+    return false
+  }
+}
+
+/** 上报前叠加唤醒标记：能被唤醒的会话带 remoteWindow=true（服务端据此放行 8348） */
+function withWake(states) {
+  return states.map((s) => (s.remoteWindow === true || !wakeMarked(s.sessionId) ? s : Object.assign({}, s, { remoteWindow: true })))
 }
 
 /**
@@ -1063,8 +1124,13 @@ async function main(argv) {
     if (event === 'SessionEnd' && payload.reason === 'clear') {
       const old = readState(sessionId)
       if (old && old.pid) markClear(old.pid, sessionId)
+      // /clear 引起的结束交给别名改绑，不写已结束记录
+      removeState(sessionId)
+    } else {
+      // 先记下已结束（设备再回复时续跑要知道 cwd），再删状态。状态文件没有就用这次事件里的 cwd
+      const old = readState(sessionId)
+      retireState(old || { sessionId: sessionId, tool: tool, cwd: payload.cwd || '', interactive: false })
     }
-    removeState(sessionId)
     // 删完补发一次剩余会话，否则这条会一直挂在屏上等服务端失联 TTL。
     // 关掉最后一个窗口时发的是空列表——服务端放行空快照，该机器该工具随即从屏上消失
     await reportIfAuthorized(tool, Date.now())
@@ -1165,12 +1231,15 @@ module.exports = {
   main,
   mapEvent,
   processName,
+  parentPid,
   readAiTitle,
   readAliveStates,
   readCodexThreadName,
   readGitBranch,
   readState,
   removeState,
+  retireState,
+  writeEnded,
   resolveAgentPid,
   resolveTitle,
   takeNotice,
@@ -1192,4 +1261,8 @@ module.exports = {
   noteWindow,
   readWindow,
   windowOwns,
+  wakeMarked,
+  withWake,
+  detectInteractive,
+  pidExists,
 }