muonroi-cli 1.2.0

package/dist/storage/transcript.js

@@ -0,0 +1,269 @@
+import { getCompactionSummaryText } from "../orchestrator/compaction";
+import { getDatabase, withTransaction } from "./db";
+import { extractToolResultFromOutput, getOutputKind, isOutputSuccess } from "./tool-results";
+import { buildEffectiveTranscript } from "./transcript-view";
+function loadMessageRows(sessionId) {
+    return getDatabase()
+        .prepare(`
+    SELECT session_id, seq, role, message_json, created_at
+    FROM messages
+    WHERE session_id = ?
+    ORDER BY seq ASC
+  `)
+        .all(sessionId);
+}
+function toPersistedCompaction(row) {
+    if (!row)
+        return null;
+    return {
+        firstKeptSeq: row.first_kept_seq,
+        summary: row.summary,
+        tokensBefore: row.tokens_before,
+        createdAt: new Date(row.created_at),
+    };
+}
+export function loadLatestCompaction(sessionId) {
+    const row = getDatabase()
+        .prepare(`
+    SELECT session_id, first_kept_seq, summary, tokens_before, created_at
+    FROM compactions
+    WHERE session_id = ?
+    ORDER BY id DESC
+    LIMIT 1
+  `)
+        .get(sessionId);
+    return toPersistedCompaction(row);
+}
+function buildEffectiveMessageRecords(sessionId) {
+    const rows = loadMessageRows(sessionId);
+    const messages = rows.map((row) => JSON.parse(row.message_json));
+    const seqs = rows.map((row) => row.seq);
+    const timestamps = rows.map((row) => new Date(row.created_at));
+    const transcript = buildEffectiveTranscript(messages, seqs, timestamps, loadLatestCompaction(sessionId));
+    return transcript.messages.map((message, index) => ({
+        message,
+        seq: transcript.seqs[index],
+        timestamp: transcript.timestamps[index],
+    }));
+}
+export function loadRawTranscript(sessionId) {
+    return loadMessageRows(sessionId).map((row) => JSON.parse(row.message_json));
+}
+export function loadTranscriptState(sessionId) {
+    const rows = loadMessageRows(sessionId);
+    return buildEffectiveTranscript(rows.map((row) => JSON.parse(row.message_json)), rows.map((row) => row.seq), rows.map((row) => new Date(row.created_at)), loadLatestCompaction(sessionId));
+}
+export function loadTranscript(sessionId) {
+    return loadTranscriptState(sessionId).messages;
+}
+export function getNextMessageSequence(sessionId) {
+    return getNextSequence(getDatabase(), sessionId);
+}
+export function appendMessages(sessionId, messages) {
+    if (messages.length === 0)
+        return [];
+    const insertedSeqs = [];
+    withTransaction((db) => {
+        const nextSeq = getNextSequence(db, sessionId);
+        const insertMessage = db.prepare(`
+      INSERT INTO messages (session_id, seq, role, message_json, created_at)
+      VALUES (?, ?, ?, ?, ?)
+    `);
+        const insertToolCall = db.prepare(`
+      INSERT OR IGNORE INTO tool_calls (
+        session_id, message_seq, tool_call_id, tool_name, args_json, status, started_at, completed_at
+      ) VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+    `);
+        const updateToolCall = db.prepare(`
+      UPDATE tool_calls
+      SET tool_name = ?, args_json = ?, status = ?, completed_at = ?
+      WHERE session_id = ? AND tool_call_id = ?
+    `);
+        const selectToolCall = db.prepare(`
+      SELECT id, tool_call_id, tool_name, args_json
+      FROM tool_calls
+      WHERE session_id = ? AND tool_call_id = ?
+    `);
+        const insertToolResult = db.prepare(`
+      INSERT INTO tool_results (tool_call_row_id, output_kind, output_json, success, created_at)
+      VALUES (?, ?, ?, ?, ?)
+    `);
+        const updateSession = db.prepare(`
+      UPDATE sessions
+      SET updated_at = ?
+      WHERE id = ?
+    `);
+        messages.forEach((message, index) => {
+            const seq = nextSeq + index;
+            const createdAt = new Date().toISOString();
+            insertedSeqs.push(seq);
+            insertMessage.run(sessionId, seq, message.role, JSON.stringify(message), createdAt);
+            if (message.role === "assistant" && Array.isArray(message.content)) {
+                for (const part of message.content) {
+                    if (part.type !== "tool-call")
+                        continue;
+                    insertToolCall.run(sessionId, seq, part.toolCallId, part.toolName, JSON.stringify(part.input ?? {}), "completed", createdAt, createdAt);
+                    updateToolCall.run(part.toolName, JSON.stringify(part.input ?? {}), "completed", createdAt, sessionId, part.toolCallId);
+                }
+            }
+            if (message.role === "tool" && Array.isArray(message.content)) {
+                for (const part of message.content) {
+                    if (part.type !== "tool-result")
+                        continue;
+                    let toolCall = selectToolCall.get(sessionId, part.toolCallId);
+                    if (!toolCall) {
+                        insertToolCall.run(sessionId, seq, part.toolCallId, part.toolName, "{}", "completed", createdAt, createdAt);
+                        toolCall = selectToolCall.get(sessionId, part.toolCallId);
+                    }
+                    if (!toolCall)
+                        continue;
+                    const extracted = extractToolResultFromOutput(part.output);
+                    insertToolResult.run(toolCall.id, getOutputKind(part.output), JSON.stringify(extracted ?? part.output), extracted ? Number(extracted.success) : Number(isOutputSuccess(part.output)), createdAt);
+                }
+            }
+        });
+        updateSession.run(new Date().toISOString(), sessionId);
+    });
+    return insertedSeqs;
+}
+export function appendSystemMessage(sessionId, content) {
+    return appendMessages(sessionId, [{ role: "system", content }])[0] ?? null;
+}
+export function appendCompaction(sessionId, firstKeptSeq, summary, tokensBefore) {
+    withTransaction((db) => {
+        db.prepare(`
+      INSERT INTO compactions (session_id, first_kept_seq, summary, tokens_before, created_at)
+      VALUES (?, ?, ?, ?, ?)
+    `).run(sessionId, firstKeptSeq, summary, tokensBefore, new Date().toISOString());
+        db.prepare(`
+      UPDATE sessions
+      SET updated_at = ?
+      WHERE id = ?
+    `).run(new Date().toISOString(), sessionId);
+    });
+}
+export function buildChatEntries(sessionId) {
+    const toolResults = loadStoredToolResults(sessionId);
+    const callMap = new Map();
+    const entries = [];
+    for (const row of buildEffectiveMessageRecords(sessionId)) {
+        const { message, timestamp } = row;
+        if (message.role === "user") {
+            const content = renderUserContent(message.content);
+            if (content) {
+                entries.push({ type: "user", content, timestamp });
+            }
+            continue;
+        }
+        if (message.role === "system") {
+            const content = getCompactionSummaryText(message) ?? (typeof message.content === "string" ? message.content.trim() : "");
+            if (content) {
+                entries.push({ type: "assistant", content, timestamp });
+            }
+            continue;
+        }
+        if (message.role === "assistant") {
+            const text = renderAssistantContent(message.content, callMap);
+            if (text) {
+                entries.push({ type: "assistant", content: text, timestamp });
+            }
+            continue;
+        }
+        if (message.role === "tool" && Array.isArray(message.content)) {
+            for (const part of message.content) {
+                if (part.type !== "tool-result")
+                    continue;
+                const toolCall = callMap.get(part.toolCallId) ?? toFallbackToolCall(part.toolCallId, part.toolName);
+                const toolResult = toolResults.get(part.toolCallId) ??
+                    extractToolResultFromOutput(part.output) ?? {
+                    success: isOutputSuccess(part.output),
+                    output: JSON.stringify(part.output),
+                };
+                entries.push({
+                    type: "tool_result",
+                    content: toolResult.success ? toolResult.output || "Success" : toolResult.error || "Error",
+                    timestamp,
+                    toolCall,
+                    toolResult,
+                });
+            }
+        }
+    }
+    return entries;
+}
+function getNextSequence(db, sessionId) {
+    const row = db
+        .prepare(`
+    SELECT COALESCE(MAX(seq), 0) AS max_seq
+    FROM messages
+    WHERE session_id = ?
+  `)
+        .get(sessionId);
+    return (row?.max_seq ?? 0) + 1;
+}
+function renderUserContent(content) {
+    if (typeof content === "string")
+        return content.trim();
+    if (!Array.isArray(content))
+        return "";
+    return content
+        .map((part) => {
+        if (part.type === "text")
+            return part.text;
+        if (part.type === "image")
+            return "[Image]";
+        if (part.type === "file")
+            return part.filename ? `[File: ${part.filename}]` : "[File]";
+        return "";
+    })
+        .filter(Boolean)
+        .join("\n")
+        .trim();
+}
+function renderAssistantContent(content, callMap) {
+    if (typeof content === "string")
+        return content.trim();
+    if (!Array.isArray(content))
+        return "";
+    const textParts = [];
+    for (const part of content) {
+        if (part.type === "text") {
+            textParts.push(part.text);
+            continue;
+        }
+        if (part.type === "tool-call") {
+            callMap.set(part.toolCallId, {
+                id: part.toolCallId,
+                type: "function",
+                function: {
+                    name: part.toolName,
+                    arguments: JSON.stringify(part.input ?? {}),
+                },
+            });
+        }
+    }
+    return textParts.join("").trim();
+}
+function loadStoredToolResults(sessionId) {
+    const rows = getDatabase()
+        .prepare(`
+    SELECT tc.tool_call_id, tr.output_json
+    FROM tool_results tr
+    JOIN tool_calls tc ON tc.id = tr.tool_call_row_id
+    WHERE tc.session_id = ?
+    ORDER BY tr.id ASC
+  `)
+        .all(sessionId);
+    return new Map(rows.map((row) => [row.tool_call_id, JSON.parse(row.output_json)]));
+}
+function toFallbackToolCall(toolCallId, toolName) {
+    return {
+        id: toolCallId,
+        type: "function",
+        function: {
+            name: toolName,
+            arguments: "{}",
+        },
+    };
+}
+//# sourceMappingURL=transcript.js.map

package/dist/storage/transcript.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transcript.js","sourceRoot":"","sources":["../../src/storage/transcript.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,wBAAwB,EAAE,MAAM,4BAA4B,CAAC;AAEtE,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,MAAM,CAAC;AACpD,OAAO,EAAE,2BAA2B,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAC7F,OAAO,EAAE,wBAAwB,EAAwD,MAAM,mBAAmB,CAAC;AAoCnH,SAAS,eAAe,CAAC,SAAiB;IACxC,OAAO,WAAW,EAAE;SACjB,OAAO,CAAC;;;;;GAKV,CAAC;SACC,GAAG,CAAC,SAAS,CAAiB,CAAC;AACpC,CAAC;AAED,SAAS,qBAAqB,CAAC,GAA8B;IAC3D,IAAI,CAAC,GAAG;QAAE,OAAO,IAAI,CAAC;IACtB,OAAO;QACL,YAAY,EAAE,GAAG,CAAC,cAAc;QAChC,OAAO,EAAE,GAAG,CAAC,OAAO;QACpB,YAAY,EAAE,GAAG,CAAC,aAAa;QAC/B,SAAS,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC;KACpC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,SAAiB;IACpD,MAAM,GAAG,GAAG,WAAW,EAAE;SACtB,OAAO,CAAC;;;;;;GAMV,CAAC;SACC,GAAG,CAAC,SAAS,CAA8B,CAAC;IAE/C,OAAO,qBAAqB,CAAC,GAAG,CAAC,CAAC;AACpC,CAAC;AAED,SAAS,4BAA4B,CAAC,SAAiB;IACrD,MAAM,IAAI,GAAG,eAAe,CAAC,SAAS,CAAC,CAAC;IACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,YAAY,CAAiB,CAAC,CAAC;IACjF,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IACxC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;IAC/D,MAAM,UAAU,GAAG,wBAAwB,CAAC,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,oBAAoB,CAAC,SAAS,CAAC,CAAC,CAAC;IAEzG,OAAO,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;QAClD,OAAO;QACP,GAAG,EAAE,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC;QAC3B,SAAS,EAAE,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC;KACxC,CAAC,CAAC,CAAC;AACN,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,SAAiB;IACjD,OAAO,eAAe,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,YAAY,CAAiB,CAAC,CAAC;AAC/F,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,SAAiB;IACnD,MAAM,IAAI,GAAG,eAAe,CAAC,SAAS,CAAC,CAAC;IACxC,OAAO,wBAAwB,CAC7B,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,YAAY,CAAiB,CAAC,EAC/D,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAC1B,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,EAC3C,oBAAoB,CAAC,SAAS,CAAC,CAChC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,SAAiB;IAC9C,OAAO,mBAAmB,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC;AACjD,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,SAAiB;IACtD,OAAO,eAAe,CAAC,WAAW,EAAE,EAAE,SAAS,CAAC,CAAC;AACnD,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,SAAiB,EAAE,QAAwB;IACxE,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAErC,MAAM,YAAY,GAAa,EAAE,CAAC;IAClC,eAAe,CAAC,CAAC,EAAE,EAAE,EAAE;QACrB,MAAM,OAAO,GAAG,eAAe,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QAC/C,MAAM,aAAa,GAAG,EAAE,CAAC,OAAO,CAAC;;;KAGhC,CAAC,CAAC;QACH,MAAM,cAAc,GAAG,EAAE,CAAC,OAAO,CAAC;;;;KAIjC,CAAC,CAAC;QACH,MAAM,cAAc,GAAG,EAAE,CAAC,OAAO,CAAC;;;;KAIjC,CAAC,CAAC;QACH,MAAM,cAAc,GAAG,EAAE,CAAC,OAAO,CAAC;;;;KAIjC,CAAC,CAAC;QACH,MAAM,gBAAgB,GAAG,EAAE,CAAC,OAAO,CAAC;;;KAGnC,CAAC,CAAC;QACH,MAAM,aAAa,GAAG,EAAE,CAAC,OAAO,CAAC;;;;KAIhC,CAAC,CAAC;QAEH,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;YAClC,MAAM,GAAG,GAAG,OAAO,GAAG,KAAK,CAAC;YAC5B,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YAC3C,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACvB,aAAa,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,EAAE,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,SAAS,CAAC,CAAC;YAEpF,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;gBACnE,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBACnC,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW;wBAAE,SAAS;oBACxC,cAAc,CAAC,GAAG,CAChB,SAAS,EACT,GAAG,EACH,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,QAAQ,EACb,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,EAChC,WAAW,EACX,SAAS,EACT,SAAS,CACV,CAAC;oBACF,cAAc,CAAC,GAAG,CAChB,IAAI,CAAC,QAAQ,EACb,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,EAChC,WAAW,EACX,SAAS,EACT,SAAS,EACT,IAAI,CAAC,UAAU,CAChB,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC9D,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBACnC,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa;wBAAE,SAAS;oBAC1C,IAAI,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,UAAU,CAAkC,CAAC;oBAC/F,IAAI,CAAC,QAAQ,EAAE,CAAC;wBACd,cAAc,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;wBAC5G,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,UAAU,CAAkC,CAAC;oBAC7F,CAAC;oBACD,IAAI,CAAC,QAAQ;wBAAE,SAAS;oBAExB,MAAM,SAAS,GAAG,2BAA2B,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;oBAC3D,gBAAgB,CAAC,GAAG,CAClB,QAAQ,CAAC,EAAE,EACX,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,EAC1B,IAAI,CAAC,SAAS,CAAC,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,EACxC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,EAC5E,SAAS,CACV,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,aAAa,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,SAAS,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,SAAiB,EAAE,OAAe;IACpE,OAAO,cAAc,CAAC,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC;AAC7E,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,SAAiB,EAAE,YAAoB,EAAE,OAAe,EAAE,YAAoB;IAC7G,eAAe,CAAC,CAAC,EAAE,EAAE,EAAE;QACrB,EAAE,CAAC,OAAO,CAAC;;;KAGV,CAAC,CAAC,GAAG,CAAC,SAAS,EAAE,YAAY,EAAE,OAAO,EAAE,YAAY,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC;QAEjF,EAAE,CAAC,OAAO,CAAC;;;;KAIV,CAAC,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,SAAS,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,SAAiB;IAChD,MAAM,WAAW,GAAG,qBAAqB,CAAC,SAAS,CAAC,CAAC;IACrD,MAAM,OAAO,GAAG,IAAI,GAAG,EAAoB,CAAC;IAC5C,MAAM,OAAO,GAAgB,EAAE,CAAC;IAEhC,KAAK,MAAM,GAAG,IAAI,4BAA4B,CAAC,SAAS,CAAC,EAAE,CAAC;QAC1D,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,GAAG,CAAC;QAEnC,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC5B,MAAM,OAAO,GAAG,iBAAiB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACnD,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;YACrD,CAAC;YACD,SAAS;QACX,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC9B,MAAM,OAAO,GACX,wBAAwB,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAC3G,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;YAC1D,CAAC;YACD,SAAS;QACX,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACjC,MAAM,IAAI,GAAG,sBAAsB,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC9D,IAAI,IAAI,EAAE,CAAC;gBACT,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;YAChE,CAAC;YACD,SAAS;QACX,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC9D,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBACnC,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa;oBAAE,SAAS;gBAC1C,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,kBAAkB,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACpG,MAAM,UAAU,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC;oBACjD,2BAA2B,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI;oBAC1C,OAAO,EAAE,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC;oBACrC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC;iBACpC,CAAC;gBACJ,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,aAAa;oBACnB,OAAO,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,IAAI,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK,IAAI,OAAO;oBAC1F,SAAS;oBACT,QAAQ;oBACR,UAAU;iBACX,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,eAAe,CAAC,EAAkC,EAAE,SAAiB;IAC5E,MAAM,GAAG,GAAG,EAAE;SACX,OAAO,CAAC;;;;GAIV,CAAC;SACC,GAAG,CAAC,SAAS,CAAoC,CAAC;IAErD,OAAO,CAAC,GAAG,EAAE,OAAO,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,iBAAiB,CAAC,OAAgC;IACzD,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,OAAO,CAAC,IAAI,EAAE,CAAC;IACvD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;QAAE,OAAO,EAAE,CAAC;IAEvC,OAAO,OAAO;SACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACZ,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM;YAAE,OAAO,IAAI,CAAC,IAAI,CAAC;QAC3C,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO;YAAE,OAAO,SAAS,CAAC;QAC5C,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM;YAAE,OAAO,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC;QACvF,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC;SACD,MAAM,CAAC,OAAO,CAAC;SACf,IAAI,CAAC,IAAI,CAAC;SACV,IAAI,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,sBAAsB,CAAC,OAAgC,EAAE,OAA8B;IAC9F,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,OAAO,CAAC,IAAI,EAAE,CAAC;IACvD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;QAAE,OAAO,EAAE,CAAC;IAEvC,MAAM,SAAS,GAAa,EAAE,CAAC;IAC/B,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;QAC3B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACzB,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1B,SAAS;QACX,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC9B,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE;gBAC3B,EAAE,EAAE,IAAI,CAAC,UAAU;gBACnB,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACR,IAAI,EAAE,IAAI,CAAC,QAAQ;oBACnB,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;iBAC5C;aACF,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;AACnC,CAAC;AAED,SAAS,qBAAqB,CAAC,SAAiB;IAC9C,MAAM,IAAI,GAAG,WAAW,EAAE;SACvB,OAAO,CAAC;;;;;;GAMV,CAAC;SACC,GAAG,CAAC,SAAS,CAA0B,CAAC;IAE3C,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,CAAe,CAAC,CAAC,CAAC,CAAC;AACnG,CAAC;AAED,SAAS,kBAAkB,CAAC,UAAkB,EAAE,QAAgB;IAC9D,OAAO;QACL,EAAE,EAAE,UAAU;QACd,IAAI,EAAE,UAAU;QAChB,QAAQ,EAAE;YACR,IAAI,EAAE,QAAQ;YACd,SAAS,EAAE,IAAI;SAChB;KACF,CAAC;AACJ,CAAC"}

package/dist/storage/transcript.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/storage/transcript.test.js

@@ -0,0 +1,22 @@
+import { describe, expect, it } from "vitest";
+import { extractToolResultFromOutput } from "./tool-results";
+describe("transcript media tool results", () => {
+    it("preserves media metadata when stored tool output is normalized", () => {
+        const mediaResult = {
+            success: true,
+            output: "Generated 1 image.\n- /tmp/example.png",
+            media: [
+                {
+                    kind: "image",
+                    path: "/tmp/example.png",
+                    url: "https://example.com/generated.png",
+                    sourcePath: "/tmp/source.png",
+                    prompt: "Create a new hero image",
+                    modelId: "muonroi-test-image",
+                },
+            ],
+        };
+        expect(extractToolResultFromOutput(mediaResult)).toEqual(mediaResult);
+    });
+});
+//# sourceMappingURL=transcript.test.js.map

package/dist/storage/transcript.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transcript.test.js","sourceRoot":"","sources":["../../src/storage/transcript.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE9C,OAAO,EAAE,2BAA2B,EAAE,MAAM,gBAAgB,CAAC;AAE7D,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC7C,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,MAAM,WAAW,GAAe;YAC9B,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,wCAAwC;YAChD,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,kBAAkB;oBACxB,GAAG,EAAE,mCAAmC;oBACxC,UAAU,EAAE,iBAAiB;oBAC7B,MAAM,EAAE,yBAAyB;oBACjC,OAAO,EAAE,oBAAoB;iBAC9B;aACF;SACF,CAAC;QAEF,MAAM,CAAC,2BAA2B,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/storage/usage-cap.d.ts

@@ -0,0 +1,34 @@
+/**
+ * TUI-owned monthly usage state stored at ~/.muonroi-cli/usage.json.
+ *
+ * Phase 0 scope: schema + atomic IO + boot read + month-rollover only.
+ * No enforcement, no thresholds, no auto-downgrade — those land in Phase 1 USAGE-02..05/07.
+ *
+ * Architecture anti-pattern 4: this file is owned exclusively by the TUI process.
+ */
+export interface UsageState {
+    current_month_utc: string;
+    current_usd: number;
+    reservations: Array<{
+        id: string;
+        usd: number;
+        createdAtMs: number;
+        model?: string;
+        provider?: string;
+        est_input_tokens?: number;
+        est_output_tokens?: number;
+    }>;
+    thresholds_fired_this_month?: number[];
+}
+/**
+ * Load usage state from ~/.muonroi-cli/usage.json.
+ * - If absent: writes and returns a fresh default state.
+ * - If month has rolled over: resets current_usd to 0 and rewrites the file.
+ * Accepts an optional homeOverride for test isolation.
+ */
+export declare function loadUsage(homeOverride?: string): Promise<UsageState>;
+/**
+ * Persist usage state atomically to ~/.muonroi-cli/usage.json.
+ * Accepts an optional homeOverride for test isolation.
+ */
+export declare function saveUsage(state: UsageState, homeOverride?: string): Promise<void>;

package/dist/storage/usage-cap.js

@@ -0,0 +1,54 @@
+import * as os from "node:os";
+import * as path from "node:path";
+import { atomicReadJSON, atomicWriteJSON } from "./atomic-io.js";
+function currentMonthUTC(d = new Date()) {
+    const yyyy = d.getUTCFullYear();
+    const mm = String(d.getUTCMonth() + 1).padStart(2, "0");
+    return `${yyyy}-${mm}`;
+}
+/**
+ * Resolve the ~/.muonroi-cli/ home directory.
+ * Priority: explicit override → MUONROI_CLI_HOME env → os.homedir()/.muonroi-cli
+ */
+function muonroiHome(override) {
+    return override ?? process.env.MUONROI_CLI_HOME ?? path.join(os.homedir(), ".muonroi-cli");
+}
+function defaultState() {
+    return { current_month_utc: currentMonthUTC(), current_usd: 0, reservations: [], thresholds_fired_this_month: [] };
+}
+/**
+ * Load usage state from ~/.muonroi-cli/usage.json.
+ * - If absent: writes and returns a fresh default state.
+ * - If month has rolled over: resets current_usd to 0 and rewrites the file.
+ * Accepts an optional homeOverride for test isolation.
+ */
+export async function loadUsage(homeOverride) {
+    const filePath = path.join(muonroiHome(homeOverride), "usage.json");
+    const existing = await atomicReadJSON(filePath);
+    if (!existing) {
+        const state = defaultState();
+        await atomicWriteJSON(filePath, state);
+        return state;
+    }
+    // Month rollover: reset spend if the stored month is stale
+    if (existing.current_month_utc !== currentMonthUTC()) {
+        const reset = {
+            current_month_utc: currentMonthUTC(),
+            current_usd: 0,
+            reservations: [],
+            thresholds_fired_this_month: [],
+        };
+        await atomicWriteJSON(filePath, reset);
+        return reset;
+    }
+    return existing;
+}
+/**
+ * Persist usage state atomically to ~/.muonroi-cli/usage.json.
+ * Accepts an optional homeOverride for test isolation.
+ */
+export async function saveUsage(state, homeOverride) {
+    const filePath = path.join(muonroiHome(homeOverride), "usage.json");
+    await atomicWriteJSON(filePath, state);
+}
+//# sourceMappingURL=usage-cap.js.map

package/dist/storage/usage-cap.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-cap.js","sourceRoot":"","sources":["../../src/storage/usage-cap.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAyBjE,SAAS,eAAe,CAAC,IAAU,IAAI,IAAI,EAAE;IAC3C,MAAM,IAAI,GAAG,CAAC,CAAC,cAAc,EAAE,CAAC;IAChC,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IACxD,OAAO,GAAG,IAAI,IAAI,EAAE,EAAE,CAAC;AACzB,CAAC;AAED;;;GAGG;AACH,SAAS,WAAW,CAAC,QAAiB;IACpC,OAAO,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,cAAc,CAAC,CAAC;AAC7F,CAAC;AAED,SAAS,YAAY;IACnB,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,2BAA2B,EAAE,EAAE,EAAE,CAAC;AACrH,CAAC;AAED;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,SAAS,CAAC,YAAqB;IACnD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,EAAE,YAAY,CAAC,CAAC;IACpE,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAa,QAAQ,CAAC,CAAC;IAC5D,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,KAAK,GAAG,YAAY,EAAE,CAAC;QAC7B,MAAM,eAAe,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACvC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,2DAA2D;IAC3D,IAAI,QAAQ,CAAC,iBAAiB,KAAK,eAAe,EAAE,EAAE,CAAC;QACrD,MAAM,KAAK,GAAe;YACxB,iBAAiB,EAAE,eAAe,EAAE;YACpC,WAAW,EAAE,CAAC;YACd,YAAY,EAAE,EAAE;YAChB,2BAA2B,EAAE,EAAE;SAChC,CAAC;QACF,MAAM,eAAe,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACvC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,SAAS,CAAC,KAAiB,EAAE,YAAqB;IACtE,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,EAAE,YAAY,CAAC,CAAC;IACpE,MAAM,eAAe,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;AACzC,CAAC"}

package/dist/storage/usage-cap.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/storage/usage-cap.test.js

@@ -0,0 +1,51 @@
+import { promises as fs } from "node:fs";
+import * as os from "node:os";
+import * as path from "node:path";
+import { afterEach, beforeEach, describe, expect, it } from "vitest";
+import { atomicWriteJSON } from "./atomic-io.js";
+import { loadUsage, saveUsage } from "./usage-cap.js";
+let tmpDir;
+beforeEach(async () => {
+    tmpDir = await fs.mkdtemp(path.join(os.tmpdir(), "usage-test-"));
+});
+afterEach(async () => {
+    await fs.rm(tmpDir, { recursive: true, force: true });
+});
+function currentMonthUTC() {
+    const d = new Date();
+    const yyyy = d.getUTCFullYear();
+    const mm = String(d.getUTCMonth() + 1).padStart(2, "0");
+    return `${yyyy}-${mm}`;
+}
+describe("loadUsage", () => {
+    it("Test 7: bootstraps usage.json when absent", async () => {
+        const state = await loadUsage(tmpDir);
+        expect(state.current_month_utc).toBe(currentMonthUTC());
+        expect(state.current_usd).toBe(0);
+        expect(state.reservations).toEqual([]);
+        // File must be written
+        const raw = await fs.readFile(path.join(tmpDir, "usage.json"), "utf8");
+        const parsed = JSON.parse(raw);
+        expect(parsed.current_usd).toBe(0);
+    });
+    it("Test 9: auto-resets on stale month", async () => {
+        // Write a state with a past month
+        await atomicWriteJSON(path.join(tmpDir, "usage.json"), {
+            current_month_utc: "2026-03",
+            current_usd: 5.5,
+            reservations: [],
+        });
+        const state = await loadUsage(tmpDir);
+        expect(state.current_month_utc).toBe(currentMonthUTC());
+        expect(state.current_usd).toBe(0);
+    });
+});
+describe("saveUsage", () => {
+    it("Test 8: round-trips via loadUsage", async () => {
+        const initial = await loadUsage(tmpDir);
+        await saveUsage({ ...initial, current_usd: 1.23 }, tmpDir);
+        const loaded = await loadUsage(tmpDir);
+        expect(loaded.current_usd).toBe(1.23);
+    });
+});
+//# sourceMappingURL=usage-cap.test.js.map

package/dist/storage/usage-cap.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-cap.test.js","sourceRoot":"","sources":["../../src/storage/usage-cap.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAEtD,IAAI,MAAc,CAAC;AAEnB,UAAU,CAAC,KAAK,IAAI,EAAE;IACpB,MAAM,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,aAAa,CAAC,CAAC,CAAC;AACnE,CAAC,CAAC,CAAC;AAEH,SAAS,CAAC,KAAK,IAAI,EAAE;IACnB,MAAM,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;AACxD,CAAC,CAAC,CAAC;AAEH,SAAS,eAAe;IACtB,MAAM,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC;IACrB,MAAM,IAAI,GAAG,CAAC,CAAC,cAAc,EAAE,CAAC;IAChC,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IACxD,OAAO,GAAG,IAAI,IAAI,EAAE,EAAE,CAAC;AACzB,CAAC;AAED,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,CAAC;QACtC,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC;QACxD,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAEvC,uBAAuB;QACvB,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,EAAE,MAAM,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,kCAAkC;QAClC,MAAM,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,EAAE;YACrD,iBAAiB,EAAE,SAAS;YAC5B,WAAW,EAAE,GAAG;YAChB,YAAY,EAAE,EAAE;SACjB,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,CAAC;QACtC,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC;QACxD,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,SAAS,CAAC,EAAE,GAAG,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,MAAM,CAAC,CAAC;QAC3D,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/storage/usage.d.ts

@@ -0,0 +1,11 @@
+import type { UsageEvent, UsageSource } from "../types/index";
+export interface TokenUsageLike {
+    inputTokens?: number;
+    outputTokens?: number;
+    totalTokens?: number;
+    cacheReadTokens?: number;
+    cacheCreationTokens?: number;
+}
+export declare function recordUsageEvent(sessionId: string, source: UsageSource, model: string, usage?: TokenUsageLike, messageSeq?: number | null, pilActive?: boolean, enrichmentDelta?: number): void;
+export declare function getSessionTotalTokens(sessionId: string): number;
+export declare function listSessionUsage(sessionId: string): UsageEvent[];

package/dist/storage/usage.js

@@ -0,0 +1,65 @@
+import { getModelInfo } from "../models/registry.js";
+import { getDatabase } from "./db";
+export function recordUsageEvent(sessionId, source, model, usage, messageSeq, pilActive = false, enrichmentDelta = 0) {
+    if (!usage)
+        return;
+    const inputTokens = usage.inputTokens ?? 0;
+    const outputTokens = usage.outputTokens ?? 0;
+    const totalTokens = usage.totalTokens ?? inputTokens + outputTokens;
+    if (inputTokens <= 0 && outputTokens <= 0 && totalTokens <= 0)
+        return;
+    const cacheReadTokens = usage.cacheReadTokens ?? 0;
+    const cacheCreationTokens = usage.cacheCreationTokens ?? 0;
+    const costMicros = estimateCostMicros(model, inputTokens, outputTokens, cacheReadTokens);
+    getDatabase()
+        .prepare(`
+    INSERT INTO usage_events (
+      session_id, message_seq, source, model, input_tokens, output_tokens, total_tokens, cost_micros, created_at,
+      pil_active, enrichment_delta, cache_read_tokens, cache_creation_tokens
+    ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+  `)
+        .run(sessionId, messageSeq ?? null, source, model, inputTokens, outputTokens, totalTokens, costMicros, new Date().toISOString(), pilActive ? 1 : 0, enrichmentDelta, cacheReadTokens, cacheCreationTokens);
+}
+export function getSessionTotalTokens(sessionId) {
+    const row = getDatabase()
+        .prepare(`
+    SELECT COALESCE(SUM(total_tokens), 0) AS total_tokens
+    FROM usage_events
+    WHERE session_id = ?
+  `)
+        .get(sessionId);
+    return row?.total_tokens ?? 0;
+}
+export function listSessionUsage(sessionId) {
+    const rows = getDatabase()
+        .prepare(`
+    SELECT id, session_id, message_seq, source, model, input_tokens, output_tokens, total_tokens, cost_micros, created_at, cache_read_tokens, cache_creation_tokens
+    FROM usage_events
+    WHERE session_id = ?
+    ORDER BY id ASC
+  `)
+        .all(sessionId);
+    return rows.map((row) => ({
+        id: row.id,
+        sessionId: row.session_id,
+        messageSeq: row.message_seq,
+        source: row.source,
+        model: row.model,
+        inputTokens: row.input_tokens,
+        outputTokens: row.output_tokens,
+        totalTokens: row.total_tokens,
+        costMicros: row.cost_micros,
+        createdAt: new Date(row.created_at),
+        cacheReadTokens: row.cache_read_tokens ?? 0,
+        cacheCreationTokens: row.cache_creation_tokens ?? 0,
+    }));
+}
+function estimateCostMicros(model, inputTokens, outputTokens, cacheReadTokens = 0) {
+    const info = getModelInfo(model);
+    if (!info)
+        return 0;
+    const cachedPrice = info.cachedInputPrice ?? info.inputPrice * 0.1;
+    const nonCachedInput = Math.max(0, inputTokens - cacheReadTokens);
+    return Math.round(nonCachedInput * info.inputPrice + cacheReadTokens * cachedPrice + outputTokens * info.outputPrice);
+}
+//# sourceMappingURL=usage.js.map

package/dist/storage/usage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage.js","sourceRoot":"","sources":["../../src/storage/usage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAErD,OAAO,EAAE,WAAW,EAAE,MAAM,MAAM,CAAC;AAyBnC,MAAM,UAAU,gBAAgB,CAC9B,SAAiB,EACjB,MAAmB,EACnB,KAAa,EACb,KAAsB,EACtB,UAA0B,EAC1B,SAAS,GAAG,KAAK,EACjB,eAAe,GAAG,CAAC;IAEnB,IAAI,CAAC,KAAK;QAAE,OAAO;IAEnB,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC;IAC3C,MAAM,YAAY,GAAG,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC;IAC7C,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,IAAI,WAAW,GAAG,YAAY,CAAC;IACpE,IAAI,WAAW,IAAI,CAAC,IAAI,YAAY,IAAI,CAAC,IAAI,WAAW,IAAI,CAAC;QAAE,OAAO;IAEtE,MAAM,eAAe,GAAG,KAAK,CAAC,eAAe,IAAI,CAAC,CAAC;IACnD,MAAM,mBAAmB,GAAG,KAAK,CAAC,mBAAmB,IAAI,CAAC,CAAC;IAC3D,MAAM,UAAU,GAAG,kBAAkB,CAAC,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,eAAe,CAAC,CAAC;IACzF,WAAW,EAAE;SACV,OAAO,CAAC;;;;;GAKV,CAAC;SACC,GAAG,CACF,SAAS,EACT,UAAU,IAAI,IAAI,EAClB,MAAM,EACN,KAAK,EACL,WAAW,EACX,YAAY,EACZ,WAAW,EACX,UAAU,EACV,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EACxB,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EACjB,eAAe,EACf,eAAe,EACf,mBAAmB,CACpB,CAAC;AACN,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,SAAiB;IACrD,MAAM,GAAG,GAAG,WAAW,EAAE;SACtB,OAAO,CAAC;;;;GAIV,CAAC;SACC,GAAG,CAAC,SAAS,CAAyC,CAAC;IAE1D,OAAO,GAAG,EAAE,YAAY,IAAI,CAAC,CAAC;AAChC,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,SAAiB;IAChD,MAAM,IAAI,GAAG,WAAW,EAAE;SACvB,OAAO,CAAC;;;;;GAKV,CAAC;SACC,GAAG,CAAC,SAAS,CAAe,CAAC;IAEhC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACxB,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,UAAU,EAAE,GAAG,CAAC,WAAW;QAC3B,MAAM,EAAE,GAAG,CAAC,MAAM;QAClB,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,WAAW,EAAE,GAAG,CAAC,YAAY;QAC7B,YAAY,EAAE,GAAG,CAAC,aAAa;QAC/B,WAAW,EAAE,GAAG,CAAC,YAAY;QAC7B,UAAU,EAAE,GAAG,CAAC,WAAW;QAC3B,SAAS,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC;QACnC,eAAe,EAAE,GAAG,CAAC,iBAAiB,IAAI,CAAC;QAC3C,mBAAmB,EAAE,GAAG,CAAC,qBAAqB,IAAI,CAAC;KACpD,CAAC,CAAC,CAAC;AACN,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAa,EAAE,WAAmB,EAAE,YAAoB,EAAE,eAAe,GAAG,CAAC;IACvG,MAAM,IAAI,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;IACjC,IAAI,CAAC,IAAI;QAAE,OAAO,CAAC,CAAC;IACpB,MAAM,WAAW,GAAG,IAAI,CAAC,gBAAgB,IAAI,IAAI,CAAC,UAAU,GAAG,GAAG,CAAC;IACnE,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,GAAG,eAAe,CAAC,CAAC;IAClE,OAAO,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,GAAG,eAAe,GAAG,WAAW,GAAG,YAAY,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC;AACxH,CAAC"}

package/dist/storage/workspaces.d.ts

@@ -0,0 +1,9 @@
+import type { WorkspaceInfo } from "../types/index";
+export interface ResolvedWorkspace {
+    scopeKey: string;
+    canonicalPath: string;
+    gitRoot: string | null;
+    displayName: string;
+}
+export declare function ensureWorkspace(cwd: string): WorkspaceInfo;
+export declare function resolveWorkspace(cwd: string): ResolvedWorkspace;

package/dist/storage/workspaces.js

@@ -0,0 +1,60 @@
+import { createHash } from "crypto";
+import fs from "fs";
+import path from "path";
+import { findGitRoot } from "../utils/git-root";
+import { getDatabase } from "./db";
+export function ensureWorkspace(cwd) {
+    const resolved = resolveWorkspace(cwd);
+    const now = new Date().toISOString();
+    const id = createHash("sha1").update(resolved.scopeKey).digest("hex").slice(0, 16);
+    const db = getDatabase();
+    db.prepare(`
+    INSERT INTO workspaces (id, scope_key, canonical_path, git_root, display_name, last_seen_at)
+    VALUES (@id, @scope_key, @canonical_path, @git_root, @display_name, @last_seen_at)
+    ON CONFLICT(scope_key) DO UPDATE SET
+      canonical_path = excluded.canonical_path,
+      git_root = excluded.git_root,
+      display_name = excluded.display_name,
+      last_seen_at = excluded.last_seen_at
+  `).run({
+        id,
+        scope_key: resolved.scopeKey,
+        canonical_path: resolved.canonicalPath,
+        git_root: resolved.gitRoot,
+        display_name: resolved.displayName,
+        last_seen_at: now,
+    });
+    const row = db
+        .prepare(`
+    SELECT id, scope_key, canonical_path, git_root, display_name, last_seen_at
+    FROM workspaces
+    WHERE scope_key = ?
+  `)
+        .get(resolved.scopeKey);
+    if (!row) {
+        throw new Error(`Failed to resolve workspace for ${cwd}`);
+    }
+    return toWorkspaceInfo(row);
+}
+export function resolveWorkspace(cwd) {
+    const canonicalPath = fs.realpathSync.native(cwd);
+    const gitRoot = findGitRoot(canonicalPath);
+    const scopePath = gitRoot || canonicalPath;
+    return {
+        scopeKey: scopePath,
+        canonicalPath,
+        gitRoot,
+        displayName: path.basename(scopePath) || "workspace",
+    };
+}
+function toWorkspaceInfo(row) {
+    return {
+        id: row.id,
+        scopeKey: row.scope_key,
+        canonicalPath: row.canonical_path,
+        gitRoot: row.git_root,
+        displayName: row.display_name,
+        lastSeenAt: new Date(row.last_seen_at),
+    };
+}
+//# sourceMappingURL=workspaces.js.map

package/dist/storage/workspaces.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspaces.js","sourceRoot":"","sources":["../../src/storage/workspaces.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AAExB,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,WAAW,EAAE,MAAM,MAAM,CAAC;AAkBnC,MAAM,UAAU,eAAe,CAAC,GAAW;IACzC,MAAM,QAAQ,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC;IACvC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACrC,MAAM,EAAE,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACnF,MAAM,EAAE,GAAG,WAAW,EAAE,CAAC;IAEzB,EAAE,CAAC,OAAO,CAAC;;;;;;;;GAQV,CAAC,CAAC,GAAG,CAAC;QACL,EAAE;QACF,SAAS,EAAE,QAAQ,CAAC,QAAQ;QAC5B,cAAc,EAAE,QAAQ,CAAC,aAAa;QACtC,QAAQ,EAAE,QAAQ,CAAC,OAAO;QAC1B,YAAY,EAAE,QAAQ,CAAC,WAAW;QAClC,YAAY,EAAE,GAAG;KAClB,CAAC,CAAC;IAEH,MAAM,GAAG,GAAG,EAAE;SACX,OAAO,CAAC;;;;GAIV,CAAC;SACC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAA6B,CAAC;IAEtD,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,MAAM,IAAI,KAAK,CAAC,mCAAmC,GAAG,EAAE,CAAC,CAAC;IAC5D,CAAC;IAED,OAAO,eAAe,CAAC,GAAG,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,GAAW;IAC1C,MAAM,aAAa,GAAG,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAClD,MAAM,OAAO,GAAG,WAAW,CAAC,aAAa,CAAC,CAAC;IAC3C,MAAM,SAAS,GAAG,OAAO,IAAI,aAAa,CAAC;IAE3C,OAAO;QACL,QAAQ,EAAE,SAAS;QACnB,aAAa;QACb,OAAO;QACP,WAAW,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,WAAW;KACrD,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,GAAiB;IACxC,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,QAAQ,EAAE,GAAG,CAAC,SAAS;QACvB,aAAa,EAAE,GAAG,CAAC,cAAc;QACjC,OAAO,EAAE,GAAG,CAAC,QAAQ;QACrB,WAAW,EAAE,GAAG,CAAC,YAAY;QAC7B,UAAU,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC;KACvC,CAAC;AACJ,CAAC"}