@google/gemini-cli-core 0.0.3-preview.4

package/dist/src/mcp/token-storage/base-token-storage.test.js

@@ -0,0 +1,160 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach } from 'vitest';
+import { BaseTokenStorage } from './base-token-storage.js';
+class TestTokenStorage extends BaseTokenStorage {
+    storage = new Map();
+    async getCredentials(serverName) {
+        return this.storage.get(serverName) || null;
+    }
+    async setCredentials(credentials) {
+        this.validateCredentials(credentials);
+        this.storage.set(credentials.serverName, credentials);
+    }
+    async deleteCredentials(serverName) {
+        this.storage.delete(serverName);
+    }
+    async listServers() {
+        return Array.from(this.storage.keys());
+    }
+    async getAllCredentials() {
+        return new Map(this.storage);
+    }
+    async clearAll() {
+        this.storage.clear();
+    }
+    validateCredentials(credentials) {
+        super.validateCredentials(credentials);
+    }
+    isTokenExpired(credentials) {
+        return super.isTokenExpired(credentials);
+    }
+    sanitizeServerName(serverName) {
+        return super.sanitizeServerName(serverName);
+    }
+}
+describe('BaseTokenStorage', () => {
+    let storage;
+    beforeEach(() => {
+        storage = new TestTokenStorage();
+    });
+    describe('validateCredentials', () => {
+        it('should validate valid credentials', () => {
+            const credentials = {
+                serverName: 'test-server',
+                token: {
+                    accessToken: 'access-token',
+                    tokenType: 'Bearer',
+                },
+                updatedAt: Date.now(),
+            };
+            expect(() => storage.validateCredentials(credentials)).not.toThrow();
+        });
+        it('should throw for missing server name', () => {
+            const credentials = {
+                serverName: '',
+                token: {
+                    accessToken: 'access-token',
+                    tokenType: 'Bearer',
+                },
+                updatedAt: Date.now(),
+            };
+            expect(() => storage.validateCredentials(credentials)).toThrow('Server name is required');
+        });
+        it('should throw for missing token', () => {
+            const credentials = {
+                serverName: 'test-server',
+                token: null,
+                updatedAt: Date.now(),
+            };
+            expect(() => storage.validateCredentials(credentials)).toThrow('Token is required');
+        });
+        it('should throw for missing access token', () => {
+            const credentials = {
+                serverName: 'test-server',
+                token: {
+                    accessToken: '',
+                    tokenType: 'Bearer',
+                },
+                updatedAt: Date.now(),
+            };
+            expect(() => storage.validateCredentials(credentials)).toThrow('Access token is required');
+        });
+        it('should throw for missing token type', () => {
+            const credentials = {
+                serverName: 'test-server',
+                token: {
+                    accessToken: 'access-token',
+                    tokenType: '',
+                },
+                updatedAt: Date.now(),
+            };
+            expect(() => storage.validateCredentials(credentials)).toThrow('Token type is required');
+        });
+    });
+    describe('isTokenExpired', () => {
+        it('should return false for tokens without expiry', () => {
+            const credentials = {
+                serverName: 'test-server',
+                token: {
+                    accessToken: 'access-token',
+                    tokenType: 'Bearer',
+                },
+                updatedAt: Date.now(),
+            };
+            expect(storage.isTokenExpired(credentials)).toBe(false);
+        });
+        it('should return false for valid tokens', () => {
+            const credentials = {
+                serverName: 'test-server',
+                token: {
+                    accessToken: 'access-token',
+                    tokenType: 'Bearer',
+                    expiresAt: Date.now() + 3600000,
+                },
+                updatedAt: Date.now(),
+            };
+            expect(storage.isTokenExpired(credentials)).toBe(false);
+        });
+        it('should return true for expired tokens', () => {
+            const credentials = {
+                serverName: 'test-server',
+                token: {
+                    accessToken: 'access-token',
+                    tokenType: 'Bearer',
+                    expiresAt: Date.now() - 3600000,
+                },
+                updatedAt: Date.now(),
+            };
+            expect(storage.isTokenExpired(credentials)).toBe(true);
+        });
+        it('should apply 5-minute buffer for expiry check', () => {
+            const fourMinutesFromNow = Date.now() + 4 * 60 * 1000;
+            const credentials = {
+                serverName: 'test-server',
+                token: {
+                    accessToken: 'access-token',
+                    tokenType: 'Bearer',
+                    expiresAt: fourMinutesFromNow,
+                },
+                updatedAt: Date.now(),
+            };
+            expect(storage.isTokenExpired(credentials)).toBe(true);
+        });
+    });
+    describe('sanitizeServerName', () => {
+        it('should keep valid characters', () => {
+            expect(storage.sanitizeServerName('test-server.example_123')).toBe('test-server.example_123');
+        });
+        it('should replace invalid characters with underscore', () => {
+            expect(storage.sanitizeServerName('test@server#example')).toBe('test_server_example');
+        });
+        it('should handle special characters', () => {
+            expect(storage.sanitizeServerName('test server/example:123')).toBe('test_server_example_123');
+        });
+    });
+});
+//# sourceMappingURL=base-token-storage.test.js.map

package/dist/src/mcp/token-storage/base-token-storage.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base-token-storage.test.js","sourceRoot":"","sources":["../../../../src/mcp/token-storage/base-token-storage.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC1D,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAG3D,MAAM,gBAAiB,SAAQ,gBAAgB;IACrC,OAAO,GAAG,IAAI,GAAG,EAA4B,CAAC;IAEtD,KAAK,CAAC,cAAc,CAAC,UAAkB;QACrC,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,WAA6B;QAChD,IAAI,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC;QACtC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;IACxD,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,UAAkB;QACxC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,WAAW;QACf,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,iBAAiB;QACrB,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAEQ,mBAAmB,CAAC,WAA6B;QACxD,KAAK,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC;IACzC,CAAC;IAEQ,cAAc,CAAC,WAA6B;QACnD,OAAO,KAAK,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;IAC3C,CAAC;IAEQ,kBAAkB,CAAC,UAAkB;QAC5C,OAAO,KAAK,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC;IAC9C,CAAC;CACF;AAED,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,OAAyB,CAAC;IAE9B,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,GAAG,IAAI,gBAAgB,EAAE,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,WAAW,GAAqB;gBACpC,UAAU,EAAE,aAAa;gBACzB,KAAK,EAAE;oBACL,WAAW,EAAE,cAAc;oBAC3B,SAAS,EAAE,QAAQ;iBACpB;gBACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC;YAEF,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,WAAW,GAAG;gBAClB,UAAU,EAAE,EAAE;gBACd,KAAK,EAAE;oBACL,WAAW,EAAE,cAAc;oBAC3B,SAAS,EAAE,QAAQ;iBACpB;gBACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACF,CAAC;YAEtB,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAC5D,yBAAyB,CAC1B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,WAAW,GAAG;gBAClB,UAAU,EAAE,aAAa;gBACzB,KAAK,EAAE,IAA6B;gBACpC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACF,CAAC;YAEtB,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAC5D,mBAAmB,CACpB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,WAAW,GAAG;gBAClB,UAAU,EAAE,aAAa;gBACzB,KAAK,EAAE;oBACL,WAAW,EAAE,EAAE;oBACf,SAAS,EAAE,QAAQ;iBACpB;gBACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACF,CAAC;YAEtB,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAC5D,0BAA0B,CAC3B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,WAAW,GAAG;gBAClB,UAAU,EAAE,aAAa;gBACzB,KAAK,EAAE;oBACL,WAAW,EAAE,cAAc;oBAC3B,SAAS,EAAE,EAAE;iBACd;gBACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACF,CAAC;YAEtB,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAC5D,wBAAwB,CACzB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,WAAW,GAAqB;gBACpC,UAAU,EAAE,aAAa;gBACzB,KAAK,EAAE;oBACL,WAAW,EAAE,cAAc;oBAC3B,SAAS,EAAE,QAAQ;iBACpB;gBACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC;YAEF,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,WAAW,GAAqB;gBACpC,UAAU,EAAE,aAAa;gBACzB,KAAK,EAAE;oBACL,WAAW,EAAE,cAAc;oBAC3B,SAAS,EAAE,QAAQ;oBACnB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO;iBAChC;gBACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC;YAEF,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,WAAW,GAAqB;gBACpC,UAAU,EAAE,aAAa;gBACzB,KAAK,EAAE;oBACL,WAAW,EAAE,cAAc;oBAC3B,SAAS,EAAE,QAAQ;oBACnB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO;iBAChC;gBACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC;YAEF,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,kBAAkB,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;YACtD,MAAM,WAAW,GAAqB;gBACpC,UAAU,EAAE,aAAa;gBACzB,KAAK,EAAE;oBACL,WAAW,EAAE,cAAc;oBAC3B,SAAS,EAAE,QAAQ;oBACnB,SAAS,EAAE,kBAAkB;iBAC9B;gBACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC;YAEF,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAChE,yBAAyB,CAC1B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAC5D,qBAAqB,CACtB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAChE,yBAAyB,CAC1B,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/mcp/token-storage/types.d.ts

@@ -0,0 +1,34 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Interface for OAuth tokens.
+ */
+export interface OAuthToken {
+    accessToken: string;
+    refreshToken?: string;
+    expiresAt?: number;
+    tokenType: string;
+    scope?: string;
+}
+/**
+ * Interface for stored OAuth credentials.
+ */
+export interface OAuthCredentials {
+    serverName: string;
+    token: OAuthToken;
+    clientId?: string;
+    tokenUrl?: string;
+    mcpServerUrl?: string;
+    updatedAt: number;
+}
+export interface TokenStorage {
+    getCredentials(serverName: string): Promise<OAuthCredentials | null>;
+    setCredentials(credentials: OAuthCredentials): Promise<void>;
+    deleteCredentials(serverName: string): Promise<void>;
+    listServers(): Promise<string[]>;
+    getAllCredentials(): Promise<Map<string, OAuthCredentials>>;
+    clearAll(): Promise<void>;
+}

package/dist/src/mcp/token-storage/types.js

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};
+//# sourceMappingURL=types.js.map

package/dist/src/mcp/token-storage/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../src/mcp/token-storage/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG"}

package/dist/src/mocks/msw.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare const server: import("msw/node").SetupServerApi;

package/dist/src/mocks/msw.js

@@ -0,0 +1,8 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { setupServer } from 'msw/node';
+export const server = setupServer();
+//# sourceMappingURL=msw.js.map

package/dist/src/mocks/msw.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"msw.js","sourceRoot":"","sources":["../../../src/mocks/msw.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAEvC,MAAM,CAAC,MAAM,MAAM,GAAG,WAAW,EAAE,CAAC"}

package/dist/src/prompts/mcp-prompts.d.ts

@@ -0,0 +1,8 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config } from '../config/config.js';
+import type { DiscoveredMCPPrompt } from '../tools/mcp-client.js';
+export declare function getMCPServerPrompts(config: Config, serverName: string): DiscoveredMCPPrompt[];

package/dist/src/prompts/mcp-prompts.js

@@ -0,0 +1,13 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export function getMCPServerPrompts(config, serverName) {
+    const promptRegistry = config.getPromptRegistry();
+    if (!promptRegistry) {
+        return [];
+    }
+    return promptRegistry.getPromptsByServer(serverName);
+}
+//# sourceMappingURL=mcp-prompts.js.map

package/dist/src/prompts/mcp-prompts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-prompts.js","sourceRoot":"","sources":["../../../src/prompts/mcp-prompts.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH,MAAM,UAAU,mBAAmB,CACjC,MAAc,EACd,UAAkB;IAElB,MAAM,cAAc,GAAG,MAAM,CAAC,iBAAiB,EAAE,CAAC;IAClD,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,OAAO,cAAc,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC;AACvD,CAAC"}

package/dist/src/prompts/prompt-registry.d.ts

@@ -0,0 +1,34 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { DiscoveredMCPPrompt } from '../tools/mcp-client.js';
+export declare class PromptRegistry {
+    private prompts;
+    /**
+     * Registers a prompt definition.
+     * @param prompt - The prompt object containing schema and execution logic.
+     */
+    registerPrompt(prompt: DiscoveredMCPPrompt): void;
+    /**
+     * Returns an array of all registered and discovered prompt instances.
+     */
+    getAllPrompts(): DiscoveredMCPPrompt[];
+    /**
+     * Get the definition of a specific prompt.
+     */
+    getPrompt(name: string): DiscoveredMCPPrompt | undefined;
+    /**
+     * Returns an array of prompts registered from a specific MCP server.
+     */
+    getPromptsByServer(serverName: string): DiscoveredMCPPrompt[];
+    /**
+     * Clears all the prompts from the registry.
+     */
+    clear(): void;
+    /**
+     * Removes all prompts from a specific server.
+     */
+    removePromptsByServer(serverName: string): void;
+}

package/dist/src/prompts/prompt-registry.js

@@ -0,0 +1,63 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export class PromptRegistry {
+    prompts = new Map();
+    /**
+     * Registers a prompt definition.
+     * @param prompt - The prompt object containing schema and execution logic.
+     */
+    registerPrompt(prompt) {
+        if (this.prompts.has(prompt.name)) {
+            const newName = `${prompt.serverName}_${prompt.name}`;
+            console.warn(`Prompt with name "${prompt.name}" is already registered. Renaming to "${newName}".`);
+            this.prompts.set(newName, { ...prompt, name: newName });
+        }
+        else {
+            this.prompts.set(prompt.name, prompt);
+        }
+    }
+    /**
+     * Returns an array of all registered and discovered prompt instances.
+     */
+    getAllPrompts() {
+        return Array.from(this.prompts.values()).sort((a, b) => a.name.localeCompare(b.name));
+    }
+    /**
+     * Get the definition of a specific prompt.
+     */
+    getPrompt(name) {
+        return this.prompts.get(name);
+    }
+    /**
+     * Returns an array of prompts registered from a specific MCP server.
+     */
+    getPromptsByServer(serverName) {
+        const serverPrompts = [];
+        for (const prompt of this.prompts.values()) {
+            if (prompt.serverName === serverName) {
+                serverPrompts.push(prompt);
+            }
+        }
+        return serverPrompts.sort((a, b) => a.name.localeCompare(b.name));
+    }
+    /**
+     * Clears all the prompts from the registry.
+     */
+    clear() {
+        this.prompts.clear();
+    }
+    /**
+     * Removes all prompts from a specific server.
+     */
+    removePromptsByServer(serverName) {
+        for (const [name, prompt] of this.prompts.entries()) {
+            if (prompt.serverName === serverName) {
+                this.prompts.delete(name);
+            }
+        }
+    }
+}
+//# sourceMappingURL=prompt-registry.js.map

package/dist/src/prompts/prompt-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-registry.js","sourceRoot":"","sources":["../../../src/prompts/prompt-registry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,MAAM,OAAO,cAAc;IACjB,OAAO,GAAqC,IAAI,GAAG,EAAE,CAAC;IAE9D;;;OAGG;IACH,cAAc,CAAC,MAA2B;QACxC,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,GAAG,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YACtD,OAAO,CAAC,IAAI,CACV,qBAAqB,MAAM,CAAC,IAAI,yCAAyC,OAAO,IAAI,CACrF,CAAC;YACF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAC1D,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACrD,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAC7B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,IAAY;QACpB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,UAAkB;QACnC,MAAM,aAAa,GAA0B,EAAE,CAAC;QAChD,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,IAAI,MAAM,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;gBACrC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;QACD,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IACpE,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,qBAAqB,CAAC,UAAkB;QACtC,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YACpD,IAAI,MAAM,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;gBACrC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;IACH,CAAC;CACF"}

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

@@ -0,0 +1,150 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type Config } from '../config/config.js';
+import { type Status } from '../core/coreToolScheduler.js';
+import { type ThoughtSummary } from '../core/turn.js';
+import type { PartListUnion } from '@google/genai';
+/**
+ * Token usage summary for a message or conversation.
+ */
+export interface TokensSummary {
+    input: number;
+    output: number;
+    cached: number;
+    thoughts?: number;
+    tool?: number;
+    total: number;
+}
+/**
+ * Base fields common to all messages.
+ */
+export interface BaseMessageRecord {
+    id: string;
+    timestamp: string;
+    content: string;
+}
+/**
+ * Record of a tool call execution within a conversation.
+ */
+export interface ToolCallRecord {
+    id: string;
+    name: string;
+    args: Record<string, unknown>;
+    result?: PartListUnion | null;
+    status: Status;
+    timestamp: string;
+    displayName?: string;
+    description?: string;
+    resultDisplay?: string;
+    renderOutputAsMarkdown?: boolean;
+}
+/**
+ * Message type and message type-specific fields.
+ */
+export type ConversationRecordExtra = {
+    type: 'user';
+} | {
+    type: 'gemini';
+    toolCalls?: ToolCallRecord[];
+    thoughts?: Array<ThoughtSummary & {
+        timestamp: string;
+    }>;
+    tokens?: TokensSummary | null;
+    model?: string;
+};
+/**
+ * A single message record in a conversation.
+ */
+export type MessageRecord = BaseMessageRecord & ConversationRecordExtra;
+/**
+ * Complete conversation record stored in session files.
+ */
+export interface ConversationRecord {
+    sessionId: string;
+    projectHash: string;
+    startTime: string;
+    lastUpdated: string;
+    messages: MessageRecord[];
+}
+/**
+ * Data structure for resuming an existing session.
+ */
+export interface ResumedSessionData {
+    conversation: ConversationRecord;
+    filePath: string;
+}
+/**
+ * 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 declare class ChatRecordingService {
+    private conversationFile;
+    private cachedLastConvData;
+    private sessionId;
+    private projectHash;
+    private queuedThoughts;
+    private queuedTokens;
+    private config;
+    constructor(config: Config);
+    /**
+     * 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?: ResumedSessionData): void;
+    private getLastMessage;
+    private newMessage;
+    /**
+     * Records a message in the conversation.
+     */
+    recordMessage(message: {
+        type: ConversationRecordExtra['type'];
+        content: string;
+        append?: boolean;
+    }): void;
+    /**
+     * Records a thought from the assistant's reasoning process.
+     */
+    recordThought(thought: ThoughtSummary): void;
+    /**
+     * Updates the tokens for the last message in the conversation (which should be by Gemini).
+     */
+    recordMessageTokens(tokens: {
+        input: number;
+        output: number;
+        cached: number;
+        thoughts?: number;
+        tool?: number;
+        total: number;
+    }): void;
+    /**
+     * Adds tool calls to the last message in the conversation (which should be by Gemini).
+     */
+    recordToolCalls(toolCalls: ToolCallRecord[]): void;
+    /**
+     * Loads up the conversation record from disk.
+     */
+    private readConversation;
+    /**
+     * Saves the conversation record; overwrites the file.
+     */
+    private writeConversation;
+    /**
+     * Convenient helper for updating the conversation without file reading and writing and time
+     * updating boilerplate.
+     */
+    private updateConversation;
+    /**
+     * Deletes a session file by session ID.
+     */
+    deleteSession(sessionId: string): void;
+}