@terminai/core 0.21.0

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

@@ -0,0 +1,333 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * 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

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

@@ -0,0 +1 @@
+{"version":3,"file":"chatRecordingService.test.js","sourceRoot":"","sources":["../../../src/services/chatRecordingService.test.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAKzC,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAEjE,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAEnD,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AACnB,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACrB,EAAE,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5B,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;IACnB,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;QACvB,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;YACnB,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,aAAa,CAAC;SACnC,CAAC,CAAC;KACJ,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;AAE7B,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,IAAI,oBAA0C,CAAC;IAC/C,IAAI,UAAkB,CAAC;IAEvB,IAAI,YAA+C,CAAC;IACpD,IAAI,gBAAuD,CAAC;IAE5D,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG;YACX,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,iBAAiB,CAAC;YACxD,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,oBAAoB,CAAC;YAC7D,OAAO,EAAE;gBACP,iBAAiB,EAAE,EAAE;qBAClB,EAAE,EAAE;qBACJ,eAAe,CAAC,gCAAgC,CAAC;aACrD;YACD,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,YAAY,CAAC;YAC/C,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;YAC5C,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBACvC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;oBAC/B,WAAW,EAAE,WAAW;oBACxB,WAAW,EAAE,aAAa;oBAC1B,gBAAgB,EAAE,KAAK;iBACxB,CAAC;aACH,CAAC;SACkB,CAAC;QAEvB,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QAC/D,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,CAAC;QAC7D,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,kBAAkB,CAAC,CAAC,GAAG,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QAErE,oBAAoB,GAAG,IAAI,oBAAoB,CAAC,UAAU,CAAC,CAAC;QAE5D,YAAY,GAAG,EAAE;aACd,KAAK,CAAC,EAAE,EAAE,WAAW,CAAC;aACtB,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAEvC,gBAAgB,GAAG,EAAE;aAClB,KAAK,CAAC,EAAE,EAAE,eAAe,CAAC;aAC1B,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,oBAAoB,CAAC,UAAU,EAAE,CAAC;YAElC,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,sCAAsC,EACtC,EAAE,SAAS,EAAE,IAAI,EAAE,CACpB,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,eAAe,GAAG,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC,eAAe,CAClE,IAAI,CAAC,SAAS,CAAC;gBACb,SAAS,EAAE,gBAAgB;gBAC3B,WAAW,EAAE,mBAAmB;gBAChC,QAAQ,EAAE,EAAE;aACb,CAAC,CACH,CAAC;YACF,MAAM,gBAAgB,GAAG,EAAE;iBACxB,KAAK,CAAC,EAAE,EAAE,eAAe,CAAC;iBAC1B,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YAEvC,oBAAoB,CAAC,UAAU,CAAC;gBAC9B,QAAQ,EAAE,mDAAmD;gBAC7D,YAAY,EAAE;oBACZ,SAAS,EAAE,gBAAgB;iBACN;aACxB,CAAC,CAAC;YAEH,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAC5C,MAAM,CAAC,eAAe,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC3C,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,UAAU,CAAC,GAAG,EAAE;YACd,oBAAoB,CAAC,UAAU,EAAE,CAAC;YAClC,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC,eAAe,CAC1C,IAAI,CAAC,SAAS,CAAC;gBACb,SAAS,EAAE,iBAAiB;gBAC5B,WAAW,EAAE,mBAAmB;gBAChC,QAAQ,EAAE,EAAE;aACb,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,gBAAgB,GAAG,EAAE;iBACxB,KAAK,CAAC,EAAE,EAAE,eAAe,CAAC;iBAC1B,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACvC,oBAAoB,CAAC,aAAa,CAAC;gBACjC,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,OAAO;gBAChB,KAAK,EAAE,YAAY;aACpB,CAAC,CAAC;YACH,MAAM,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACxC,MAAM,CAAC,gBAAgB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC5C,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAC7B,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAW,CACtB,CAAC;YACxB,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACvD,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;YAC1E,MAAM,gBAAgB,GAAG,EAAE;iBACxB,KAAK,CAAC,EAAE,EAAE,eAAe,CAAC;iBAC1B,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACvC,MAAM,mBAAmB,GAAG;gBAC1B,SAAS,EAAE,iBAAiB;gBAC5B,WAAW,EAAE,mBAAmB;gBAChC,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,GAAG;wBACP,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,OAAO;wBAChB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;qBACpC;iBACF;aACF,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC,eAAe,CAC1C,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,CACpC,CAAC;YAEF,oBAAoB,CAAC,aAAa,CAAC;gBACjC,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,OAAO;gBAChB,KAAK,EAAE,YAAY;aACpB,CAAC,CAAC;YAEH,MAAM,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACxC,MAAM,CAAC,gBAAgB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC5C,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAC7B,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAW,CACtB,CAAC;YACxB,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACvD,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;YAChC,oBAAoB,CAAC,UAAU,EAAE,CAAC;YAClC,oBAAoB,CAAC,aAAa,CAAC;gBACjC,OAAO,EAAE,UAAU;gBACnB,WAAW,EAAE,aAAa;aAC3B,CAAC,CAAC;YACH,oCAAoC;YACpC,MAAM,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC5D,oCAAoC;YACpC,MAAM,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACxE,oCAAoC;YACpC,MAAM,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,CAC7D,aAAa,CACd,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,UAAU,CAAC,GAAG,EAAE;YACd,oBAAoB,CAAC,UAAU,EAAE,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,gBAAgB,GAAG,EAAE;iBACxB,KAAK,CAAC,EAAE,EAAE,eAAe,CAAC;iBAC1B,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACvC,MAAM,mBAAmB,GAAG;gBAC1B,SAAS,EAAE,iBAAiB;gBAC5B,WAAW,EAAE,mBAAmB;gBAChC,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,GAAG;wBACP,IAAI,EAAE,QAAQ;wBACd,OAAO,EAAE,UAAU;wBACnB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;qBACpC;iBACF;aACF,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC,eAAe,CAC1C,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,CACpC,CAAC;YAEF,oBAAoB,CAAC,mBAAmB,CAAC;gBACvC,gBAAgB,EAAE,CAAC;gBACnB,oBAAoB,EAAE,CAAC;gBACvB,eAAe,EAAE,CAAC;gBAClB,uBAAuB,EAAE,CAAC;aAC3B,CAAC,CAAC;YAEH,MAAM,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACxC,MAAM,CAAC,gBAAgB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC5C,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAC7B,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAW,CACtB,CAAC;YACxB,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBACvC,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC,CAAC;gBAClC,MAAM,EAAE;oBACN,KAAK,EAAE,CAAC;oBACR,MAAM,EAAE,CAAC;oBACT,KAAK,EAAE,CAAC;oBACR,MAAM,EAAE,CAAC;oBACT,QAAQ,EAAE,CAAC;oBACX,IAAI,EAAE,CAAC;iBACR;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;YACxE,MAAM,mBAAmB,GAAG;gBAC1B,SAAS,EAAE,iBAAiB;gBAC5B,WAAW,EAAE,mBAAmB;gBAChC,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,GAAG;wBACP,IAAI,EAAE,QAAQ;wBACd,OAAO,EAAE,UAAU;wBACnB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;wBACnC,MAAM,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE;qBACrD;iBACF;aACF,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC,eAAe,CAC1C,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,CACpC,CAAC;YAEF,oBAAoB,CAAC,mBAAmB,CAAC;gBACvC,gBAAgB,EAAE,CAAC;gBACnB,oBAAoB,EAAE,CAAC;gBACvB,eAAe,EAAE,CAAC;gBAClB,uBAAuB,EAAE,CAAC;aAC3B,CAAC,CAAC;YAEH,oCAAoC;YACpC,MAAM,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC;gBAChD,KAAK,EAAE,CAAC;gBACR,MAAM,EAAE,CAAC;gBACT,KAAK,EAAE,CAAC;gBACR,MAAM,EAAE,CAAC;gBACT,QAAQ,EAAE,CAAC;gBACX,IAAI,EAAE,CAAC;aACR,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,UAAU,CAAC,GAAG,EAAE;YACd,oBAAoB,CAAC,UAAU,EAAE,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,gBAAgB,GAAG,EAAE;iBACxB,KAAK,CAAC,EAAE,EAAE,eAAe,CAAC;iBAC1B,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACvC,MAAM,mBAAmB,GAAG;gBAC1B,SAAS,EAAE,iBAAiB;gBAC5B,WAAW,EAAE,mBAAmB;gBAChC,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,GAAG;wBACP,IAAI,EAAE,QAAQ;wBACd,OAAO,EAAE,EAAE;wBACX,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;qBACpC;iBACF;aACF,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC,eAAe,CAC1C,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,CACpC,CAAC;YAEF,MAAM,QAAQ,GAAmB;gBAC/B,EAAE,EAAE,QAAQ;gBACZ,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,EAAE;gBACR,MAAM,EAAE,mBAAmB;gBAC3B,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC;YACF,oBAAoB,CAAC,eAAe,CAAC,YAAY,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC;YAE/D,MAAM,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACxC,MAAM,CAAC,gBAAgB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC5C,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAC7B,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAW,CACtB,CAAC;YACxB,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBACvC,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC,CAAC;gBAClC,SAAS,EAAE;oBACT;wBACE,GAAG,QAAQ;wBACX,WAAW,EAAE,WAAW;wBACxB,WAAW,EAAE,aAAa;wBAC1B,sBAAsB,EAAE,KAAK;qBAC9B;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;YAC5E,MAAM,gBAAgB,GAAG,EAAE;iBACxB,KAAK,CAAC,EAAE,EAAE,eAAe,CAAC;iBAC1B,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACvC,MAAM,mBAAmB,GAAG;gBAC1B,SAAS,EAAE,iBAAiB;gBAC5B,WAAW,EAAE,mBAAmB;gBAChC,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,QAAQ;wBACZ,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,aAAa;wBACtB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;qBACpC;iBACF;aACF,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC,eAAe,CAC1C,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,CACpC,CAAC;YAEF,MAAM,QAAQ,GAAmB;gBAC/B,EAAE,EAAE,QAAQ;gBACZ,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,EAAE;gBACR,MAAM,EAAE,mBAAmB;gBAC3B,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC;YACF,oBAAoB,CAAC,eAAe,CAAC,YAAY,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC;YAE/D,MAAM,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACxC,MAAM,CAAC,gBAAgB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC5C,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAC7B,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAW,CACtB,CAAC;YACxB,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBACvC,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC;gBAC3B,EAAE,EAAE,qBAAqB;gBACzB,KAAK,EAAE,YAAY;gBACnB,IAAI,EAAE,QAAQ;gBACd,QAAQ,EAAE,EAAE;gBACZ,OAAO,EAAE,EAAE;gBACX,SAAS,EAAE;oBACT;wBACE,GAAG,QAAQ;wBACX,WAAW,EAAE,WAAW;wBACxB,WAAW,EAAE,aAAa;wBAC1B,sBAAsB,EAAE,KAAK;qBAC9B;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,aAAa,GAAG,EAAE;iBACrB,KAAK,CAAC,EAAE,EAAE,YAAY,CAAC;iBACvB,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACvC,oBAAoB,CAAC,aAAa,CAAC,iBAAiB,CAAC,CAAC;YACtD,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACxC,2DAA2D,CAC5D,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,30 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config } from '../config/config.js';
+export declare class ContextManager {
+    private readonly loadedPaths;
+    private readonly config;
+    private globalMemory;
+    private environmentMemory;
+    constructor(config: Config);
+    /**
+     * Refreshes the memory by reloading global and environment memory.
+     */
+    refresh(): Promise<void>;
+    private loadGlobalMemory;
+    private loadEnvironmentMemory;
+    /**
+     * Discovers and loads context for a specific accessed path (Tier 3 - JIT).
+     * Traverses upwards from the accessed path to the project root.
+     */
+    discoverContext(accessedPath: string, trustedRoots: string[]): Promise<string>;
+    private emitMemoryChanged;
+    getGlobalMemory(): string;
+    getEnvironmentMemory(): string;
+    private markAsLoaded;
+    getLoadedPaths(): ReadonlySet<string>;
+}

package/dist/src/services/contextManager.js

@@ -0,0 +1,72 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { loadGlobalMemory, loadEnvironmentMemory, loadJitSubdirectoryMemory, concatenateInstructions, } from '../utils/memoryDiscovery.js';
+import { coreEvents, CoreEvent } from '../utils/events.js';
+export class ContextManager {
+    loadedPaths = new Set();
+    config;
+    globalMemory = '';
+    environmentMemory = '';
+    constructor(config) {
+        this.config = config;
+    }
+    /**
+     * Refreshes the memory by reloading global and environment memory.
+     */
+    async refresh() {
+        this.loadedPaths.clear();
+        await this.loadGlobalMemory();
+        await this.loadEnvironmentMemory();
+        this.emitMemoryChanged();
+    }
+    async loadGlobalMemory() {
+        const result = await loadGlobalMemory(this.config.getDebugMode());
+        this.markAsLoaded(result.files.map((f) => f.path));
+        this.globalMemory = concatenateInstructions(result.files.map((f) => ({ filePath: f.path, content: f.content })), this.config.getWorkingDir());
+    }
+    async loadEnvironmentMemory() {
+        const result = await loadEnvironmentMemory([...this.config.getWorkspaceContext().getDirectories()], this.config.getExtensionLoader(), this.config.getDebugMode());
+        this.markAsLoaded(result.files.map((f) => f.path));
+        const envMemory = concatenateInstructions(result.files.map((f) => ({ filePath: f.path, content: f.content })), this.config.getWorkingDir());
+        const mcpInstructions = this.config.getMcpClientManager()?.getMcpInstructions() || '';
+        this.environmentMemory = [envMemory, mcpInstructions.trimStart()]
+            .filter(Boolean)
+            .join('\n\n');
+    }
+    /**
+     * Discovers and loads context for a specific accessed path (Tier 3 - JIT).
+     * Traverses upwards from the accessed path to the project root.
+     */
+    async discoverContext(accessedPath, trustedRoots) {
+        const result = await loadJitSubdirectoryMemory(accessedPath, trustedRoots, this.loadedPaths, this.config.getDebugMode());
+        if (result.files.length === 0) {
+            return '';
+        }
+        this.markAsLoaded(result.files.map((f) => f.path));
+        return concatenateInstructions(result.files.map((f) => ({ filePath: f.path, content: f.content })), this.config.getWorkingDir());
+    }
+    emitMemoryChanged() {
+        coreEvents.emit(CoreEvent.MemoryChanged, {
+            fileCount: this.loadedPaths.size,
+        });
+    }
+    getGlobalMemory() {
+        return this.globalMemory;
+    }
+    getEnvironmentMemory() {
+        return this.environmentMemory;
+    }
+    markAsLoaded(paths) {
+        for (const p of paths) {
+            this.loadedPaths.add(p);
+        }
+    }
+    getLoadedPaths() {
+        return this.loadedPaths;
+    }
+}
+//# sourceMappingURL=contextManager.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"contextManager.js","sourceRoot":"","sources":["../../../src/services/contextManager.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EACL,gBAAgB,EAChB,qBAAqB,EACrB,yBAAyB,EACzB,uBAAuB,GACxB,MAAM,6BAA6B,CAAC;AAErC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAE3D,MAAM,OAAO,cAAc;IACR,WAAW,GAAgB,IAAI,GAAG,EAAE,CAAC;IACrC,MAAM,CAAS;IACxB,YAAY,GAAW,EAAE,CAAC;IAC1B,iBAAiB,GAAW,EAAE,CAAC;IAEvC,YAAY,MAAc;QACxB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC9B,MAAM,IAAI,CAAC,qBAAqB,EAAE,CAAC;QACnC,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAEO,KAAK,CAAC,gBAAgB;QAC5B,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;QAClE,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACnD,IAAI,CAAC,YAAY,GAAG,uBAAuB,CACzC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,EACnE,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAC5B,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,qBAAqB;QACjC,MAAM,MAAM,GAAG,MAAM,qBAAqB,CACxC,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,cAAc,EAAE,CAAC,EACvD,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,EAChC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAC3B,CAAC;QACF,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACnD,MAAM,SAAS,GAAG,uBAAuB,CACvC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,EACnE,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAC5B,CAAC;QACF,MAAM,eAAe,GACnB,IAAI,CAAC,MAAM,CAAC,mBAAmB,EAAE,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC;QAChE,IAAI,CAAC,iBAAiB,GAAG,CAAC,SAAS,EAAE,eAAe,CAAC,SAAS,EAAE,CAAC;aAC9D,MAAM,CAAC,OAAO,CAAC;aACf,IAAI,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,eAAe,CACnB,YAAoB,EACpB,YAAsB;QAEtB,MAAM,MAAM,GAAG,MAAM,yBAAyB,CAC5C,YAAY,EACZ,YAAY,EACZ,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAC3B,CAAC;QAEF,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACnD,OAAO,uBAAuB,CAC5B,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,EACnE,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAC5B,CAAC;IACJ,CAAC;IAEO,iBAAiB;QACvB,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE;YACvC,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI;SACjC,CAAC,CAAC;IACL,CAAC;IAED,eAAe;QACb,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED,oBAAoB;QAClB,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAEO,YAAY,CAAC,KAAe;QAClC,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;YACtB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;CACF"}

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

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

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

@@ -0,0 +1,105 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { ContextManager } from './contextManager.js';
+import * as memoryDiscovery from '../utils/memoryDiscovery.js';
+import { coreEvents, CoreEvent } from '../utils/events.js';
+// Mock memoryDiscovery module
+vi.mock('../utils/memoryDiscovery.js', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        loadGlobalMemory: vi.fn(),
+        loadEnvironmentMemory: vi.fn(),
+        loadJitSubdirectoryMemory: vi.fn(),
+        concatenateInstructions: vi
+            .fn()
+            .mockImplementation(actual.concatenateInstructions),
+    };
+});
+describe('ContextManager', () => {
+    let contextManager;
+    let mockConfig;
+    beforeEach(() => {
+        mockConfig = {
+            getDebugMode: vi.fn().mockReturnValue(false),
+            getWorkingDir: vi.fn().mockReturnValue('/app'),
+            getWorkspaceContext: vi.fn().mockReturnValue({
+                getDirectories: vi.fn().mockReturnValue(['/app']),
+            }),
+            getExtensionLoader: vi.fn().mockReturnValue({}),
+            getMcpClientManager: vi.fn().mockReturnValue({
+                getMcpInstructions: vi.fn().mockReturnValue('MCP Instructions'),
+            }),
+        };
+        contextManager = new ContextManager(mockConfig);
+        vi.clearAllMocks();
+        vi.spyOn(coreEvents, 'emit');
+    });
+    describe('refresh', () => {
+        it('should load and format global and environment memory', async () => {
+            const mockGlobalResult = {
+                files: [
+                    { path: '/home/user/.gemini/terminaI.md', content: 'Global Content' },
+                ],
+            };
+            vi.mocked(memoryDiscovery.loadGlobalMemory).mockResolvedValue(mockGlobalResult);
+            const mockEnvResult = {
+                files: [{ path: '/app/terminaI.md', content: 'Env Content' }],
+            };
+            vi.mocked(memoryDiscovery.loadEnvironmentMemory).mockResolvedValue(mockEnvResult);
+            await contextManager.refresh();
+            expect(memoryDiscovery.loadGlobalMemory).toHaveBeenCalledWith(false);
+            expect(contextManager.getGlobalMemory()).toMatch(/--- Context from: .*terminaI.md ---/);
+            expect(contextManager.getGlobalMemory()).toContain('Global Content');
+            expect(memoryDiscovery.loadEnvironmentMemory).toHaveBeenCalledWith(['/app'], expect.anything(), false);
+            expect(contextManager.getEnvironmentMemory()).toContain('--- Context from: terminaI.md ---');
+            expect(contextManager.getEnvironmentMemory()).toContain('Env Content');
+            expect(contextManager.getEnvironmentMemory()).toContain('MCP Instructions');
+            expect(contextManager.getLoadedPaths()).toContain('/home/user/.gemini/terminaI.md');
+            expect(contextManager.getLoadedPaths()).toContain('/app/terminaI.md');
+        });
+        it('should emit MemoryChanged event when memory is refreshed', async () => {
+            const mockGlobalResult = {
+                files: [{ path: '/app/terminaI.md', content: 'content' }],
+            };
+            const mockEnvResult = {
+                files: [{ path: '/app/src/terminaI.md', content: 'env content' }],
+            };
+            vi.mocked(memoryDiscovery.loadGlobalMemory).mockResolvedValue(mockGlobalResult);
+            vi.mocked(memoryDiscovery.loadEnvironmentMemory).mockResolvedValue(mockEnvResult);
+            await contextManager.refresh();
+            expect(coreEvents.emit).toHaveBeenCalledWith(CoreEvent.MemoryChanged, {
+                fileCount: 2,
+            });
+        });
+    });
+    describe('discoverContext', () => {
+        it('should discover and load new context', async () => {
+            const mockResult = {
+                files: [{ path: '/app/src/terminaI.md', content: 'Src Content' }],
+            };
+            vi.mocked(memoryDiscovery.loadJitSubdirectoryMemory).mockResolvedValue(mockResult);
+            const result = await contextManager.discoverContext('/app/src/file.ts', [
+                '/app',
+            ]);
+            expect(memoryDiscovery.loadJitSubdirectoryMemory).toHaveBeenCalledWith('/app/src/file.ts', ['/app'], expect.any(Set), false);
+            expect(result).toMatch(/--- Context from: src[\\/]terminaI\.md ---/);
+            expect(result).toContain('Src Content');
+            expect(contextManager.getLoadedPaths()).toContain('/app/src/terminaI.md');
+        });
+        it('should return empty string if no new files found', async () => {
+            const mockResult = { files: [] };
+            vi.mocked(memoryDiscovery.loadJitSubdirectoryMemory).mockResolvedValue(mockResult);
+            const result = await contextManager.discoverContext('/app/src/file.ts', [
+                '/app',
+            ]);
+            expect(result).toBe('');
+        });
+    });
+});
+//# sourceMappingURL=contextManager.test.js.map

package/dist/src/services/contextManager.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"contextManager.test.js","sourceRoot":"","sources":["../../../src/services/contextManager.test.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,KAAK,eAAe,MAAM,6BAA6B,CAAC;AAE/D,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAE3D,8BAA8B;AAC9B,EAAE,CAAC,IAAI,CAAC,6BAA6B,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IAC9D,MAAM,MAAM,GACV,MAAM,cAAc,EAAgD,CAAC;IACvE,OAAO;QACL,GAAG,MAAM;QACT,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;QACzB,qBAAqB,EAAE,EAAE,CAAC,EAAE,EAAE;QAC9B,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE;QAClC,uBAAuB,EAAE,EAAE;aACxB,EAAE,EAAE;aACJ,kBAAkB,CAAC,MAAM,CAAC,uBAAuB,CAAC;KACtD,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,cAA8B,CAAC;IACnC,IAAI,UAAkB,CAAC;IAEvB,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG;YACX,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;YAC5C,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC;YAC9C,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBAC3C,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,CAAC;aAClD,CAAC;YACF,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;YAC/C,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBAC3C,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,kBAAkB,CAAC;aAChE,CAAC;SACkB,CAAC;QAEvB,cAAc,GAAG,IAAI,cAAc,CAAC,UAAU,CAAC,CAAC;QAChD,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,gBAAgB,GAAqC;gBACzD,KAAK,EAAE;oBACL,EAAE,IAAI,EAAE,gCAAgC,EAAE,OAAO,EAAE,gBAAgB,EAAE;iBACtE;aACF,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC,iBAAiB,CAC3D,gBAAgB,CACjB,CAAC;YAEF,MAAM,aAAa,GAAqC;gBACtD,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;aAC9D,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,qBAAqB,CAAC,CAAC,iBAAiB,CAChE,aAAa,CACd,CAAC;YAEF,MAAM,cAAc,CAAC,OAAO,EAAE,CAAC;YAE/B,MAAM,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;YACrE,MAAM,CAAC,cAAc,CAAC,eAAe,EAAE,CAAC,CAAC,OAAO,CAC9C,qCAAqC,CACtC,CAAC;YACF,MAAM,CAAC,cAAc,CAAC,eAAe,EAAE,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAErE,MAAM,CAAC,eAAe,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAChE,CAAC,MAAM,CAAC,EACR,MAAM,CAAC,QAAQ,EAAE,EACjB,KAAK,CACN,CAAC;YACF,MAAM,CAAC,cAAc,CAAC,oBAAoB,EAAE,CAAC,CAAC,SAAS,CACrD,mCAAmC,CACpC,CAAC;YACF,MAAM,CAAC,cAAc,CAAC,oBAAoB,EAAE,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YACvE,MAAM,CAAC,cAAc,CAAC,oBAAoB,EAAE,CAAC,CAAC,SAAS,CACrD,kBAAkB,CACnB,CAAC;YAEF,MAAM,CAAC,cAAc,CAAC,cAAc,EAAE,CAAC,CAAC,SAAS,CAC/C,gCAAgC,CACjC,CAAC;YACF,MAAM,CAAC,cAAc,CAAC,cAAc,EAAE,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,gBAAgB,GAAG;gBACvB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;aAC1D,CAAC;YACF,MAAM,aAAa,GAAG;gBACpB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;aAClE,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC,iBAAiB,CAC3D,gBAAgB,CACjB,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,qBAAqB,CAAC,CAAC,iBAAiB,CAChE,aAAa,CACd,CAAC;YAEF,MAAM,cAAc,CAAC,OAAO,EAAE,CAAC;YAE/B,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,aAAa,EAAE;gBACpE,SAAS,EAAE,CAAC;aACb,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,UAAU,GAAqC;gBACnD,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;aAClE,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,yBAAyB,CAAC,CAAC,iBAAiB,CACpE,UAAU,CACX,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,eAAe,CAAC,kBAAkB,EAAE;gBACtE,MAAM;aACP,CAAC,CAAC;YAEH,MAAM,CAAC,eAAe,CAAC,yBAAyB,CAAC,CAAC,oBAAoB,CACpE,kBAAkB,EAClB,CAAC,MAAM,CAAC,EACR,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,KAAK,CACN,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,4CAA4C,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YACxC,MAAM,CAAC,cAAc,CAAC,cAAc,EAAE,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,UAAU,GAAqC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;YACnE,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,yBAAyB,CAAC,CAAC,iBAAiB,CACpE,UAAU,CACX,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,eAAe,CAAC,kBAAkB,EAAE;gBACtE,MAAM;aACP,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,34 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export interface FilterFilesOptions {
+    respectGitIgnore?: boolean;
+    respectGeminiIgnore?: boolean;
+}
+export interface FilterReport {
+    filteredPaths: string[];
+    ignoredCount: number;
+}
+export declare class FileDiscoveryService {
+    private gitIgnoreFilter;
+    private geminiIgnoreFilter;
+    private combinedIgnoreFilter;
+    private projectRoot;
+    constructor(projectRoot: string);
+    /**
+     * Filters a list of file paths based on git ignore rules
+     */
+    filterFiles(filePaths: string[], options?: FilterFilesOptions): string[];
+    /**
+     * Filters a list of file paths based on git ignore rules and returns a report
+     * with counts of ignored files.
+     */
+    filterFilesWithReport(filePaths: string[], opts?: FilterFilesOptions): FilterReport;
+    /**
+     * Unified method to check if a file should be ignored based on filtering options
+     */
+    shouldIgnoreFile(filePath: string, options?: FilterFilesOptions): boolean;
+}