agentmomo 0.1.1 → 0.1.2

package/dist/server.js

@@ -47464,6 +47464,7 @@ var CLAUDE_CONFIG_PATH = process.env.APPDATA ? path3.join(process.env.APPDATA, "
 var AGENTSMOMO_DIST_DIR = process.env.AGENTSMOMO_DIST_DIR;
 var WRAPPER_JS = AGENTSMOMO_DIST_DIR ? path3.join(AGENTSMOMO_DIST_DIR, "stdio-wrapper.js") : path3.join(process.cwd(), "proxy", "dist", "stdio-wrapper.js");
 var HOOK_JS = AGENTSMOMO_DIST_DIR ? path3.join(AGENTSMOMO_DIST_DIR, "claude-code-hook.cjs") : path3.join(process.cwd(), "proxy", "dist", "claude-code-hook.cjs");
+var GEMINI_HOOK_JS = AGENTSMOMO_DIST_DIR ? path3.join(AGENTSMOMO_DIST_DIR, "gemini-cli-hook.cjs") : path3.join(process.cwd(), "proxy", "dist", "gemini-cli-hook.cjs");
 function projectSettingsPath(projectPath) {
   return path3.join(projectPath, ".claude", "settings.json");
 }
@@ -47622,11 +47623,112 @@ app.post("/api/event", requireAuth, async (req, res) => {
     };
     broadcast(subEvent);
   }
+  if (event.event === "call_start" && event.source === "gemini-cli" && event.toolName === "complete_task" && event.metadata?.subAgentId && typeof event.metadata.subAgentId === "string") {
+    const subAgentId = event.metadata.subAgentId;
+    const subAgentName = event.metadata.subAgentName ?? subAgentId;
+    const folderPart = typeof event.metadata.cwd === "string" ? path3.basename(event.metadata.cwd) : "Gemini CLI";
+    const subEvent = {
+      agentId: subAgentId,
+      agentName: `${folderPart} \xB7 ${subAgentName}`,
+      source: "gemini-cli-subagent",
+      event: "agent_register",
+      toolName: "",
+      timestamp: event.timestamp,
+      requestId: `${event.requestId}-sub-register`,
+      parentId: event.agentId,
+      isSubAgent: true,
+      metadata: event.metadata
+    };
+    broadcast(subEvent);
+  }
   res.json({ ok: true });
 });
 app.get("/api/health", (_req, res) => {
   res.json({ status: "ok", agents: getAllAgents().length });
 });
+app.get("/.well-known/agent.json", (_req, res) => {
+  res.json({
+    name: "agentmomo",
+    description: "3D visualization dashboard for AI coding agents. Monitor active agents, view real-time activity, and manage the dashboard.",
+    url: `http://localhost:${PORT}`,
+    version: "1.0.0",
+    capabilities: {
+      streaming: false,
+      pushNotifications: false
+    },
+    skills: [
+      {
+        id: "list-agents",
+        name: "List Agents",
+        description: "Get all active agents and their current states (idle, working, break, error, offline)"
+      },
+      {
+        id: "get-activity",
+        name: "Get Activity",
+        description: "Get server health status and agent count"
+      },
+      {
+        id: "reset-dashboard",
+        name: "Reset Dashboard",
+        description: "Clear all agents from the visualization dashboard"
+      }
+    ],
+    defaultInputModes: ["text/plain"],
+    defaultOutputModes: ["application/json"]
+  });
+});
+app.post("/a2a/tasks/send", import_express2.default.json(), (req, res) => {
+  const { message } = req.body ?? {};
+  if (!message || !message.parts || !Array.isArray(message.parts)) {
+    res.status(400).json({ error: "Invalid A2A task message" });
+    return;
+  }
+  const textParts = message.parts.filter((p) => !p.type || p.type === "text").map((p) => p.text ?? "").join(" ").toLowerCase();
+  let result;
+  if (textParts.includes("reset") || textParts.includes("clear")) {
+    clearAllAgents();
+    broadcast({
+      agentId: "__system__",
+      agentName: "system",
+      source: "manual",
+      event: "agent_register",
+      toolName: "",
+      timestamp: Date.now(),
+      requestId: "a2a-clear"
+    });
+    result = { action: "reset", success: true, message: "Dashboard cleared" };
+  } else if (textParts.includes("list") || textParts.includes("agents") || textParts.includes("status")) {
+    const agents2 = getAllAgents();
+    result = {
+      action: "list-agents",
+      count: agents2.length,
+      agents: agents2.map((a) => ({
+        id: a.id,
+        name: a.name,
+        source: a.source,
+        state: a.state,
+        currentTool: a.currentTool,
+        totalCalls: a.totalCalls,
+        totalErrors: a.totalErrors
+      }))
+    };
+  } else {
+    result = {
+      action: "health",
+      status: "ok",
+      agents: getAllAgents().length
+    };
+  }
+  res.json({
+    id: req.body.id ?? `a2a-${Date.now()}`,
+    status: { state: "completed" },
+    artifacts: [
+      {
+        parts: [{ type: "text", text: JSON.stringify(result, null, 2) }]
+      }
+    ]
+  });
+});
 app.post(
   "/api/payments/create-checkout-session",
   requireAuth,
@@ -48113,7 +48215,7 @@ app.post("/api/claude-code-config", (req, res) => {
     if (!cfg.hooks) cfg.hooks = {};
     const hookCommandParts = [`node "${HOOK_JS.replace(/\\/g, "/")}"`];
     if (apiKey && typeof apiKey === "string" && apiKey.startsWith("amk_")) {
-      hookCommandParts.push(`--api-key "${apiKey}"`);
+      hookCommandParts.push(`--api-key ${apiKey}`);
     }
     const hookEntry = {
       type: "command",
@@ -48161,6 +48263,107 @@ app.post("/api/build-hook", (_req, res) => {
     }
   );
 });
+function geminiSettingsPath(projectPath) {
+  return path3.join(projectPath, ".gemini", "settings.json");
+}
+app.get("/api/gemini-cli-config", (req, res) => {
+  const projectPath = req.query.project || process.cwd();
+  const configPath = geminiSettingsPath(projectPath);
+  const hookBuilt = fs2.existsSync(GEMINI_HOOK_JS);
+  const configExists = fs2.existsSync(configPath);
+  let config2 = null;
+  let hooksInstalled = false;
+  if (configExists) {
+    try {
+      config2 = JSON.parse(fs2.readFileSync(configPath, "utf-8"));
+      const before = config2?.hooks?.BeforeTool ?? [];
+      hooksInstalled = before.some(
+        (entry) => entry.hooks?.some(
+          (h) => String(h.command ?? "").includes("gemini-cli-hook")
+        )
+      );
+    } catch {
+      config2 = null;
+    }
+  }
+  res.json({
+    configPath,
+    projectPath,
+    exists: configExists,
+    config: config2,
+    hooksInstalled,
+    hookPath: GEMINI_HOOK_JS,
+    hookBuilt
+  });
+});
+app.post("/api/gemini-cli-config", (req, res) => {
+  const {
+    action,
+    projectPath: pp,
+    apiKey
+  } = req.body;
+  const projectPath = pp || process.cwd();
+  const configPath = geminiSettingsPath(projectPath);
+  try {
+    const configDir = path3.dirname(configPath);
+    if (!fs2.existsSync(configDir)) fs2.mkdirSync(configDir, { recursive: true });
+    let cfg = {};
+    if (fs2.existsSync(configPath)) {
+      cfg = JSON.parse(fs2.readFileSync(configPath, "utf-8"));
+    }
+    if (!cfg.hooks) cfg.hooks = {};
+    const hookCommandParts = [`node "${GEMINI_HOOK_JS.replace(/\\/g, "/")}"`];
+    if (apiKey && typeof apiKey === "string" && apiKey.startsWith("amk_")) {
+      hookCommandParts.push(`--api-key ${apiKey}`);
+    }
+    const hookEntry = {
+      name: "agentmomo",
+      type: "command",
+      command: hookCommandParts.join(" "),
+      timeout: 2e3
+    };
+    const matcher = { matcher: ".*", hooks: [hookEntry] };
+    const sessionMatcher = { matcher: ".*", hooks: [hookEntry] };
+    if (action === "install") {
+      cfg.hooks.BeforeTool = [matcher];
+      cfg.hooks.AfterTool = [matcher];
+      cfg.hooks.SessionStart = [sessionMatcher];
+      if (!cfg.experimental) cfg.experimental = {};
+      cfg.experimental.enableAgents = true;
+    } else {
+      const strip = (arr) => (arr ?? []).filter(
+        (e) => !e.hooks?.some(
+          (h) => String(h.command ?? "").includes("gemini-cli-hook")
+        )
+      );
+      cfg.hooks.BeforeTool = strip(cfg.hooks.BeforeTool ?? []);
+      cfg.hooks.AfterTool = strip(cfg.hooks.AfterTool ?? []);
+      cfg.hooks.SessionStart = strip(cfg.hooks.SessionStart ?? []);
+      if (!cfg.hooks.BeforeTool.length) delete cfg.hooks.BeforeTool;
+      if (!cfg.hooks.AfterTool.length) delete cfg.hooks.AfterTool;
+      if (!cfg.hooks.SessionStart.length) delete cfg.hooks.SessionStart;
+      if (!Object.keys(cfg.hooks).length) delete cfg.hooks;
+    }
+    fs2.writeFileSync(configPath, JSON.stringify(cfg, null, 2), "utf-8");
+    res.json({ ok: true, configPath });
+  } catch (err) {
+    res.status(500).json({ error: err.message });
+  }
+});
+app.post("/api/build-gemini-hook", (_req, res) => {
+  if (AGENTSMOMO_DIST_DIR) {
+    res.json({ ok: true });
+    return;
+  }
+  exec(
+    `npx esbuild proxy/gemini-cli-hook.ts --bundle --platform=node --format=cjs --outfile=proxy/dist/gemini-cli-hook.cjs`,
+    { cwd: process.cwd() },
+    (err, _stdout, stderr) => {
+      if (err) res.status(500).json({ error: stderr || err.message });
+      else res.json({ ok: true });
+    }
+  );
+});
 app.get("/api/subagents", (req, res) => {
   const projectPath = req.query.project || process.cwd();
   const agentsDir = path3.join(projectPath, ".claude", "agents");
@@ -48216,7 +48419,7 @@ if (config.httpProxy.targets.length > 0) {
 if (fs2.existsSync(INDEX_HTML)) {
   app.use(import_express2.default.static(DIST_DIR));
   app.get("*", (req, res, next) => {
-    if (req.path.startsWith("/api") || req.path.startsWith("/proxy") || req.path.startsWith("/ws")) {
+    if (req.path.startsWith("/api") || req.path.startsWith("/proxy") || req.path.startsWith("/ws") || req.path.startsWith("/.well-known") || req.path.startsWith("/a2a")) {
       next();
       return;
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agentmomo",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "3D visualization of MCP agent activity — watch your AI tools work in a virtual office",
   "type": "module",
   "bin": {
@@ -18,9 +18,10 @@
     "start": "tsx proxy/server.ts",
     "build:wrapper": "npx esbuild proxy/stdio-wrapper.ts --bundle --platform=node --format=esm --outfile=proxy/dist/stdio-wrapper.js",
     "build:hook": "npx esbuild proxy/claude-code-hook.ts --bundle --platform=node --format=cjs --outfile=proxy/dist/claude-code-hook.cjs",
+    "build:gemini-hook": "npx esbuild proxy/gemini-cli-hook.ts --bundle --platform=node --format=cjs --outfile=proxy/dist/gemini-cli-hook.cjs",
     "build:server": "npx esbuild proxy/server.ts --bundle --platform=node --format=esm --outfile=dist/server.js",
     "build:cli": "npx esbuild bin/cli.ts --bundle --platform=node --format=esm --banner:js=\"#!/usr/bin/env node\" --outfile=dist/cli.js",
-    "build:npm": "npm run build && npm run build:server && npm run build:cli && npx esbuild proxy/stdio-wrapper.ts --bundle --platform=node --format=esm --outfile=dist/stdio-wrapper.js && npx esbuild proxy/claude-code-hook.ts --bundle --platform=node --format=cjs --outfile=dist/claude-code-hook.cjs",
+    "build:npm": "npm run build && npm run build:server && npm run build:cli && npx esbuild proxy/stdio-wrapper.ts --bundle --platform=node --format=esm --outfile=dist/stdio-wrapper.js && npx esbuild proxy/claude-code-hook.ts --bundle --platform=node --format=cjs --outfile=dist/claude-code-hook.cjs && npx esbuild proxy/gemini-cli-hook.ts --bundle --platform=node --format=cjs --outfile=dist/gemini-cli-hook.cjs",
     "preview": "vite preview"
   },
   "dependencies": {
@@ -32,6 +33,7 @@
     "@radix-ui/react-dialog": "^1.1.15",
     "@radix-ui/react-dropdown-menu": "^2.1.16",
     "@radix-ui/react-scroll-area": "^1.2.10",
+    "@radix-ui/react-select": "^2.2.6",
     "@radix-ui/react-separator": "^1.1.8",
     "@radix-ui/react-slider": "^1.3.6",
     "@radix-ui/react-slot": "^1.2.4",