@vybestack/llxprt-code 0.1.20 → 0.1.22

package/dist/src/ui/utils/secureInputHandler.test.js

@@ -0,0 +1,274 @@
+/**
+ * @license
+ * Copyright 2025 Vybestack LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach } from 'vitest';
+import { SecureInputHandler } from './secureInputHandler.js';
+import * as os from 'os';
+import * as path from 'path';
+import { writeFileSync } from 'fs';
+describe('SecureInputHandler', () => {
+    let handler;
+    beforeEach(() => {
+        handler = new SecureInputHandler();
+    });
+    describe('shouldUseSecureMode', () => {
+        it('should detect /key command with argument', () => {
+            expect(handler.shouldUseSecureMode('/key abc')).toBe(true);
+            expect(handler.shouldUseSecureMode('/key test-key-123')).toBe(true);
+            expect(handler.shouldUseSecureMode('  /key   secret  ')).toBe(true);
+        });
+        it('should detect /key command even without argument', () => {
+            expect(handler.shouldUseSecureMode('/key')).toBe(true);
+            expect(handler.shouldUseSecureMode('/key ')).toBe(true);
+            expect(handler.shouldUseSecureMode('/ke')).toBe(false);
+            expect(handler.shouldUseSecureMode('key abc')).toBe(false);
+        });
+        it('should not detect other commands', () => {
+            expect(handler.shouldUseSecureMode('/help')).toBe(false);
+            expect(handler.shouldUseSecureMode('/clear')).toBe(false);
+            expect(handler.shouldUseSecureMode('hello world')).toBe(false);
+        });
+    });
+    describe('processInput', () => {
+        it('should mask API key in /key command', () => {
+            const input = '/key my-secret-api-key';
+            const processed = handler.processInput(input);
+            // The key portion is 'my-secret-api-key' which is 17 characters
+            // First 2: 'my', Last 2: 'ey', Middle: 13 asterisks
+            expect(processed).toBe('/key my*************ey');
+            expect(handler.isInSecureMode()).toBe(true);
+            expect(handler.getActualValue()).toBe(input);
+        });
+        it('should mask short API keys completely', () => {
+            const input = '/key abc123';
+            const processed = handler.processInput(input);
+            expect(processed).toBe('/key ******');
+            expect(handler.getActualValue()).toBe(input);
+        });
+        it('should not mask non-key commands', () => {
+            const input = '/help me with this';
+            const processed = handler.processInput(input);
+            expect(processed).toBe(input);
+            expect(handler.isInSecureMode()).toBe(false);
+        });
+        it('should handle progressive typing', () => {
+            // Simulate typing character by character
+            handler.reset();
+            // Each test should simulate the full input progression
+            handler.reset();
+            expect(handler.processInput('/')).toBe('/');
+            expect(handler.isInSecureMode()).toBe(false);
+            handler.reset();
+            expect(handler.processInput('/k')).toBe('/k');
+            expect(handler.isInSecureMode()).toBe(false);
+            handler.reset();
+            expect(handler.processInput('/ke')).toBe('/ke');
+            expect(handler.isInSecureMode()).toBe(false);
+            // Secure mode starts at /key
+            handler.reset();
+            expect(handler.processInput('/key')).toBe('/key');
+            expect(handler.isInSecureMode()).toBe(true);
+            handler.reset();
+            expect(handler.processInput('/key ')).toBe('/key ');
+            expect(handler.isInSecureMode()).toBe(true);
+            // Masking starts when key content appears
+            handler.reset();
+            expect(handler.processInput('/key a')).toBe('/key *');
+            expect(handler.isInSecureMode()).toBe(true);
+            handler.reset();
+            expect(handler.processInput('/key ab')).toBe('/key **');
+            handler.reset();
+            expect(handler.processInput('/key abc')).toBe('/key ***');
+            // 'abcd1234' is 8 characters, so it should be fully masked
+            handler.reset();
+            expect(handler.processInput('/key abcd1234')).toBe('/key ********');
+        });
+        it('should exit secure mode when text is cleared', () => {
+            handler.processInput('/key secret');
+            expect(handler.isInSecureMode()).toBe(true);
+            handler.processInput('');
+            expect(handler.isInSecureMode()).toBe(false);
+            expect(handler.getActualValue()).toBe('');
+        });
+    });
+    describe('sanitizeForHistory', () => {
+        it('should sanitize /key commands for history', () => {
+            const command = '/key my-very-secret-api-key-12345';
+            const sanitized = handler.sanitizeForHistory(command);
+            // The key portion is 'my-very-secret-api-key-12345' which is 28 characters
+            // First 2: 'my', Last 2: '45', Middle: 24 asterisks
+            expect(sanitized).toBe('/key my************************45');
+            expect(sanitized).not.toContain('secret');
+        });
+        it('should not sanitize other commands', () => {
+            const commands = ['/help', '/clear', 'normal text', '/auth login'];
+            commands.forEach((cmd) => {
+                expect(handler.sanitizeForHistory(cmd)).toBe(cmd);
+            });
+        });
+        it('should handle edge cases', () => {
+            expect(handler.sanitizeForHistory('')).toBe('');
+            expect(handler.sanitizeForHistory('/key')).toBe('/key');
+            expect(handler.sanitizeForHistory('/key ')).toBe('/key ');
+        });
+    });
+    describe('reset', () => {
+        it('should clear all secure state', () => {
+            handler.processInput('/key secret-key');
+            expect(handler.isInSecureMode()).toBe(true);
+            expect(handler.getActualValue()).toBe('/key secret-key');
+            handler.reset();
+            expect(handler.isInSecureMode()).toBe(false);
+            expect(handler.getActualValue()).toBe('');
+        });
+    });
+    describe('masking behavior', () => {
+        it('should show first and last 2 characters for long keys', () => {
+            const testCases = [
+                { input: '/key 123456789', expected: '/key 12*****89' }, // 9 chars: 12 + 5 asterisks + 89
+                { input: '/key abcdefghijklmnop', expected: '/key ab************op' }, // 16 chars: ab + 12 asterisks + op
+                { input: '/key AAAAAAAAAA', expected: '/key AA******AA' }, // 10 chars: AA + 6 asterisks + AA
+            ];
+            testCases.forEach(({ input, expected }) => {
+                handler.reset();
+                expect(handler.processInput(input)).toBe(expected);
+            });
+        });
+        it('should mask pasted API keys immediately', () => {
+            // Simulate pasting a full command with API key
+            handler.reset();
+            const pastedContent = '/key sk-proj-abcdefghijklmnopqrstuvwxyz123456789';
+            const masked = handler.processInput(pastedContent);
+            expect(masked).toBe('/key sk***************************************89');
+            expect(handler.isInSecureMode()).toBe(true);
+            expect(handler.getActualValue()).toBe(pastedContent);
+        });
+        it('should handle multi-line pasted content correctly', () => {
+            // Simulate pasting API key with newline at the end
+            handler.reset();
+            const pastedContent = '/key sk-proj-abcdefghijklmnopqrstuvwxyz123456789\n';
+            const masked = handler.processInput(pastedContent);
+            expect(masked).toBe('/key sk***************************************89\n');
+            expect(handler.isInSecureMode()).toBe(true);
+            expect(handler.getActualValue()).toBe(pastedContent);
+        });
+        it('should handle progressive input then paste scenario', () => {
+            // Simulate typing "/key " then pasting content with newline
+            handler.reset();
+            // First, user types "/key "
+            let result = handler.processInput('/key ');
+            expect(result).toBe('/key ');
+            expect(handler.isInSecureMode()).toBe(true);
+            // Then user pastes a key with newline
+            result = handler.processInput('/key sk-proj-abcdefghijklmnopqrstuvwxyz123456789\n');
+            expect(result).toBe('/key sk***************************************89\n');
+            expect(handler.isInSecureMode()).toBe(true);
+        });
+        it('should not mask space after /key command', () => {
+            // When user types "/key " the space should not be masked
+            handler.reset();
+            const result = handler.processInput('/key ');
+            expect(result).toBe('/key '); // Should NOT be '/key *'
+            expect(handler.isInSecureMode()).toBe(true);
+        });
+        it('should handle type /key[space] then paste scenario correctly', () => {
+            // This is the exact scenario the user reported
+            handler.reset();
+            // Step 1: User types "/key "
+            let result = handler.processInput('/key ');
+            expect(result).toBe('/key '); // Space should not be masked
+            // Step 2: User pastes content from pbcopy (with trailing newline)
+            result = handler.processInput('/key mySecretAPIKeyFromFile\n');
+            expect(result).toBe('/key my******************le\n'); // Should be masked with newline preserved
+            // Verify we can get the actual value
+            expect(handler.getActualValue()).toBe('/key mySecretAPIKeyFromFile\n');
+        });
+        it('should mask API key with carriage return after the key', () => {
+            // Test with a carriage return (CR) after the key
+            handler.reset();
+            const keyWithCR = '/key mySecretKey123\r';
+            const result = handler.processInput(keyWithCR);
+            // The key should be masked and the carriage return preserved
+            expect(result).toBe('/key my**********23\r');
+            expect(handler.getActualValue()).toBe(keyWithCR);
+        });
+        it('should debug CR issue - what actually happens', () => {
+            handler.reset();
+            // Simulate the exact scenario: type "/key " then paste with CR
+            const afterTyping = '/key ';
+            const afterPaste = '/key mySecretKey123\r';
+            // First, user types "/key "
+            const result1 = handler.processInput(afterTyping);
+            const debug1 = {
+                input: JSON.stringify(afterTyping),
+                output: JSON.stringify(result1),
+                isSecure: handler.isInSecureMode(),
+            };
+            // Then paste happens
+            const result2 = handler.processInput(afterPaste);
+            const debug2 = {
+                input: JSON.stringify(afterPaste),
+                output: JSON.stringify(result2),
+                isSecure: handler.isInSecureMode(),
+                shouldBeSecure: handler.shouldUseSecureMode(afterPaste),
+            };
+            // What does the regex actually match?
+            const match = afterPaste.match(/^\/key\s+([\s\S]*)/);
+            const debug3 = {
+                matched: !!match,
+                groups: match ? match.map((g) => JSON.stringify(g)) : null,
+            };
+            // Also test with actual line ending characters
+            const withLF = '/key mySecretKey123\n';
+            const withCRLF = '/key mySecretKey123\r\n';
+            const withMultipleCR = '/key mySecretKey123\r\r';
+            const lfResult = handler.processInput(withLF);
+            const crlfResult = handler.processInput(withCRLF);
+            const multiCRResult = handler.processInput(withMultipleCR);
+            // Write debug output to a file
+            const debugOutput = JSON.stringify({
+                afterTyping: debug1,
+                afterPaste: debug2,
+                regexMatch: debug3,
+                lineEndingTests: {
+                    LF: { input: withLF, output: lfResult },
+                    CRLF: { input: withCRLF, output: crlfResult },
+                    multiCR: { input: withMultipleCR, output: multiCRResult },
+                },
+            }, null, 2);
+            // Use os.tmpdir() for cross-platform temp directory
+            const tmpPath = path.join(os.tmpdir(), 'cr-debug.json');
+            writeFileSync(tmpPath, debugOutput);
+            // Expectations
+            expect(result1).toBe('/key ');
+            expect(result2).toBe('/key my**********23\r');
+            expect(lfResult).toBe('/key my**********23\n');
+            expect(crlfResult).toBe('/key my**********23\r\n');
+            expect(multiCRResult).toBe('/key my**********23\r\r');
+        });
+        it('should mask only the key part when content has newline in the middle', () => {
+            // Simulate pasting API key with newline and additional content
+            handler.reset();
+            const pastedContent = '/key sk-proj-secret123\nsome other text';
+            const masked = handler.processInput(pastedContent);
+            expect(masked).toBe('/key sk*************23\nsome other text');
+            expect(handler.isInSecureMode()).toBe(true);
+            expect(handler.getActualValue()).toBe(pastedContent);
+        });
+        it('should mask short keys completely', () => {
+            const testCases = [
+                { input: '/key a', expected: '/key *' },
+                { input: '/key ab', expected: '/key **' },
+                { input: '/key abc', expected: '/key ***' },
+                { input: '/key 12345678', expected: '/key ********' },
+            ];
+            testCases.forEach(({ input, expected }) => {
+                handler.reset();
+                expect(handler.processInput(input)).toBe(expected);
+            });
+        });
+    });
+});
+//# sourceMappingURL=secureInputHandler.test.js.map

package/dist/src/ui/utils/secureInputHandler.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"secureInputHandler.test.js","sourceRoot":"","sources":["../../../../src/ui/utils/secureInputHandler.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AAEnC,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,IAAI,OAA2B,CAAC;IAEhC,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,GAAG,IAAI,kBAAkB,EAAE,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC3D,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACpE,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvD,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACxD,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACvD,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACzD,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1D,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,KAAK,GAAG,wBAAwB,CAAC;YACvC,MAAM,SAAS,GAAG,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;YAE9C,gEAAgE;YAChE,oDAAoD;YACpD,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YACjD,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5C,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,KAAK,GAAG,aAAa,CAAC;YAC5B,MAAM,SAAS,GAAG,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;YAE9C,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACtC,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,KAAK,GAAG,oBAAoB,CAAC;YACnC,MAAM,SAAS,GAAG,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;YAE9C,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC9B,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,yCAAyC;YACzC,OAAO,CAAC,KAAK,EAAE,CAAC;YAEhB,uDAAuD;YACvD,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAE7C,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAE7C,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChD,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAE7C,6BAA6B;YAC7B,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAClD,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAE5C,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAE5C,0CAA0C;YAC1C,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACtD,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAE5C,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAExD,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAE1D,2DAA2D;YAC3D,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,OAAO,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;YACpC,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAE5C,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;YACzB,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC7C,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,OAAO,GAAG,mCAAmC,CAAC;YACpD,MAAM,SAAS,GAAG,OAAO,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;YAEtD,2EAA2E;YAC3E,oDAAoD;YACpD,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC;YAC5D,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,QAAQ,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,aAAa,CAAC,CAAC;YAEnE,QAAQ,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;gBACvB,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAClC,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAChD,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACxD,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,OAAO,CAAC,YAAY,CAAC,iBAAiB,CAAC,CAAC;YACxC,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5C,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAEzD,OAAO,CAAC,KAAK,EAAE,CAAC;YAEhB,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC7C,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,MAAM,SAAS,GAAG;gBAChB,EAAE,KAAK,EAAE,gBAAgB,EAAE,QAAQ,EAAE,gBAAgB,EAAE,EAAE,iCAAiC;gBAC1F,EAAE,KAAK,EAAE,uBAAuB,EAAE,QAAQ,EAAE,uBAAuB,EAAE,EAAE,mCAAmC;gBAC1G,EAAE,KAAK,EAAE,iBAAiB,EAAE,QAAQ,EAAE,iBAAiB,EAAE,EAAE,kCAAkC;aAC9F,CAAC;YAEF,SAAS,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE;gBACxC,OAAO,CAAC,KAAK,EAAE,CAAC;gBAChB,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,+CAA+C;YAC/C,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,aAAa,GAAG,kDAAkD,CAAC;YACzE,MAAM,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;YACxE,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5C,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,mDAAmD;YACnD,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,aAAa,GACjB,oDAAoD,CAAC;YACvD,MAAM,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,oDAAoD,CAAC,CAAC;YAC1E,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5C,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,4DAA4D;YAC5D,OAAO,CAAC,KAAK,EAAE,CAAC;YAEhB,4BAA4B;YAC5B,IAAI,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC7B,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAE5C,sCAAsC;YACtC,MAAM,GAAG,OAAO,CAAC,YAAY,CAC3B,oDAAoD,CACrD,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,oDAAoD,CAAC,CAAC;YAC1E,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,yDAAyD;YACzD,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,yBAAyB;YACvD,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACtE,+CAA+C;YAC/C,OAAO,CAAC,KAAK,EAAE,CAAC;YAEhB,6BAA6B;YAC7B,IAAI,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,6BAA6B;YAE3D,kEAAkE;YAClE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,+BAA+B,CAAC,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC,CAAC,0CAA0C;YAEhG,qCAAqC;YACrC,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,iDAAiD;YACjD,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,SAAS,GAAG,uBAAuB,CAAC;YAC1C,MAAM,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;YAE/C,6DAA6D;YAC7D,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;YAC7C,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,OAAO,CAAC,KAAK,EAAE,CAAC;YAEhB,+DAA+D;YAC/D,MAAM,WAAW,GAAG,OAAO,CAAC;YAC5B,MAAM,UAAU,GAAG,uBAAuB,CAAC;YAE3C,4BAA4B;YAC5B,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;YAClD,MAAM,MAAM,GAAG;gBACb,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;gBAClC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;gBAC/B,QAAQ,EAAE,OAAO,CAAC,cAAc,EAAE;aACnC,CAAC;YAEF,qBAAqB;YACrB,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;YACjD,MAAM,MAAM,GAAG;gBACb,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;gBACjC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;gBAC/B,QAAQ,EAAE,OAAO,CAAC,cAAc,EAAE;gBAClC,cAAc,EAAE,OAAO,CAAC,mBAAmB,CAAC,UAAU,CAAC;aACxD,CAAC;YAEF,sCAAsC;YACtC,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;YACrD,MAAM,MAAM,GAAG;gBACb,OAAO,EAAE,CAAC,CAAC,KAAK;gBAChB,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI;aAC3D,CAAC;YAEF,+CAA+C;YAC/C,MAAM,MAAM,GAAG,uBAAuB,CAAC;YACvC,MAAM,QAAQ,GAAG,yBAAyB,CAAC;YAC3C,MAAM,cAAc,GAAG,yBAAyB,CAAC;YAEjD,MAAM,QAAQ,GAAG,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YAC9C,MAAM,UAAU,GAAG,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAClD,MAAM,aAAa,GAAG,OAAO,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC;YAE3D,+BAA+B;YAC/B,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAChC;gBACE,WAAW,EAAE,MAAM;gBACnB,UAAU,EAAE,MAAM;gBAClB,UAAU,EAAE,MAAM;gBAClB,eAAe,EAAE;oBACf,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE;oBACvC,IAAI,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE;oBAC7C,OAAO,EAAE,EAAE,KAAK,EAAE,cAAc,EAAE,MAAM,EAAE,aAAa,EAAE;iBAC1D;aACF,EACD,IAAI,EACJ,CAAC,CACF,CAAC;YACF,oDAAoD;YACpD,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,eAAe,CAAC,CAAC;YACxD,aAAa,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;YAEpC,eAAe;YACf,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC9B,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;YAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;YAC/C,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YACnD,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;YAC9E,+DAA+D;YAC/D,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,aAAa,GAAG,yCAAyC,CAAC;YAChE,MAAM,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,yCAAyC,CAAC,CAAC;YAC/D,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5C,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,SAAS,GAAG;gBAChB,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE;gBACvC,EAAE,KAAK,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE;gBACzC,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,UAAU,EAAE;gBAC3C,EAAE,KAAK,EAAE,eAAe,EAAE,QAAQ,EAAE,eAAe,EAAE;aACtD,CAAC;YAEF,SAAS,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE;gBACxC,OAAO,CAAC,KAAK,EAAE,CAAC;gBAChB,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/ui/utils/updateCheck.test.d.ts

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

package/dist/src/ui/utils/updateCheck.test.js

@@ -0,0 +1,202 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { vi, describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { checkForUpdates, FETCH_TIMEOUT_MS } from './updateCheck.js';
+const getPackageJson = vi.hoisted(() => vi.fn());
+vi.mock('../../utils/package.js', () => ({
+    getPackageJson,
+}));
+const updateNotifier = vi.hoisted(() => vi.fn());
+vi.mock('update-notifier', () => ({
+    default: updateNotifier,
+}));
+describe('checkForUpdates', () => {
+    beforeEach(() => {
+        vi.useFakeTimers();
+        vi.resetAllMocks();
+        vi.clearAllTimers();
+        // Clear DEV environment variable before each test
+        delete process.env.DEV;
+    });
+    afterEach(() => {
+        vi.useRealTimers();
+        vi.restoreAllMocks();
+    });
+    it('should return null when running from source (DEV=true)', async () => {
+        process.env.DEV = 'true';
+        getPackageJson.mockResolvedValue({
+            name: 'test-package',
+            version: '1.0.0',
+        });
+        updateNotifier.mockReturnValue({
+            fetchInfo: vi
+                .fn()
+                .mockResolvedValue({ current: '1.0.0', latest: '1.1.0' }),
+        });
+        const result = await checkForUpdates();
+        expect(result).toBeNull();
+        expect(getPackageJson).not.toHaveBeenCalled();
+        expect(updateNotifier).not.toHaveBeenCalled();
+    });
+    it('should return null if package.json is missing', async () => {
+        getPackageJson.mockResolvedValue(null);
+        const result = await checkForUpdates();
+        expect(result).toBeNull();
+    });
+    it('should return null if there is no update', async () => {
+        getPackageJson.mockResolvedValue({
+            name: 'test-package',
+            version: '1.0.0',
+        });
+        updateNotifier.mockReturnValue({
+            fetchInfo: vi.fn().mockResolvedValue(null),
+        });
+        const result = await checkForUpdates();
+        expect(result).toBeNull();
+    });
+    it('should return a message if a newer version is available', async () => {
+        getPackageJson.mockResolvedValue({
+            name: 'test-package',
+            version: '1.0.0',
+        });
+        updateNotifier.mockReturnValue({
+            fetchInfo: vi
+                .fn()
+                .mockResolvedValue({ current: '1.0.0', latest: '1.1.0' }),
+        });
+        const result = await checkForUpdates();
+        expect(result?.message).toContain('1.0.0 → 1.1.0');
+        expect(result?.update).toEqual({ current: '1.0.0', latest: '1.1.0' });
+    });
+    it('should return null if the latest version is the same as the current version', async () => {
+        getPackageJson.mockResolvedValue({
+            name: 'test-package',
+            version: '1.0.0',
+        });
+        updateNotifier.mockReturnValue({
+            fetchInfo: vi
+                .fn()
+                .mockResolvedValue({ current: '1.0.0', latest: '1.0.0' }),
+        });
+        const result = await checkForUpdates();
+        expect(result).toBeNull();
+    });
+    it('should return null if the latest version is older than the current version', async () => {
+        getPackageJson.mockResolvedValue({
+            name: 'test-package',
+            version: '1.1.0',
+        });
+        updateNotifier.mockReturnValue({
+            fetchInfo: vi
+                .fn()
+                .mockResolvedValue({ current: '1.1.0', latest: '1.0.0' }),
+        });
+        const result = await checkForUpdates();
+        expect(result).toBeNull();
+    });
+    it('should return null if fetchInfo rejects', async () => {
+        getPackageJson.mockResolvedValue({
+            name: 'test-package',
+            version: '1.0.0',
+        });
+        updateNotifier.mockReturnValue({
+            fetchInfo: vi.fn().mockRejectedValue(new Error('Timeout')),
+        });
+        const result = await checkForUpdates();
+        expect(result).toBeNull();
+    });
+    it('should handle errors gracefully', async () => {
+        getPackageJson.mockRejectedValue(new Error('test error'));
+        const result = await checkForUpdates();
+        expect(result).toBeNull();
+    });
+    it('should return null if update notifier fails', async () => {
+        getPackageJson.mockResolvedValue({
+            name: 'test-package',
+            version: '1.0.0',
+        });
+        updateNotifier.mockReturnValue({
+            fetchInfo: vi.fn(async () => {
+                throw new Error('Network error');
+            }),
+        });
+        const result = await checkForUpdates();
+        expect(result).toBeNull();
+    });
+    describe('nightly updates', () => {
+        it('should notify for a newer nightly version when current is nightly', async () => {
+            getPackageJson.mockResolvedValue({
+                name: 'test-package',
+                version: '1.2.3-nightly.1',
+            });
+            const fetchInfoMock = vi.fn().mockImplementation(({ distTag }) => {
+                if (distTag === 'nightly') {
+                    return Promise.resolve({
+                        latest: '1.2.3-nightly.2',
+                        current: '1.2.3-nightly.1',
+                    });
+                }
+                if (distTag === 'latest') {
+                    return Promise.resolve({
+                        latest: '1.2.3',
+                        current: '1.2.3-nightly.1',
+                    });
+                }
+                return Promise.resolve(null);
+            });
+            updateNotifier.mockImplementation(({ pkg, distTag }) => ({
+                fetchInfo: () => fetchInfoMock({ pkg, distTag }),
+            }));
+            const result = await checkForUpdates();
+            expect(result?.message).toContain('1.2.3-nightly.1 → 1.2.3-nightly.2');
+            expect(result?.update.latest).toBe('1.2.3-nightly.2');
+        });
+        it('should prefer a newer stable version over an older nightly', async () => {
+            getPackageJson.mockResolvedValue({
+                name: 'test-package',
+                version: '1.2.3-nightly.1',
+            });
+            const fetchInfoMock = vi.fn().mockImplementation(({ distTag }) => {
+                if (distTag === 'nightly') {
+                    return Promise.resolve({
+                        latest: '1.2.3-nightly.2',
+                        current: '1.2.3-nightly.1',
+                    });
+                }
+                if (distTag === 'latest') {
+                    return Promise.resolve({
+                        latest: '1.3.0',
+                        current: '1.2.3-nightly.1',
+                    });
+                }
+                return Promise.resolve(null);
+            });
+            updateNotifier.mockImplementation(({ pkg, distTag }) => ({
+                fetchInfo: () => fetchInfoMock({ pkg, distTag }),
+            }));
+            const result = await checkForUpdates();
+            expect(result?.message).toContain('1.2.3-nightly.1 → 1.3.0');
+            expect(result?.update.latest).toBe('1.3.0');
+        });
+        it('should handle timeout gracefully for nightly checks', async () => {
+            vi.useFakeTimers();
+            getPackageJson.mockResolvedValue({
+                name: 'test-package',
+                version: '1.2.3-nightly.1',
+            });
+            updateNotifier.mockImplementation(() => ({
+                fetchInfo: () => new Promise(() => { }), // Never resolves
+            }));
+            const checkPromise = checkForUpdates();
+            // Advance timers to trigger the timeout
+            await vi.advanceTimersByTimeAsync(FETCH_TIMEOUT_MS + 100);
+            const result = await checkPromise;
+            expect(result).toBeNull();
+            vi.useRealTimers();
+        });
+    });
+});
+//# sourceMappingURL=updateCheck.test.js.map

package/dist/src/ui/utils/updateCheck.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"updateCheck.test.js","sourceRoot":"","sources":["../../../../src/ui/utils/updateCheck.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAErE,MAAM,cAAc,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACjD,EAAE,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,CAAC;IACvC,cAAc;CACf,CAAC,CAAC,CAAC;AAEJ,MAAM,cAAc,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACjD,EAAE,CAAC,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,CAAC;IAChC,OAAO,EAAE,cAAc;CACxB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,cAAc,EAAE,CAAC;QACpB,kDAAkD;QAClD,OAAO,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,OAAO,CAAC,GAAG,CAAC,GAAG,GAAG,MAAM,CAAC;QACzB,cAAc,CAAC,iBAAiB,CAAC;YAC/B,IAAI,EAAE,cAAc;YACpB,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;QACH,cAAc,CAAC,eAAe,CAAC;YAC7B,SAAS,EAAE,EAAE;iBACV,EAAE,EAAE;iBACJ,iBAAiB,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;SAC5D,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC1B,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC9C,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,cAAc,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,cAAc,CAAC,iBAAiB,CAAC;YAC/B,IAAI,EAAE,cAAc;YACpB,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;QACH,cAAc,CAAC,eAAe,CAAC;YAC7B,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;SAC3C,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,cAAc,CAAC,iBAAiB,CAAC;YAC/B,IAAI,EAAE,cAAc;YACpB,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;QACH,cAAc,CAAC,eAAe,CAAC;YAC7B,SAAS,EAAE,EAAE;iBACV,EAAE,EAAE;iBACJ,iBAAiB,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;SAC5D,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;QAC3F,cAAc,CAAC,iBAAiB,CAAC;YAC/B,IAAI,EAAE,cAAc;YACpB,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;QACH,cAAc,CAAC,eAAe,CAAC;YAC7B,SAAS,EAAE,EAAE;iBACV,EAAE,EAAE;iBACJ,iBAAiB,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;SAC5D,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4EAA4E,EAAE,KAAK,IAAI,EAAE;QAC1F,cAAc,CAAC,iBAAiB,CAAC;YAC/B,IAAI,EAAE,cAAc;YACpB,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;QACH,cAAc,CAAC,eAAe,CAAC;YAC7B,SAAS,EAAE,EAAE;iBACV,EAAE,EAAE;iBACJ,iBAAiB,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;SAC5D,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,cAAc,CAAC,iBAAiB,CAAC;YAC/B,IAAI,EAAE,cAAc;YACpB,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;QACH,cAAc,CAAC,eAAe,CAAC;YAC7B,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;SAC3D,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,cAAc,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;QAC1D,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,cAAc,CAAC,iBAAiB,CAAC;YAC/B,IAAI,EAAE,cAAc;YACpB,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;QACH,cAAc,CAAC,eAAe,CAAC;YAC7B,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;gBAC1B,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;YACnC,CAAC,CAAC;SACH,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YACjF,cAAc,CAAC,iBAAiB,CAAC;gBAC/B,IAAI,EAAE,cAAc;gBACpB,OAAO,EAAE,iBAAiB;aAC3B,CAAC,CAAC;YAEH,MAAM,aAAa,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;gBAC/D,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;oBAC1B,OAAO,OAAO,CAAC,OAAO,CAAC;wBACrB,MAAM,EAAE,iBAAiB;wBACzB,OAAO,EAAE,iBAAiB;qBAC3B,CAAC,CAAC;gBACL,CAAC;gBACD,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;oBACzB,OAAO,OAAO,CAAC,OAAO,CAAC;wBACrB,MAAM,EAAE,OAAO;wBACf,OAAO,EAAE,iBAAiB;qBAC3B,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC/B,CAAC,CAAC,CAAC;YAEH,cAAc,CAAC,kBAAkB,CAAC,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;gBACvD,SAAS,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC;aACjD,CAAC,CAAC,CAAC;YAEJ,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;YACvC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,mCAAmC,CAAC,CAAC;YACvE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,cAAc,CAAC,iBAAiB,CAAC;gBAC/B,IAAI,EAAE,cAAc;gBACpB,OAAO,EAAE,iBAAiB;aAC3B,CAAC,CAAC;YAEH,MAAM,aAAa,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;gBAC/D,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;oBAC1B,OAAO,OAAO,CAAC,OAAO,CAAC;wBACrB,MAAM,EAAE,iBAAiB;wBACzB,OAAO,EAAE,iBAAiB;qBAC3B,CAAC,CAAC;gBACL,CAAC;gBACD,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;oBACzB,OAAO,OAAO,CAAC,OAAO,CAAC;wBACrB,MAAM,EAAE,OAAO;wBACf,OAAO,EAAE,iBAAiB;qBAC3B,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC/B,CAAC,CAAC,CAAC;YAEH,cAAc,CAAC,kBAAkB,CAAC,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;gBACvD,SAAS,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC;aACjD,CAAC,CAAC,CAAC;YAEJ,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;YACvC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,EAAE,CAAC,aAAa,EAAE,CAAC;YACnB,cAAc,CAAC,iBAAiB,CAAC;gBAC/B,IAAI,EAAE,cAAc;gBACpB,OAAO,EAAE,iBAAiB;aAC3B,CAAC,CAAC;YAEH,cAAc,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;gBACvC,SAAS,EAAE,GAAG,EAAE,CAAC,IAAI,OAAO,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,EAAE,iBAAiB;aAC1D,CAAC,CAAC,CAAC;YAEJ,MAAM,YAAY,GAAG,eAAe,EAAE,CAAC;YAEvC,wCAAwC;YACxC,MAAM,EAAE,CAAC,wBAAwB,CAAC,gBAAgB,GAAG,GAAG,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;YAE1B,EAAE,CAAC,aAAa,EAAE,CAAC;QACrB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/utils/ConversationContext.test.d.ts

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

package/dist/src/utils/ConversationContext.test.js

@@ -0,0 +1,64 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach } from 'vitest';
+import { ConversationContext, } from './ConversationContext.js';
+describe('ConversationContextManager', () => {
+    // Reset the singleton before each test to ensure isolation
+    beforeEach(() => {
+        ConversationContext.reset();
+    });
+    it('should initialize with a new context when getContext is called for the first time', () => {
+        const context = ConversationContext.getContext();
+        expect(context.conversationId).toBeDefined();
+        expect(context.conversationId).toMatch(/^conv_/);
+        expect(context.parentId).toBeUndefined();
+    });
+    it('should start a new conversation with a new ID', () => {
+        const firstContext = ConversationContext.getContext();
+        ConversationContext.startNewConversation();
+        const secondContext = ConversationContext.getContext();
+        expect(secondContext.conversationId).toBeDefined();
+        expect(secondContext.conversationId).not.toEqual(firstContext.conversationId);
+        expect(secondContext.parentId).toBeUndefined();
+    });
+    it('should correctly set and retrieve the parentId', () => {
+        ConversationContext.startNewConversation(); // Ensure a conversation is active
+        const parentId = 'parent_12345';
+        ConversationContext.setParentId(parentId);
+        const context = ConversationContext.getContext();
+        expect(context.parentId).toEqual(parentId);
+    });
+    it('should correctly restore a full context object', () => {
+        const newContext = {
+            conversationId: 'restored_conv_abc',
+            parentId: 'restored_parent_def',
+        };
+        ConversationContext.setContext(newContext);
+        const restoredContext = ConversationContext.getContext();
+        expect(restoredContext).toEqual(newContext);
+    });
+    it('should reset the context and start fresh', () => {
+        // Set an initial context
+        ConversationContext.setContext({
+            conversationId: 'initial_conv',
+            parentId: 'initial_parent',
+        });
+        // Reset the context
+        ConversationContext.reset();
+        // Get the context again, which should trigger a new one to be created
+        const freshContext = ConversationContext.getContext();
+        expect(freshContext.conversationId).toBeDefined();
+        expect(freshContext.conversationId).not.toEqual('initial_conv');
+        expect(freshContext.parentId).toBeUndefined();
+    });
+    it('should not set parentId if no conversation is active (after a reset)', () => {
+        ConversationContext.reset();
+        ConversationContext.setParentId('should_not_be_set');
+        const context = ConversationContext.getContext(); // This will initialize a new context
+        expect(context.parentId).toBeUndefined();
+    });
+});
+//# sourceMappingURL=ConversationContext.test.js.map

package/dist/src/utils/ConversationContext.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ConversationContext.test.js","sourceRoot":"","sources":["../../../src/utils/ConversationContext.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC1D,OAAO,EACL,mBAAmB,GAEpB,MAAM,0BAA0B,CAAC;AAElC,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;IAC1C,2DAA2D;IAC3D,UAAU,CAAC,GAAG,EAAE;QACd,mBAAmB,CAAC,KAAK,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mFAAmF,EAAE,GAAG,EAAE;QAC3F,MAAM,OAAO,GAAG,mBAAmB,CAAC,UAAU,EAAE,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7C,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,YAAY,GAAG,mBAAmB,CAAC,UAAU,EAAE,CAAC;QACtD,mBAAmB,CAAC,oBAAoB,EAAE,CAAC;QAC3C,MAAM,aAAa,GAAG,mBAAmB,CAAC,UAAU,EAAE,CAAC;QAEvD,MAAM,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC,WAAW,EAAE,CAAC;QACnD,MAAM,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,OAAO,CAC9C,YAAY,CAAC,cAAc,CAC5B,CAAC;QACF,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,mBAAmB,CAAC,oBAAoB,EAAE,CAAC,CAAC,kCAAkC;QAC9E,MAAM,QAAQ,GAAG,cAAc,CAAC;QAChC,mBAAmB,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,OAAO,GAAG,mBAAmB,CAAC,UAAU,EAAE,CAAC;QAEjD,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,UAAU,GAAyB;YACvC,cAAc,EAAE,mBAAmB;YACnC,QAAQ,EAAE,qBAAqB;SAChC,CAAC;QACF,mBAAmB,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;QAC3C,MAAM,eAAe,GAAG,mBAAmB,CAAC,UAAU,EAAE,CAAC;QAEzD,MAAM,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,yBAAyB;QACzB,mBAAmB,CAAC,UAAU,CAAC;YAC7B,cAAc,EAAE,cAAc;YAC9B,QAAQ,EAAE,gBAAgB;SAC3B,CAAC,CAAC;QAEH,oBAAoB;QACpB,mBAAmB,CAAC,KAAK,EAAE,CAAC;QAE5B,sEAAsE;QACtE,MAAM,YAAY,GAAG,mBAAmB,CAAC,UAAU,EAAE,CAAC;QAEtD,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,WAAW,EAAE,CAAC;QAClD,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QAChE,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;QAC9E,mBAAmB,CAAC,KAAK,EAAE,CAAC;QAC5B,mBAAmB,CAAC,WAAW,CAAC,mBAAmB,CAAC,CAAC;QACrD,MAAM,OAAO,GAAG,mBAAmB,CAAC,UAAU,EAAE,CAAC,CAAC,qCAAqC;QAEvF,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/utils/gitUtils.test.d.ts

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