olakai-cli 0.1.14 → 0.2.0

package/dist/index.js

@@ -2186,6 +2186,121 @@ function registerActivityCommand(program2) {
 import * as fs2 from "fs";
 import * as path2 from "path";
 import * as readline from "readline";
+
+// src/commands/monitor-transcript.ts
+var SKILL_REGEX = /^\/([\w-]+)(?:\s|$)/;
+function detectSkill(userMessage) {
+  if (typeof userMessage !== "string") return void 0;
+  const trimmed = userMessage.trimStart();
+  if (!trimmed) return void 0;
+  const match = SKILL_REGEX.exec(trimmed);
+  if (!match) return void 0;
+  return match[1];
+}
+function extractTextContent(content) {
+  if (typeof content === "string") return content;
+  if (!Array.isArray(content)) return "";
+  const parts = [];
+  for (const block of content) {
+    if (block?.type === "text" && typeof block.text === "string") {
+      parts.push(block.text);
+    }
+  }
+  return parts.join("\n").trim();
+}
+function isMetaUserMessage(line, text) {
+  if (line.isMeta === true) return true;
+  if (!text) return true;
+  return text.includes("<command-name>") || text.includes("<local-command-caveat>") || text.includes("<command-message>");
+}
+function parseTimestamp(ts) {
+  if (typeof ts !== "string" || !ts) return NaN;
+  return Date.parse(ts);
+}
+function isCompactionEntry(line) {
+  if (line.type !== "system") return false;
+  const subtype = line.subtype ?? "";
+  return subtype === "compact_boundary" || subtype === "pre_compact" || subtype === "post_compact" || /compact/i.test(subtype);
+}
+function parseTranscript(raw) {
+  const empty = {
+    prompt: "",
+    response: "",
+    tokens: 0,
+    inputTokens: 0,
+    outputTokens: 0,
+    modelName: null,
+    numTurns: 0
+  };
+  if (!raw) return empty;
+  const lines = raw.split("\n");
+  let lastUserText = "";
+  let lastUserTimestamp = NaN;
+  let lastAssistantText = "";
+  let lastAssistantTimestamp = NaN;
+  let lastAssistantModel = null;
+  let lastAssistantInputTokens = 0;
+  let lastAssistantOutputTokens = 0;
+  let numTurns = 0;
+  let currentTurnUserTimestamp = NaN;
+  for (const rawLine of lines) {
+    if (!rawLine) continue;
+    let parsed;
+    try {
+      parsed = JSON.parse(rawLine);
+    } catch {
+      continue;
+    }
+    if (isCompactionEntry(parsed)) continue;
+    if (parsed.isSidechain === true) continue;
+    if (parsed.type === "user" && parsed.message) {
+      const text = extractTextContent(parsed.message.content);
+      if (isMetaUserMessage(parsed, text)) continue;
+      lastUserText = text;
+      lastUserTimestamp = parseTimestamp(parsed.timestamp);
+      currentTurnUserTimestamp = lastUserTimestamp;
+    } else if (parsed.type === "assistant" && parsed.message) {
+      const text = extractTextContent(parsed.message.content);
+      numTurns += 1;
+      if (text) lastAssistantText = text;
+      if (typeof parsed.message.model === "string") {
+        lastAssistantModel = parsed.message.model;
+      }
+      const usage = parsed.message.usage;
+      if (usage) {
+        const input = (usage.input_tokens ?? 0) + (usage.cache_creation_input_tokens ?? 0) + (usage.cache_read_input_tokens ?? 0);
+        const output = usage.output_tokens ?? 0;
+        lastAssistantInputTokens = input;
+        lastAssistantOutputTokens = output;
+      }
+      const ts = parseTimestamp(parsed.timestamp);
+      if (!Number.isNaN(ts)) {
+        lastAssistantTimestamp = ts;
+      }
+    }
+  }
+  const result = {
+    prompt: lastUserText,
+    response: lastAssistantText,
+    tokens: lastAssistantInputTokens + lastAssistantOutputTokens,
+    inputTokens: lastAssistantInputTokens,
+    outputTokens: lastAssistantOutputTokens,
+    modelName: lastAssistantModel,
+    numTurns
+  };
+  if (!Number.isNaN(currentTurnUserTimestamp) && !Number.isNaN(lastAssistantTimestamp) && lastAssistantTimestamp >= currentTurnUserTimestamp) {
+    result.latencyMs = Math.round(
+      lastAssistantTimestamp - currentTurnUserTimestamp
+    );
+  }
+  const skill = detectSkill(lastUserText);
+  if (skill) {
+    result.skill = skill;
+  }
+  return result;
+}
+
+// src/commands/monitor.ts
 var CLAUDE_DIR = ".claude";
 var SETTINGS_FILE = "settings.json";
 var MONITOR_CONFIG_FILE = "olakai-monitor.json";
@@ -2201,8 +2316,36 @@ var HOOK_DEFINITIONS = {
         }
       ]
     }
+  ],
+  SubagentStop: [
+    {
+      matcher: "",
+      hooks: [
+        {
+          type: "command",
+          command: "olakai monitor hook subagent-stop"
+        }
+      ]
+    }
   ]
 };
+function mergeHooksSettings(existing, definitions = HOOK_DEFINITIONS) {
+  const merged = {
+    ...existing ?? {}
+  };
+  for (const [event, defaultEntries] of Object.entries(definitions)) {
+    const existingEntries = merged[event] ?? [];
+    const hasOlakaiHook = existingEntries.some(
+      (e) => e.hooks.some((h) => h.command.includes(OLAKAI_HOOK_MARKER))
+    );
+    if (hasOlakaiHook) {
+      merged[event] = existingEntries;
+    } else {
+      merged[event] = [...existingEntries, ...defaultEntries];
+    }
+  }
+  return merged;
+}
 function prompt(question) {
   const rl = readline.createInterface({
     input: process.stdin,
@@ -2346,16 +2489,7 @@ Select agent (1-${agents.length}): `);
   }
   const settingsPath = getSettingsPath(projectRoot);
   const existingSettings = readJsonFile(settingsPath) ?? {};
-  const mergedHooks = {
-    ...existingSettings.hooks ?? {}
-  };
-  for (const [event, entries] of Object.entries(HOOK_DEFINITIONS)) {
-    const existing = mergedHooks[event] ?? [];
-    const filtered = existing.filter(
-      (e) => !e.hooks.some((h) => h.command.includes(OLAKAI_HOOK_MARKER))
-    );
-    mergedHooks[event] = [...filtered, ...entries];
-  }
+  const mergedHooks = mergeHooksSettings(existingSettings.hooks);
   const updatedSettings = {
     ...existingSettings,
     hooks: mergedHooks
@@ -2410,6 +2544,53 @@ async function createNewAgent(defaultName) {
   });
   return agent;
 }
+function debugLog(label, data) {
+  if (process.env.OLAKAI_MONITOR_DEBUG !== "1") return;
+  try {
+    const logPath = `/tmp/olakai-monitor-debug-${process.pid}.log`;
+    const line = `[${(/* @__PURE__ */ new Date()).toISOString()}] ${label}: ${typeof data === "string" ? data : JSON.stringify(data, null, 2)}
+`;
+    fs2.appendFileSync(logPath, line, "utf-8");
+  } catch {
+  }
+}
+function extractFromTranscript(transcriptPath) {
+  const empty = {
+    prompt: "",
+    response: "",
+    tokens: 0,
+    inputTokens: 0,
+    outputTokens: 0,
+    modelName: null,
+    numTurns: 0
+  };
+  if (!transcriptPath) return empty;
+  let raw;
+  try {
+    raw = fs2.readFileSync(transcriptPath, "utf-8");
+  } catch (err) {
+    debugLog("transcript-read-failed", {
+      transcriptPath,
+      error: err.message
+    });
+    return empty;
+  }
+  return parseTranscript(raw);
+}
+function extractSubagentName(event) {
+  const candidates = [
+    event.agent_name,
+    event.subagent_type,
+    event.agent_type,
+    event.tool_input?.subagent_type
+  ];
+  for (const value of candidates) {
+    if (typeof value === "string" && value.trim()) {
+      return value.trim();
+    }
+  }
+  return void 0;
+}
 async function hookCommand(event) {
   try {
     const config = loadMonitorConfig();
@@ -2417,22 +2598,26 @@ async function hookCommand(event) {
       return;
     }
     const stdinData = await readStdin(3e3);
+    debugLog("stdin-raw", stdinData);
     let eventData = {};
     if (stdinData) {
       try {
         eventData = JSON.parse(stdinData);
       } catch {
+        debugLog("stdin-parse-failed", stdinData);
         return;
       }
     }
+    debugLog("event-parsed", { event, eventData });
     const payload = buildPayload(event, eventData, config);
     if (!payload) {
       return;
     }
+    debugLog("payload-built", payload);
     const controller = new AbortController();
     const timeoutId = setTimeout(() => controller.abort(), 5e3);
     try {
-      await fetch(config.monitoringEndpoint, {
+      const res = await fetch(config.monitoringEndpoint, {
         method: "POST",
         headers: {
           "x-api-key": config.apiKey,
@@ -2441,33 +2626,55 @@ async function hookCommand(event) {
         body: JSON.stringify([payload]),
         signal: controller.signal
       });
+      debugLog("response-status", {
+        status: res.status,
+        ok: res.ok
+      });
     } finally {
       clearTimeout(timeoutId);
     }
-  } catch {
+  } catch (err) {
+    debugLog("hook-exception", err.message);
   }
 }
 function buildPayload(event, eventData, config) {
   const sessionId = eventData.session_id ?? `claude-code-${Date.now()}`;
   switch (event) {
-    case "stop": {
+    case "stop":
+    case "subagent-stop": {
+      const extracted = extractFromTranscript(eventData.transcript_path);
+      const isSubagent = event === "subagent-stop";
+      const customData = {
+        hookEvent: eventData.hook_event_name ?? (isSubagent ? "SubagentStop" : "Stop"),
+        sessionId,
+        transcriptPath: eventData.transcript_path ?? "",
+        cwd: eventData.cwd ?? "",
+        stopHookActive: eventData.stop_hook_active ?? false,
+        inputTokens: extracted.inputTokens,
+        outputTokens: extracted.outputTokens,
+        numTurns: extracted.numTurns
+      };
+      if (typeof extracted.latencyMs === "number") {
+        customData.latencyMs = extracted.latencyMs;
+      }
+      if (isSubagent) {
+        const subagent = extractSubagentName(eventData);
+        if (subagent) {
+          customData.subagent = subagent;
+        }
+      } else if (extracted.skill) {
+        customData.skill = extracted.skill;
+      }
       return {
-        prompt: eventData.prompt ?? "",
-        response: eventData.response ?? "",
+        prompt: extracted.prompt,
+        response: extracted.response,
         chatId: sessionId,
+        // Top-level source so the backend ExternalPromptRequestSchema
+        // picks it up (matches regex ^[a-z0-9_-]+$).
         source: config.source,
-        modelName: eventData.model,
-        tokens: eventData.total_tokens_in !== void 0 && eventData.total_tokens_out !== void 0 ? eventData.total_tokens_in + eventData.total_tokens_out : void 0,
-        requestTime: eventData.duration_api_ms,
-        customData: {
-          hookEvent: "Stop",
-          stopReason: eventData.stop_reason ?? "",
-          totalTokensIn: eventData.total_tokens_in ?? 0,
-          totalTokensOut: eventData.total_tokens_out ?? 0,
-          totalCost: eventData.total_cost ?? 0,
-          durationMs: eventData.duration_ms ?? 0,
-          numTurns: eventData.num_turns ?? 0
-        }
+        modelName: extracted.modelName ?? void 0,
+        tokens: extracted.tokens,
+        customData
       };
     }
     default: