@trukhinyuri/gemini-pp 0.27.0-nightly.20260121.97aac696f

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

@@ -0,0 +1,188 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { act } from 'react';
+import { renderHook } from '../../test-utils/render.js';
+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: 'gemini', // Replaced HistoryItemType.Gemini
+            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: 'gemini', // Replaced HistoryItemType.Gemini
+            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: 'gemini', // Replaced HistoryItemType.Gemini
+            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: 'gemini', // Replaced HistoryItemType.Gemini
+            text: 'Gemini 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('Gemini 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: 'gemini', // Replaced HistoryItemType.Gemini
+            text: 'Gemini 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('Gemini response');
+        expect(result.current.history[2].text).toBe('Message 1');
+    });
+    it('should use Date.now() as default baseTimestamp if not provided', () => {
+        const { result } = renderHook(() => useHistory());
+        const before = Date.now();
+        const itemData = {
+            type: 'user',
+            text: 'Default timestamp test',
+        };
+        act(() => {
+            result.current.addItem(itemData);
+        });
+        const after = Date.now();
+        expect(result.current.history).toHaveLength(1);
+        // ID should be >= before + 1 (since counter starts at 0 and increments to 1)
+        expect(result.current.history[0].id).toBeGreaterThanOrEqual(before + 1);
+        expect(result.current.history[0].id).toBeLessThanOrEqual(after + 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,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AACxD,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,QAAQ,EAAE,kCAAkC;YAClD,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,QAAQ,EAAE,kCAAkC;YAClD,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,QAAQ,EAAE,kCAAkC;YAClD,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,QAAQ,EAAE,kCAAkC;YAClD,IAAI,EAAE,iBAAiB;SACxB,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,iBAAiB,CAAC,CAAC;QAC/D,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,QAAQ,EAAE,kCAAkC;YAClD,IAAI,EAAE,iBAAiB;SACxB,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,iBAAiB,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC3D,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,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC1B,MAAM,QAAQ,GAA4B;YACxC,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,wBAAwB;SAC/B,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEzB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/C,6EAA6E;QAC7E,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,sBAAsB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACxE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,mBAAmB,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type ActiveHook } from '../types.js';
+export declare const useHookDisplayState: () => ActiveHook[];

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

@@ -0,0 +1,83 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { useState, useEffect, useRef } from 'react';
+import { coreEvents, CoreEvent, } from '@google/gemini-cli-core';
+import {} from '../types.js';
+import { WARNING_PROMPT_DURATION_MS } from '../constants.js';
+export const useHookDisplayState = () => {
+    const [activeHooks, setActiveHooks] = useState([]);
+    // Track start times independently of render state to calculate duration in event handlers
+    // Key: `${hookName}:${eventName}` -> Stack of StartTimes (FIFO)
+    const hookStartTimes = useRef(new Map());
+    // Track active timeouts to clear them on unmount
+    const timeouts = useRef(new Set());
+    useEffect(() => {
+        const activeTimeouts = timeouts.current;
+        const startTimes = hookStartTimes.current;
+        const handleHookStart = (payload) => {
+            const key = `${payload.hookName}:${payload.eventName}`;
+            const now = Date.now();
+            // Add start time to ref
+            if (!startTimes.has(key)) {
+                startTimes.set(key, []);
+            }
+            startTimes.get(key).push(now);
+            setActiveHooks((prev) => [
+                ...prev,
+                {
+                    name: payload.hookName,
+                    eventName: payload.eventName,
+                    index: payload.hookIndex,
+                    total: payload.totalHooks,
+                },
+            ]);
+        };
+        const handleHookEnd = (payload) => {
+            const key = `${payload.hookName}:${payload.eventName}`;
+            const starts = startTimes.get(key);
+            const startTime = starts?.shift(); // Get the earliest start time for this hook type
+            // Cleanup empty arrays in map
+            if (starts && starts.length === 0) {
+                startTimes.delete(key);
+            }
+            const now = Date.now();
+            // Default to immediate removal if start time not found (defensive)
+            const elapsed = startTime ? now - startTime : WARNING_PROMPT_DURATION_MS;
+            const remaining = WARNING_PROMPT_DURATION_MS - elapsed;
+            const removeHook = () => {
+                setActiveHooks((prev) => {
+                    const index = prev.findIndex((h) => h.name === payload.hookName && h.eventName === payload.eventName);
+                    if (index === -1)
+                        return prev;
+                    const newHooks = [...prev];
+                    newHooks.splice(index, 1);
+                    return newHooks;
+                });
+            };
+            if (remaining > 0) {
+                const timeoutId = setTimeout(() => {
+                    removeHook();
+                    activeTimeouts.delete(timeoutId);
+                }, remaining);
+                activeTimeouts.add(timeoutId);
+            }
+            else {
+                removeHook();
+            }
+        };
+        coreEvents.on(CoreEvent.HookStart, handleHookStart);
+        coreEvents.on(CoreEvent.HookEnd, handleHookEnd);
+        return () => {
+            coreEvents.off(CoreEvent.HookStart, handleHookStart);
+            coreEvents.off(CoreEvent.HookEnd, handleHookEnd);
+            // Clear all pending timeouts
+            activeTimeouts.forEach(clearTimeout);
+            activeTimeouts.clear();
+        };
+    }, []);
+    return activeHooks;
+};
+//# sourceMappingURL=useHookDisplayState.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"useHookDisplayState.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useHookDisplayState.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AACpD,OAAO,EACL,UAAU,EACV,SAAS,GAGV,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAmB,MAAM,aAAa,CAAC;AAC9C,OAAO,EAAE,0BAA0B,EAAE,MAAM,iBAAiB,CAAC;AAE7D,MAAM,CAAC,MAAM,mBAAmB,GAAG,GAAG,EAAE;IACtC,MAAM,CAAC,WAAW,EAAE,cAAc,CAAC,GAAG,QAAQ,CAAe,EAAE,CAAC,CAAC;IAEjE,0FAA0F;IAC1F,gEAAgE;IAChE,MAAM,cAAc,GAAG,MAAM,CAAwB,IAAI,GAAG,EAAE,CAAC,CAAC;IAEhE,iDAAiD;IACjD,MAAM,QAAQ,GAAG,MAAM,CAAsB,IAAI,GAAG,EAAE,CAAC,CAAC;IAExD,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,cAAc,GAAG,QAAQ,CAAC,OAAO,CAAC;QACxC,MAAM,UAAU,GAAG,cAAc,CAAC,OAAO,CAAC;QAE1C,MAAM,eAAe,GAAG,CAAC,OAAyB,EAAE,EAAE;YACpD,MAAM,GAAG,GAAG,GAAG,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YACvD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAEvB,wBAAwB;YACxB,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBACzB,UAAU,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAC1B,CAAC;YACD,UAAU,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE/B,cAAc,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;gBACvB,GAAG,IAAI;gBACP;oBACE,IAAI,EAAE,OAAO,CAAC,QAAQ;oBACtB,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,KAAK,EAAE,OAAO,CAAC,SAAS;oBACxB,KAAK,EAAE,OAAO,CAAC,UAAU;iBAC1B;aACF,CAAC,CAAC;QACL,CAAC,CAAC;QAEF,MAAM,aAAa,GAAG,CAAC,OAAuB,EAAE,EAAE;YAChD,MAAM,GAAG,GAAG,GAAG,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YACvD,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACnC,MAAM,SAAS,GAAG,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,iDAAiD;YAEpF,8BAA8B;YAC9B,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAClC,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACzB,CAAC;YAED,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACvB,mEAAmE;YACnE,MAAM,OAAO,GAAG,SAAS,CAAC,CAAC,CAAC,GAAG,GAAG,SAAS,CAAC,CAAC,CAAC,0BAA0B,CAAC;YACzE,MAAM,SAAS,GAAG,0BAA0B,GAAG,OAAO,CAAC;YAEvD,MAAM,UAAU,GAAG,GAAG,EAAE;gBACtB,cAAc,CAAC,CAAC,IAAI,EAAE,EAAE;oBACtB,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAC1B,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,QAAQ,IAAI,CAAC,CAAC,SAAS,KAAK,OAAO,CAAC,SAAS,CACnE,CAAC;oBACF,IAAI,KAAK,KAAK,CAAC,CAAC;wBAAE,OAAO,IAAI,CAAC;oBAC9B,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;oBAC3B,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;oBAC1B,OAAO,QAAQ,CAAC;gBAClB,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;YAEF,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;gBAClB,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;oBAChC,UAAU,EAAE,CAAC;oBACb,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBACnC,CAAC,EAAE,SAAS,CAAC,CAAC;gBACd,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAChC,CAAC;iBAAM,CAAC;gBACN,UAAU,EAAE,CAAC;YACf,CAAC;QACH,CAAC,CAAC;QAEF,UAAU,CAAC,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;QACpD,UAAU,CAAC,EAAE,CAAC,SAAS,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;QAEhD,OAAO,GAAG,EAAE;YACV,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;YACrD,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;YACjD,6BAA6B;YAC7B,cAAc,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YACrC,cAAc,CAAC,KAAK,EAAE,CAAC;QACzB,CAAC,CAAC;IACJ,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,WAAW,CAAC;AACrB,CAAC,CAAC"}

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

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

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

@@ -0,0 +1,180 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { renderHook } from '../../test-utils/render.js';
+import { describe, it, expect, vi, afterEach, beforeEach } from 'vitest';
+import { useHookDisplayState } from './useHookDisplayState.js';
+import { coreEvents, CoreEvent, } from '@google/gemini-cli-core';
+import { act } from 'react';
+describe('useHookDisplayState', () => {
+    beforeEach(() => {
+        vi.useFakeTimers();
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+        vi.useRealTimers();
+        coreEvents.removeAllListeners(CoreEvent.HookStart);
+        coreEvents.removeAllListeners(CoreEvent.HookEnd);
+    });
+    it('should initialize with empty hooks', () => {
+        const { result } = renderHook(() => useHookDisplayState());
+        expect(result.current).toEqual([]);
+    });
+    it('should add a hook when HookStart event is emitted', () => {
+        const { result } = renderHook(() => useHookDisplayState());
+        const payload = {
+            hookName: 'test-hook',
+            eventName: 'before-agent',
+            hookIndex: 1,
+            totalHooks: 1,
+        };
+        act(() => {
+            coreEvents.emitHookStart(payload);
+        });
+        expect(result.current).toHaveLength(1);
+        expect(result.current[0]).toMatchObject({
+            name: 'test-hook',
+            eventName: 'before-agent',
+        });
+    });
+    it('should remove a hook immediately if duration > 1s', () => {
+        const { result } = renderHook(() => useHookDisplayState());
+        const startPayload = {
+            hookName: 'test-hook',
+            eventName: 'before-agent',
+        };
+        act(() => {
+            coreEvents.emitHookStart(startPayload);
+        });
+        // Advance time by 1.1 seconds
+        act(() => {
+            vi.advanceTimersByTime(1100);
+        });
+        const endPayload = {
+            hookName: 'test-hook',
+            eventName: 'before-agent',
+            success: true,
+        };
+        act(() => {
+            coreEvents.emitHookEnd(endPayload);
+        });
+        expect(result.current).toHaveLength(0);
+    });
+    it('should delay removal if duration < 1s', () => {
+        const { result } = renderHook(() => useHookDisplayState());
+        const startPayload = {
+            hookName: 'test-hook',
+            eventName: 'before-agent',
+        };
+        act(() => {
+            coreEvents.emitHookStart(startPayload);
+        });
+        // Advance time by only 100ms
+        act(() => {
+            vi.advanceTimersByTime(100);
+        });
+        const endPayload = {
+            hookName: 'test-hook',
+            eventName: 'before-agent',
+            success: true,
+        };
+        act(() => {
+            coreEvents.emitHookEnd(endPayload);
+        });
+        // Should still be present
+        expect(result.current).toHaveLength(1);
+        // Advance remaining time (900ms needed, let's go 950ms)
+        act(() => {
+            vi.advanceTimersByTime(950);
+        });
+        expect(result.current).toHaveLength(0);
+    });
+    it('should handle multiple hooks correctly', () => {
+        const { result } = renderHook(() => useHookDisplayState());
+        act(() => {
+            coreEvents.emitHookStart({ hookName: 'h1', eventName: 'e1' });
+        });
+        act(() => {
+            vi.advanceTimersByTime(500);
+        });
+        act(() => {
+            coreEvents.emitHookStart({ hookName: 'h2', eventName: 'e1' });
+        });
+        expect(result.current).toHaveLength(2);
+        // End h1 (total time 500ms -> needs 500ms delay)
+        act(() => {
+            coreEvents.emitHookEnd({
+                hookName: 'h1',
+                eventName: 'e1',
+                success: true,
+            });
+        });
+        // h1 still there
+        expect(result.current).toHaveLength(2);
+        // Advance 600ms. h1 should disappear. h2 has been running for 600ms.
+        act(() => {
+            vi.advanceTimersByTime(600);
+        });
+        expect(result.current).toHaveLength(1);
+        expect(result.current[0].name).toBe('h2');
+        // End h2 (total time 600ms -> needs 400ms delay)
+        act(() => {
+            coreEvents.emitHookEnd({
+                hookName: 'h2',
+                eventName: 'e1',
+                success: true,
+            });
+        });
+        expect(result.current).toHaveLength(1);
+        act(() => {
+            vi.advanceTimersByTime(500);
+        });
+        expect(result.current).toHaveLength(0);
+    });
+    it('should handle interleaved hooks with same name and event', () => {
+        const { result } = renderHook(() => useHookDisplayState());
+        const hook = { hookName: 'same-hook', eventName: 'same-event' };
+        // Start Hook 1 at t=0
+        act(() => {
+            coreEvents.emitHookStart(hook);
+        });
+        // Advance to t=500
+        act(() => {
+            vi.advanceTimersByTime(500);
+        });
+        // Start Hook 2 at t=500
+        act(() => {
+            coreEvents.emitHookStart(hook);
+        });
+        expect(result.current).toHaveLength(2);
+        expect(result.current[0].name).toBe('same-hook');
+        expect(result.current[1].name).toBe('same-hook');
+        // End Hook 1 at t=600 (Duration 600ms -> delay 400ms)
+        act(() => {
+            vi.advanceTimersByTime(100);
+            coreEvents.emitHookEnd({ ...hook, success: true });
+        });
+        // Both still visible (Hook 1 pending removal in 400ms)
+        expect(result.current).toHaveLength(2);
+        // Advance 400ms (t=1000). Hook 1 should be removed.
+        act(() => {
+            vi.advanceTimersByTime(400);
+        });
+        expect(result.current).toHaveLength(1);
+        // End Hook 2 at t=1100 (Duration: 1100 - 500 = 600ms -> delay 400ms)
+        act(() => {
+            vi.advanceTimersByTime(100);
+            coreEvents.emitHookEnd({ ...hook, success: true });
+        });
+        // Hook 2 still visible (pending removal in 400ms)
+        expect(result.current).toHaveLength(1);
+        // Advance 400ms (t=1500). Hook 2 should be removed.
+        act(() => {
+            vi.advanceTimersByTime(400);
+        });
+        expect(result.current).toHaveLength(0);
+    });
+});
+//# sourceMappingURL=useHookDisplayState.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"useHookDisplayState.test.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useHookDisplayState.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AACxD,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAC/D,OAAO,EACL,UAAU,EACV,SAAS,GAGV,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAE5B,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;QACrB,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,UAAU,CAAC,kBAAkB,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACnD,UAAU,CAAC,kBAAkB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAE3D,MAAM,OAAO,GAAqB;YAChC,QAAQ,EAAE,WAAW;YACrB,SAAS,EAAE,cAAc;YACzB,SAAS,EAAE,CAAC;YACZ,UAAU,EAAE,CAAC;SACd,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,UAAU,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YACtC,IAAI,EAAE,WAAW;YACjB,SAAS,EAAE,cAAc;SAC1B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAE3D,MAAM,YAAY,GAAqB;YACrC,QAAQ,EAAE,WAAW;YACrB,SAAS,EAAE,cAAc;SAC1B,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,UAAU,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,8BAA8B;QAC9B,GAAG,CAAC,GAAG,EAAE;YACP,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,MAAM,UAAU,GAAmB;YACjC,QAAQ,EAAE,WAAW;YACrB,SAAS,EAAE,cAAc;YACzB,OAAO,EAAE,IAAI;SACd,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,UAAU,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAE3D,MAAM,YAAY,GAAqB;YACrC,QAAQ,EAAE,WAAW;YACrB,SAAS,EAAE,cAAc;SAC1B,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,UAAU,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,6BAA6B;QAC7B,GAAG,CAAC,GAAG,EAAE;YACP,EAAE,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,MAAM,UAAU,GAAmB;YACjC,QAAQ,EAAE,WAAW;YACrB,SAAS,EAAE,cAAc;YACzB,OAAO,EAAE,IAAI;SACd,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,UAAU,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,0BAA0B;QAC1B,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAEvC,wDAAwD;QACxD,GAAG,CAAC,GAAG,EAAE;YACP,EAAE,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAE3D,GAAG,CAAC,GAAG,EAAE;YACP,UAAU,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,GAAG,CAAC,GAAG,EAAE;YACP,EAAE,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,GAAG,CAAC,GAAG,EAAE;YACP,UAAU,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAEvC,iDAAiD;QACjD,GAAG,CAAC,GAAG,EAAE;YACP,UAAU,CAAC,WAAW,CAAC;gBACrB,QAAQ,EAAE,IAAI;gBACd,SAAS,EAAE,IAAI;gBACf,OAAO,EAAE,IAAI;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,iBAAiB;QACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAEvC,qEAAqE;QACrE,GAAG,CAAC,GAAG,EAAE;YACP,EAAE,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE1C,iDAAiD;QACjD,GAAG,CAAC,GAAG,EAAE;YACP,UAAU,CAAC,WAAW,CAAC;gBACrB,QAAQ,EAAE,IAAI;gBACd,SAAS,EAAE,IAAI;gBACf,OAAO,EAAE,IAAI;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAEvC,GAAG,CAAC,GAAG,EAAE;YACP,EAAE,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAC3D,MAAM,IAAI,GAAG,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC;QAEhE,sBAAsB;QACtB,GAAG,CAAC,GAAG,EAAE;YACP,UAAU,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,mBAAmB;QACnB,GAAG,CAAC,GAAG,EAAE;YACP,EAAE,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,wBAAwB;QACxB,GAAG,CAAC,GAAG,EAAE;YACP,UAAU,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAEjD,sDAAsD;QACtD,GAAG,CAAC,GAAG,EAAE;YACP,EAAE,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;YAC5B,UAAU,CAAC,WAAW,CAAC,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,uDAAuD;QACvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAEvC,oDAAoD;QACpD,GAAG,CAAC,GAAG,EAAE;YACP,EAAE,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAEvC,qEAAqE;QACrE,GAAG,CAAC,GAAG,EAAE;YACP,EAAE,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;YAC5B,UAAU,CAAC,WAAW,CAAC,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,kDAAkD;QAClD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAEvC,oDAAoD;QACpD,GAAG,CAAC,GAAG,EAAE;YACP,EAAE,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACzC,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,67 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { useCallback, useEffect, useState, useSyncExternalStore, useRef, } from 'react';
+import { IdeClient, IDEConnectionStatus, ideContextStore, } from '@google/gemini-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();
+        };
+        // eslint-disable-next-line @typescript-eslint/no-floating-promises
+        (async () => {
+            const ideClient = await IdeClient.getInstance();
+            ideClient.addTrustChangeListener(handleTrustChange);
+            ideClient.addStatusChangeListener(handleStatusChange);
+            setConnectionStatus(ideClient.getConnectionStatus().status);
+        })();
+        return () => {
+            // eslint-disable-next-line @typescript-eslint/no-floating-promises
+            (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,yBAAyB,CAAC;AACjC,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,mEAAmE;QACnE,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,mEAAmE;YACnE,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 {};