@cocorograph/hub-agent 0.5.4 → 0.5.6

package/package.json

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

package/src/agents.mjs

@@ -0,0 +1,79 @@
+/**
+ * Claude Code の subagent (`~/.claude/agents/<category>/<name>.md`) を集計する。
+ *
+ * - 各 .md の先頭 H1 から title / description を抽出 ("Title — description")
+ * - category=ディレクトリ名、name=basename
+ *
+ * 移植元: D00000_cockpit/webapp/lib/agents.ts
+ */
+import { promises as fs } from "node:fs"
+import os from "node:os"
+import path from "node:path"
+
+const AGENTS_DIR = process.env.CLAUDE_AGENTS_DIR || path.join(os.homedir(), ".claude", "agents")
+
+function parseH1(text) {
+  for (const line of text.split("\n")) {
+    const m = line.match(/^#\s+(.+)$/)
+    if (!m) continue
+    const heading = m[1].trim()
+    const dashIdx = heading.search(/\s*[-—–]\s*/)
+    if (dashIdx > 0) {
+      const title = heading.slice(0, dashIdx).trim()
+      const description = heading.slice(dashIdx).replace(/^[\s\-—–]+/, "").trim()
+      return { title, description }
+    }
+    return { title: heading, description: "" }
+  }
+  return { title: "", description: "" }
+}
+
+export async function listAgents() {
+  const agents = []
+  let categories
+  try {
+    categories = await fs.readdir(AGENTS_DIR)
+  } catch {
+    return []
+  }
+  for (const cat of categories) {
+    if (cat.startsWith(".") || cat.startsWith("_")) continue
+    const catPath = path.join(AGENTS_DIR, cat)
+    let st
+    try {
+      st = await fs.stat(catPath)
+    } catch {
+      continue
+    }
+    if (!st.isDirectory()) continue
+    let files
+    try {
+      files = await fs.readdir(catPath)
+    } catch {
+      continue
+    }
+    for (const f of files) {
+      if (!f.endsWith(".md")) continue
+      if (f.startsWith("_")) continue
+      const fp = path.join(catPath, f)
+      try {
+        const text = await fs.readFile(fp, "utf-8")
+        const { title, description } = parseH1(text)
+        const baseName = f.replace(/\.md$/, "")
+        agents.push({
+          name: baseName,
+          category: cat,
+          title: title || baseName,
+          description: description || "",
+          path: `${cat}/${baseName}`,
+        })
+      } catch {
+        /* skip */
+      }
+    }
+  }
+  agents.sort(
+    (a, b) => a.category.localeCompare(b.category) || a.name.localeCompare(b.name),
+  )
+  return agents
+}

package/src/main.mjs

@@ -20,10 +20,12 @@ import { readConfig } from "./config.mjs"
 import { loadPlugins, runHookBroadcast, runHookChain } from "./plugin-loader.mjs"
 import { WsClient } from "./ws-client.mjs"
 import { PtyBridge } from "./pty-bridge.mjs"
+import { listAgents } from "./agents.mjs"
 import { listSkills } from "./skills.mjs"
 import { listSessionStates } from "./state.mjs"
 import {
   createSession as createTmuxSession,
+  createWorktreeDir,
   execTmux,
   killManySessions,
   killSession as killTmuxSession,
@@ -313,6 +315,53 @@ async function dispatch(msg, ctx) {
         }
         return
       }
+      case "worktree.create": {
+        // body: { request_id, dir, branch, claude_cmd? }
+        const dir = (msg.dir || "").trim()
+        const branch = (msg.branch || "").trim()
+        if (!dir || !branch) {
+          ctx.client.send({
+            type: "worktree.create.result",
+            request_id: msg.request_id,
+            ok: false,
+            error: "dir and branch required",
+          })
+          return
+        }
+        if (!/^[A-Za-z0-9._\-/]+$/.test(branch) || branch.startsWith("-")) {
+          ctx.client.send({
+            type: "worktree.create.result",
+            request_id: msg.request_id,
+            ok: false,
+            error: "invalid branch name",
+          })
+          return
+        }
+        try {
+          const { wtName, wtPath, createdBranch } = await createWorktreeDir(dir, branch)
+          // worktree dir で tmux session を作成 (claude_cmd は createSession の default)
+          await createTmuxSession(wtName, wtPath, {
+            claudeCmd: typeof msg.claude_cmd === "string" ? msg.claude_cmd : undefined,
+          })
+          ctx.client.send({
+            type: "worktree.create.result",
+            request_id: msg.request_id,
+            ok: true,
+            wt_name: wtName,
+            wt_path: wtPath,
+            branch,
+            created_branch: createdBranch,
+          })
+        } catch (err) {
+          ctx.client.send({
+            type: "worktree.create.result",
+            request_id: msg.request_id,
+            ok: false,
+            error: err.message,
+          })
+        }
+        return
+      }
       case "tmux.kill_session": {
         const names = Array.isArray(msg.session_names)
           ? msg.session_names
@@ -357,6 +406,24 @@ async function dispatch(msg, ctx) {
         }
         return
       }
+      case "agents.request": {
+        try {
+          const agents = await listAgents()
+          ctx.client.send({
+            type: "agents.response",
+            request_id: msg.request_id,
+            agents,
+          })
+        } catch (err) {
+          ctx.client.send({
+            type: "agents.response",
+            request_id: msg.request_id,
+            agents: [],
+            error: err.message,
+          })
+        }
+        return
+      }
       case "usage.request": {
         try {
           const [usage, sessions] = await Promise.all([

package/src/tmux.mjs

@@ -33,6 +33,72 @@ function sanitizeTmuxName(s) {
   return s.replace(/[.:]/g, "-")
 }
 
+/**
+ * git branch 名から worktree dir 名 (= tmux session 名) を導出する。
+ * 例: "feature/multi-area" -> "feature-multi-area"
+ *     "fix.typo:v2"        -> "fix-typo-v2"
+ *
+ * 移植元: D00000_cockpit/webapp/lib/worktrees.ts (sanitizeBranchToWtName)
+ */
+export function sanitizeBranchToWtName(branch) {
+  return branch
+    .replace(/[\s/:.]+/g, "-")
+    .replace(/[^A-Za-z0-9_-]/g, "")
+    .replace(/^-+|-+$/g, "")
+}
+
+async function branchExists(repoDir, branch) {
+  try {
+    await execFileP("git", [
+      "-C",
+      repoDir,
+      "show-ref",
+      "--verify",
+      "--quiet",
+      `refs/heads/${branch}`,
+    ])
+    return true
+  } catch {
+    return false
+  }
+}
+
+/**
+ * `<repoDir>/.claude/worktrees/<wtName>` に worktree を作成する。
+ * branch が既存ならそれを checkout、無ければ `-b` で新規作成。
+ *
+ * 移植元: D00000_cockpit/webapp/lib/worktrees.ts (createWorktree)
+ */
+export async function createWorktreeDir(parentDir, branch) {
+  const repoDir = path.resolve(HUB_PROJECTS_BASE, parentDir)
+  if (repoDir !== HUB_PROJECTS_BASE && !repoDir.startsWith(HUB_PROJECTS_BASE + path.sep)) {
+    throw new Error("dir outside projects base")
+  }
+  // .git の存在チェック (file or dir)
+  try {
+    await fs.stat(path.join(repoDir, ".git"))
+  } catch {
+    throw new Error("not a git repository")
+  }
+  const wtName = sanitizeBranchToWtName(branch)
+  if (!wtName) throw new Error("invalid branch name")
+  const wtPath = path.join(repoDir, ".claude", "worktrees", wtName)
+  try {
+    await fs.stat(wtPath)
+    throw new Error("worktree already exists")
+  } catch (err) {
+    if (err.message === "worktree already exists") throw err
+    // ENOENT は期待動作
+  }
+  const exists = await branchExists(repoDir, branch)
+  if (exists) {
+    await execFileP("git", ["-C", repoDir, "worktree", "add", wtPath, branch])
+  } else {
+    await execFileP("git", ["-C", repoDir, "worktree", "add", "-b", branch, wtPath])
+  }
+  return { wtName, wtPath, createdBranch: !exists }
+}
+
 /**
  * `~/hub/projects/<project>/.claude/worktrees/<wt>` を走査し、
  * worktree_session_name → parent_session_name の Map を返す。