@cocorograph/hub-agent 0.6.78 → 0.6.81

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cocorograph/hub-agent",
-  "version": "0.6.78",
+  "version": "0.6.81",
   "description": "Hub Hosted Cockpit のローカル常駐 agent。Hub と outbound WSS で接続し、ローカルの tmux/pty を中継する。",
   "type": "module",
   "license": "UNLICENSED",

package/scripts/install.ps1

@@ -237,15 +237,33 @@ function Set-Systemd($distro) {
   }
 }
 
-# Register a logon task that wakes the distro (its systemd starts hub-agent).
+# Register a logon task that keeps the distro ALIVE (so its systemd --user
+# hub-agent stays online).
+#
+# 旧実装は `wsl ... /bin/true` で distro を一瞬起こすだけだった。これだと WSL2 の
+# アイドルシャットダウンが数十秒で distro ごと畳んでしまい (agent.log に
+# `SIGTERM shutting down` が出て `wsl -l -v` が Stopped になる)、hub-agent も停止して
+# オフラインに戻る。systemd --user + linger は「distro が起動している間」しか効かない
+# ため、distro 自体を起こし続ける常駐プロセスが必要。
+#
+# 対策: ログオン中ずっと `sleep infinity` を WSL 内で走らせて distro を起動状態に保つ。
+# 併せて hub-agent.service を start しておく (enable + linger 済みなら冪等)。
 function Register-BootTask($distro, $user) {
-  Write-Step "Registering autostart on logon (Task Scheduler)"
-  $action  = New-ScheduledTaskAction -Execute "wsl.exe" -Argument "-d $distro -u $user -- /bin/true"
+  Write-Step "Registering autostart on logon (Task Scheduler, keep-alive)"
+  # distro を起動し続ける keep-alive。systemctl start は冪等 (既に active なら no-op)。
+  $keepalive = "systemctl --user start hub-agent.service 2>/dev/null; exec sleep infinity"
+  $action  = New-ScheduledTaskAction -Execute "wsl.exe" `
+    -Argument "-d $distro -u $user -- bash -lc `"$keepalive`""
   $trigger = New-ScheduledTaskTrigger -AtLogOn
-  $set     = New-ScheduledTaskSettingsSet -AllowStartIfOnBatteries -DontStopIfGoingOnBatteries -StartWhenAvailable
+  # ExecutionTimeLimit=0 で無制限 (既定 72h で停止されると distro が落ちる)。
+  # keep-alive が万一終了したら 1 分後に最大 3 回まで再起動。多重起動は抑止。
+  $set     = New-ScheduledTaskSettingsSet -AllowStartIfOnBatteries -DontStopIfGoingOnBatteries `
+    -StartWhenAvailable -MultipleInstances IgnoreNew `
+    -RestartInterval (New-TimeSpan -Minutes 1) -RestartCount 3 `
+    -ExecutionTimeLimit (New-TimeSpan -Seconds 0)
   Register-ScheduledTask -TaskName $TASK_NAME -Action $action -Trigger $trigger -Settings $set `
     -RunLevel Limited -Force | Out-Null
-  Write-Ok "Logon task '$TASK_NAME' registered"
+  Write-Ok "Logon task '$TASK_NAME' registered (keeps WSL distro alive)"
 }
 
 # Run the existing install.sh inside WSL.
@@ -315,6 +333,16 @@ function Main {
   # --- autostart ---
   Register-BootTask $distro $user
 
+  # セットアップ完了時点でオンライン化まで完結させる: keep-alive タスクを今すぐ起動し、
+  # distro を起動状態に保つ (再ログオンを待たずに hub-agent が常駐する)。
+  Write-Step "Starting keep-alive now (online without re-logon)"
+  try {
+    Start-ScheduledTask -TaskName $TASK_NAME
+    Write-Ok "Keep-alive started; hub-agent should come online shortly"
+  } catch {
+    Write-Warn "Start-ScheduledTask failed ($($_.Exception.Message)); ログオンし直すと起動します"
+  }
+
   Write-Host ""
   Write-Ok "Setup complete. Check the online status in the Hub UI:"
   Write-Host "    https://hub.cocorograph.com/user/cockpit/agents" -ForegroundColor Cyan

package/src/main.mjs

@@ -25,6 +25,7 @@ import { WsClient } from "./ws-client.mjs"
 import { PtyBridge } from "./pty-bridge.mjs"
 import { ClaudeStreamBridge } from "./claude-stream-bridge.mjs"
 import { UploadManager } from "./claude-upload.mjs"
+import { requestSelfUninstall } from "./service-install.mjs"
 import {
   decideSessionRotation,
   fetchSessionHistory,
@@ -656,6 +657,8 @@ export async function startDaemon({ version, ptyModule, claudeSdk } = {}) {
   }
   process.on("SIGINT", () => shutdown("SIGINT"))
   process.on("SIGTERM", () => shutdown("SIGTERM"))
+  // dispatch から呼べるように shutdown を ctx に載せる (agent.shutdown 用)。
+  ctx.requestShutdown = shutdown
 
   // 0.6.2 fix: 例外で silent 終了しないよう最後のセーフティネット。
   // Node 24 で unhandledRejection は default で process を kill する仕様のため、
@@ -1136,6 +1139,21 @@ async function dispatch(msg, ctx) {
       case "error":
         ctx.logger.warn({ msg }, "hub error")
         return
+      case "agent.shutdown": {
+        // Cockpit の「エージェント削除 (ゴミ箱)」由来。この端末をエージェントとして
+        // 撤去する: サービス停止・無効化・autostart 撤去のみ。ユーザーデータ
+        // (~/.hub / ~/.claude / プロジェクト) には触れない。
+        ctx.logger.info({ msg }, "agent.shutdown received — self-uninstalling")
+        try {
+          // cgroup の外で動く detached プロセスにクリーンアップを委譲してから exit。
+          requestSelfUninstall()
+        } catch (err) {
+          ctx.logger.warn({ err: err?.message }, "requestSelfUninstall failed")
+        }
+        // 本体は即停止 (detached cleanup が disable/stop を完走させる)。
+        await ctx.requestShutdown?.("agent.shutdown")
+        return
+      }
       case "pty.attach": {
         const stream_id = msg.stream_id
         try {

package/src/service-install.mjs

@@ -19,6 +19,9 @@ import { readConfig } from "./config.mjs"
 
 const PLIST_LABEL = "co.cocorograph.hub-agent"
 const SYSTEMD_UNIT_NAME = "hub-agent.service"
+// Windows(WSL) のログオン keep-alive タスク名 (install.ps1 の $TASK_NAME と一致)。
+// WSL からは Windows 連携 (interop) で schtasks.exe を叩いて削除できる。
+const WSL_BOOT_TASK_NAME = "HubAgentWSLBoot"
 
 function repoTemplatesDir() {
   // src/service-install.mjs → ../templates
@@ -323,7 +326,14 @@ export async function installService({ bin } = {}) {
     // 無い間 (= WSL2 のログオンタスク経由の boot や headless) に常駐しない。
     const linger = enableLinger()
     run("systemctl", ["--user", "daemon-reload"])
-    run("systemctl", ["--user", "enable", "--now", SYSTEMD_UNIT_NAME])
+    run("systemctl", ["--user", "enable", SYSTEMD_UNIT_NAME])
+    // `enable --now` は既に起動中のサービスを再起動しない (--now は停止中なら start、
+    // 起動中なら no-op)。distro を起動し続ける keep-alive と組み合わさると、再 enroll で
+    // agent.json のトークンを書き換えても古いプロセスが旧トークンのまま生き続け、
+    // (同一ホスト重複解消で旧 agent 行が消えるため) WS 403 でオフラインになる。
+    // restart は停止中なら起動・起動中なら再起動するので、install-service のたびに
+    // 必ず最新の agent.json でプロセスを入れ替える。
+    run("systemctl", ["--user", "restart", SYSTEMD_UNIT_NAME])
     return {
       platform: "linux",
       path: dest,
@@ -409,6 +419,75 @@ export async function uninstallService() {
   throw new Error(`unsupported platform: ${process.platform}`)
 }
 
+/**
+ * Cockpit の「エージェント削除 (ゴミ箱)」を受けて、この端末を **エージェントとして
+ * 撤去** する。サービス停止・無効化・autostart 撤去のみを行い、ユーザーデータ
+ * (`~/.hub` / `~/.claude` / プロジェクト) には一切触れない。
+ *
+ * 重要: hub-agent.service は `Restart=always`。自プロセスから `systemctl --user
+ * stop` すると自分の cgroup ごと殺され、後続のクリーンアップが完走しない。そこで
+ * **サービスの cgroup の外** で動く detached プロセスにクリーンアップを委譲し、
+ * 本体はすぐ exit する (呼び出し側が exit する)。
+ *
+ * 撤去内容:
+ *   - systemd user unit: disable --now + unit ファイル削除 + daemon-reload
+ *   - linger: loginctl disable-linger
+ *   - Windows(WSL): schtasks.exe で keep-alive タスク削除 (interop。非WSLでは no-op)
+ *
+ * @param {{spawnSync?: typeof spawnSync}} [opts] テスト用に spawn を注入可能。
+ * @returns {boolean} クリーンアップ用 detached プロセスの起動に成功したら true。
+ */
+export function requestSelfUninstall(opts = {}) {
+  const spawn = opts.spawnSync ?? spawnSync
+
+  if (process.platform === "linux") {
+    const user = currentUsername()
+    const unit = linuxUnitPath()
+    // 1 秒待ってから (本体 exit を待つ) 撤去。各手順は失敗しても続行 (|| true)。
+    const script = [
+      "sleep 1",
+      `systemctl --user disable --now ${SYSTEMD_UNIT_NAME} 2>/dev/null || true`,
+      `rm -f '${unit}' 2>/dev/null || true`,
+      "systemctl --user daemon-reload 2>/dev/null || true",
+      `loginctl disable-linger ${user} 2>/dev/null || true`,
+      // WSL のみ有効 (Windows interop)。schtasks.exe が無い環境では黙って失敗。
+      `schtasks.exe /delete /tn ${WSL_BOOT_TASK_NAME} /f >/dev/null 2>&1 || true`,
+    ].join("; ")
+
+    // systemd-run --user --scope で別 scope(=別 cgroup) に逃がす。これにより
+    // hub-agent.service を stop しても掃除プロセスは生き残って完走する。
+    const r = spawn(
+      "systemd-run",
+      ["--user", "--scope", "--collect", "--quiet", "bash", "-lc", script],
+      { stdio: "ignore", detached: true },
+    )
+    if (!r.error && r.status === 0) return true
+    // systemd-run が無い/失敗時は setsid で cgroup から外して代替。
+    const r2 = spawn("setsid", ["bash", "-lc", script], {
+      stdio: "ignore",
+      detached: true,
+    })
+    return !r2.error
+  }
+
+  if (process.platform === "darwin") {
+    const uid = ensureUnixUid()
+    const dest = macPlistPath()
+    // launchctl bootout は job の全プロセスを殺すため、setsid で job から切り離した
+    // detached プロセスに委譲する。
+    const script =
+      `sleep 1; launchctl bootout gui/${uid} '${dest}' 2>/dev/null || true; ` +
+      `rm -f '${dest}' 2>/dev/null || true`
+    const r = spawn("bash", ["-lc", `setsid bash -lc "${script}" >/dev/null 2>&1 &`], {
+      stdio: "ignore",
+      detached: true,
+    })
+    return !r.error
+  }
+
+  return false
+}
+
 export const _internal = {
   expandTemplate,
   detectHubAgentBin,
@@ -420,4 +499,5 @@ export const _internal = {
   bootstrapWithRetry,
   currentUsername,
   enableLinger,
+  requestSelfUninstall,
 }