@zgsm/costrict-cli 0.21.0-nightly.20251213.977248e09

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

@@ -0,0 +1,237 @@
+import { jsx as _jsx } from "react/jsx-runtime";
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { act } from 'react';
+import { render } from '../../test-utils/render.js';
+import { Text } from 'ink';
+import { usePhraseCycler, PHRASE_CHANGE_INTERVAL_MS, INTERACTIVE_SHELL_WAITING_PHRASE, } from './usePhraseCycler.js';
+import { INFORMATIVE_TIPS } from '../constants/tips.js';
+import { WITTY_LOADING_PHRASES } from '../constants/wittyPhrases.js';
+// Test component to consume the hook
+const TestComponent = ({ isActive, isWaiting, isInteractiveShellWaiting = false, lastOutputTime = 0, customPhrases, }) => {
+    const phrase = usePhraseCycler(isActive, isWaiting, isInteractiveShellWaiting, lastOutputTime, customPhrases);
+    return _jsx(Text, { children: phrase });
+};
+describe('usePhraseCycler', () => {
+    beforeEach(() => {
+        vi.useFakeTimers();
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    it('should initialize with a witty phrase when not active and not waiting', () => {
+        vi.spyOn(Math, 'random').mockImplementation(() => 0.5); // Always witty
+        const { lastFrame } = render(_jsx(TestComponent, { isActive: false, isWaiting: false }));
+        expect(WITTY_LOADING_PHRASES).toContain(lastFrame());
+    });
+    it('should show "Waiting for user confirmation..." when isWaiting is true', async () => {
+        const { lastFrame, rerender } = render(_jsx(TestComponent, { isActive: true, isWaiting: false }));
+        rerender(_jsx(TestComponent, { isActive: true, isWaiting: true }));
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(0);
+        });
+        expect(lastFrame()).toBe('Waiting for user confirmation...');
+    });
+    it('should show interactive shell waiting message when isInteractiveShellWaiting is true after 5s', async () => {
+        vi.spyOn(Math, 'random').mockImplementation(() => 0.5); // Always witty
+        const { lastFrame, rerender } = render(_jsx(TestComponent, { isActive: true, isWaiting: false }));
+        rerender(_jsx(TestComponent, { isActive: true, isWaiting: false, isInteractiveShellWaiting: true, lastOutputTime: 1 }));
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(0);
+        });
+        // Should still be showing a witty phrase or tip initially
+        expect([...WITTY_LOADING_PHRASES, ...INFORMATIVE_TIPS]).toContain(lastFrame());
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(5000);
+        });
+        expect(lastFrame()).toBe(INTERACTIVE_SHELL_WAITING_PHRASE);
+    });
+    it('should reset interactive shell waiting timer when lastOutputTime changes', async () => {
+        vi.spyOn(Math, 'random').mockImplementation(() => 0.5); // Always witty
+        const { lastFrame, rerender } = render(_jsx(TestComponent, { isActive: true, isWaiting: false, isInteractiveShellWaiting: true, lastOutputTime: 1000 }));
+        // Advance 3 seconds
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(3000);
+        });
+        // Should still be witty phrase or tip
+        expect([...WITTY_LOADING_PHRASES, ...INFORMATIVE_TIPS]).toContain(lastFrame());
+        // Update lastOutputTime
+        rerender(_jsx(TestComponent, { isActive: true, isWaiting: false, isInteractiveShellWaiting: true, lastOutputTime: 4000 }));
+        // Advance another 3 seconds (total 6s from start, but only 3s from last output)
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(3000);
+        });
+        // Should STILL be witty phrase or tip because timer reset
+        expect([...WITTY_LOADING_PHRASES, ...INFORMATIVE_TIPS]).toContain(lastFrame());
+        // Advance another 2 seconds (total 5s from last output)
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(2000);
+        });
+        expect(lastFrame()).toBe(INTERACTIVE_SHELL_WAITING_PHRASE);
+    });
+    it('should prioritize interactive shell waiting over normal waiting after 5s', async () => {
+        const { lastFrame, rerender } = render(_jsx(TestComponent, { isActive: true, isWaiting: true }));
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(0);
+        });
+        expect(lastFrame()).toBe('Waiting for user confirmation...');
+        rerender(_jsx(TestComponent, { isActive: true, isWaiting: true, isInteractiveShellWaiting: true, lastOutputTime: 1 }));
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(5000);
+        });
+        expect(lastFrame()).toBe(INTERACTIVE_SHELL_WAITING_PHRASE);
+    });
+    it('should not cycle phrases if isActive is false and not waiting', async () => {
+        const { lastFrame } = render(_jsx(TestComponent, { isActive: false, isWaiting: false }));
+        const initialPhrase = lastFrame();
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(PHRASE_CHANGE_INTERVAL_MS * 2);
+        });
+        expect(lastFrame()).toBe(initialPhrase);
+    });
+    it('should show a tip on first activation, then a witty phrase', async () => {
+        vi.spyOn(Math, 'random').mockImplementation(() => 0.99); // Subsequent phrases are witty
+        const { lastFrame } = render(_jsx(TestComponent, { isActive: true, isWaiting: false }));
+        // Initial phrase on first activation should be a tip
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(0);
+        });
+        expect(INFORMATIVE_TIPS).toContain(lastFrame());
+        // After the first interval, it should be a witty phrase
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(PHRASE_CHANGE_INTERVAL_MS + 100);
+        });
+        expect(WITTY_LOADING_PHRASES).toContain(lastFrame());
+    });
+    it('should cycle through phrases when isActive is true and not waiting', async () => {
+        vi.spyOn(Math, 'random').mockImplementation(() => 0.5); // Always witty for subsequent phrases
+        const { lastFrame } = render(_jsx(TestComponent, { isActive: true, isWaiting: false }));
+        // Initial phrase on first activation will be a tip, not necessarily from witty phrases
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(0);
+        });
+        // First activation shows a tip, so we can't guarantee it's in WITTY_LOADING_PHRASES
+        // After the first interval, it should follow the random pattern (witty phrases due to mock)
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(PHRASE_CHANGE_INTERVAL_MS + 100);
+        });
+        expect(WITTY_LOADING_PHRASES).toContain(lastFrame());
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(PHRASE_CHANGE_INTERVAL_MS);
+        });
+        expect(WITTY_LOADING_PHRASES).toContain(lastFrame());
+    });
+    it('should reset to a phrase when isActive becomes true after being false', async () => {
+        const customPhrases = ['Phrase A', 'Phrase B'];
+        let callCount = 0;
+        vi.spyOn(Math, 'random').mockImplementation(() => {
+            // For custom phrases, only 1 Math.random call is made per update.
+            // 0 -> index 0 ('Phrase A')
+            // 0.99 -> index 1 ('Phrase B')
+            const val = callCount % 2 === 0 ? 0 : 0.99;
+            callCount++;
+            return val;
+        });
+        const { lastFrame, rerender } = render(_jsx(TestComponent, { isActive: false, isWaiting: false, customPhrases: customPhrases }));
+        // Activate -> On first activation will show tip on initial call, then first interval will use first mock value for 'Phrase A'
+        rerender(_jsx(TestComponent, { isActive: true, isWaiting: false, customPhrases: customPhrases }));
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(PHRASE_CHANGE_INTERVAL_MS); // First interval after initial state -> callCount 0 -> 'Phrase A'
+        });
+        expect(customPhrases).toContain(lastFrame()); // Should be one of the custom phrases
+        // Second interval -> callCount 1 -> returns 0.99 -> 'Phrase B'
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(PHRASE_CHANGE_INTERVAL_MS);
+        });
+        expect(customPhrases).toContain(lastFrame()); // Should be one of the custom phrases
+        // Deactivate -> resets to first phrase in sequence
+        rerender(_jsx(TestComponent, { isActive: false, isWaiting: false, customPhrases: customPhrases }));
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(0);
+        });
+        // The phrase should be the first phrase after reset
+        expect(customPhrases).toContain(lastFrame());
+        // Activate again -> this will show a tip on first activation, then cycle from where mock is
+        rerender(_jsx(TestComponent, { isActive: true, isWaiting: false, customPhrases: customPhrases }));
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(PHRASE_CHANGE_INTERVAL_MS); // First interval after re-activation -> should contain phrase
+        });
+        expect(customPhrases).toContain(lastFrame()); // Should be one of the custom phrases
+    });
+    it('should clear phrase interval on unmount when active', () => {
+        const { unmount } = render(_jsx(TestComponent, { isActive: true, isWaiting: false }));
+        const clearIntervalSpy = vi.spyOn(global, 'clearInterval');
+        unmount();
+        expect(clearIntervalSpy).toHaveBeenCalledOnce();
+    });
+    it('should use custom phrases when provided', async () => {
+        const customPhrases = ['Custom Phrase 1', 'Custom Phrase 2'];
+        const randomMock = vi.spyOn(Math, 'random');
+        const { lastFrame, rerender } = render(_jsx(TestComponent, { isActive: true, isWaiting: false, customPhrases: customPhrases }));
+        // After first interval, it should use custom phrases
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(PHRASE_CHANGE_INTERVAL_MS + 100);
+        });
+        randomMock.mockReturnValue(0);
+        rerender(_jsx(TestComponent, { isActive: true, isWaiting: false, customPhrases: customPhrases }));
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(PHRASE_CHANGE_INTERVAL_MS + 100);
+        });
+        expect(customPhrases).toContain(lastFrame());
+        randomMock.mockReturnValue(0.99);
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(PHRASE_CHANGE_INTERVAL_MS);
+        });
+        expect(customPhrases).toContain(lastFrame());
+        // Test fallback to default phrases.
+        randomMock.mockRestore();
+        vi.spyOn(Math, 'random').mockReturnValue(0.5); // Always witty
+        rerender(_jsx(TestComponent, { isActive: true, isWaiting: false, customPhrases: [] }));
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(PHRASE_CHANGE_INTERVAL_MS); // Wait for first cycle
+        });
+        expect(WITTY_LOADING_PHRASES).toContain(lastFrame());
+    });
+    it('should fall back to witty phrases if custom phrases are an empty array', async () => {
+        vi.spyOn(Math, 'random').mockImplementation(() => 0.5); // Always witty for subsequent phrases
+        const { lastFrame } = render(_jsx(TestComponent, { isActive: true, isWaiting: false, customPhrases: [] }));
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(0); // First activation will be a tip
+        });
+        // First activation shows a tip, so we can't guarantee it's in WITTY_LOADING_PHRASES
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(PHRASE_CHANGE_INTERVAL_MS); // Next phrase after tip
+        });
+        expect(WITTY_LOADING_PHRASES).toContain(lastFrame());
+    });
+    it('should reset phrase when transitioning from waiting to active', async () => {
+        vi.spyOn(Math, 'random').mockImplementation(() => 0.5); // Always witty for subsequent phrases
+        const { lastFrame, rerender } = render(_jsx(TestComponent, { isActive: true, isWaiting: false }));
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(0); // First activation will be a tip
+        });
+        // First activation shows a tip, so we can't guarantee it's in WITTY_LOADING_PHRASES
+        // Cycle to a different phrase (should be witty due to mock)
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(PHRASE_CHANGE_INTERVAL_MS);
+        });
+        expect(WITTY_LOADING_PHRASES).toContain(lastFrame());
+        // Go to waiting state
+        rerender(_jsx(TestComponent, { isActive: false, isWaiting: true }));
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(0);
+        });
+        expect(lastFrame()).toBe('Waiting for user confirmation...');
+        // Go back to active cycling - should pick a phrase based on the logic (witty due to mock)
+        rerender(_jsx(TestComponent, { isActive: true, isWaiting: false }));
+        await act(async () => {
+            await vi.advanceTimersByTimeAsync(PHRASE_CHANGE_INTERVAL_MS); // Skip the tip and get next phrase
+        });
+        expect(WITTY_LOADING_PHRASES).toContain(lastFrame());
+    });
+});
+//# sourceMappingURL=usePhraseCycler.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"usePhraseCycler.test.js","sourceRoot":"","sources":["../../../../src/ui/hooks/usePhraseCycler.test.tsx"],"names":[],"mappings":";AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AACpD,OAAO,EAAE,IAAI,EAAE,MAAM,KAAK,CAAC;AAC3B,OAAO,EACL,eAAe,EACf,yBAAyB,EACzB,gCAAgC,GACjC,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AAErE,qCAAqC;AACrC,MAAM,aAAa,GAAG,CAAC,EACrB,QAAQ,EACR,SAAS,EACT,yBAAyB,GAAG,KAAK,EACjC,cAAc,GAAG,CAAC,EAClB,aAAa,GAOd,EAAE,EAAE;IACH,MAAM,MAAM,GAAG,eAAe,CAC5B,QAAQ,EACR,SAAS,EACT,yBAAyB,EACzB,cAAc,EACd,aAAa,CACd,CAAC;IACF,OAAO,KAAC,IAAI,cAAE,MAAM,GAAQ,CAAC;AAC/B,CAAC,CAAC;AAEF,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,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;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC/E,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,eAAe;QACvE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAC1B,KAAC,aAAa,IAAC,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,GAAI,CACrD,CAAC;QACF,MAAM,CAAC,qBAAqB,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,MAAM,CACpC,KAAC,aAAa,IAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,GAAI,CACpD,CAAC;QACF,QAAQ,CAAC,KAAC,aAAa,IAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,GAAI,CAAC,CAAC;QAC7D,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+FAA+F,EAAE,KAAK,IAAI,EAAE;QAC7G,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,eAAe;QACvE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,MAAM,CACpC,KAAC,aAAa,IAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,GAAI,CACpD,CAAC;QACF,QAAQ,CACN,KAAC,aAAa,IACZ,QAAQ,EAAE,IAAI,EACd,SAAS,EAAE,KAAK,EAChB,yBAAyB,EAAE,IAAI,EAC/B,cAAc,EAAE,CAAC,GACjB,CACH,CAAC;QACF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QACH,0DAA0D;QAC1D,MAAM,CAAC,CAAC,GAAG,qBAAqB,EAAE,GAAG,gBAAgB,CAAC,CAAC,CAAC,SAAS,CAC/D,SAAS,EAAE,CACZ,CAAC;QAEF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACxF,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,eAAe;QACvE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,MAAM,CACpC,KAAC,aAAa,IACZ,QAAQ,EAAE,IAAI,EACd,SAAS,EAAE,KAAK,EAChB,yBAAyB,EAAE,IAAI,EAC/B,cAAc,EAAE,IAAI,GACpB,CACH,CAAC;QAEF,oBAAoB;QACpB,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QACH,sCAAsC;QACtC,MAAM,CAAC,CAAC,GAAG,qBAAqB,EAAE,GAAG,gBAAgB,CAAC,CAAC,CAAC,SAAS,CAC/D,SAAS,EAAE,CACZ,CAAC;QAEF,wBAAwB;QACxB,QAAQ,CACN,KAAC,aAAa,IACZ,QAAQ,EAAE,IAAI,EACd,SAAS,EAAE,KAAK,EAChB,yBAAyB,EAAE,IAAI,EAC/B,cAAc,EAAE,IAAI,GACpB,CACH,CAAC;QAEF,gFAAgF;QAChF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QACH,0DAA0D;QAC1D,MAAM,CAAC,CAAC,GAAG,qBAAqB,EAAE,GAAG,gBAAgB,CAAC,CAAC,CAAC,SAAS,CAC/D,SAAS,EAAE,CACZ,CAAC;QAEF,wDAAwD;QACxD,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACxF,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,MAAM,CACpC,KAAC,aAAa,IAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,GAAI,CACnD,CAAC;QACF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;QAE7D,QAAQ,CACN,KAAC,aAAa,IACZ,QAAQ,EAAE,IAAI,EACd,SAAS,EAAE,IAAI,EACf,yBAAyB,EAAE,IAAI,EAC/B,cAAc,EAAE,CAAC,GACjB,CACH,CAAC;QACF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAC1B,KAAC,aAAa,IAAC,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,GAAI,CACrD,CAAC;QACF,MAAM,aAAa,GAAG,SAAS,EAAE,CAAC;QAClC,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,yBAAyB,GAAG,CAAC,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,+BAA+B;QACxF,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAC1B,KAAC,aAAa,IAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,GAAI,CACpD,CAAC;QAEF,qDAAqD;QACrD,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,gBAAgB,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC;QAEhD,wDAAwD;QACxD,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,yBAAyB,GAAG,GAAG,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,qBAAqB,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;QAClF,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,sCAAsC;QAC9F,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAC1B,KAAC,aAAa,IAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,GAAI,CACpD,CAAC;QACF,uFAAuF;QACvF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QACH,oFAAoF;QAEpF,4FAA4F;QAC5F,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,yBAAyB,GAAG,GAAG,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,qBAAqB,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC;QAErD,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,yBAAyB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,qBAAqB,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,aAAa,GAAG,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;QAC/C,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE;YAC/C,kEAAkE;YAClE,4BAA4B;YAC5B,+BAA+B;YAC/B,MAAM,GAAG,GAAG,SAAS,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC3C,SAAS,EAAE,CAAC;YACZ,OAAO,GAAG,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,MAAM,CACpC,KAAC,aAAa,IACZ,QAAQ,EAAE,KAAK,EACf,SAAS,EAAE,KAAK,EAChB,aAAa,EAAE,aAAa,GAC5B,CACH,CAAC;QAEF,8HAA8H;QAC9H,QAAQ,CACN,KAAC,aAAa,IACZ,QAAQ,EAAE,IAAI,EACd,SAAS,EAAE,KAAK,EAChB,aAAa,EAAE,aAAa,GAC5B,CACH,CAAC;QACF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,yBAAyB,CAAC,CAAC,CAAC,kEAAkE;QAClI,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,sCAAsC;QAEpF,+DAA+D;QAC/D,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,yBAAyB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,sCAAsC;QAEpF,mDAAmD;QACnD,QAAQ,CACN,KAAC,aAAa,IACZ,QAAQ,EAAE,KAAK,EACf,SAAS,EAAE,KAAK,EAChB,aAAa,EAAE,aAAa,GAC5B,CACH,CAAC;QACF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QACH,oDAAoD;QACpD,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC;QAE7C,4FAA4F;QAC5F,QAAQ,CACN,KAAC,aAAa,IACZ,QAAQ,EAAE,IAAI,EACd,SAAS,EAAE,KAAK,EAChB,aAAa,EAAE,aAAa,GAC5B,CACH,CAAC;QACF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,yBAAyB,CAAC,CAAC,CAAC,8DAA8D;QAC9H,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,sCAAsC;IACtF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CACxB,KAAC,aAAa,IAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,GAAI,CACpD,CAAC;QACF,MAAM,gBAAgB,GAAG,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;QAC3D,OAAO,EAAE,CAAC;QACV,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,EAAE,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,aAAa,GAAG,CAAC,iBAAiB,EAAE,iBAAiB,CAAC,CAAC;QAC7D,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QAE5C,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,MAAM,CACpC,KAAC,aAAa,IACZ,QAAQ,EAAE,IAAI,EACd,SAAS,EAAE,KAAK,EAChB,aAAa,EAAE,aAAa,GAC5B,CACH,CAAC;QAEF,qDAAqD;QACrD,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,yBAAyB,GAAG,GAAG,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QAEH,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC9B,QAAQ,CACN,KAAC,aAAa,IACZ,QAAQ,EAAE,IAAI,EACd,SAAS,EAAE,KAAK,EAChB,aAAa,EAAE,aAAa,GAC5B,CACH,CAAC;QACF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,yBAAyB,GAAG,GAAG,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC;QAE7C,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QACjC,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,yBAAyB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC;QAE7C,oCAAoC;QACpC,UAAU,CAAC,WAAW,EAAE,CAAC;QACzB,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,eAAe;QAE9D,QAAQ,CACN,KAAC,aAAa,IAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE,GAAI,CACvE,CAAC;QACF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,yBAAyB,CAAC,CAAC,CAAC,uBAAuB;QACvF,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,qBAAqB,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,sCAAsC;QAC9F,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAC1B,KAAC,aAAa,IAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE,GAAI,CACvE,CAAC;QACF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC,CAAC,iCAAiC;QACzE,CAAC,CAAC,CAAC;QACH,oFAAoF;QAEpF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,yBAAyB,CAAC,CAAC,CAAC,wBAAwB;QACxF,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,qBAAqB,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,sCAAsC;QAC9F,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,MAAM,CACpC,KAAC,aAAa,IAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,GAAI,CACpD,CAAC;QACF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC,CAAC,iCAAiC;QACzE,CAAC,CAAC,CAAC;QACH,oFAAoF;QAEpF,4DAA4D;QAC5D,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,yBAAyB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,qBAAqB,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC;QAErD,sBAAsB;QACtB,QAAQ,CAAC,KAAC,aAAa,IAAC,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,GAAI,CAAC,CAAC;QAC9D,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;QAE7D,0FAA0F;QAC1F,QAAQ,CAAC,KAAC,aAAa,IAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,GAAI,CAAC,CAAC;QAC9D,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,EAAE,CAAC,wBAAwB,CAAC,yBAAyB,CAAC,CAAC,CAAC,mCAAmC;QACnG,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,qBAAqB,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,16 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type Config } from '@google/gemini-cli-core';
+export interface PrivacyState {
+    isLoading: boolean;
+    error?: string;
+    isFreeTier?: boolean;
+    dataCollectionOptIn?: boolean;
+}
+export declare const usePrivacySettings: (config: Config) => {
+    privacyState: PrivacyState;
+    updateDataCollectionOptIn: (optIn: boolean) => Promise<void>;
+};

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

@@ -0,0 +1,103 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { useState, useEffect, useCallback } from 'react';
+import { UserTierId, getCodeAssistServer, } from '@google/gemini-cli-core';
+export const usePrivacySettings = (config) => {
+    const [privacyState, setPrivacyState] = useState({
+        isLoading: true,
+    });
+    useEffect(() => {
+        const fetchInitialState = async () => {
+            setPrivacyState({
+                isLoading: true,
+            });
+            try {
+                const server = getCodeAssistServerOrFail(config);
+                const tier = server.userTier;
+                if (tier === undefined) {
+                    throw new Error('Could not determine user tier.');
+                }
+                if (tier !== UserTierId.FREE) {
+                    // We don't need to fetch opt-out info since non-free tier
+                    // data gathering is already worked out some other way.
+                    setPrivacyState({
+                        isLoading: false,
+                        isFreeTier: false,
+                    });
+                    return;
+                }
+                const optIn = await getRemoteDataCollectionOptIn(server);
+                setPrivacyState({
+                    isLoading: false,
+                    isFreeTier: true,
+                    dataCollectionOptIn: optIn,
+                });
+            }
+            catch (e) {
+                setPrivacyState({
+                    isLoading: false,
+                    error: e instanceof Error ? e.message : String(e),
+                });
+            }
+        };
+        // eslint-disable-next-line @typescript-eslint/no-floating-promises
+        fetchInitialState();
+    }, [config]);
+    const updateDataCollectionOptIn = useCallback(async (optIn) => {
+        try {
+            const server = getCodeAssistServerOrFail(config);
+            const updatedOptIn = await setRemoteDataCollectionOptIn(server, optIn);
+            setPrivacyState({
+                isLoading: false,
+                isFreeTier: true,
+                dataCollectionOptIn: updatedOptIn,
+            });
+        }
+        catch (e) {
+            setPrivacyState({
+                isLoading: false,
+                error: e instanceof Error ? e.message : String(e),
+            });
+        }
+    }, [config]);
+    return {
+        privacyState,
+        updateDataCollectionOptIn,
+    };
+};
+function getCodeAssistServerOrFail(config) {
+    const server = getCodeAssistServer(config);
+    if (server === undefined) {
+        throw new Error('Oauth not being used');
+    }
+    else if (server.projectId === undefined) {
+        throw new Error('CodeAssist server is missing a project ID');
+    }
+    return server;
+}
+async function getRemoteDataCollectionOptIn(server) {
+    try {
+        const resp = await server.getCodeAssistGlobalUserSetting();
+        return resp.freeTierDataCollectionOptin;
+    }
+    catch (error) {
+        if (error && typeof error === 'object' && 'response' in error) {
+            const gaxiosError = error;
+            if (gaxiosError.response?.status === 404) {
+                return true;
+            }
+        }
+        throw error;
+    }
+}
+async function setRemoteDataCollectionOptIn(server, optIn) {
+    const resp = await server.setCodeAssistGlobalUserSetting({
+        cloudaicompanionProject: server.projectId,
+        freeTierDataCollectionOptin: optIn,
+    });
+    return resp.freeTierDataCollectionOptin;
+}
+//# sourceMappingURL=usePrivacySettings.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"usePrivacySettings.js","sourceRoot":"","sources":["../../../../src/ui/hooks/usePrivacySettings.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AACzD,OAAO,EAGL,UAAU,EACV,mBAAmB,GACpB,MAAM,yBAAyB,CAAC;AASjC,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,MAAc,EAAE,EAAE;IACnD,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAAe;QAC7D,SAAS,EAAE,IAAI;KAChB,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,iBAAiB,GAAG,KAAK,IAAI,EAAE;YACnC,eAAe,CAAC;gBACd,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;YACH,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,yBAAyB,CAAC,MAAM,CAAC,CAAC;gBACjD,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ,CAAC;gBAC7B,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;oBACvB,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;gBACpD,CAAC;gBACD,IAAI,IAAI,KAAK,UAAU,CAAC,IAAI,EAAE,CAAC;oBAC7B,0DAA0D;oBAC1D,uDAAuD;oBACvD,eAAe,CAAC;wBACd,SAAS,EAAE,KAAK;wBAChB,UAAU,EAAE,KAAK;qBAClB,CAAC,CAAC;oBACH,OAAO;gBACT,CAAC;gBAED,MAAM,KAAK,GAAG,MAAM,4BAA4B,CAAC,MAAM,CAAC,CAAC;gBACzD,eAAe,CAAC;oBACd,SAAS,EAAE,KAAK;oBAChB,UAAU,EAAE,IAAI;oBAChB,mBAAmB,EAAE,KAAK;iBAC3B,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,eAAe,CAAC;oBACd,SAAS,EAAE,KAAK;oBAChB,KAAK,EAAE,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;iBAClD,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC;QACF,mEAAmE;QACnE,iBAAiB,EAAE,CAAC;IACtB,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;IAEb,MAAM,yBAAyB,GAAG,WAAW,CAC3C,KAAK,EAAE,KAAc,EAAE,EAAE;QACvB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,yBAAyB,CAAC,MAAM,CAAC,CAAC;YACjD,MAAM,YAAY,GAAG,MAAM,4BAA4B,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YACvE,eAAe,CAAC;gBACd,SAAS,EAAE,KAAK;gBAChB,UAAU,EAAE,IAAI;gBAChB,mBAAmB,EAAE,YAAY;aAClC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,eAAe,CAAC;gBACd,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;aAClD,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EACD,CAAC,MAAM,CAAC,CACT,CAAC;IAEF,OAAO;QACL,YAAY;QACZ,yBAAyB;KAC1B,CAAC;AACJ,CAAC,CAAC;AAEF,SAAS,yBAAyB,CAAC,MAAc;IAC/C,MAAM,MAAM,GAAG,mBAAmB,CAAC,MAAM,CAAC,CAAC;IAC3C,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAC1C,CAAC;SAAM,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;QAC1C,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;IAC/D,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,KAAK,UAAU,4BAA4B,CACzC,MAAwB;IAExB,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,8BAA8B,EAAE,CAAC;QAC3D,OAAO,IAAI,CAAC,2BAA2B,CAAC;IAC1C,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,UAAU,IAAI,KAAK,EAAE,CAAC;YAC9D,MAAM,WAAW,GAAG,KAInB,CAAC;YACF,IAAI,WAAW,CAAC,QAAQ,EAAE,MAAM,KAAK,GAAG,EAAE,CAAC;gBACzC,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,KAAK,UAAU,4BAA4B,CACzC,MAAwB,EACxB,KAAc;IAEd,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,8BAA8B,CAAC;QACvD,uBAAuB,EAAE,MAAM,CAAC,SAAS;QACzC,2BAA2B,EAAE,KAAK;KACnC,CAAC,CAAC;IACH,OAAO,IAAI,CAAC,2BAA2B,CAAC;AAC1C,CAAC"}

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

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

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

@@ -0,0 +1,104 @@
+import { jsx as _jsx } from "react/jsx-runtime";
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+import { act } from 'react';
+import { render } from '../../test-utils/render.js';
+import { UserTierId, getCodeAssistServer } from '@google/gemini-cli-core';
+import { usePrivacySettings } from './usePrivacySettings.js';
+import { waitFor } from '../../test-utils/async.js';
+// Mock the dependencies
+vi.mock('@google/gemini-cli-core', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        getCodeAssistServer: vi.fn(),
+    };
+});
+describe('usePrivacySettings', () => {
+    const mockConfig = {};
+    beforeEach(() => {
+        vi.clearAllMocks();
+    });
+    const renderPrivacySettingsHook = () => {
+        let hookResult;
+        function TestComponent() {
+            hookResult = usePrivacySettings(mockConfig);
+            return null;
+        }
+        render(_jsx(TestComponent, {}));
+        return {
+            result: {
+                get current() {
+                    return hookResult;
+                },
+            },
+        };
+    };
+    it('should throw error when content generator is not a CodeAssistServer', async () => {
+        vi.mocked(getCodeAssistServer).mockReturnValue(undefined);
+        const { result } = renderPrivacySettingsHook();
+        await waitFor(() => {
+            expect(result.current.privacyState.isLoading).toBe(false);
+        });
+        expect(result.current.privacyState.error).toBe('Oauth not being used');
+    });
+    it('should handle paid tier users correctly', async () => {
+        // Mock paid tier response
+        vi.mocked(getCodeAssistServer).mockReturnValue({
+            projectId: 'test-project-id',
+            userTier: UserTierId.STANDARD,
+        });
+        const { result } = renderPrivacySettingsHook();
+        await waitFor(() => {
+            expect(result.current.privacyState.isLoading).toBe(false);
+        });
+        expect(result.current.privacyState.error).toBeUndefined();
+        expect(result.current.privacyState.isFreeTier).toBe(false);
+        expect(result.current.privacyState.dataCollectionOptIn).toBeUndefined();
+    });
+    it('should throw error when CodeAssistServer has no projectId', async () => {
+        vi.mocked(getCodeAssistServer).mockReturnValue({
+            userTier: UserTierId.FREE,
+        });
+        const { result } = renderPrivacySettingsHook();
+        await waitFor(() => {
+            expect(result.current.privacyState.isLoading).toBe(false);
+        });
+        expect(result.current.privacyState.error).toBe('CodeAssist server is missing a project ID');
+    });
+    it('should update data collection opt-in setting', async () => {
+        const mockCodeAssistServer = {
+            projectId: 'test-project-id',
+            getCodeAssistGlobalUserSetting: vi.fn().mockResolvedValue({
+                freeTierDataCollectionOptin: true,
+            }),
+            setCodeAssistGlobalUserSetting: vi.fn().mockResolvedValue({
+                freeTierDataCollectionOptin: false,
+            }),
+            userTier: UserTierId.FREE,
+        };
+        vi.mocked(getCodeAssistServer).mockReturnValue(mockCodeAssistServer);
+        const { result } = renderPrivacySettingsHook();
+        // Wait for initial load
+        await waitFor(() => {
+            expect(result.current.privacyState.isLoading).toBe(false);
+        });
+        // Update the setting
+        await act(async () => {
+            await result.current.updateDataCollectionOptIn(false);
+        });
+        // Wait for update to complete
+        await waitFor(() => {
+            expect(result.current.privacyState.dataCollectionOptIn).toBe(false);
+        });
+        expect(mockCodeAssistServer.setCodeAssistGlobalUserSetting).toHaveBeenCalledWith({
+            cloudaicompanionProject: 'test-project-id',
+            freeTierDataCollectionOptin: false,
+        });
+    });
+});
+//# sourceMappingURL=usePrivacySettings.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"usePrivacySettings.test.js","sourceRoot":"","sources":["../../../../src/ui/hooks/usePrivacySettings.test.tsx"],"names":[],"mappings":";AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAC5B,OAAO,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AAEpD,OAAO,EAAE,UAAU,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAC1E,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,OAAO,EAAE,MAAM,2BAA2B,CAAC;AAEpD,wBAAwB;AACxB,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IAC1D,MAAM,MAAM,GACV,MAAM,cAAc,EAA4C,CAAC;IACnE,OAAO;QACL,GAAG,MAAM;QACT,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE;KAC7B,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,MAAM,UAAU,GAAG,EAAuB,CAAC;IAE3C,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,MAAM,yBAAyB,GAAG,GAAG,EAAE;QACrC,IAAI,UAAiD,CAAC;QACtD,SAAS,aAAa;YACpB,UAAU,GAAG,kBAAkB,CAAC,UAAU,CAAC,CAAC;YAC5C,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,CAAC,KAAC,aAAa,KAAG,CAAC,CAAC;QAC1B,OAAO;YACL,MAAM,EAAE;gBACN,IAAI,OAAO;oBACT,OAAO,UAAU,CAAC;gBACpB,CAAC;aACF;SACF,CAAC;IACJ,CAAC,CAAC;IAEF,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QAE1D,MAAM,EAAE,MAAM,EAAE,GAAG,yBAAyB,EAAE,CAAC;QAE/C,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,0BAA0B;QAC1B,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;YAC7C,SAAS,EAAE,iBAAiB;YAC5B,QAAQ,EAAE,UAAU,CAAC,QAAQ;SACC,CAAC,CAAC;QAElC,MAAM,EAAE,MAAM,EAAE,GAAG,yBAAyB,EAAE,CAAC;QAE/C,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC,CAAC,aAAa,EAAE,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;YAC7C,QAAQ,EAAE,UAAU,CAAC,IAAI;SACK,CAAC,CAAC;QAElC,MAAM,EAAE,MAAM,EAAE,GAAG,yBAAyB,EAAE,CAAC;QAE/C,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAC5C,2CAA2C,CAC5C,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,oBAAoB,GAAG;YAC3B,SAAS,EAAE,iBAAiB;YAC5B,8BAA8B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBACxD,2BAA2B,EAAE,IAAI;aAClC,CAAC;YACF,8BAA8B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBACxD,2BAA2B,EAAE,KAAK;aACnC,CAAC;YACF,QAAQ,EAAE,UAAU,CAAC,IAAI;SACK,CAAC;QACjC,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC,oBAAoB,CAAC,CAAC;QAErE,MAAM,EAAE,MAAM,EAAE,GAAG,yBAAyB,EAAE,CAAC;QAE/C,wBAAwB;QACxB,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,qBAAqB;QACrB,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,8BAA8B;QAC9B,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,MAAM,CACJ,oBAAoB,CAAC,8BAA8B,CACpD,CAAC,oBAAoB,CAAC;YACrB,uBAAuB,EAAE,iBAAiB;YAC1C,2BAA2B,EAAE,KAAK;SACnC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,23 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config } from '@google/gemini-cli-core';
+import type { TextBuffer } from '../components/shared/text-buffer.js';
+export declare const PROMPT_COMPLETION_MIN_LENGTH = 5;
+export declare const PROMPT_COMPLETION_DEBOUNCE_MS = 250;
+export interface PromptCompletion {
+    text: string;
+    isLoading: boolean;
+    isActive: boolean;
+    accept: () => void;
+    clear: () => void;
+    markSelected: (selectedText: string) => void;
+}
+export interface UsePromptCompletionOptions {
+    buffer: TextBuffer;
+    config?: Config;
+    enabled: boolean;
+}
+export declare function usePromptCompletion({ buffer, config, enabled, }: UsePromptCompletionOptions): PromptCompletion;