@cocorograph/hub-agent 0.6.79 → 0.6.82

package/package.json

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

package/src/config.mjs

@@ -7,10 +7,34 @@
  * られるように)。`$HUB_AGENT_CONFIG_DIR` を export しておくと、その値が
  * 優先される (mosh / 専用ユーザー用)。
  */
-import { promises as fs, constants as fsConsts } from "node:fs"
+import { promises as fs, constants as fsConsts, readFileSync } from "node:fs"
 import path from "node:path"
 import os from "node:os"
 
+/**
+ * 実行プラットフォームを返す。Hub の UI が OS 別コマンドを出し分けるために使う。
+ *
+ * 重要: Windows のエージェントは WSL2 内で動くため `process.platform` は "linux" を
+ * 返す。素の Linux と区別するため、WSL を検出して "wsl" を返す。
+ *
+ * @returns {"darwin"|"wsl"|"linux"|"win32"|string}
+ */
+export function detectPlatform() {
+  if (process.platform === "darwin") return "darwin"
+  if (process.platform === "win32") return "win32"
+  if (process.platform === "linux") {
+    if (process.env.WSL_DISTRO_NAME || process.env.WSL_INTEROP) return "wsl"
+    try {
+      const v = readFileSync("/proc/version", "utf-8").toLowerCase()
+      if (v.includes("microsoft") || v.includes("wsl")) return "wsl"
+    } catch {
+      /* /proc/version が読めない環境は素の linux 扱い */
+    }
+    return "linux"
+  }
+  return process.platform || "unknown"
+}
+
 function resolveConfigDir() {
   if (process.env.HUB_AGENT_CONFIG_DIR) return process.env.HUB_AGENT_CONFIG_DIR
   return path.join(os.homedir(), ".hub")

package/src/enroll.mjs

@@ -8,7 +8,7 @@
  */
 import os from "node:os"
 
-import { hasConfig, writeConfig } from "./config.mjs"
+import { detectPlatform, hasConfig, writeConfig } from "./config.mjs"
 
 const DEFAULT_HUB_URL = process.env.HUB_URL || "https://hub.cocorograph.com"
 
@@ -41,6 +41,7 @@ export async function enroll(enrollmentToken, opts = {}) {
       enrollment_token: enrollmentToken,
       hostname,
       version,
+      platform: detectPlatform(),
     }),
   })
 

package/src/main.mjs

@@ -19,12 +19,13 @@ import path from "node:path"
 
 import pino from "pino"
 
-import { readConfig, writeConfig } from "./config.mjs"
+import { detectPlatform, readConfig, writeConfig } from "./config.mjs"
 import { loadPlugins, runHookBroadcast, runHookChain } from "./plugin-loader.mjs"
 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,
@@ -404,6 +405,7 @@ export async function startDaemon({ version, ptyModule, claudeSdk } = {}) {
   const client = new WsClient(config, {
     logger,
     version,
+    platform: detectPlatform(),
     bundleVersion,
     bundleVersionProvider: readBundleVersionSync,
     bundleManifestPath: BUNDLE_MANIFEST_PATH,
@@ -656,6 +658,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 +1140,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,
 }

package/src/ws-client.mjs

@@ -84,6 +84,7 @@ export class WsClient extends EventEmitter {
     this.bundleWatcher = null
     this.bundleWatchDebounceTimer = null
     this.hostname = opts.hostname || os.hostname()
+    this.platform = opts.platform || "unknown"
     this.ws = null
     this.heartbeatTimer = null
     this.reconnectTimer = null
@@ -190,6 +191,7 @@ export class WsClient extends EventEmitter {
       hostname: this.hostname,
       version: this.version,
       bundle_version: this.bundleVersion,
+      platform: this.platform,
     })
     // Plan ε: 毎回 reconnect 直後に backend ↔ agent の stream_id を能動同期する
     // (orphan stream の即時 kill 用)。response は main.mjs の dispatch が拾い、