@perplexity-cli/perplexity-cli 0.1.0 → 0.1.3

package/dist/src/ui/hooks/useGitBranchName.test.js

@@ -0,0 +1,189 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
+import { act } from 'react';
+import { renderHook, waitFor } from '@testing-library/react';
+import { useGitBranchName } from './useGitBranchName.js';
+import { fs, vol } from 'memfs'; // For mocking fs
+import { isCommandAvailable, execCommand, } from '@perplexity-cli/perplexity-cli-core';
+// Mock @perplexity-cli/perplexity-cli-core
+vi.mock('@perplexity-cli/perplexity-cli-core', async () => {
+    const original = await vi.importActual('@perplexity-cli/perplexity-cli-core');
+    return {
+        ...original,
+        execCommand: vi.fn(),
+        isCommandAvailable: vi.fn(),
+    };
+});
+// Mock fs and fs/promises
+vi.mock('node:fs', async () => {
+    const memfs = await vi.importActual('memfs');
+    return {
+        ...memfs.fs,
+        default: memfs.fs,
+    };
+});
+vi.mock('node:fs/promises', async () => {
+    const memfs = await vi.importActual('memfs');
+    return memfs.fs.promises;
+});
+const CWD = '/test/project';
+const GIT_LOGS_HEAD_PATH = `${CWD}/.git/logs/HEAD`;
+describe('useGitBranchName', () => {
+    beforeEach(() => {
+        vol.reset(); // Reset in-memory filesystem
+        vol.fromJSON({
+            [GIT_LOGS_HEAD_PATH]: 'ref: refs/heads/main',
+        });
+        vi.useFakeTimers(); // Use fake timers for async operations
+        isCommandAvailable.mockReturnValue({ available: true });
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+        vi.clearAllTimers();
+    });
+    it('should return branch name', async () => {
+        execCommand.mockResolvedValueOnce({
+            stdout: 'main\n',
+            stderr: '',
+            code: 0,
+        });
+        const { result, rerender } = renderHook(() => useGitBranchName(CWD));
+        await act(async () => {
+            vi.runAllTimers(); // Advance timers to trigger useEffect and exec callback
+            rerender(); // Rerender to get the updated state
+        });
+        expect(result.current).toBe('main');
+    });
+    it('should return undefined if git command fails', async () => {
+        execCommand.mockRejectedValue(new Error('Git error'));
+        const { result, rerender } = renderHook(() => useGitBranchName(CWD));
+        expect(result.current).toBeUndefined();
+        await act(async () => {
+            vi.runAllTimers();
+            rerender();
+        });
+        expect(result.current).toBeUndefined();
+    });
+    it('should return short commit hash if branch is HEAD (detached state)', async () => {
+        execCommand.mockImplementation(async (_command, args) => {
+            if (args?.includes('--abbrev-ref')) {
+                return { stdout: 'HEAD\n', stderr: '', code: 0 };
+            }
+            else if (args?.includes('--short')) {
+                return { stdout: 'a1b2c3d\n', stderr: '', code: 0 };
+            }
+            return { stdout: '', stderr: '', code: 0 };
+        });
+        const { result, rerender } = renderHook(() => useGitBranchName(CWD));
+        await act(async () => {
+            vi.runAllTimers();
+            rerender();
+        });
+        expect(result.current).toBe('a1b2c3d');
+    });
+    it('should return undefined if branch is HEAD and getting commit hash fails', async () => {
+        execCommand.mockImplementation(async (_command, args) => {
+            if (args?.includes('--abbrev-ref')) {
+                return { stdout: 'HEAD\n', stderr: '', code: 0 };
+            }
+            else if (args?.includes('--short')) {
+                throw new Error('Git error');
+            }
+            return { stdout: '', stderr: '', code: 0 };
+        });
+        const { result, rerender } = renderHook(() => useGitBranchName(CWD));
+        await act(async () => {
+            vi.runAllTimers();
+            rerender();
+        });
+        expect(result.current).toBeUndefined();
+    });
+    it('should update branch name when .git/HEAD changes', async ({ skip }) => {
+        skip(); // TODO: fix
+        execCommand
+            .mockResolvedValueOnce({
+            stdout: 'main\n',
+            stderr: '',
+            code: 0,
+        })
+            .mockResolvedValueOnce({
+            stdout: 'develop\n',
+            stderr: '',
+            code: 0,
+        });
+        const { result, rerender } = renderHook(() => useGitBranchName(CWD));
+        await act(async () => {
+            vi.runAllTimers();
+            rerender();
+        });
+        expect(result.current).toBe('main');
+        // Simulate file change event
+        // Ensure the watcher is set up before triggering the change
+        await act(async () => {
+            fs.writeFileSync(GIT_LOGS_HEAD_PATH, 'ref: refs/heads/develop'); // Trigger watcher
+            vi.runAllTimers(); // Process timers for watcher and exec
+            rerender();
+        });
+        await waitFor(() => {
+            expect(result.current).toBe('develop');
+        });
+    });
+    it('should handle watcher setup error silently', async () => {
+        // Remove .git/logs/HEAD to cause an error in fs.watch setup
+        vol.unlinkSync(GIT_LOGS_HEAD_PATH);
+        execCommand.mockResolvedValue({
+            stdout: 'main\n',
+            stderr: '',
+            code: 0,
+        });
+        const { result, rerender } = renderHook(() => useGitBranchName(CWD));
+        await act(async () => {
+            vi.runAllTimers();
+            rerender();
+        });
+        expect(result.current).toBe('main'); // Branch name should still be fetched initially
+        execCommand.mockResolvedValueOnce({
+            stdout: 'develop\n',
+            stderr: '',
+            code: 0,
+        });
+        // This write would trigger the watcher if it was set up
+        // but since it failed, the branch name should not update
+        // We need to create the file again for writeFileSync to not throw
+        vol.fromJSON({
+            [GIT_LOGS_HEAD_PATH]: 'ref: refs/heads/develop',
+        });
+        await act(async () => {
+            fs.writeFileSync(GIT_LOGS_HEAD_PATH, 'ref: refs/heads/develop');
+            vi.runAllTimers();
+            rerender();
+        });
+        // Branch name should not change because watcher setup failed
+        expect(result.current).toBe('main');
+    });
+    it('should cleanup watcher on unmount', async ({ skip }) => {
+        skip(); // TODO: fix
+        const closeMock = vi.fn();
+        const watchMock = vi.spyOn(fs, 'watch').mockReturnValue({
+            close: closeMock,
+        });
+        execCommand.mockResolvedValue({
+            stdout: 'main\n',
+            stderr: '',
+            code: 0,
+        });
+        const { unmount, rerender } = renderHook(() => useGitBranchName(CWD));
+        await act(async () => {
+            vi.runAllTimers();
+            rerender();
+        });
+        unmount();
+        expect(watchMock).toHaveBeenCalledWith(GIT_LOGS_HEAD_PATH, expect.any(Function));
+        expect(closeMock).toHaveBeenCalled();
+    });
+});
+//# sourceMappingURL=useGitBranchName.test.js.map

package/dist/src/ui/hooks/useGitBranchName.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useGitBranchName.test.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useGitBranchName.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC,CAAC,iBAAiB;AAClD,OAAO,EACL,kBAAkB,EAClB,WAAW,GACZ,MAAM,qCAAqC,CAAC;AAE7C,2CAA2C;AAC3C,EAAE,CAAC,IAAI,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;IACxD,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC,YAAY,CAEpC,qCAAqC,CAAC,CAAC;IACzC,OAAO;QACL,GAAG,QAAQ;QACX,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;QACpB,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;KAC5B,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,0BAA0B;AAC1B,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,IAAI,EAAE;IAC5B,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,YAAY,CAAyB,OAAO,CAAC,CAAC;IACrE,OAAO;QACL,GAAG,KAAK,CAAC,EAAE;QACX,OAAO,EAAE,KAAK,CAAC,EAAE;KAClB,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,KAAK,IAAI,EAAE;IACrC,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,YAAY,CAAyB,OAAO,CAAC,CAAC;IACrE,OAAO,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC;AAC3B,CAAC,CAAC,CAAC;AAEH,MAAM,GAAG,GAAG,eAAe,CAAC;AAC5B,MAAM,kBAAkB,GAAG,GAAG,GAAG,iBAAiB,CAAC;AAEnD,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,UAAU,CAAC,GAAG,EAAE;QACd,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC,6BAA6B;QAC1C,GAAG,CAAC,QAAQ,CAAC;YACX,CAAC,kBAAkB,CAAC,EAAE,sBAAsB;SAC7C,CAAC,CAAC;QACH,EAAE,CAAC,aAAa,EAAE,CAAC,CAAC,uCAAuC;QAC1D,kBAA2B,CAAC,eAAe,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;QACrB,EAAE,CAAC,cAAc,EAAE,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACxC,WAAoB,CAAC,qBAAqB,CAAC;YAC1C,MAAM,EAAE,QAAQ;YAChB,MAAM,EAAE,EAAE;YACV,IAAI,EAAE,CAAC;SACR,CAAC,CAAC;QACH,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;QAErE,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,EAAE,CAAC,YAAY,EAAE,CAAC,CAAC,wDAAwD;YAC3E,QAAQ,EAAE,CAAC,CAAC,oCAAoC;QAClD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC3D,WAAoB,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;QAEhE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;QACrE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;QAEvC,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,EAAE,CAAC,YAAY,EAAE,CAAC;YAClB,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;QACjF,WAAoB,CAAC,kBAAkB,CACtC,KAAK,EAAE,QAAgB,EAAE,IAA+B,EAAE,EAAE;YAC1D,IAAI,IAAI,EAAE,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;gBACnC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YACnD,CAAC;iBAAM,IAAI,IAAI,EAAE,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBACrC,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YACtD,CAAC;YACD,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QAC7C,CAAC,CACF,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;QACrE,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,EAAE,CAAC,YAAY,EAAE,CAAC;YAClB,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACtF,WAAoB,CAAC,kBAAkB,CACtC,KAAK,EAAE,QAAgB,EAAE,IAA+B,EAAE,EAAE;YAC1D,IAAI,IAAI,EAAE,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;gBACnC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YACnD,CAAC;iBAAM,IAAI,IAAI,EAAE,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBACrC,MAAM,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YAC/B,CAAC;YACD,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QAC7C,CAAC,CACF,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;QACrE,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,EAAE,CAAC,YAAY,EAAE,CAAC;YAClB,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE;QACxE,IAAI,EAAE,CAAC,CAAC,YAAY;QACnB,WAAoB;aAClB,qBAAqB,CAAC;YACrB,MAAM,EAAE,QAAQ;YAChB,MAAM,EAAE,EAAE;YACV,IAAI,EAAE,CAAC;SACR,CAAC;aACD,qBAAqB,CAAC;YACrB,MAAM,EAAE,WAAW;YACnB,MAAM,EAAE,EAAE;YACV,IAAI,EAAE,CAAC;SACR,CAAC,CAAC;QAEL,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;QAErE,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,EAAE,CAAC,YAAY,EAAE,CAAC;YAClB,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEpC,6BAA6B;QAC7B,4DAA4D;QAC5D,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,EAAE,CAAC,aAAa,CAAC,kBAAkB,EAAE,yBAAyB,CAAC,CAAC,CAAC,kBAAkB;YACnF,EAAE,CAAC,YAAY,EAAE,CAAC,CAAC,sCAAsC;YACzD,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,4DAA4D;QAC5D,GAAG,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;QAElC,WAAoB,CAAC,iBAAiB,CAAC;YACtC,MAAM,EAAE,QAAQ;YAChB,MAAM,EAAE,EAAE;YACV,IAAI,EAAE,CAAC;SACR,CAAC,CAAC;QAEH,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;QAErE,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,EAAE,CAAC,YAAY,EAAE,CAAC;YAClB,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,gDAAgD;QAEpF,WAAoB,CAAC,qBAAqB,CAAC;YAC1C,MAAM,EAAE,WAAW;YACnB,MAAM,EAAE,EAAE;YACV,IAAI,EAAE,CAAC;SACR,CAAC,CAAC;QAEH,wDAAwD;QACxD,yDAAyD;QACzD,kEAAkE;QAClE,GAAG,CAAC,QAAQ,CAAC;YACX,CAAC,kBAAkB,CAAC,EAAE,yBAAyB;SAChD,CAAC,CAAC;QAEH,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,EAAE,CAAC,aAAa,CAAC,kBAAkB,EAAE,yBAAyB,CAAC,CAAC;YAChE,EAAE,CAAC,YAAY,EAAE,CAAC;YAClB,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,6DAA6D;QAC7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE;QACzD,IAAI,EAAE,CAAC,CAAC,YAAY;QACpB,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,MAAM,SAAS,GAAG,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC,eAAe,CAAC;YACtD,KAAK,EAAE,SAAS;SACyB,CAAC,CAAC;QAE5C,WAAoB,CAAC,iBAAiB,CAAC;YACtC,MAAM,EAAE,QAAQ;YAChB,MAAM,EAAE,EAAE;YACV,IAAI,EAAE,CAAC;SACR,CAAC,CAAC;QAEH,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;QAEtE,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,EAAE,CAAC,YAAY,EAAE,CAAC;YAClB,QAAQ,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,OAAO,EAAE,CAAC;QACV,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,kBAAkB,EAClB,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CACrB,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAC;IACvC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/ui/hooks/useHistoryManager.d.ts

@@ -0,0 +1,22 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { HistoryItem } from '../types.js';
+type HistoryItemUpdater = (prevItem: HistoryItem) => Partial<Omit<HistoryItem, 'id'>>;
+export interface UseHistoryManagerReturn {
+    history: HistoryItem[];
+    addItem: (itemData: Omit<HistoryItem, 'id'>, baseTimestamp: number) => number;
+    updateItem: (id: number, updates: Partial<Omit<HistoryItem, 'id'>> | HistoryItemUpdater) => void;
+    clearItems: () => void;
+    loadHistory: (newHistory: HistoryItem[]) => void;
+}
+/**
+ * Custom hook to manage the chat history state.
+ *
+ * Encapsulates the history array, message ID generation, adding items,
+ * updating items, and clearing the history.
+ */
+export declare function useHistory(): UseHistoryManagerReturn;
+export {};

package/dist/src/ui/hooks/useHistoryManager.js

@@ -0,0 +1,72 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { useState, useRef, useCallback } from 'react';
+/**
+ * Custom hook to manage the chat history state.
+ *
+ * Encapsulates the history array, message ID generation, adding items,
+ * updating items, and clearing the history.
+ */
+export function useHistory() {
+    const [history, setHistory] = useState([]);
+    const messageIdCounterRef = useRef(0);
+    // Generates a unique message ID based on a timestamp and a counter.
+    const getNextMessageId = useCallback((baseTimestamp) => {
+        messageIdCounterRef.current += 1;
+        return baseTimestamp + messageIdCounterRef.current;
+    }, []);
+    const loadHistory = useCallback((newHistory) => {
+        setHistory(newHistory);
+    }, []);
+    // Adds a new item to the history state with a unique ID.
+    const addItem = useCallback((itemData, baseTimestamp) => {
+        const id = getNextMessageId(baseTimestamp);
+        const newItem = { ...itemData, id };
+        setHistory((prevHistory) => {
+            if (prevHistory.length > 0) {
+                const lastItem = prevHistory[prevHistory.length - 1];
+                // Prevent adding duplicate consecutive user messages
+                if (lastItem.type === 'user' &&
+                    newItem.type === 'user' &&
+                    lastItem.text === newItem.text) {
+                    return prevHistory; // Don't add the duplicate
+                }
+            }
+            return [...prevHistory, newItem];
+        });
+        return id; // Return the generated ID (even if not added, to keep signature)
+    }, [getNextMessageId]);
+    /**
+     * Updates an existing history item identified by its ID.
+     * @deprecated Prefer not to update history item directly as we are currently
+     * rendering all history items in <Static /> for performance reasons. Only use
+     * if ABSOLUTELY NECESSARY
+     */
+    //
+    const updateItem = useCallback((id, updates) => {
+        setHistory((prevHistory) => prevHistory.map((item) => {
+            if (item.id === id) {
+                // Apply updates based on whether it's an object or a function
+                const newUpdates = typeof updates === 'function' ? updates(item) : updates;
+                return { ...item, ...newUpdates };
+            }
+            return item;
+        }));
+    }, []);
+    // Clears the entire history state and resets the ID counter.
+    const clearItems = useCallback(() => {
+        setHistory([]);
+        messageIdCounterRef.current = 0;
+    }, []);
+    return {
+        history,
+        addItem,
+        updateItem,
+        clearItems,
+        loadHistory,
+    };
+}
+//# sourceMappingURL=useHistoryManager.js.map

package/dist/src/ui/hooks/useHistoryManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useHistoryManager.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useHistoryManager.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AAmBtD;;;;;GAKG;AACH,MAAM,UAAU,UAAU;IACxB,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,GAAG,QAAQ,CAAgB,EAAE,CAAC,CAAC;IAC1D,MAAM,mBAAmB,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;IAEtC,oEAAoE;IACpE,MAAM,gBAAgB,GAAG,WAAW,CAAC,CAAC,aAAqB,EAAU,EAAE;QACrE,mBAAmB,CAAC,OAAO,IAAI,CAAC,CAAC;QACjC,OAAO,aAAa,GAAG,mBAAmB,CAAC,OAAO,CAAC;IACrD,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,WAAW,GAAG,WAAW,CAAC,CAAC,UAAyB,EAAE,EAAE;QAC5D,UAAU,CAAC,UAAU,CAAC,CAAC;IACzB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,yDAAyD;IACzD,MAAM,OAAO,GAAG,WAAW,CACzB,CAAC,QAAiC,EAAE,aAAqB,EAAU,EAAE;QACnE,MAAM,EAAE,GAAG,gBAAgB,CAAC,aAAa,CAAC,CAAC;QAC3C,MAAM,OAAO,GAAgB,EAAE,GAAG,QAAQ,EAAE,EAAE,EAAiB,CAAC;QAEhE,UAAU,CAAC,CAAC,WAAW,EAAE,EAAE;YACzB,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,MAAM,QAAQ,GAAG,WAAW,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBACrD,qDAAqD;gBACrD,IACE,QAAQ,CAAC,IAAI,KAAK,MAAM;oBACxB,OAAO,CAAC,IAAI,KAAK,MAAM;oBACvB,QAAQ,CAAC,IAAI,KAAK,OAAO,CAAC,IAAI,EAC9B,CAAC;oBACD,OAAO,WAAW,CAAC,CAAC,0BAA0B;gBAChD,CAAC;YACH,CAAC;YACD,OAAO,CAAC,GAAG,WAAW,EAAE,OAAO,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QACH,OAAO,EAAE,CAAC,CAAC,iEAAiE;IAC9E,CAAC,EACD,CAAC,gBAAgB,CAAC,CACnB,CAAC;IAEF;;;;;OAKG;IACH,EAAE;IACF,MAAM,UAAU,GAAG,WAAW,CAC5B,CACE,EAAU,EACV,OAA8D,EAC9D,EAAE;QACF,UAAU,CAAC,CAAC,WAAW,EAAE,EAAE,CACzB,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,IAAI,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;gBACnB,8DAA8D;gBAC9D,MAAM,UAAU,GACd,OAAO,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBAC1D,OAAO,EAAE,GAAG,IAAI,EAAE,GAAG,UAAU,EAAiB,CAAC;YACnD,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CACH,CAAC;IACJ,CAAC,EACD,EAAE,CACH,CAAC;IAEF,6DAA6D;IAC7D,MAAM,UAAU,GAAG,WAAW,CAAC,GAAG,EAAE;QAClC,UAAU,CAAC,EAAE,CAAC,CAAC;QACf,mBAAmB,CAAC,OAAO,GAAG,CAAC,CAAC;IAClC,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO;QACL,OAAO;QACP,OAAO;QACP,UAAU;QACV,UAAU;QACV,WAAW;KACZ,CAAC;AACJ,CAAC"}

package/dist/src/ui/hooks/useHistoryManager.test.d.ts

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

package/dist/src/ui/hooks/useHistoryManager.test.js

@@ -0,0 +1,171 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { renderHook, act } from '@testing-library/react';
+import { useHistory } from './useHistoryManager.js';
+describe('useHistoryManager', () => {
+    it('should initialize with an empty history', () => {
+        const { result } = renderHook(() => useHistory());
+        expect(result.current.history).toEqual([]);
+    });
+    it('should add an item to history with a unique ID', () => {
+        const { result } = renderHook(() => useHistory());
+        const timestamp = Date.now();
+        const itemData = {
+            type: 'user', // Replaced HistoryItemType.User
+            text: 'Hello',
+        };
+        act(() => {
+            result.current.addItem(itemData, timestamp);
+        });
+        expect(result.current.history).toHaveLength(1);
+        expect(result.current.history[0]).toEqual(expect.objectContaining({
+            ...itemData,
+            id: expect.any(Number),
+        }));
+        // Basic check that ID incorporates timestamp
+        expect(result.current.history[0].id).toBeGreaterThanOrEqual(timestamp);
+    });
+    it('should generate unique IDs for items added with the same base timestamp', () => {
+        const { result } = renderHook(() => useHistory());
+        const timestamp = Date.now();
+        const itemData1 = {
+            type: 'user', // Replaced HistoryItemType.User
+            text: 'First',
+        };
+        const itemData2 = {
+            type: 'perplexity', // Replaced HistoryItemType.Perplexity
+            text: 'Second',
+        };
+        let id1;
+        let id2;
+        act(() => {
+            id1 = result.current.addItem(itemData1, timestamp);
+            id2 = result.current.addItem(itemData2, timestamp);
+        });
+        expect(result.current.history).toHaveLength(2);
+        expect(id1).not.toEqual(id2);
+        expect(result.current.history[0].id).toEqual(id1);
+        expect(result.current.history[1].id).toEqual(id2);
+        // IDs should be sequential based on the counter
+        expect(id2).toBeGreaterThan(id1);
+    });
+    it('should update an existing history item', () => {
+        const { result } = renderHook(() => useHistory());
+        const timestamp = Date.now();
+        const initialItem = {
+            type: 'perplexity', // Replaced HistoryItemType.Perplexity
+            text: 'Initial content',
+        };
+        let itemId;
+        act(() => {
+            itemId = result.current.addItem(initialItem, timestamp);
+        });
+        const updatedText = 'Updated content';
+        act(() => {
+            result.current.updateItem(itemId, { text: updatedText });
+        });
+        expect(result.current.history).toHaveLength(1);
+        expect(result.current.history[0]).toEqual({
+            ...initialItem,
+            id: itemId,
+            text: updatedText,
+        });
+    });
+    it('should not change history if updateHistoryItem is called with a nonexistent ID', () => {
+        const { result } = renderHook(() => useHistory());
+        const timestamp = Date.now();
+        const itemData = {
+            type: 'user', // Replaced HistoryItemType.User
+            text: 'Hello',
+        };
+        act(() => {
+            result.current.addItem(itemData, timestamp);
+        });
+        const originalHistory = [...result.current.history]; // Clone before update attempt
+        act(() => {
+            result.current.updateItem(99999, { text: 'Should not apply' }); // Nonexistent ID
+        });
+        expect(result.current.history).toEqual(originalHistory);
+    });
+    it('should clear the history', () => {
+        const { result } = renderHook(() => useHistory());
+        const timestamp = Date.now();
+        const itemData1 = {
+            type: 'user', // Replaced HistoryItemType.User
+            text: 'First',
+        };
+        const itemData2 = {
+            type: 'perplexity', // Replaced HistoryItemType.Perplexity
+            text: 'Second',
+        };
+        act(() => {
+            result.current.addItem(itemData1, timestamp);
+            result.current.addItem(itemData2, timestamp);
+        });
+        expect(result.current.history).toHaveLength(2);
+        act(() => {
+            result.current.clearItems();
+        });
+        expect(result.current.history).toEqual([]);
+    });
+    it('should not add consecutive duplicate user messages', () => {
+        const { result } = renderHook(() => useHistory());
+        const timestamp = Date.now();
+        const itemData1 = {
+            type: 'user', // Replaced HistoryItemType.User
+            text: 'Duplicate message',
+        };
+        const itemData2 = {
+            type: 'user', // Replaced HistoryItemType.User
+            text: 'Duplicate message',
+        };
+        const itemData3 = {
+            type: 'perplexity', // Replaced HistoryItemType.Perplexity
+            text: 'Perplexity response',
+        };
+        const itemData4 = {
+            type: 'user', // Replaced HistoryItemType.User
+            text: 'Another user message',
+        };
+        act(() => {
+            result.current.addItem(itemData1, timestamp);
+            result.current.addItem(itemData2, timestamp + 1); // Same text, different timestamp
+            result.current.addItem(itemData3, timestamp + 2);
+            result.current.addItem(itemData4, timestamp + 3);
+        });
+        expect(result.current.history).toHaveLength(3);
+        expect(result.current.history[0].text).toBe('Duplicate message');
+        expect(result.current.history[1].text).toBe('Perplexity response');
+        expect(result.current.history[2].text).toBe('Another user message');
+    });
+    it('should add duplicate user messages if they are not consecutive', () => {
+        const { result } = renderHook(() => useHistory());
+        const timestamp = Date.now();
+        const itemData1 = {
+            type: 'user', // Replaced HistoryItemType.User
+            text: 'Message 1',
+        };
+        const itemData2 = {
+            type: 'perplexity', // Replaced HistoryItemType.Perplexity
+            text: 'Perplexity response',
+        };
+        const itemData3 = {
+            type: 'user', // Replaced HistoryItemType.User
+            text: 'Message 1', // Duplicate text, but not consecutive
+        };
+        act(() => {
+            result.current.addItem(itemData1, timestamp);
+            result.current.addItem(itemData2, timestamp + 1);
+            result.current.addItem(itemData3, timestamp + 2);
+        });
+        expect(result.current.history).toHaveLength(3);
+        expect(result.current.history[0].text).toBe('Message 1');
+        expect(result.current.history[1].text).toBe('Perplexity response');
+        expect(result.current.history[2].text).toBe('Message 1');
+    });
+});
+//# sourceMappingURL=useHistoryManager.test.js.map

package/dist/src/ui/hooks/useHistoryManager.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useHistoryManager.test.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useHistoryManager.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAGpD,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,EAAE,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,EAAE,CAAC,CAAC;QAClD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,QAAQ,GAA4B;YACxC,IAAI,EAAE,MAAM,EAAE,gCAAgC;YAC9C,IAAI,EAAE,OAAO;SACd,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CACvC,MAAM,CAAC,gBAAgB,CAAC;YACtB,GAAG,QAAQ;YACX,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SACvB,CAAC,CACH,CAAC;QACF,6CAA6C;QAC7C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,sBAAsB,CAAC,SAAS,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yEAAyE,EAAE,GAAG,EAAE;QACjF,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,EAAE,CAAC,CAAC;QAClD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,SAAS,GAA4B;YACzC,IAAI,EAAE,MAAM,EAAE,gCAAgC;YAC9C,IAAI,EAAE,OAAO;SACd,CAAC;QACF,MAAM,SAAS,GAA4B;YACzC,IAAI,EAAE,YAAY,EAAE,sCAAsC;YAC1D,IAAI,EAAE,QAAQ;SACf,CAAC;QAEF,IAAI,GAAY,CAAC;QACjB,IAAI,GAAY,CAAC;QAEjB,GAAG,CAAC,GAAG,EAAE;YACP,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;YACnD,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAClD,gDAAgD;QAChD,MAAM,CAAC,GAAG,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,EAAE,CAAC,CAAC;QAClD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,WAAW,GAA4B;YAC3C,IAAI,EAAE,YAAY,EAAE,sCAAsC;YAC1D,IAAI,EAAE,iBAAiB;SACxB,CAAC;QACF,IAAI,MAAe,CAAC;QAEpB,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,iBAAiB,CAAC;QACtC,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACxC,GAAG,WAAW;YACd,EAAE,EAAE,MAAM;YACV,IAAI,EAAE,WAAW;SAClB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;QACxF,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,EAAE,CAAC,CAAC;QAClD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,QAAQ,GAA4B;YACxC,IAAI,EAAE,MAAM,EAAE,gCAAgC;YAC9C,IAAI,EAAE,OAAO;SACd,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,MAAM,eAAe,GAAG,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,8BAA8B;QAEnF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC,iBAAiB;QACnF,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,EAAE,CAAC,CAAC;QAClD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,SAAS,GAA4B;YACzC,IAAI,EAAE,MAAM,EAAE,gCAAgC;YAC9C,IAAI,EAAE,OAAO;SACd,CAAC;QACF,MAAM,SAAS,GAA4B;YACzC,IAAI,EAAE,YAAY,EAAE,sCAAsC;YAC1D,IAAI,EAAE,QAAQ;SACf,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;YAC7C,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAE/C,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,EAAE,CAAC,CAAC;QAClD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,SAAS,GAA4B;YACzC,IAAI,EAAE,MAAM,EAAE,gCAAgC;YAC9C,IAAI,EAAE,mBAAmB;SAC1B,CAAC;QACF,MAAM,SAAS,GAA4B;YACzC,IAAI,EAAE,MAAM,EAAE,gCAAgC;YAC9C,IAAI,EAAE,mBAAmB;SAC1B,CAAC;QACF,MAAM,SAAS,GAA4B;YACzC,IAAI,EAAE,YAAY,EAAE,sCAAsC;YAC1D,IAAI,EAAE,qBAAqB;SAC5B,CAAC;QACF,MAAM,SAAS,GAA4B;YACzC,IAAI,EAAE,MAAM,EAAE,gCAAgC;YAC9C,IAAI,EAAE,sBAAsB;SAC7B,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;YAC7C,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,iCAAiC;YACnF,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;YACjD,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACjE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACnE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,EAAE,CAAC,CAAC;QAClD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,SAAS,GAA4B;YACzC,IAAI,EAAE,MAAM,EAAE,gCAAgC;YAC9C,IAAI,EAAE,WAAW;SAClB,CAAC;QACF,MAAM,SAAS,GAA4B;YACzC,IAAI,EAAE,YAAY,EAAE,sCAAsC;YAC1D,IAAI,EAAE,qBAAqB;SAC5B,CAAC;QACF,MAAM,SAAS,GAA4B;YACzC,IAAI,EAAE,MAAM,EAAE,gCAAgC;YAC9C,IAAI,EAAE,WAAW,EAAE,sCAAsC;SAC1D,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;YAC7C,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;YACjD,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACnE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/ui/hooks/useIdeTrustListener.d.ts

@@ -0,0 +1,16 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export type RestartReason = 'NONE' | 'CONNECTION_CHANGE' | 'TRUST_CHANGE';
+/**
+ * This hook listens for trust status updates from the IDE companion extension.
+ * It provides the current trust status from the IDE and a reason if a restart
+ * is needed because the trust state has changed.
+ */
+export declare function useIdeTrustListener(): {
+    isIdeTrusted: boolean | undefined;
+    needsRestart: boolean;
+    restartReason: RestartReason;
+};

package/dist/src/ui/hooks/useIdeTrustListener.js

@@ -0,0 +1,65 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { useCallback, useEffect, useState, useSyncExternalStore, useRef, } from 'react';
+import { IdeClient, IDEConnectionStatus, ideContextStore, } from '@perplexity-cli/perplexity-cli-core';
+import { useSettings } from '../contexts/SettingsContext.js';
+import { isWorkspaceTrusted } from '../../config/trustedFolders.js';
+/**
+ * This hook listens for trust status updates from the IDE companion extension.
+ * It provides the current trust status from the IDE and a reason if a restart
+ * is needed because the trust state has changed.
+ */
+export function useIdeTrustListener() {
+    const settings = useSettings();
+    const [connectionStatus, setConnectionStatus] = useState(IDEConnectionStatus.Disconnected);
+    const previousTrust = useRef(undefined);
+    const [restartReason, setRestartReason] = useState('NONE');
+    const [needsRestart, setNeedsRestart] = useState(false);
+    const subscribe = useCallback((onStoreChange) => {
+        const handleStatusChange = (state) => {
+            setConnectionStatus(state.status);
+            setRestartReason('CONNECTION_CHANGE');
+            // Also notify useSyncExternalStore that the data has changed
+            onStoreChange();
+        };
+        const handleTrustChange = () => {
+            setRestartReason('TRUST_CHANGE');
+            onStoreChange();
+        };
+        (async () => {
+            const ideClient = await IdeClient.getInstance();
+            ideClient.addTrustChangeListener(handleTrustChange);
+            ideClient.addStatusChangeListener(handleStatusChange);
+            setConnectionStatus(ideClient.getConnectionStatus().status);
+        })();
+        return () => {
+            (async () => {
+                const ideClient = await IdeClient.getInstance();
+                ideClient.removeTrustChangeListener(handleTrustChange);
+                ideClient.removeStatusChangeListener(handleStatusChange);
+            })();
+        };
+    }, []);
+    const getSnapshot = () => {
+        if (connectionStatus !== IDEConnectionStatus.Connected) {
+            return undefined;
+        }
+        return ideContextStore.get()?.workspaceState?.isTrusted;
+    };
+    const isIdeTrusted = useSyncExternalStore(subscribe, getSnapshot);
+    useEffect(() => {
+        const currentTrust = isWorkspaceTrusted(settings.merged).isTrusted;
+        // Trigger a restart if the overall trust status for the CLI has changed,
+        // but not on the initial trust value.
+        if (previousTrust.current !== undefined &&
+            previousTrust.current !== currentTrust) {
+            setNeedsRestart(true);
+        }
+        previousTrust.current = currentTrust;
+    }, [isIdeTrusted, settings.merged]);
+    return { isIdeTrusted, needsRestart, restartReason };
+}
+//# sourceMappingURL=useIdeTrustListener.js.map

package/dist/src/ui/hooks/useIdeTrustListener.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useIdeTrustListener.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useIdeTrustListener.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,WAAW,EACX,SAAS,EACT,QAAQ,EACR,oBAAoB,EACpB,MAAM,GACP,MAAM,OAAO,CAAC;AACf,OAAO,EACL,SAAS,EACT,mBAAmB,EACnB,eAAe,GAEhB,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAAE,WAAW,EAAE,MAAM,gCAAgC,CAAC;AAC7D,OAAO,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAIpE;;;;GAIG;AACH,MAAM,UAAU,mBAAmB;IACjC,MAAM,QAAQ,GAAG,WAAW,EAAE,CAAC;IAC/B,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,QAAQ,CACtD,mBAAmB,CAAC,YAAY,CACjC,CAAC;IACF,MAAM,aAAa,GAAG,MAAM,CAAsB,SAAS,CAAC,CAAC;IAC7D,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAAgB,MAAM,CAAC,CAAC;IAC1E,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAExD,MAAM,SAAS,GAAG,WAAW,CAAC,CAAC,aAAyB,EAAE,EAAE;QAC1D,MAAM,kBAAkB,GAAG,CAAC,KAAyB,EAAE,EAAE;YACvD,mBAAmB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAClC,gBAAgB,CAAC,mBAAmB,CAAC,CAAC;YACtC,6DAA6D;YAC7D,aAAa,EAAE,CAAC;QAClB,CAAC,CAAC;QAEF,MAAM,iBAAiB,GAAG,GAAG,EAAE;YAC7B,gBAAgB,CAAC,cAAc,CAAC,CAAC;YACjC,aAAa,EAAE,CAAC;QAClB,CAAC,CAAC;QAEF,CAAC,KAAK,IAAI,EAAE;YACV,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,WAAW,EAAE,CAAC;YAChD,SAAS,CAAC,sBAAsB,CAAC,iBAAiB,CAAC,CAAC;YACpD,SAAS,CAAC,uBAAuB,CAAC,kBAAkB,CAAC,CAAC;YACtD,mBAAmB,CAAC,SAAS,CAAC,mBAAmB,EAAE,CAAC,MAAM,CAAC,CAAC;QAC9D,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,GAAG,EAAE;YACV,CAAC,KAAK,IAAI,EAAE;gBACV,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,WAAW,EAAE,CAAC;gBAChD,SAAS,CAAC,yBAAyB,CAAC,iBAAiB,CAAC,CAAC;gBACvD,SAAS,CAAC,0BAA0B,CAAC,kBAAkB,CAAC,CAAC;YAC3D,CAAC,CAAC,EAAE,CAAC;QACP,CAAC,CAAC;IACJ,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,WAAW,GAAG,GAAG,EAAE;QACvB,IAAI,gBAAgB,KAAK,mBAAmB,CAAC,SAAS,EAAE,CAAC;YACvD,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO,eAAe,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,SAAS,CAAC;IAC1D,CAAC,CAAC;IAEF,MAAM,YAAY,GAAG,oBAAoB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;IAElE,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,YAAY,GAAG,kBAAkB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC;QACnE,yEAAyE;QACzE,sCAAsC;QACtC,IACE,aAAa,CAAC,OAAO,KAAK,SAAS;YACnC,aAAa,CAAC,OAAO,KAAK,YAAY,EACtC,CAAC;YACD,eAAe,CAAC,IAAI,CAAC,CAAC;QACxB,CAAC;QACD,aAAa,CAAC,OAAO,GAAG,YAAY,CAAC;IACvC,CAAC,EAAE,CAAC,YAAY,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;IAEpC,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,aAAa,EAAE,CAAC;AACvD,CAAC"}

package/dist/src/ui/hooks/useIdeTrustListener.test.d.ts

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