@google/gemini-cli-core 0.0.1

package/dist/src/services/chatRecordingService.js

@@ -0,0 +1,330 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import {} from '../config/config.js';
+import {} from '../core/coreToolScheduler.js';
+import {} from '../core/turn.js';
+import { getProjectHash } from '../utils/paths.js';
+import path from 'node:path';
+import fs from 'node:fs';
+import { randomUUID } from 'node:crypto';
+/**
+ * Service for automatically recording chat conversations to disk.
+ *
+ * This service provides comprehensive conversation recording that captures:
+ * - All user and assistant messages
+ * - Tool calls and their execution results
+ * - Token usage statistics
+ * - Assistant thoughts and reasoning
+ *
+ * Sessions are stored as JSON files in ~/.gemini/tmp/<project_hash>/chats/
+ */
+export class ChatRecordingService {
+    conversationFile = null;
+    cachedLastConvData = null;
+    sessionId;
+    projectHash;
+    queuedThoughts = [];
+    queuedTokens = null;
+    config;
+    constructor(config) {
+        this.config = config;
+        this.sessionId = config.getSessionId();
+        this.projectHash = getProjectHash(config.getProjectRoot());
+    }
+    /**
+     * Initializes the chat recording service: creates a new conversation file and associates it with
+     * this service instance, or resumes from an existing session if resumedSessionData is provided.
+     */
+    initialize(resumedSessionData) {
+        try {
+            if (resumedSessionData) {
+                // Resume from existing session
+                this.conversationFile = resumedSessionData.filePath;
+                this.sessionId = resumedSessionData.conversation.sessionId;
+                // Update the session ID in the existing file
+                this.updateConversation((conversation) => {
+                    conversation.sessionId = this.sessionId;
+                });
+                // Clear any cached data to force fresh reads
+                this.cachedLastConvData = null;
+            }
+            else {
+                // Create new session
+                const chatsDir = path.join(this.config.storage.getProjectTempDir(), 'chats');
+                fs.mkdirSync(chatsDir, { recursive: true });
+                const timestamp = new Date()
+                    .toISOString()
+                    .slice(0, 16)
+                    .replace(/:/g, '-');
+                const filename = `session-${timestamp}-${this.sessionId.slice(0, 8)}.json`;
+                this.conversationFile = path.join(chatsDir, filename);
+                this.writeConversation({
+                    sessionId: this.sessionId,
+                    projectHash: this.projectHash,
+                    startTime: new Date().toISOString(),
+                    lastUpdated: new Date().toISOString(),
+                    messages: [],
+                });
+            }
+            // Clear any queued data since this is a fresh start
+            this.queuedThoughts = [];
+            this.queuedTokens = null;
+        }
+        catch (error) {
+            console.error('Error initializing chat recording service:', error);
+            throw error;
+        }
+    }
+    getLastMessage(conversation) {
+        return conversation.messages.at(-1);
+    }
+    newMessage(type, content) {
+        return {
+            id: randomUUID(),
+            timestamp: new Date().toISOString(),
+            type,
+            content,
+        };
+    }
+    /**
+     * Records a message in the conversation.
+     */
+    recordMessage(message) {
+        if (!this.conversationFile)
+            return;
+        try {
+            this.updateConversation((conversation) => {
+                const msg = this.newMessage(message.type, message.content);
+                if (msg.type === 'gemini') {
+                    // If it's a new Gemini message then incorporate any queued thoughts.
+                    conversation.messages.push({
+                        ...msg,
+                        thoughts: this.queuedThoughts,
+                        tokens: this.queuedTokens,
+                        model: message.model,
+                    });
+                    this.queuedThoughts = [];
+                    this.queuedTokens = null;
+                }
+                else {
+                    // Or else just add it.
+                    conversation.messages.push(msg);
+                }
+            });
+        }
+        catch (error) {
+            console.error('Error saving message:', error);
+            throw error;
+        }
+    }
+    /**
+     * Records a thought from the assistant's reasoning process.
+     */
+    recordThought(thought) {
+        if (!this.conversationFile)
+            return;
+        try {
+            this.queuedThoughts.push({
+                ...thought,
+                timestamp: new Date().toISOString(),
+            });
+        }
+        catch (error) {
+            console.error('Error saving thought:', error);
+            throw error;
+        }
+    }
+    /**
+     * Updates the tokens for the last message in the conversation (which should be by Gemini).
+     */
+    recordMessageTokens(respUsageMetadata) {
+        if (!this.conversationFile)
+            return;
+        try {
+            const tokens = {
+                input: respUsageMetadata.promptTokenCount ?? 0,
+                output: respUsageMetadata.candidatesTokenCount ?? 0,
+                cached: respUsageMetadata.cachedContentTokenCount ?? 0,
+                thoughts: respUsageMetadata.thoughtsTokenCount ?? 0,
+                tool: respUsageMetadata.toolUsePromptTokenCount ?? 0,
+                total: respUsageMetadata.totalTokenCount ?? 0,
+            };
+            this.updateConversation((conversation) => {
+                const lastMsg = this.getLastMessage(conversation);
+                // If the last message already has token info, it's because this new token info is for a
+                // new message that hasn't been recorded yet.
+                if (lastMsg && lastMsg.type === 'gemini' && !lastMsg.tokens) {
+                    lastMsg.tokens = tokens;
+                    this.queuedTokens = null;
+                }
+                else {
+                    this.queuedTokens = tokens;
+                }
+            });
+        }
+        catch (error) {
+            console.error('Error updating message tokens:', error);
+            throw error;
+        }
+    }
+    /**
+     * Adds tool calls to the last message in the conversation (which should be by Gemini).
+     * This method enriches tool calls with metadata from the ToolRegistry.
+     */
+    recordToolCalls(model, toolCalls) {
+        if (!this.conversationFile)
+            return;
+        // Enrich tool calls with metadata from the ToolRegistry
+        const toolRegistry = this.config.getToolRegistry();
+        const enrichedToolCalls = toolCalls.map((toolCall) => {
+            const toolInstance = toolRegistry.getTool(toolCall.name);
+            return {
+                ...toolCall,
+                displayName: toolInstance?.displayName || toolCall.name,
+                description: toolInstance?.description || '',
+                renderOutputAsMarkdown: toolInstance?.isOutputMarkdown || false,
+            };
+        });
+        try {
+            this.updateConversation((conversation) => {
+                const lastMsg = this.getLastMessage(conversation);
+                // If a tool call was made, but the last message isn't from Gemini, it's because Gemini is
+                // calling tools without starting the message with text.  So the user submits a prompt, and
+                // Gemini immediately calls a tool (maybe with some thinking first).  In that case, create
+                // a new empty Gemini message.
+                // Also if there are any queued thoughts, it means this tool call(s) is from a new Gemini
+                // message--because it's thought some more since we last, if ever, created a new Gemini
+                // message from tool calls, when we dequeued the thoughts.
+                if (!lastMsg ||
+                    lastMsg.type !== 'gemini' ||
+                    this.queuedThoughts.length > 0) {
+                    const newMsg = {
+                        ...this.newMessage('gemini', ''),
+                        // This isn't strictly necessary, but TypeScript apparently can't
+                        // tell that the first parameter to newMessage() becomes the
+                        // resulting message's type, and so it thinks that toolCalls may
+                        // not be present.  Confirming the type here satisfies it.
+                        type: 'gemini',
+                        toolCalls: enrichedToolCalls,
+                        thoughts: this.queuedThoughts,
+                        model,
+                    };
+                    // If there are any queued thoughts join them to this message.
+                    if (this.queuedThoughts.length > 0) {
+                        newMsg.thoughts = this.queuedThoughts;
+                        this.queuedThoughts = [];
+                    }
+                    // If there's any queued tokens info join it to this message.
+                    if (this.queuedTokens) {
+                        newMsg.tokens = this.queuedTokens;
+                        this.queuedTokens = null;
+                    }
+                    conversation.messages.push(newMsg);
+                }
+                else {
+                    // The last message is an existing Gemini message that we need to update.
+                    // Update any existing tool call entries.
+                    if (!lastMsg.toolCalls) {
+                        lastMsg.toolCalls = [];
+                    }
+                    lastMsg.toolCalls = lastMsg.toolCalls.map((toolCall) => {
+                        // If there are multiple tool calls with the same ID, this will take the first one.
+                        const incomingToolCall = toolCalls.find((tc) => tc.id === toolCall.id);
+                        if (incomingToolCall) {
+                            // Merge in the new data to keep preserve thoughts, etc., that were assigned to older
+                            // versions of the tool call.
+                            return { ...toolCall, ...incomingToolCall };
+                        }
+                        else {
+                            return toolCall;
+                        }
+                    });
+                    // Add any new tools calls that aren't in the message yet.
+                    for (const toolCall of enrichedToolCalls) {
+                        const existingToolCall = lastMsg.toolCalls.find((tc) => tc.id === toolCall.id);
+                        if (!existingToolCall) {
+                            lastMsg.toolCalls.push(toolCall);
+                        }
+                    }
+                }
+            });
+        }
+        catch (error) {
+            console.error('Error adding tool call to message:', error);
+            throw error;
+        }
+    }
+    /**
+     * Loads up the conversation record from disk.
+     */
+    readConversation() {
+        try {
+            this.cachedLastConvData = fs.readFileSync(this.conversationFile, 'utf8');
+            return JSON.parse(this.cachedLastConvData);
+        }
+        catch (error) {
+            if (error.code !== 'ENOENT') {
+                console.error('Error reading conversation file:', error);
+                throw error;
+            }
+            // Placeholder empty conversation if file doesn't exist.
+            return {
+                sessionId: this.sessionId,
+                projectHash: this.projectHash,
+                startTime: new Date().toISOString(),
+                lastUpdated: new Date().toISOString(),
+                messages: [],
+            };
+        }
+    }
+    /**
+     * Saves the conversation record; overwrites the file.
+     */
+    writeConversation(conversation) {
+        try {
+            if (!this.conversationFile)
+                return;
+            // Don't write the file yet until there's at least one message.
+            if (conversation.messages.length === 0)
+                return;
+            // Only write the file if this change would change the file.
+            if (this.cachedLastConvData !== JSON.stringify(conversation, null, 2)) {
+                conversation.lastUpdated = new Date().toISOString();
+                const newContent = JSON.stringify(conversation, null, 2);
+                this.cachedLastConvData = newContent;
+                fs.writeFileSync(this.conversationFile, newContent);
+            }
+        }
+        catch (error) {
+            console.error('Error writing conversation file:', error);
+            throw error;
+        }
+    }
+    /**
+     * Convenient helper for updating the conversation without file reading and writing and time
+     * updating boilerplate.
+     */
+    updateConversation(updateFn) {
+        const conversation = this.readConversation();
+        updateFn(conversation);
+        this.writeConversation(conversation);
+    }
+    /**
+     * Deletes a session file by session ID.
+     */
+    deleteSession(sessionId) {
+        try {
+            const chatsDir = path.join(this.config.storage.getProjectTempDir(), 'chats');
+            const sessionPath = path.join(chatsDir, `${sessionId}.json`);
+            fs.unlinkSync(sessionPath);
+        }
+        catch (error) {
+            console.error('Error deleting session:', error);
+            throw error;
+        }
+    }
+}
+//# sourceMappingURL=chatRecordingService.js.map

package/dist/src/services/chatRecordingService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatRecordingService.js","sourceRoot":"","sources":["../../../src/services/chatRecordingService.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAe,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAe,MAAM,8BAA8B,CAAC;AAC3D,OAAO,EAAuB,MAAM,iBAAiB,CAAC;AACtD,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAmFzC;;;;;;;;;;GAUG;AACH,MAAM,OAAO,oBAAoB;IACvB,gBAAgB,GAAkB,IAAI,CAAC;IACvC,kBAAkB,GAAkB,IAAI,CAAC;IACzC,SAAS,CAAS;IAClB,WAAW,CAAS;IACpB,cAAc,GAAkD,EAAE,CAAC;IACnE,YAAY,GAAyB,IAAI,CAAC;IAC1C,MAAM,CAAS;IAEvB,YAAY,MAAc;QACxB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,YAAY,EAAE,CAAC;QACvC,IAAI,CAAC,WAAW,GAAG,cAAc,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED;;;OAGG;IACH,UAAU,CAAC,kBAAuC;QAChD,IAAI,CAAC;YACH,IAAI,kBAAkB,EAAE,CAAC;gBACvB,+BAA+B;gBAC/B,IAAI,CAAC,gBAAgB,GAAG,kBAAkB,CAAC,QAAQ,CAAC;gBACpD,IAAI,CAAC,SAAS,GAAG,kBAAkB,CAAC,YAAY,CAAC,SAAS,CAAC;gBAE3D,6CAA6C;gBAC7C,IAAI,CAAC,kBAAkB,CAAC,CAAC,YAAY,EAAE,EAAE;oBACvC,YAAY,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;gBAC1C,CAAC,CAAC,CAAC;gBAEH,6CAA6C;gBAC7C,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC;YACjC,CAAC;iBAAM,CAAC;gBACN,qBAAqB;gBACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CACxB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,EACvC,OAAO,CACR,CAAC;gBACF,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAE5C,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE;qBACzB,WAAW,EAAE;qBACb,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;qBACZ,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;gBACtB,MAAM,QAAQ,GAAG,WAAW,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAC3D,CAAC,EACD,CAAC,CACF,OAAO,CAAC;gBACT,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;gBAEtD,IAAI,CAAC,iBAAiB,CAAC;oBACrB,SAAS,EAAE,IAAI,CAAC,SAAS;oBACzB,WAAW,EAAE,IAAI,CAAC,WAAW;oBAC7B,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACnC,WAAW,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACrC,QAAQ,EAAE,EAAE;iBACb,CAAC,CAAC;YACL,CAAC;YAED,oDAAoD;YACpD,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;YACzB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAC3B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,4CAA4C,EAAE,KAAK,CAAC,CAAC;YACnE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEO,cAAc,CACpB,YAAgC;QAEhC,OAAO,YAAY,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACtC,CAAC;IAEO,UAAU,CAChB,IAAqC,EACrC,OAAsB;QAEtB,OAAO;YACL,EAAE,EAAE,UAAU,EAAE;YAChB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,IAAI;YACJ,OAAO;SACR,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,OAIb;QACC,IAAI,CAAC,IAAI,CAAC,gBAAgB;YAAE,OAAO;QAEnC,IAAI,CAAC;YACH,IAAI,CAAC,kBAAkB,CAAC,CAAC,YAAY,EAAE,EAAE;gBACvC,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;gBAC3D,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;oBAC1B,qEAAqE;oBACrE,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC;wBACzB,GAAG,GAAG;wBACN,QAAQ,EAAE,IAAI,CAAC,cAAc;wBAC7B,MAAM,EAAE,IAAI,CAAC,YAAY;wBACzB,KAAK,EAAE,OAAO,CAAC,KAAK;qBACrB,CAAC,CAAC;oBACH,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;oBACzB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;gBAC3B,CAAC;qBAAM,CAAC;oBACN,uBAAuB;oBACvB,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAClC,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAK,CAAC,CAAC;YAC9C,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,OAAuB;QACnC,IAAI,CAAC,IAAI,CAAC,gBAAgB;YAAE,OAAO;QAEnC,IAAI,CAAC;YACH,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC;gBACvB,GAAG,OAAO;gBACV,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAK,CAAC,CAAC;YAC9C,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,mBAAmB,CACjB,iBAAuD;QAEvD,IAAI,CAAC,IAAI,CAAC,gBAAgB;YAAE,OAAO;QAEnC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG;gBACb,KAAK,EAAE,iBAAiB,CAAC,gBAAgB,IAAI,CAAC;gBAC9C,MAAM,EAAE,iBAAiB,CAAC,oBAAoB,IAAI,CAAC;gBACnD,MAAM,EAAE,iBAAiB,CAAC,uBAAuB,IAAI,CAAC;gBACtD,QAAQ,EAAE,iBAAiB,CAAC,kBAAkB,IAAI,CAAC;gBACnD,IAAI,EAAE,iBAAiB,CAAC,uBAAuB,IAAI,CAAC;gBACpD,KAAK,EAAE,iBAAiB,CAAC,eAAe,IAAI,CAAC;aAC9C,CAAC;YACF,IAAI,CAAC,kBAAkB,CAAC,CAAC,YAAY,EAAE,EAAE;gBACvC,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;gBAClD,wFAAwF;gBACxF,6CAA6C;gBAC7C,IAAI,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;oBAC5D,OAAO,CAAC,MAAM,GAAG,MAAM,CAAC;oBACxB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;gBAC3B,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC;gBAC7B,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAC;YACvD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,eAAe,CAAC,KAAa,EAAE,SAA2B;QACxD,IAAI,CAAC,IAAI,CAAC,gBAAgB;YAAE,OAAO;QAEnC,wDAAwD;QACxD,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;QACnD,MAAM,iBAAiB,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;YACnD,MAAM,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YACzD,OAAO;gBACL,GAAG,QAAQ;gBACX,WAAW,EAAE,YAAY,EAAE,WAAW,IAAI,QAAQ,CAAC,IAAI;gBACvD,WAAW,EAAE,YAAY,EAAE,WAAW,IAAI,EAAE;gBAC5C,sBAAsB,EAAE,YAAY,EAAE,gBAAgB,IAAI,KAAK;aAChE,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,IAAI,CAAC,kBAAkB,CAAC,CAAC,YAAY,EAAE,EAAE;gBACvC,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;gBAClD,0FAA0F;gBAC1F,2FAA2F;gBAC3F,0FAA0F;gBAC1F,8BAA8B;gBAC9B,yFAAyF;gBACzF,uFAAuF;gBACvF,0DAA0D;gBAC1D,IACE,CAAC,OAAO;oBACR,OAAO,CAAC,IAAI,KAAK,QAAQ;oBACzB,IAAI,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,EAC9B,CAAC;oBACD,MAAM,MAAM,GAAkB;wBAC5B,GAAG,IAAI,CAAC,UAAU,CAAC,QAAiB,EAAE,EAAE,CAAC;wBACzC,iEAAiE;wBACjE,4DAA4D;wBAC5D,gEAAgE;wBAChE,0DAA0D;wBAC1D,IAAI,EAAE,QAAiB;wBACvB,SAAS,EAAE,iBAAiB;wBAC5B,QAAQ,EAAE,IAAI,CAAC,cAAc;wBAC7B,KAAK;qBACN,CAAC;oBACF,8DAA8D;oBAC9D,IAAI,IAAI,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACnC,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC;wBACtC,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;oBAC3B,CAAC;oBACD,6DAA6D;oBAC7D,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;wBACtB,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC;wBAClC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;oBAC3B,CAAC;oBACD,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACrC,CAAC;qBAAM,CAAC;oBACN,yEAAyE;oBAEzE,yCAAyC;oBACzC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;wBACvB,OAAO,CAAC,SAAS,GAAG,EAAE,CAAC;oBACzB,CAAC;oBACD,OAAO,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;wBACrD,mFAAmF;wBACnF,MAAM,gBAAgB,GAAG,SAAS,CAAC,IAAI,CACrC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,QAAQ,CAAC,EAAE,CAC9B,CAAC;wBACF,IAAI,gBAAgB,EAAE,CAAC;4BACrB,qFAAqF;4BACrF,6BAA6B;4BAC7B,OAAO,EAAE,GAAG,QAAQ,EAAE,GAAG,gBAAgB,EAAE,CAAC;wBAC9C,CAAC;6BAAM,CAAC;4BACN,OAAO,QAAQ,CAAC;wBAClB,CAAC;oBACH,CAAC,CAAC,CAAC;oBAEH,0DAA0D;oBAC1D,KAAK,MAAM,QAAQ,IAAI,iBAAiB,EAAE,CAAC;wBACzC,MAAM,gBAAgB,GAAG,OAAO,CAAC,SAAS,CAAC,IAAI,CAC7C,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,QAAQ,CAAC,EAAE,CAC9B,CAAC;wBACF,IAAI,CAAC,gBAAgB,EAAE,CAAC;4BACtB,OAAO,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;wBACnC,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;YAC3D,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACK,gBAAgB;QACtB,IAAI,CAAC;YACH,IAAI,CAAC,kBAAkB,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,gBAAiB,EAAE,MAAM,CAAC,CAAC;YAC1E,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAC7C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAK,KAA+B,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACvD,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;gBACzD,MAAM,KAAK,CAAC;YACd,CAAC;YAED,wDAAwD;YACxD,OAAO;gBACL,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,WAAW,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACrC,QAAQ,EAAE,EAAE;aACb,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,YAAgC;QACxD,IAAI,CAAC;YACH,IAAI,CAAC,IAAI,CAAC,gBAAgB;gBAAE,OAAO;YACnC,+DAA+D;YAC/D,IAAI,YAAY,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO;YAE/C,4DAA4D;YAC5D,IAAI,IAAI,CAAC,kBAAkB,KAAK,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;gBACtE,YAAY,CAAC,WAAW,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;gBACpD,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;gBACzD,IAAI,CAAC,kBAAkB,GAAG,UAAU,CAAC;gBACrC,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,gBAAgB,EAAE,UAAU,CAAC,CAAC;YACtD,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;YACzD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,kBAAkB,CACxB,QAAoD;QAEpD,MAAM,YAAY,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC7C,QAAQ,CAAC,YAAY,CAAC,CAAC;QACvB,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,SAAiB;QAC7B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CACxB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,EACvC,OAAO,CACR,CAAC;YACF,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,SAAS,OAAO,CAAC,CAAC;YAC7D,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QAC7B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;YAChD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF"}

package/dist/src/services/chatRecordingService.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/services/chatRecordingService.test.js

@@ -0,0 +1,332 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { expect, it, describe, vi, beforeEach, afterEach } from 'vitest';
+import fs from 'node:fs';
+import path from 'node:path';
+import { randomUUID } from 'node:crypto';
+import { ChatRecordingService } from './chatRecordingService.js';
+import { getProjectHash } from '../utils/paths.js';
+vi.mock('node:fs');
+vi.mock('node:path');
+vi.mock('node:crypto', () => ({
+    randomUUID: vi.fn(),
+    createHash: vi.fn(() => ({
+        update: vi.fn(() => ({
+            digest: vi.fn(() => 'mocked-hash'),
+        })),
+    })),
+}));
+vi.mock('../utils/paths.js');
+describe('ChatRecordingService', () => {
+    let chatRecordingService;
+    let mockConfig;
+    let mkdirSyncSpy;
+    let writeFileSyncSpy;
+    beforeEach(() => {
+        mockConfig = {
+            getSessionId: vi.fn().mockReturnValue('test-session-id'),
+            getProjectRoot: vi.fn().mockReturnValue('/test/project/root'),
+            storage: {
+                getProjectTempDir: vi
+                    .fn()
+                    .mockReturnValue('/test/project/root/.gemini/tmp'),
+            },
+            getModel: vi.fn().mockReturnValue('gemini-pro'),
+            getDebugMode: vi.fn().mockReturnValue(false),
+            getToolRegistry: vi.fn().mockReturnValue({
+                getTool: vi.fn().mockReturnValue({
+                    displayName: 'Test Tool',
+                    description: 'A test tool',
+                    isOutputMarkdown: false,
+                }),
+            }),
+        };
+        vi.mocked(getProjectHash).mockReturnValue('test-project-hash');
+        vi.mocked(randomUUID).mockReturnValue('this-is-a-test-uuid');
+        vi.mocked(path.join).mockImplementation((...args) => args.join('/'));
+        chatRecordingService = new ChatRecordingService(mockConfig);
+        mkdirSyncSpy = vi
+            .spyOn(fs, 'mkdirSync')
+            .mockImplementation(() => undefined);
+        writeFileSyncSpy = vi
+            .spyOn(fs, 'writeFileSync')
+            .mockImplementation(() => undefined);
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    describe('initialize', () => {
+        it('should create a new session if none is provided', () => {
+            chatRecordingService.initialize();
+            expect(mkdirSyncSpy).toHaveBeenCalledWith('/test/project/root/.gemini/tmp/chats', { recursive: true });
+            expect(writeFileSyncSpy).not.toHaveBeenCalled();
+        });
+        it('should resume from an existing session if provided', () => {
+            const readFileSyncSpy = vi.spyOn(fs, 'readFileSync').mockReturnValue(JSON.stringify({
+                sessionId: 'old-session-id',
+                projectHash: 'test-project-hash',
+                messages: [],
+            }));
+            const writeFileSyncSpy = vi
+                .spyOn(fs, 'writeFileSync')
+                .mockImplementation(() => undefined);
+            chatRecordingService.initialize({
+                filePath: '/test/project/root/.gemini/tmp/chats/session.json',
+                conversation: {
+                    sessionId: 'old-session-id',
+                },
+            });
+            expect(mkdirSyncSpy).not.toHaveBeenCalled();
+            expect(readFileSyncSpy).toHaveBeenCalled();
+            expect(writeFileSyncSpy).not.toHaveBeenCalled();
+        });
+    });
+    describe('recordMessage', () => {
+        beforeEach(() => {
+            chatRecordingService.initialize();
+            vi.spyOn(fs, 'readFileSync').mockReturnValue(JSON.stringify({
+                sessionId: 'test-session-id',
+                projectHash: 'test-project-hash',
+                messages: [],
+            }));
+        });
+        it('should record a new message', () => {
+            const writeFileSyncSpy = vi
+                .spyOn(fs, 'writeFileSync')
+                .mockImplementation(() => undefined);
+            chatRecordingService.recordMessage({
+                type: 'user',
+                content: 'Hello',
+                model: 'gemini-pro',
+            });
+            expect(mkdirSyncSpy).toHaveBeenCalled();
+            expect(writeFileSyncSpy).toHaveBeenCalled();
+            const conversation = JSON.parse(writeFileSyncSpy.mock.calls[0][1]);
+            expect(conversation.messages).toHaveLength(1);
+            expect(conversation.messages[0].content).toBe('Hello');
+            expect(conversation.messages[0].type).toBe('user');
+        });
+        it('should create separate messages when recording multiple messages', () => {
+            const writeFileSyncSpy = vi
+                .spyOn(fs, 'writeFileSync')
+                .mockImplementation(() => undefined);
+            const initialConversation = {
+                sessionId: 'test-session-id',
+                projectHash: 'test-project-hash',
+                messages: [
+                    {
+                        id: '1',
+                        type: 'user',
+                        content: 'Hello',
+                        timestamp: new Date().toISOString(),
+                    },
+                ],
+            };
+            vi.spyOn(fs, 'readFileSync').mockReturnValue(JSON.stringify(initialConversation));
+            chatRecordingService.recordMessage({
+                type: 'user',
+                content: 'World',
+                model: 'gemini-pro',
+            });
+            expect(mkdirSyncSpy).toHaveBeenCalled();
+            expect(writeFileSyncSpy).toHaveBeenCalled();
+            const conversation = JSON.parse(writeFileSyncSpy.mock.calls[0][1]);
+            expect(conversation.messages).toHaveLength(2);
+            expect(conversation.messages[0].content).toBe('Hello');
+            expect(conversation.messages[1].content).toBe('World');
+        });
+    });
+    describe('recordThought', () => {
+        it('should queue a thought', () => {
+            chatRecordingService.initialize();
+            chatRecordingService.recordThought({
+                subject: 'Thinking',
+                description: 'Thinking...',
+            });
+            // @ts-expect-error private property
+            expect(chatRecordingService.queuedThoughts).toHaveLength(1);
+            // @ts-expect-error private property
+            expect(chatRecordingService.queuedThoughts[0].subject).toBe('Thinking');
+            // @ts-expect-error private property
+            expect(chatRecordingService.queuedThoughts[0].description).toBe('Thinking...');
+        });
+    });
+    describe('recordMessageTokens', () => {
+        beforeEach(() => {
+            chatRecordingService.initialize();
+        });
+        it('should update the last message with token info', () => {
+            const writeFileSyncSpy = vi
+                .spyOn(fs, 'writeFileSync')
+                .mockImplementation(() => undefined);
+            const initialConversation = {
+                sessionId: 'test-session-id',
+                projectHash: 'test-project-hash',
+                messages: [
+                    {
+                        id: '1',
+                        type: 'gemini',
+                        content: 'Response',
+                        timestamp: new Date().toISOString(),
+                    },
+                ],
+            };
+            vi.spyOn(fs, 'readFileSync').mockReturnValue(JSON.stringify(initialConversation));
+            chatRecordingService.recordMessageTokens({
+                promptTokenCount: 1,
+                candidatesTokenCount: 2,
+                totalTokenCount: 3,
+                cachedContentTokenCount: 0,
+            });
+            expect(mkdirSyncSpy).toHaveBeenCalled();
+            expect(writeFileSyncSpy).toHaveBeenCalled();
+            const conversation = JSON.parse(writeFileSyncSpy.mock.calls[0][1]);
+            expect(conversation.messages[0]).toEqual({
+                ...initialConversation.messages[0],
+                tokens: {
+                    input: 1,
+                    output: 2,
+                    total: 3,
+                    cached: 0,
+                    thoughts: 0,
+                    tool: 0,
+                },
+            });
+        });
+        it('should queue token info if the last message already has tokens', () => {
+            const initialConversation = {
+                sessionId: 'test-session-id',
+                projectHash: 'test-project-hash',
+                messages: [
+                    {
+                        id: '1',
+                        type: 'gemini',
+                        content: 'Response',
+                        timestamp: new Date().toISOString(),
+                        tokens: { input: 1, output: 1, total: 2, cached: 0 },
+                    },
+                ],
+            };
+            vi.spyOn(fs, 'readFileSync').mockReturnValue(JSON.stringify(initialConversation));
+            chatRecordingService.recordMessageTokens({
+                promptTokenCount: 2,
+                candidatesTokenCount: 2,
+                totalTokenCount: 4,
+                cachedContentTokenCount: 0,
+            });
+            // @ts-expect-error private property
+            expect(chatRecordingService.queuedTokens).toEqual({
+                input: 2,
+                output: 2,
+                total: 4,
+                cached: 0,
+                thoughts: 0,
+                tool: 0,
+            });
+        });
+    });
+    describe('recordToolCalls', () => {
+        beforeEach(() => {
+            chatRecordingService.initialize();
+        });
+        it('should add new tool calls to the last message', () => {
+            const writeFileSyncSpy = vi
+                .spyOn(fs, 'writeFileSync')
+                .mockImplementation(() => undefined);
+            const initialConversation = {
+                sessionId: 'test-session-id',
+                projectHash: 'test-project-hash',
+                messages: [
+                    {
+                        id: '1',
+                        type: 'gemini',
+                        content: '',
+                        timestamp: new Date().toISOString(),
+                    },
+                ],
+            };
+            vi.spyOn(fs, 'readFileSync').mockReturnValue(JSON.stringify(initialConversation));
+            const toolCall = {
+                id: 'tool-1',
+                name: 'testTool',
+                args: {},
+                status: 'awaiting_approval',
+                timestamp: new Date().toISOString(),
+            };
+            chatRecordingService.recordToolCalls('gemini-pro', [toolCall]);
+            expect(mkdirSyncSpy).toHaveBeenCalled();
+            expect(writeFileSyncSpy).toHaveBeenCalled();
+            const conversation = JSON.parse(writeFileSyncSpy.mock.calls[0][1]);
+            expect(conversation.messages[0]).toEqual({
+                ...initialConversation.messages[0],
+                toolCalls: [
+                    {
+                        ...toolCall,
+                        displayName: 'Test Tool',
+                        description: 'A test tool',
+                        renderOutputAsMarkdown: false,
+                    },
+                ],
+            });
+        });
+        it('should create a new message if the last message is not from gemini', () => {
+            const writeFileSyncSpy = vi
+                .spyOn(fs, 'writeFileSync')
+                .mockImplementation(() => undefined);
+            const initialConversation = {
+                sessionId: 'test-session-id',
+                projectHash: 'test-project-hash',
+                messages: [
+                    {
+                        id: 'a-uuid',
+                        type: 'user',
+                        content: 'call a tool',
+                        timestamp: new Date().toISOString(),
+                    },
+                ],
+            };
+            vi.spyOn(fs, 'readFileSync').mockReturnValue(JSON.stringify(initialConversation));
+            const toolCall = {
+                id: 'tool-1',
+                name: 'testTool',
+                args: {},
+                status: 'awaiting_approval',
+                timestamp: new Date().toISOString(),
+            };
+            chatRecordingService.recordToolCalls('gemini-pro', [toolCall]);
+            expect(mkdirSyncSpy).toHaveBeenCalled();
+            expect(writeFileSyncSpy).toHaveBeenCalled();
+            const conversation = JSON.parse(writeFileSyncSpy.mock.calls[0][1]);
+            expect(conversation.messages).toHaveLength(2);
+            expect(conversation.messages[1]).toEqual({
+                ...conversation.messages[1],
+                id: 'this-is-a-test-uuid',
+                model: 'gemini-pro',
+                type: 'gemini',
+                thoughts: [],
+                content: '',
+                toolCalls: [
+                    {
+                        ...toolCall,
+                        displayName: 'Test Tool',
+                        description: 'A test tool',
+                        renderOutputAsMarkdown: false,
+                    },
+                ],
+            });
+        });
+    });
+    describe('deleteSession', () => {
+        it('should delete the session file', () => {
+            const unlinkSyncSpy = vi
+                .spyOn(fs, 'unlinkSync')
+                .mockImplementation(() => undefined);
+            chatRecordingService.deleteSession('test-session-id');
+            expect(unlinkSyncSpy).toHaveBeenCalledWith('/test/project/root/.gemini/tmp/chats/test-session-id.json');
+        });
+    });
+});
+//# sourceMappingURL=chatRecordingService.test.js.map