@cocorograph/hub-agent 0.6.58 → 0.6.60

package/package.json

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

package/src/main.mjs

@@ -10,7 +10,8 @@
  *
  * 仕様書: ナレッジ/インフラ/cockpit-hub-hosted-integration-spec (id=6080)
  */
-import { readFileSync, watch as fsWatch } from "node:fs"
+import { spawn as _childSpawn } from "node:child_process"
+import { existsSync, readFileSync, watch as fsWatch } from "node:fs"
 import { mkdir, readFile, readdir, rename, unlink, writeFile } from "node:fs/promises"
 import { randomUUID } from "node:crypto"
 import os from "node:os"
@@ -68,6 +69,7 @@ import {
   removeWorktree as removeWorktreeDir,
   resumeWithMessage,
   setTmuxGlobalEnv,
+  setTuiModel,
 } from "./tmux.mjs"
 import { TuiPermissionBridge } from "./tui-permission-bridge.mjs"
 import { TuiViewerRegistry } from "./tui-viewer-registry.mjs"
@@ -293,6 +295,57 @@ export async function syncTmuxProfileEnv(profile, log) {
   )
 }
 
+/**
+ * `setup_hub_ai.py --silent` を best-effort でキックし、全プロファイルの settings.json へ
+ * bundle hooks (cockpit_permission_bridge 等) を反映させる。
+ *
+ * 新規プロファイル追加 (profile.add) 直後に呼ぶ。provisionProfile は
+ * `~/.claude/settings.json` を新プロファイルへコピーするが、コピー元が古い / setup 未走の
+ * タイミングだと新プロファイルにフックが欠けうる。setup_hub_ai は全プロファイル対応 (案1)
+ * なので 1 回叩けば新プロファイルの初回セッションでもフックが揃う。
+ *
+ * 失敗・未 setup 環境 (スクリプト不在 / python3 無し) でも **例外を投げない**。detached +
+ * stdio ignore で daemon をブロックしない。session_start.py の auto-update と同じ
+ * `--silent` で headless 実行する。
+ *
+ * @param {{logger?: object, spawnImpl?: Function, existsImpl?: Function, homedir?: string}} [opts]
+ *   spawnImpl / existsImpl / homedir はテスト用の注入口 (既定は node 標準)。
+ * @returns {boolean} spawn を試みたら true、スクリプト不在等でスキップしたら false。
+ */
+export function kickSetupHubAi({
+  logger,
+  spawnImpl,
+  existsImpl,
+  homedir,
+} = {}) {
+  try {
+    const home = homedir || os.homedir()
+    const setupScript = path.join(home, ".claude", "scripts", "setup_hub_ai.py")
+    const exists = existsImpl || existsSync
+    if (!exists(setupScript)) {
+      logger?.info?.(
+        { setupScript },
+        "kickSetupHubAi: setup_hub_ai.py 不在 - skip",
+      )
+      return false
+    }
+    const doSpawn = spawnImpl || _childSpawn
+    const child = doSpawn("python3", [setupScript, "--silent"], {
+      detached: true,
+      stdio: "ignore",
+    })
+    child?.on?.("error", (e) =>
+      logger?.warn?.({ err: e?.message }, "kickSetupHubAi: spawn error"),
+    )
+    child?.unref?.()
+    logger?.info?.({}, "kickSetupHubAi: setup_hub_ai --silent を起動 (hooks 同期)")
+    return true
+  } catch (e) {
+    logger?.warn?.({ err: e?.message }, "kickSetupHubAi: 失敗 (無視)")
+    return false
+  }
+}
+
 export async function startDaemon({ version, ptyModule, claudeSdk } = {}) {
   const config = await readConfig()
   if (!config) {
@@ -1331,6 +1384,41 @@ async function dispatch(msg, ctx) {
         })()
         return
       }
+      case "claude.tui.setModel": {
+        // モデルバッジ選択 → 対話 claude TUI へ `/model <id>` を送ってモデルを切り替える。
+        // 権限循環 (cyclePermission) と同設計: agent が実キーを送出 → 確認プロンプトを確定 →
+        // 全ブラウザへ claude.tui.model を broadcast し、実際に動いているターミナルのモデルを
+        // 正本として全端末に同期する。jsonl の message.model は次の assistant ターンまで更新
+        // されないため、この即時 broadcast で切替直後のバッジズレを解消する。
+        const cwd = typeof msg.cwd === "string" ? msg.cwd : ""
+        const sessionName =
+          typeof msg.session_name === "string" ? msg.session_name : ""
+        if (!sessionName) return
+        // model="" は「デフォルト」= `/model default`。frontend へはそのまま空で返し、
+        // 解決後の実 id は次ターンの jsonl 由来 (message.model) に委ねる。
+        const model = typeof msg.model === "string" ? msg.model : ""
+        ;(async () => {
+          try {
+            await setTuiModel(sessionName, model || "default", { logger })
+            ctx.client.send({
+              type: "claude.tui.model",
+              cwd: cwd || undefined,
+              session_name: sessionName,
+              model,
+            })
+            logger.info(
+              { session: sessionName, cwd, model: model || "(default)" },
+              "tui model switched → notified browser",
+            )
+          } catch (err) {
+            logger.warn(
+              { err: err?.message, session: sessionName },
+              "claude.tui.setModel failed",
+            )
+          }
+        })()
+        return
+      }
       case "claude.tui.probePermission": {
         // 読み取り専用の権限モード問い合わせ (cold-load seed)。キーは送らず、ペインを
         // 読んで現在の実モードを claude.tui.permission として broadcast するだけ。
@@ -1795,6 +1883,10 @@ async function dispatch(msg, ctx) {
             needs_login: created.needsLogin,
             login_command: `CLAUDE_CONFIG_DIR=${created.configDir} claude`,
           })
+          // 新規プロファイルへ bundle hooks (cockpit_permission_bridge 等) を即時反映する
+          // best-effort キック (詳細は kickSetupHubAi の docstring 参照)。失敗・未 setup
+          // 環境でも profile.add は成功扱い (kickSetupHubAi は例外を投げない)。
+          kickSetupHubAi({ logger: ctx.logger })
         } catch (err) {
           ctx.client.send({
             type: "profile.add.result",

package/src/tmux.mjs

@@ -865,6 +865,62 @@ export async function cyclePermissionMode(name, opts = {}) {
   }
 }
 
+/**
+ * 対話 claude TUI に `/model <id>` を送ってモデルを切り替える。
+ *
+ * cockpit のモデルバッジ選択 (claude.tui.setModel) の書込側。権限循環 (cyclePermissionMode)
+ * と同じく、frontend は raw pty.data ではなく claude.tui.setModel を送り、agent 側で本関数を
+ * 実行 → 全ブラウザへ claude.tui.model を broadcast して全端末を実モデルに揃える。
+ *
+ * `/model` はフルモデルID をそのまま受理する (起動時 `claude --model <id>` と同仕様)。引数
+ * `default` は起動時の既定へ戻す。会話に出力済みの履歴があるとモデル切替時に確認プロンプトが
+ * 出るため、Enter を送って本文確定 → 少し待って追い Enter で確認を畳む (素の入力欄では空 Enter
+ * となり無害)。copy-mode に入っているとキーが奪われるので先に抜ける。ベストエフォート。
+ *
+ * @param {string} name tmux セッション名
+ * @param {string} modelArg `/model` 引数 (フルモデルID または "default")
+ * @param {{logger?:object,tmuxBin?:string}} [opts]
+ * @returns {Promise<{ok:boolean, error?:string}>}
+ */
+export async function setTuiModel(name, modelArg, opts = {}) {
+  const bin = tmuxBin(opts)
+  const arg = String(modelArg || "default").replace(/[\r\n]+/g, " ").trim()
+  if (!arg) return { ok: false, error: "empty model arg" }
+  try {
+    // copy-mode 等に入っているとキーが奪われるので、入っている時だけ抜ける。
+    try {
+      const { stdout } = await execFileP(bin, [
+        "display-message",
+        "-p",
+        "-t",
+        `${name}:`,
+        "-F",
+        "#{pane_in_mode}",
+      ])
+      if (stdout.trim() === "1") {
+        await execFileP(bin, ["send-keys", "-t", name, "-X", "cancel"])
+      }
+    } catch {
+      // pane_in_mode 取得失敗はベストエフォートで無視。
+    }
+    // `/model <id>` をリテラルで送る (-l でキー名解釈・スラッシュ補完の暴発を避ける)。
+    await execFileP(bin, ["send-keys", "-t", name, "-l", `/model ${arg}`])
+    await _delay(120)
+    // Enter で本文確定 (send-keys の離散イベントなので paste 吸収は起きにくい)。
+    await execFileP(bin, ["send-keys", "-t", name, "Enter"])
+    // prior output ありの場合に出るモデル切替の確認プロンプトを Enter で畳む。
+    await _delay(450)
+    await execFileP(bin, ["send-keys", "-t", name, "Enter"])
+    return { ok: true }
+  } catch (err) {
+    opts.logger?.warn(
+      { session: name, model: arg, err: err?.message },
+      "setTuiModel failed",
+    )
+    return { ok: false, error: err?.message || String(err) }
+  }
+}
+
 /**
  * 中断キャンセル後の入力欄復旧 (claude.tui.recoverInput / agent >= 0.6.57)。
  *