@machina.ai/cell-cli 1.8.2-rc1 → 1.11.0-rc1

package/dist/src/ui/commands/memoryCommand.test.js

@@ -0,0 +1,266 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { vi, describe, it, expect, beforeEach } from 'vitest';
+import { memoryCommand } from './memoryCommand.js';
+import { createMockCommandContext } from '../../test-utils/mockCommandContext.js';
+import { MessageType } from '../types.js';
+import { getErrorMessage, } from '@google/gemini-cli-core';
+import { loadHierarchicalGeminiMemory } from '../../config/config.js';
+vi.mock('@google/gemini-cli-core', async (importOriginal) => {
+    const original = await importOriginal();
+    return {
+        ...original,
+        getErrorMessage: vi.fn((error) => {
+            if (error instanceof Error)
+                return error.message;
+            return String(error);
+        }),
+    };
+});
+vi.mock('../../config/config.js', async (importOriginal) => {
+    const original = await importOriginal();
+    return {
+        ...original,
+        loadHierarchicalGeminiMemory: vi.fn(),
+    };
+});
+const mockLoadHierarchicalGeminiMemory = loadHierarchicalGeminiMemory;
+describe('memoryCommand', () => {
+    let mockContext;
+    const getSubCommand = (name) => {
+        const subCommand = memoryCommand.subCommands?.find((cmd) => cmd.name === name);
+        if (!subCommand) {
+            throw new Error(`/memory ${name} command not found.`);
+        }
+        return subCommand;
+    };
+    describe('/memory show', () => {
+        let showCommand;
+        let mockGetUserMemory;
+        let mockGetGeminiMdFileCount;
+        beforeEach(() => {
+            showCommand = getSubCommand('show');
+            mockGetUserMemory = vi.fn();
+            mockGetGeminiMdFileCount = vi.fn();
+            mockContext = createMockCommandContext({
+                services: {
+                    config: {
+                        getUserMemory: mockGetUserMemory,
+                        getGeminiMdFileCount: mockGetGeminiMdFileCount,
+                    },
+                },
+            });
+        });
+        it('should display a message if memory is empty', async () => {
+            if (!showCommand.action)
+                throw new Error('Command has no action');
+            mockGetUserMemory.mockReturnValue('');
+            mockGetGeminiMdFileCount.mockReturnValue(0);
+            await showCommand.action(mockContext, '');
+            expect(mockContext.ui.addItem).toHaveBeenCalledWith({
+                type: MessageType.INFO,
+                text: 'Memory is currently empty.',
+            }, expect.any(Number));
+        });
+        it('should display the memory content and file count if it exists', async () => {
+            if (!showCommand.action)
+                throw new Error('Command has no action');
+            const memoryContent = 'This is a test memory.';
+            mockGetUserMemory.mockReturnValue(memoryContent);
+            mockGetGeminiMdFileCount.mockReturnValue(1);
+            await showCommand.action(mockContext, '');
+            expect(mockContext.ui.addItem).toHaveBeenCalledWith({
+                type: MessageType.INFO,
+                text: `Current memory content from 1 file(s):\n\n---\n${memoryContent}\n---`,
+            }, expect.any(Number));
+        });
+    });
+    describe('/memory add', () => {
+        let addCommand;
+        beforeEach(() => {
+            addCommand = getSubCommand('add');
+            mockContext = createMockCommandContext();
+        });
+        it('should return an error message if no arguments are provided', () => {
+            if (!addCommand.action)
+                throw new Error('Command has no action');
+            const result = addCommand.action(mockContext, '  ');
+            expect(result).toEqual({
+                type: 'message',
+                messageType: 'error',
+                content: 'Usage: /memory add <text to remember>',
+            });
+            expect(mockContext.ui.addItem).not.toHaveBeenCalled();
+        });
+        it('should return a tool action and add an info message when arguments are provided', () => {
+            if (!addCommand.action)
+                throw new Error('Command has no action');
+            const fact = 'remember this';
+            const result = addCommand.action(mockContext, `  ${fact}  `);
+            expect(mockContext.ui.addItem).toHaveBeenCalledWith({
+                type: MessageType.INFO,
+                text: `Attempting to save to memory: "${fact}"`,
+            }, expect.any(Number));
+            expect(result).toEqual({
+                type: 'tool',
+                toolName: 'save_memory',
+                toolArgs: { fact },
+            });
+        });
+    });
+    describe('/memory refresh', () => {
+        let refreshCommand;
+        let mockSetUserMemory;
+        let mockSetGeminiMdFileCount;
+        let mockSetGeminiMdFilePaths;
+        beforeEach(() => {
+            refreshCommand = getSubCommand('refresh');
+            mockSetUserMemory = vi.fn();
+            mockSetGeminiMdFileCount = vi.fn();
+            mockSetGeminiMdFilePaths = vi.fn();
+            const mockConfig = {
+                setUserMemory: mockSetUserMemory,
+                setGeminiMdFileCount: mockSetGeminiMdFileCount,
+                setGeminiMdFilePaths: mockSetGeminiMdFilePaths,
+                getWorkingDir: () => '/test/dir',
+                getDebugMode: () => false,
+                getFileService: () => ({}),
+                getExtensions: () => [],
+                shouldLoadMemoryFromIncludeDirectories: () => false,
+                getWorkspaceContext: () => ({
+                    getDirectories: () => [],
+                }),
+                getFileFilteringOptions: () => ({
+                    ignore: [],
+                    include: [],
+                }),
+                isTrustedFolder: () => false,
+            };
+            mockContext = createMockCommandContext({
+                services: {
+                    config: mockConfig,
+                    settings: {
+                        merged: {
+                            memoryDiscoveryMaxDirs: 1000,
+                            context: {
+                                importFormat: 'tree',
+                            },
+                        },
+                    },
+                },
+                ui: {
+                    setGeminiMdFileCount: vi.fn(),
+                },
+            });
+            mockLoadHierarchicalGeminiMemory.mockClear();
+        });
+        it('should display success message when memory is refreshed with content', async () => {
+            if (!refreshCommand.action)
+                throw new Error('Command has no action');
+            const refreshResult = {
+                memoryContent: 'new memory content',
+                fileCount: 2,
+                filePaths: ['/path/one/GEMINI.md', '/path/two/GEMINI.md'],
+            };
+            mockLoadHierarchicalGeminiMemory.mockResolvedValue(refreshResult);
+            await refreshCommand.action(mockContext, '');
+            expect(mockContext.ui.addItem).toHaveBeenCalledWith({
+                type: MessageType.INFO,
+                text: 'Refreshing memory from source files...',
+            }, expect.any(Number));
+            expect(mockLoadHierarchicalGeminiMemory).toHaveBeenCalledOnce();
+            expect(mockSetUserMemory).toHaveBeenCalledWith(refreshResult.memoryContent);
+            expect(mockSetGeminiMdFileCount).toHaveBeenCalledWith(refreshResult.fileCount);
+            expect(mockSetGeminiMdFilePaths).toHaveBeenCalledWith(refreshResult.filePaths);
+            expect(mockContext.ui.setGeminiMdFileCount).toHaveBeenCalledWith(refreshResult.fileCount);
+            expect(mockContext.ui.addItem).toHaveBeenCalledWith({
+                type: MessageType.INFO,
+                text: 'Memory refreshed successfully. Loaded 18 characters from 2 file(s).',
+            }, expect.any(Number));
+        });
+        it('should display success message when memory is refreshed with no content', async () => {
+            if (!refreshCommand.action)
+                throw new Error('Command has no action');
+            const refreshResult = { memoryContent: '', fileCount: 0, filePaths: [] };
+            mockLoadHierarchicalGeminiMemory.mockResolvedValue(refreshResult);
+            await refreshCommand.action(mockContext, '');
+            expect(mockLoadHierarchicalGeminiMemory).toHaveBeenCalledOnce();
+            expect(mockSetUserMemory).toHaveBeenCalledWith('');
+            expect(mockSetGeminiMdFileCount).toHaveBeenCalledWith(0);
+            expect(mockSetGeminiMdFilePaths).toHaveBeenCalledWith([]);
+            expect(mockContext.ui.addItem).toHaveBeenCalledWith({
+                type: MessageType.INFO,
+                text: 'Memory refreshed successfully. No memory content found.',
+            }, expect.any(Number));
+        });
+        it('should display an error message if refreshing fails', async () => {
+            if (!refreshCommand.action)
+                throw new Error('Command has no action');
+            const error = new Error('Failed to read memory files.');
+            mockLoadHierarchicalGeminiMemory.mockRejectedValue(error);
+            await refreshCommand.action(mockContext, '');
+            expect(mockLoadHierarchicalGeminiMemory).toHaveBeenCalledOnce();
+            expect(mockSetUserMemory).not.toHaveBeenCalled();
+            expect(mockSetGeminiMdFileCount).not.toHaveBeenCalled();
+            expect(mockSetGeminiMdFilePaths).not.toHaveBeenCalled();
+            expect(mockContext.ui.addItem).toHaveBeenCalledWith({
+                type: MessageType.ERROR,
+                text: `Error refreshing memory: ${error.message}`,
+            }, expect.any(Number));
+            expect(getErrorMessage).toHaveBeenCalledWith(error);
+        });
+        it('should not throw if config service is unavailable', async () => {
+            if (!refreshCommand.action)
+                throw new Error('Command has no action');
+            const nullConfigContext = createMockCommandContext({
+                services: { config: null },
+            });
+            await expect(refreshCommand.action(nullConfigContext, '')).resolves.toBeUndefined();
+            expect(nullConfigContext.ui.addItem).toHaveBeenCalledWith({
+                type: MessageType.INFO,
+                text: 'Refreshing memory from source files...',
+            }, expect.any(Number));
+            expect(mockLoadHierarchicalGeminiMemory).not.toHaveBeenCalled();
+        });
+    });
+    describe('/memory list', () => {
+        let listCommand;
+        let mockGetGeminiMdfilePaths;
+        beforeEach(() => {
+            listCommand = getSubCommand('list');
+            mockGetGeminiMdfilePaths = vi.fn();
+            mockContext = createMockCommandContext({
+                services: {
+                    config: {
+                        getGeminiMdFilePaths: mockGetGeminiMdfilePaths,
+                    },
+                },
+            });
+        });
+        it('should display a message if no GEMINI.md files are found', async () => {
+            if (!listCommand.action)
+                throw new Error('Command has no action');
+            mockGetGeminiMdfilePaths.mockReturnValue([]);
+            await listCommand.action(mockContext, '');
+            expect(mockContext.ui.addItem).toHaveBeenCalledWith({
+                type: MessageType.INFO,
+                text: 'No GEMINI.md files in use.',
+            }, expect.any(Number));
+        });
+        it('should display the file count and paths if they exist', async () => {
+            if (!listCommand.action)
+                throw new Error('Command has no action');
+            const filePaths = ['/path/one/GEMINI.md', '/path/two/GEMINI.md'];
+            mockGetGeminiMdfilePaths.mockReturnValue(filePaths);
+            await listCommand.action(mockContext, '');
+            expect(mockContext.ui.addItem).toHaveBeenCalledWith({
+                type: MessageType.INFO,
+                text: `There are 2 GEMINI.md file(s) in use:\n\n${filePaths.join('\n')}`,
+            }, expect.any(Number));
+        });
+    });
+});
+//# sourceMappingURL=memoryCommand.test.js.map

package/dist/src/ui/commands/memoryCommand.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memoryCommand.test.js","sourceRoot":"","sources":["../../../../src/ui/commands/memoryCommand.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,OAAO,EAAE,wBAAwB,EAAE,MAAM,wCAAwC,CAAC;AAClF,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1C,OAAO,EACL,eAAe,GAEhB,MAAM,yBAAyB,CAAC;AAEjC,OAAO,EAAE,4BAA4B,EAAE,MAAM,wBAAwB,CAAC;AAEtE,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IAC1D,MAAM,QAAQ,GACZ,MAAM,cAAc,EAA4C,CAAC;IACnE,OAAO;QACL,GAAG,QAAQ;QACX,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAAc,EAAE,EAAE;YACxC,IAAI,KAAK,YAAY,KAAK;gBAAE,OAAO,KAAK,CAAC,OAAO,CAAC;YACjD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC,CAAC;KACH,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IACzD,MAAM,QAAQ,GACZ,MAAM,cAAc,EAA2C,CAAC;IAClE,OAAO;QACL,GAAG,QAAQ;QACX,4BAA4B,EAAE,EAAE,CAAC,EAAE,EAAE;KACtC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,MAAM,gCAAgC,GAAG,4BAAoC,CAAC;AAE9E,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,IAAI,WAA2B,CAAC;IAEhC,MAAM,aAAa,GAAG,CACpB,IAAyC,EAC3B,EAAE;QAChB,MAAM,UAAU,GAAG,aAAa,CAAC,WAAW,EAAE,IAAI,CAChD,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,IAAI,CAC3B,CAAC;QACF,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,WAAW,IAAI,qBAAqB,CAAC,CAAC;QACxD,CAAC;QACD,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;IAEF,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,IAAI,WAAyB,CAAC;QAC9B,IAAI,iBAAuB,CAAC;QAC5B,IAAI,wBAA8B,CAAC;QAEnC,UAAU,CAAC,GAAG,EAAE;YACd,WAAW,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;YAEpC,iBAAiB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,wBAAwB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAEnC,WAAW,GAAG,wBAAwB,CAAC;gBACrC,QAAQ,EAAE;oBACR,MAAM,EAAE;wBACN,aAAa,EAAE,iBAAiB;wBAChC,oBAAoB,EAAE,wBAAwB;qBAC/C;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,IAAI,CAAC,WAAW,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAElE,iBAAiB,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;YACtC,wBAAwB,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAE5C,MAAM,WAAW,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YAE1C,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD;gBACE,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,IAAI,EAAE,4BAA4B;aACnC,EACD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,IAAI,CAAC,WAAW,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAElE,MAAM,aAAa,GAAG,wBAAwB,CAAC;YAE/C,iBAAiB,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;YACjD,wBAAwB,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAE5C,MAAM,WAAW,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YAE1C,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD;gBACE,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,IAAI,EAAE,kDAAkD,aAAa,OAAO;aAC7E,EACD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,IAAI,UAAwB,CAAC;QAE7B,UAAU,CAAC,GAAG,EAAE;YACd,UAAU,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;YAClC,WAAW,GAAG,wBAAwB,EAAE,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,IAAI,CAAC,UAAU,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAEjE,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;YACpD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,IAAI,EAAE,SAAS;gBACf,WAAW,EAAE,OAAO;gBACpB,OAAO,EAAE,uCAAuC;aACjD,CAAC,CAAC;YAEH,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iFAAiF,EAAE,GAAG,EAAE;YACzF,IAAI,CAAC,UAAU,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAEjE,MAAM,IAAI,GAAG,eAAe,CAAC;YAC7B,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,WAAW,EAAE,KAAK,IAAI,IAAI,CAAC,CAAC;YAE7D,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD;gBACE,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,IAAI,EAAE,kCAAkC,IAAI,GAAG;aAChD,EACD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,IAAI,EAAE,MAAM;gBACZ,QAAQ,EAAE,aAAa;gBACvB,QAAQ,EAAE,EAAE,IAAI,EAAE;aACnB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,IAAI,cAA4B,CAAC;QACjC,IAAI,iBAAuB,CAAC;QAC5B,IAAI,wBAA8B,CAAC;QACnC,IAAI,wBAA8B,CAAC;QAEnC,UAAU,CAAC,GAAG,EAAE;YACd,cAAc,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;YAC1C,iBAAiB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,wBAAwB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACnC,wBAAwB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAEnC,MAAM,UAAU,GAAG;gBACjB,aAAa,EAAE,iBAAiB;gBAChC,oBAAoB,EAAE,wBAAwB;gBAC9C,oBAAoB,EAAE,wBAAwB;gBAC9C,aAAa,EAAE,GAAG,EAAE,CAAC,WAAW;gBAChC,YAAY,EAAE,GAAG,EAAE,CAAC,KAAK;gBACzB,cAAc,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAyB;gBAClD,aAAa,EAAE,GAAG,EAAE,CAAC,EAAE;gBACvB,sCAAsC,EAAE,GAAG,EAAE,CAAC,KAAK;gBACnD,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;oBAC1B,cAAc,EAAE,GAAG,EAAE,CAAC,EAAE;iBACzB,CAAC;gBACF,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;oBAC9B,MAAM,EAAE,EAAE;oBACV,OAAO,EAAE,EAAE;iBACZ,CAAC;gBACF,eAAe,EAAE,GAAG,EAAE,CAAC,KAAK;aAC7B,CAAC;YAEF,WAAW,GAAG,wBAAwB,CAAC;gBACrC,QAAQ,EAAE;oBACR,MAAM,EAAE,UAAU;oBAClB,QAAQ,EAAE;wBACR,MAAM,EAAE;4BACN,sBAAsB,EAAE,IAAI;4BAC5B,OAAO,EAAE;gCACP,YAAY,EAAE,MAAM;6BACrB;yBACF;qBAC2B;iBAC/B;gBACD,EAAE,EAAE;oBACF,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE;iBAC9B;aACF,CAAC,CAAC;YACH,gCAAgC,CAAC,SAAS,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACpF,IAAI,CAAC,cAAc,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAErE,MAAM,aAAa,GAAyC;gBAC1D,aAAa,EAAE,oBAAoB;gBACnC,SAAS,EAAE,CAAC;gBACZ,SAAS,EAAE,CAAC,qBAAqB,EAAE,qBAAqB,CAAC;aAC1D,CAAC;YACF,gCAAgC,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAElE,MAAM,cAAc,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YAE7C,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD;gBACE,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,IAAI,EAAE,wCAAwC;aAC/C,EACD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;YAEF,MAAM,CAAC,gCAAgC,CAAC,CAAC,oBAAoB,EAAE,CAAC;YAChE,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,aAAa,CAAC,aAAa,CAC5B,CAAC;YACF,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CACnD,aAAa,CAAC,SAAS,CACxB,CAAC;YACF,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CACnD,aAAa,CAAC,SAAS,CACxB,CAAC;YACF,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAC9D,aAAa,CAAC,SAAS,CACxB,CAAC;YAEF,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD;gBACE,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,IAAI,EAAE,qEAAqE;aAC5E,EACD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;YACvF,IAAI,CAAC,cAAc,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAErE,MAAM,aAAa,GAAG,EAAE,aAAa,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;YACzE,gCAAgC,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAElE,MAAM,cAAc,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YAE7C,MAAM,CAAC,gCAAgC,CAAC,CAAC,oBAAoB,EAAE,CAAC;YAChE,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC;YACnD,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC;YACzD,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC;YAE1D,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD;gBACE,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,IAAI,EAAE,yDAAyD;aAChE,EACD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,IAAI,CAAC,cAAc,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAErE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;YACxD,gCAAgC,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAE1D,MAAM,cAAc,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YAE7C,MAAM,CAAC,gCAAgC,CAAC,CAAC,oBAAoB,EAAE,CAAC;YAChE,MAAM,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACjD,MAAM,CAAC,wBAAwB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACxD,MAAM,CAAC,wBAAwB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAExD,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD;gBACE,IAAI,EAAE,WAAW,CAAC,KAAK;gBACvB,IAAI,EAAE,4BAA4B,KAAK,CAAC,OAAO,EAAE;aAClD,EACD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;YAEF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,IAAI,CAAC,cAAc,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAErE,MAAM,iBAAiB,GAAG,wBAAwB,CAAC;gBACjD,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE;aAC3B,CAAC,CAAC;YAEH,MAAM,MAAM,CACV,cAAc,CAAC,MAAM,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAC7C,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;YAE3B,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACvD;gBACE,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,IAAI,EAAE,wCAAwC;aAC/C,EACD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;YAEF,MAAM,CAAC,gCAAgC,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,IAAI,WAAyB,CAAC;QAC9B,IAAI,wBAA8B,CAAC;QAEnC,UAAU,CAAC,GAAG,EAAE;YACd,WAAW,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;YACpC,wBAAwB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACnC,WAAW,GAAG,wBAAwB,CAAC;gBACrC,QAAQ,EAAE;oBACR,MAAM,EAAE;wBACN,oBAAoB,EAAE,wBAAwB;qBAC/C;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,IAAI,CAAC,WAAW,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAElE,wBAAwB,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;YAE7C,MAAM,WAAW,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YAE1C,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD;gBACE,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,IAAI,EAAE,4BAA4B;aACnC,EACD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,IAAI,CAAC,WAAW,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAElE,MAAM,SAAS,GAAG,CAAC,qBAAqB,EAAE,qBAAqB,CAAC,CAAC;YACjE,wBAAwB,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YAEpD,MAAM,WAAW,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YAE1C,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD;gBACE,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,IAAI,EAAE,4CAA4C,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;aACzE,EACD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/ui/commands/privacyCommand.js

@@ -6,7 +6,7 @@
 import { CommandKind } from './types.js';
 export const privacyCommand = {
     name: 'privacy',
-    description: 'display the privacy notice',
+    description: 'Display the privacy notice',
     kind: CommandKind.BUILT_IN,
     action: () => ({
         type: 'dialog',

package/dist/src/ui/commands/privacyCommand.test.d.ts

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

package/dist/src/ui/commands/privacyCommand.test.js

@@ -0,0 +1,32 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach } from 'vitest';
+import { privacyCommand } from './privacyCommand.js';
+import {} from './types.js';
+import { createMockCommandContext } from '../../test-utils/mockCommandContext.js';
+describe('privacyCommand', () => {
+    let mockContext;
+    beforeEach(() => {
+        mockContext = createMockCommandContext();
+    });
+    it('should return a dialog action to open the privacy dialog', () => {
+        // Ensure the command has an action to test.
+        if (!privacyCommand.action) {
+            throw new Error('The privacy command must have an action.');
+        }
+        const result = privacyCommand.action(mockContext, '');
+        // Assert that the action returns the correct object to trigger the privacy dialog.
+        expect(result).toEqual({
+            type: 'dialog',
+            dialog: 'privacy',
+        });
+    });
+    it('should have the correct name and description', () => {
+        expect(privacyCommand.name).toBe('privacy');
+        expect(privacyCommand.description).toBe('Display the privacy notice');
+    });
+});
+//# sourceMappingURL=privacyCommand.test.js.map

package/dist/src/ui/commands/privacyCommand.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"privacyCommand.test.js","sourceRoot":"","sources":["../../../../src/ui/commands/privacyCommand.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC1D,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAuB,MAAM,YAAY,CAAC;AACjD,OAAO,EAAE,wBAAwB,EAAE,MAAM,wCAAwC,CAAC;AAElF,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,WAA2B,CAAC;IAEhC,UAAU,CAAC,GAAG,EAAE;QACd,WAAW,GAAG,wBAAwB,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,4CAA4C;QAC5C,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC9D,CAAC;QAED,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAEtD,mFAAmF;QACnF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,SAAS;SAClB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5C,MAAM,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/ui/commands/profileCommand.d.ts

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type SlashCommand } from './types.js';
+export declare const profileCommand: SlashCommand | null;

package/dist/src/ui/commands/profileCommand.js

@@ -0,0 +1,23 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { isDevelopment } from '../../utils/installationInfo.js';
+import { CommandKind } from './types.js';
+export const profileCommand = isDevelopment
+    ? {
+        name: 'profile',
+        kind: CommandKind.BUILT_IN,
+        description: 'Toggle the debug profile display',
+        action: async (context) => {
+            context.ui.toggleDebugProfiler();
+            return {
+                type: 'message',
+                messageType: 'info',
+                content: 'Toggled profile display.',
+            };
+        },
+    }
+    : null;
+//# sourceMappingURL=profileCommand.js.map

package/dist/src/ui/commands/profileCommand.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"profileCommand.js","sourceRoot":"","sources":["../../../../src/ui/commands/profileCommand.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,WAAW,EAAqB,MAAM,YAAY,CAAC;AAE5D,MAAM,CAAC,MAAM,cAAc,GAAwB,aAAa;IAC9D,CAAC,CAAC;QACE,IAAI,EAAE,SAAS;QACf,IAAI,EAAE,WAAW,CAAC,QAAQ;QAC1B,WAAW,EAAE,kCAAkC;QAC/C,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACxB,OAAO,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC;YACjC,OAAO;gBACL,IAAI,EAAE,SAAS;gBACf,WAAW,EAAE,MAAM;gBACnB,OAAO,EAAE,0BAA0B;aACpC,CAAC;QACJ,CAAC;KACF;IACH,CAAC,CAAC,IAAI,CAAC"}

package/dist/src/ui/commands/quitCommand.js

@@ -8,7 +8,7 @@ import { CommandKind } from './types.js';
 export const quitCommand = {
     name: 'quit',
     altNames: ['exit'],
-    description: 'exit the cli',
+    description: 'Exit the cli',
     kind: CommandKind.BUILT_IN,
     action: (context) => {
         const now = Date.now();

package/dist/src/ui/commands/quitCommand.test.d.ts

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

package/dist/src/ui/commands/quitCommand.test.js

@@ -0,0 +1,50 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { vi, describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { quitCommand } from './quitCommand.js';
+import { createMockCommandContext } from '../../test-utils/mockCommandContext.js';
+import { formatDuration } from '../utils/formatters.js';
+vi.mock('../utils/formatters.js');
+describe('quitCommand', () => {
+    beforeEach(() => {
+        vi.useFakeTimers();
+        vi.setSystemTime(new Date('2025-01-01T01:00:00Z'));
+        vi.mocked(formatDuration).mockReturnValue('1h 0m 0s');
+    });
+    afterEach(() => {
+        vi.useRealTimers();
+        vi.clearAllMocks();
+    });
+    it('returns a QuitActionReturn object with the correct messages', () => {
+        const mockContext = createMockCommandContext({
+            session: {
+                stats: {
+                    sessionStartTime: new Date('2025-01-01T00:00:00Z'),
+                },
+            },
+        });
+        if (!quitCommand.action)
+            throw new Error('Action is not defined');
+        const result = quitCommand.action(mockContext, 'quit');
+        expect(formatDuration).toHaveBeenCalledWith(3600000); // 1 hour in ms
+        expect(result).toEqual({
+            type: 'quit',
+            messages: [
+                {
+                    type: 'user',
+                    text: '/quit',
+                    id: expect.any(Number),
+                },
+                {
+                    type: 'quit',
+                    duration: '1h 0m 0s',
+                    id: expect.any(Number),
+                },
+            ],
+        });
+    });
+});
+//# sourceMappingURL=quitCommand.test.js.map

package/dist/src/ui/commands/quitCommand.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"quitCommand.test.js","sourceRoot":"","sources":["../../../../src/ui/commands/quitCommand.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,wBAAwB,EAAE,MAAM,wCAAwC,CAAC;AAClF,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AAExD,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;AAElC,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,aAAa,CAAC,IAAI,IAAI,CAAC,sBAAsB,CAAC,CAAC,CAAC;QACnD,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,WAAW,GAAG,wBAAwB,CAAC;YAC3C,OAAO,EAAE;gBACP,KAAK,EAAE;oBACL,gBAAgB,EAAE,IAAI,IAAI,CAAC,sBAAsB,CAAC;iBACnD;aACF;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAClE,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;QAEvD,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;QACrE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,IAAI,EAAE,MAAM;YACZ,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,OAAO;oBACb,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;iBACvB;gBACD;oBACE,IAAI,EAAE,MAAM;oBACZ,QAAQ,EAAE,UAAU;oBACpB,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;iBACvB;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/ui/commands/restoreCommand.test.d.ts

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

package/dist/src/ui/commands/restoreCommand.test.js

@@ -0,0 +1,190 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { vi, describe, it, expect, beforeEach, afterEach } from 'vitest';
+import * as fs from 'node:fs/promises';
+import * as os from 'node:os';
+import * as path from 'node:path';
+import { restoreCommand } from './restoreCommand.js';
+import {} from './types.js';
+import { createMockCommandContext } from '../../test-utils/mockCommandContext.js';
+import { CELL_DIR, } from '@google/gemini-cli-core';
+describe('restoreCommand', () => {
+    let mockContext;
+    let mockConfig;
+    let mockGitService;
+    let mockSetHistory;
+    let testRootDir;
+    let geminiTempDir;
+    let checkpointsDir;
+    beforeEach(async () => {
+        testRootDir = await fs.mkdtemp(path.join(os.tmpdir(), 'restore-command-test-'));
+        geminiTempDir = path.join(testRootDir, CELL_DIR);
+        checkpointsDir = path.join(geminiTempDir, 'checkpoints');
+        // The command itself creates this, but for tests it's easier to have it ready.
+        // Some tests might remove it to test error paths.
+        await fs.mkdir(checkpointsDir, { recursive: true });
+        mockSetHistory = vi.fn().mockResolvedValue(undefined);
+        mockGitService = {
+            restoreProjectFromSnapshot: vi.fn().mockResolvedValue(undefined),
+        };
+        mockConfig = {
+            getCheckpointingEnabled: vi.fn().mockReturnValue(true),
+            storage: {
+                getProjectTempCheckpointsDir: vi.fn().mockReturnValue(checkpointsDir),
+                getProjectTempDir: vi.fn().mockReturnValue(geminiTempDir),
+            },
+            getGeminiClient: vi.fn().mockReturnValue({
+                setHistory: mockSetHistory,
+            }),
+        };
+        mockContext = createMockCommandContext({
+            services: {
+                config: mockConfig,
+                git: mockGitService,
+            },
+        });
+    });
+    afterEach(async () => {
+        vi.restoreAllMocks();
+        await fs.rm(testRootDir, { recursive: true, force: true });
+    });
+    it('should return null if checkpointing is not enabled', () => {
+        vi.mocked(mockConfig.getCheckpointingEnabled).mockReturnValue(false);
+        expect(restoreCommand(mockConfig)).toBeNull();
+    });
+    it('should return the command if checkpointing is enabled', () => {
+        expect(restoreCommand(mockConfig)).toEqual(expect.objectContaining({
+            name: 'restore',
+            description: expect.any(String),
+            action: expect.any(Function),
+            completion: expect.any(Function),
+        }));
+    });
+    describe('action', () => {
+        it('should return an error if temp dir is not found', async () => {
+            vi.mocked(mockConfig.storage.getProjectTempCheckpointsDir).mockReturnValue('');
+            expect(await restoreCommand(mockConfig)?.action?.(mockContext, '')).toEqual({
+                type: 'message',
+                messageType: 'error',
+                content: 'Could not determine the .cell-cli directory path.',
+            });
+        });
+        it('should inform when no checkpoints are found if no args are passed', async () => {
+            // Remove the directory to ensure the command creates it.
+            await fs.rm(checkpointsDir, { recursive: true, force: true });
+            const command = restoreCommand(mockConfig);
+            expect(await command?.action?.(mockContext, '')).toEqual({
+                type: 'message',
+                messageType: 'info',
+                content: 'No restorable tool calls found.',
+            });
+            // Verify the directory was created by the command.
+            await expect(fs.stat(checkpointsDir)).resolves.toBeDefined();
+        });
+        it('should list available checkpoints if no args are passed', async () => {
+            await fs.writeFile(path.join(checkpointsDir, 'test1.json'), '{}');
+            await fs.writeFile(path.join(checkpointsDir, 'test2.json'), '{}');
+            const command = restoreCommand(mockConfig);
+            expect(await command?.action?.(mockContext, '')).toEqual({
+                type: 'message',
+                messageType: 'info',
+                content: 'Available tool calls to restore:\n\ntest1\ntest2',
+            });
+        });
+        it('should return an error if the specified file is not found', async () => {
+            await fs.writeFile(path.join(checkpointsDir, 'test1.json'), '{}');
+            const command = restoreCommand(mockConfig);
+            expect(await command?.action?.(mockContext, 'test2')).toEqual({
+                type: 'message',
+                messageType: 'error',
+                content: 'File not found: test2.json',
+            });
+        });
+        it('should handle file read errors gracefully', async () => {
+            const checkpointName = 'test1';
+            const checkpointPath = path.join(checkpointsDir, `${checkpointName}.json`);
+            // Create a directory instead of a file to cause a read error.
+            await fs.mkdir(checkpointPath);
+            const command = restoreCommand(mockConfig);
+            expect(await command?.action?.(mockContext, checkpointName)).toEqual({
+                type: 'message',
+                messageType: 'error',
+                content: expect.stringContaining('Could not read restorable tool calls.'),
+            });
+        });
+        it('should restore a tool call and project state', async () => {
+            const toolCallData = {
+                history: [{ type: 'user', text: 'do a thing' }],
+                clientHistory: [{ role: 'user', parts: [{ text: 'do a thing' }] }],
+                commitHash: 'abcdef123',
+                toolCall: { name: 'run_shell_command', args: 'ls' },
+            };
+            await fs.writeFile(path.join(checkpointsDir, 'my-checkpoint.json'), JSON.stringify(toolCallData));
+            const command = restoreCommand(mockConfig);
+            expect(await command?.action?.(mockContext, 'my-checkpoint')).toEqual({
+                type: 'tool',
+                toolName: 'run_shell_command',
+                toolArgs: 'ls',
+            });
+            expect(mockContext.ui.loadHistory).toHaveBeenCalledWith(toolCallData.history);
+            expect(mockSetHistory).toHaveBeenCalledWith(toolCallData.clientHistory);
+            expect(mockGitService.restoreProjectFromSnapshot).toHaveBeenCalledWith(toolCallData.commitHash);
+            expect(mockContext.ui.addItem).toHaveBeenCalledWith({
+                type: 'info',
+                text: 'Restored project to the state before the tool call.',
+            }, expect.any(Number));
+        });
+        it('should restore even if only toolCall is present', async () => {
+            const toolCallData = {
+                toolCall: { name: 'run_shell_command', args: 'ls' },
+            };
+            await fs.writeFile(path.join(checkpointsDir, 'my-checkpoint.json'), JSON.stringify(toolCallData));
+            const command = restoreCommand(mockConfig);
+            expect(await command?.action?.(mockContext, 'my-checkpoint')).toEqual({
+                type: 'tool',
+                toolName: 'run_shell_command',
+                toolArgs: 'ls',
+            });
+            expect(mockContext.ui.loadHistory).not.toHaveBeenCalled();
+            expect(mockSetHistory).not.toHaveBeenCalled();
+            expect(mockGitService.restoreProjectFromSnapshot).not.toHaveBeenCalled();
+        });
+    });
+    it('should return an error for a checkpoint file missing the toolCall property', async () => {
+        const checkpointName = 'missing-toolcall';
+        await fs.writeFile(path.join(checkpointsDir, `${checkpointName}.json`), JSON.stringify({ history: [] }));
+        const command = restoreCommand(mockConfig);
+        expect(await command?.action?.(mockContext, checkpointName)).toEqual({
+            type: 'message',
+            messageType: 'error',
+            // A more specific error message would be ideal, but for now, we can assert the current behavior.
+            content: expect.stringContaining('Could not read restorable tool calls.'),
+        });
+    });
+    describe('completion', () => {
+        it('should return an empty array if temp dir is not found', async () => {
+            vi.mocked(mockConfig.storage.getProjectTempDir).mockReturnValue('');
+            const command = restoreCommand(mockConfig);
+            expect(await command?.completion?.(mockContext, '')).toEqual([]);
+        });
+        it('should return an empty array on readdir error', async () => {
+            await fs.rm(checkpointsDir, { recursive: true, force: true });
+            const command = restoreCommand(mockConfig);
+            expect(await command?.completion?.(mockContext, '')).toEqual([]);
+        });
+        it('should return a list of checkpoint names', async () => {
+            await fs.writeFile(path.join(checkpointsDir, 'test1.json'), '{}');
+            await fs.writeFile(path.join(checkpointsDir, 'test2.json'), '{}');
+            await fs.writeFile(path.join(checkpointsDir, 'not-a-checkpoint.txt'), '{}');
+            const command = restoreCommand(mockConfig);
+            expect(await command?.completion?.(mockContext, '')).toEqual([
+                'test1',
+                'test2',
+            ]);
+        });
+    });
+});
+//# sourceMappingURL=restoreCommand.test.js.map