agentmomo 0.1.1 → 0.2.0

package/dist/gemini-cli-hook.cjs

@@ -0,0 +1,312 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+
+// proxy/gemini-cli-hook.ts
+var import_http = __toESM(require("http"), 1);
+var import_fs = __toESM(require("fs"), 1);
+var import_path = __toESM(require("path"), 1);
+var AGENTMOMO_PORT = 9001;
+var TIMEOUT_MS = 1500;
+var LOG_FILE = import_path.default.join(
+  process.env.USERPROFILE ?? process.env.HOME ?? ".",
+  ".agentmomo-hook.log"
+);
+function readArg(name) {
+  const idx = process.argv.indexOf(name);
+  if (idx !== -1 && process.argv[idx + 1]) {
+    return process.argv[idx + 1].replace(/^["']+|["']+$/g, "");
+  }
+  return void 0;
+}
+var AGENTMOMO_API_KEY = readArg("--api-key") ?? process.env.AGENTMOMO_API_KEY ?? "";
+function log(msg) {
+  try {
+    import_fs.default.appendFileSync(
+      LOG_FILE,
+      `[${(/* @__PURE__ */ new Date()).toISOString()}] [gemini] ${msg}
+`
+    );
+  } catch {
+  }
+}
+function postEvent(body) {
+  const headers = {
+    "Content-Type": "application/json",
+    "Content-Length": Buffer.byteLength(body)
+  };
+  if (AGENTMOMO_API_KEY) {
+    headers.Authorization = `Bearer ${AGENTMOMO_API_KEY}`;
+  }
+  const req = import_http.default.request(
+    {
+      hostname: "localhost",
+      port: AGENTMOMO_PORT,
+      path: "/api/event",
+      method: "POST",
+      headers,
+      timeout: TIMEOUT_MS
+    },
+    (res) => {
+      if (res.statusCode && (res.statusCode < 200 || res.statusCode >= 300)) {
+        let responseBody = "";
+        res.on("data", (chunk) => {
+          responseBody += chunk.toString();
+        });
+        res.on("end", () => {
+          log(`POST /api/event failed ${res.statusCode}: ${responseBody}`);
+        });
+      } else {
+        res.resume();
+      }
+    }
+  );
+  req.on("error", (err) => {
+    log("POST error: " + err.message);
+  });
+  req.on("timeout", () => {
+    req.destroy();
+  });
+  req.write(body);
+  req.end();
+}
+function respond() {
+  process.stdout.write("{}\n");
+}
+var TOOL_ROLES = {
+  // Read / inspect
+  read_file: "reader",
+  read_many_files: "reader",
+  list_directory: "reader",
+  get_internal_docs: "reader",
+  // Write / mutate
+  write_file: "writer",
+  replace: "writer",
+  save_memory: "writer",
+  // Execute / shell / planning
+  run_shell_command: "runner",
+  complete_task: "runner",
+  enter_plan_mode: "runner",
+  exit_plan_mode: "runner",
+  write_todos: "runner",
+  ask_user: "runner",
+  activate_skill: "runner",
+  // Search / navigate
+  grep_search: "searcher",
+  glob: "searcher",
+  google_web_search: "searcher",
+  web_fetch: "searcher"
+};
+var ROLE_LABELS = {
+  reader: "Reader",
+  writer: "Writer",
+  runner: "Runner",
+  searcher: "Searcher"
+};
+function extractToolMessage(toolName, toolInput) {
+  const str = (v, max = 80) => typeof v === "string" && v.trim() ? v.trim().slice(0, max) : void 0;
+  switch (toolName) {
+    case "read_file":
+    case "read_many_files":
+      return str(toolInput["path"] ?? toolInput["paths"]) ? `Reading ${str(toolInput["path"] ?? toolInput["paths"])}` : void 0;
+    case "list_directory":
+      return str(toolInput["path"]) ? `Listing ${str(toolInput["path"])}` : void 0;
+    case "write_file":
+      return str(toolInput["path"]) ? `Writing ${str(toolInput["path"])}` : void 0;
+    case "replace":
+      return str(toolInput["path"]) ? `Editing ${str(toolInput["path"])}` : void 0;
+    case "run_shell_command": {
+      const cmd = str(toolInput["command"], 100);
+      return cmd ? `Running: ${cmd}` : void 0;
+    }
+    case "grep_search": {
+      const pat = str(toolInput["pattern"]);
+      const dir = str(toolInput["directory"] ?? toolInput["path"], 40);
+      return pat ? `Searching "${pat}"${dir ? ` in ${dir}` : ""}` : void 0;
+    }
+    case "google_web_search":
+      return str(toolInput["query"]) ? `Web search: ${str(toolInput["query"])}` : void 0;
+    case "web_fetch":
+      return str(toolInput["url"]) ? `Fetching ${str(toolInput["url"])}` : void 0;
+    case "save_memory":
+      return "Saving to memory";
+    case "write_todos":
+      return "Updating task list";
+    case "ask_user": {
+      const q = str(toolInput["question"], 80);
+      return q ? `Asking: ${q}` : void 0;
+    }
+    case "complete_task": {
+      const result = str(toolInput["result"], 80);
+      return result ? `Completing task: ${result}` : void 0;
+    }
+    default:
+      if (toolName.startsWith("mcp_")) {
+        return `Using ${toolName.replace(/^mcp_[^_]+_/, "").replace(/_/g, " ")}`;
+      }
+      return void 0;
+  }
+}
+function extractResultText(response) {
+  if (!response || typeof response !== "object") return void 0;
+  const obj = response;
+  for (const key of ["text", "content", "output", "result", "stdout"]) {
+    if (typeof obj[key] === "string" && obj[key].trim())
+      return obj[key].trim();
+  }
+  const llm = obj.llmContent;
+  if (typeof llm === "string" && llm.trim()) return llm.trim();
+  if (llm && typeof llm === "object") {
+    const llmObj = llm;
+    if (typeof llmObj.text === "string" && llmObj.text.trim())
+      return llmObj.text.trim();
+    if (Array.isArray(llmObj.parts)) {
+      const parts = llmObj.parts.map((p) => typeof p === "string" ? p : p?.text).filter((t) => typeof t === "string" && t.trim());
+      if (parts.length > 0) return parts.join("\n").trim() || void 0;
+    }
+  }
+  return void 0;
+}
+function getRole(toolName) {
+  if (TOOL_ROLES[toolName]) return TOOL_ROLES[toolName];
+  if (toolName.startsWith("mcp_")) {
+    const parts = toolName.split("_");
+    const suffix = parts[parts.length - 1];
+    if (suffix === "read" || suffix === "get" || suffix === "list")
+      return "reader";
+    if (suffix === "write" || suffix === "create" || suffix === "update" || suffix === "delete")
+      return "writer";
+    if (suffix === "search" || suffix === "find" || suffix === "query")
+      return "searcher";
+  }
+  return "runner";
+}
+process.stdin.setEncoding("utf-8");
+var raw = "";
+process.stdin.on("data", (chunk) => {
+  raw += chunk;
+});
+process.stdin.on("end", () => {
+  respond();
+  try {
+    const payload = JSON.parse(raw);
+    log(
+      `Received: hook_event_name=${payload.hook_event_name ?? "?"} tool_name=${payload.tool_name ?? "?"} session_id=${String(payload.session_id ?? "?").slice(0, 8)}`
+    );
+    const {
+      session_id = "unknown",
+      hook_event_name = "BeforeTool",
+      tool_name = "unknown",
+      tool_input,
+      tool_response,
+      cwd,
+      timestamp
+    } = payload;
+    if (hook_event_name === "SessionStart") {
+      const sessionShort2 = String(session_id).slice(0, 8);
+      const folderName2 = cwd ? import_path.default.basename(cwd) : "Gemini CLI";
+      const event2 = {
+        agentId: `gc-${sessionShort2}-runner`,
+        agentName: `${folderName2} \xB7 Runner`,
+        source: "gemini-cli",
+        event: "agent_register",
+        toolName: "",
+        timestamp: Date.now(),
+        requestId: `${session_id}-session-start-${Date.now()}`,
+        metadata: {
+          cwd: cwd ?? "unknown",
+          sessionId: session_id,
+          hookEvent: "SessionStart"
+        }
+      };
+      postEvent(JSON.stringify(event2));
+      return;
+    }
+    if (hook_event_name === "SessionEnd") {
+      const sessionShort2 = String(session_id).slice(0, 8);
+      const folderName2 = cwd ? import_path.default.basename(cwd) : "Gemini CLI";
+      const event2 = {
+        agentId: `gc-${sessionShort2}-runner`,
+        agentName: `${folderName2} \xB7 Runner`,
+        source: "gemini-cli",
+        event: "agent_heartbeat",
+        toolName: "",
+        timestamp: Date.now(),
+        requestId: `${session_id}-session-end-${Date.now()}`,
+        metadata: {
+          cwd: cwd ?? "unknown",
+          sessionId: session_id,
+          hookEvent: "SessionEnd"
+        }
+      };
+      postEvent(JSON.stringify(event2));
+      return;
+    }
+    const isAfter = hook_event_name === "AfterTool";
+    const hasError = isAfter && tool_response?.error;
+    const role = getRole(tool_name);
+    const sessionShort = String(session_id).slice(0, 8);
+    const agentId = `gc-${sessionShort}-${role}`;
+    const folderName = cwd ? import_path.default.basename(cwd) : "Gemini CLI";
+    const agentName = `${folderName} \xB7 ${ROLE_LABELS[role]}`;
+    let subAgentMeta;
+    if (tool_name === "complete_task" && tool_input) {
+      const result = tool_input["result"];
+      const subName = typeof result === "string" && result.trim() ? result.trim().toLowerCase().replace(/[^a-z0-9]+/g, "-").replace(/^-|-$/g, "").slice(0, 24) : "subagent";
+      const subAgentId = `gc-${sessionShort}-subagent-${subName}`;
+      subAgentMeta = { subAgentId, subAgentName: subName };
+    }
+    let eventType;
+    if (isAfter && hasError) {
+      eventType = "call_error";
+    } else if (isAfter) {
+      eventType = "call_end";
+    } else {
+      eventType = "call_start";
+    }
+    const resultText = isAfter ? extractResultText(tool_response) : void 0;
+    const event = {
+      agentId,
+      agentName,
+      source: "gemini-cli",
+      event: eventType,
+      toolName: tool_name,
+      arguments: tool_input ?? {},
+      timestamp: timestamp ? new Date(timestamp).getTime() : Date.now(),
+      requestId: `${session_id}-${tool_name}-${Date.now()}`,
+      ...hasError ? { error: String(tool_response.error) } : {},
+      // On BeforeTool: attach context hint; on AfterTool: attach tool result
+      ...!isAfter && tool_input ? { message: extractToolMessage(tool_name, tool_input) } : {},
+      ...isAfter && resultText ? { message: resultText } : {},
+      metadata: {
+        cwd: cwd ?? "unknown",
+        role,
+        sessionId: session_id,
+        ...subAgentMeta ?? {}
+      }
+    };
+    postEvent(JSON.stringify(event));
+  } catch (e) {
+    log("Error: " + (e?.message ?? e));
+  }
+});

package/dist/index.html

@@ -4,15 +4,15 @@
     <meta charset="UTF-8" />
     <link rel="icon" type="image/svg+xml" href="/vite.svg" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
-    <title>AgentMomo — MCP Activity Visualizer</title>
+    <title>AgentMOMO— MCP Activity Visualizer</title>
     <link rel="preconnect" href="https://fonts.googleapis.com" />
     <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin />
     <link
       href="https://fonts.googleapis.com/css2?family=Orbitron:wght@400..900&display=swap"
       rel="stylesheet"
     />
-    <script type="module" crossorigin src="/assets/index-Cru_j148.js"></script>
-    <link rel="stylesheet" crossorigin href="/assets/index-q7eACk3a.css">
+    <script type="module" crossorigin src="/assets/index-5pVyUNvu.js"></script>
+    <link rel="stylesheet" crossorigin href="/assets/index-BjOLTppo.css">
   </head>
   <body>
     <div id="root"></div>