@cocorograph/hub-agent 0.6.79 → 0.6.81

package/package.json

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

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,
 }