@google/gemini-cli-core 0.0.1

package/dist/src/fallback/handler.js

@@ -0,0 +1,51 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { AuthType } from '../core/contentGenerator.js';
+import { DEFAULT_GEMINI_FLASH_MODEL } from '../config/models.js';
+import { logFlashFallback, FlashFallbackEvent } from '../telemetry/index.js';
+export async function handleFallback(config, failedModel, authType, error) {
+    // Applicability Checks
+    if (authType !== AuthType.LOGIN_WITH_GOOGLE)
+        return null;
+    const fallbackModel = DEFAULT_GEMINI_FLASH_MODEL;
+    if (failedModel === fallbackModel)
+        return null;
+    // Consult UI Handler for Intent
+    const fallbackModelHandler = config.fallbackModelHandler;
+    if (typeof fallbackModelHandler !== 'function')
+        return null;
+    try {
+        // Pass the specific failed model to the UI handler.
+        const intent = await fallbackModelHandler(failedModel, fallbackModel, error);
+        // Process Intent and Update State
+        switch (intent) {
+            case 'retry':
+                // Activate fallback mode. The NEXT retry attempt will pick this up.
+                activateFallbackMode(config, authType);
+                return true; // Signal retryWithBackoff to continue.
+            case 'stop':
+                activateFallbackMode(config, authType);
+                return false;
+            case 'auth':
+                return false;
+            default:
+                throw new Error(`Unexpected fallback intent received from fallbackModelHandler: "${intent}"`);
+        }
+    }
+    catch (handlerError) {
+        console.error('Fallback UI handler failed:', handlerError);
+        return null;
+    }
+}
+function activateFallbackMode(config, authType) {
+    if (!config.isInFallbackMode()) {
+        config.setFallbackMode(true);
+        if (authType) {
+            logFlashFallback(config, new FlashFallbackEvent(authType));
+        }
+    }
+}
+//# sourceMappingURL=handler.js.map

package/dist/src/fallback/handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handler.js","sourceRoot":"","sources":["../../../src/fallback/handler.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AACvD,OAAO,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AACjE,OAAO,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAE7E,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,MAAc,EACd,WAAmB,EACnB,QAAiB,EACjB,KAAe;IAEf,uBAAuB;IACvB,IAAI,QAAQ,KAAK,QAAQ,CAAC,iBAAiB;QAAE,OAAO,IAAI,CAAC;IAEzD,MAAM,aAAa,GAAG,0BAA0B,CAAC;IAEjD,IAAI,WAAW,KAAK,aAAa;QAAE,OAAO,IAAI,CAAC;IAE/C,gCAAgC;IAChC,MAAM,oBAAoB,GAAG,MAAM,CAAC,oBAAoB,CAAC;IACzD,IAAI,OAAO,oBAAoB,KAAK,UAAU;QAAE,OAAO,IAAI,CAAC;IAE5D,IAAI,CAAC;QACH,oDAAoD;QACpD,MAAM,MAAM,GAAG,MAAM,oBAAoB,CACvC,WAAW,EACX,aAAa,EACb,KAAK,CACN,CAAC;QAEF,kCAAkC;QAClC,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,OAAO;gBACV,oEAAoE;gBACpE,oBAAoB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;gBACvC,OAAO,IAAI,CAAC,CAAC,uCAAuC;YAEtD,KAAK,MAAM;gBACT,oBAAoB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;gBACvC,OAAO,KAAK,CAAC;YAEf,KAAK,MAAM;gBACT,OAAO,KAAK,CAAC;YAEf;gBACE,MAAM,IAAI,KAAK,CACb,mEAAmE,MAAM,GAAG,CAC7E,CAAC;QACN,CAAC;IACH,CAAC;IAAC,OAAO,YAAY,EAAE,CAAC;QACtB,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,YAAY,CAAC,CAAC;QAC3D,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB,CAAC,MAAc,EAAE,QAA4B;IACxE,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,EAAE,CAAC;QAC/B,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC7B,IAAI,QAAQ,EAAE,CAAC;YACb,gBAAgB,CAAC,MAAM,EAAE,IAAI,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/src/fallback/handler.test.d.ts

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

package/dist/src/fallback/handler.test.js

@@ -0,0 +1,130 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach, } from 'vitest';
+import { handleFallback } from './handler.js';
+import { AuthType } from '../core/contentGenerator.js';
+import { DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_MODEL, } from '../config/models.js';
+import { logFlashFallback } from '../telemetry/index.js';
+// Mock the telemetry logger and event class
+vi.mock('../telemetry/index.js', () => ({
+    logFlashFallback: vi.fn(),
+    FlashFallbackEvent: class {
+    },
+}));
+const MOCK_PRO_MODEL = DEFAULT_GEMINI_MODEL;
+const FALLBACK_MODEL = DEFAULT_GEMINI_FLASH_MODEL;
+const AUTH_OAUTH = AuthType.LOGIN_WITH_GOOGLE;
+const AUTH_API_KEY = AuthType.USE_GEMINI;
+const createMockConfig = (overrides = {}) => ({
+    isInFallbackMode: vi.fn(() => false),
+    setFallbackMode: vi.fn(),
+    fallbackHandler: undefined,
+    ...overrides,
+});
+describe('handleFallback', () => {
+    let mockConfig;
+    let mockHandler;
+    let consoleErrorSpy;
+    beforeEach(() => {
+        vi.clearAllMocks();
+        mockHandler = vi.fn();
+        // Default setup: OAuth user, Pro model failed, handler injected
+        mockConfig = createMockConfig({
+            fallbackModelHandler: mockHandler,
+        });
+        consoleErrorSpy = vi.spyOn(console, 'error').mockImplementation(() => { });
+    });
+    afterEach(() => {
+        consoleErrorSpy.mockRestore();
+    });
+    it('should return null immediately if authType is not OAuth', async () => {
+        const result = await handleFallback(mockConfig, MOCK_PRO_MODEL, AUTH_API_KEY);
+        expect(result).toBeNull();
+        expect(mockHandler).not.toHaveBeenCalled();
+        expect(mockConfig.setFallbackMode).not.toHaveBeenCalled();
+    });
+    it('should return null if the failed model is already the fallback model', async () => {
+        const result = await handleFallback(mockConfig, FALLBACK_MODEL, // Failed model is Flash
+        AUTH_OAUTH);
+        expect(result).toBeNull();
+        expect(mockHandler).not.toHaveBeenCalled();
+    });
+    it('should return null if no fallbackHandler is injected in config', async () => {
+        const configWithoutHandler = createMockConfig({
+            fallbackModelHandler: undefined,
+        });
+        const result = await handleFallback(configWithoutHandler, MOCK_PRO_MODEL, AUTH_OAUTH);
+        expect(result).toBeNull();
+    });
+    describe('when handler returns "retry"', () => {
+        it('should activate fallback mode, log telemetry, and return true', async () => {
+            mockHandler.mockResolvedValue('retry');
+            const result = await handleFallback(mockConfig, MOCK_PRO_MODEL, AUTH_OAUTH);
+            expect(result).toBe(true);
+            expect(mockConfig.setFallbackMode).toHaveBeenCalledWith(true);
+            expect(logFlashFallback).toHaveBeenCalled();
+        });
+    });
+    describe('when handler returns "stop"', () => {
+        it('should activate fallback mode, log telemetry, and return false', async () => {
+            mockHandler.mockResolvedValue('stop');
+            const result = await handleFallback(mockConfig, MOCK_PRO_MODEL, AUTH_OAUTH);
+            expect(result).toBe(false);
+            expect(mockConfig.setFallbackMode).toHaveBeenCalledWith(true);
+            expect(logFlashFallback).toHaveBeenCalled();
+        });
+    });
+    describe('when handler returns "auth"', () => {
+        it('should NOT activate fallback mode and return false', async () => {
+            mockHandler.mockResolvedValue('auth');
+            const result = await handleFallback(mockConfig, MOCK_PRO_MODEL, AUTH_OAUTH);
+            expect(result).toBe(false);
+            expect(mockConfig.setFallbackMode).not.toHaveBeenCalled();
+            expect(logFlashFallback).not.toHaveBeenCalled();
+        });
+    });
+    describe('when handler returns an unexpected value', () => {
+        it('should log an error and return null', async () => {
+            mockHandler.mockResolvedValue(null);
+            const result = await handleFallback(mockConfig, MOCK_PRO_MODEL, AUTH_OAUTH);
+            expect(result).toBeNull();
+            expect(consoleErrorSpy).toHaveBeenCalledWith('Fallback UI handler failed:', new Error('Unexpected fallback intent received from fallbackModelHandler: "null"'));
+            expect(mockConfig.setFallbackMode).not.toHaveBeenCalled();
+        });
+    });
+    it('should pass the correct context (failedModel, fallbackModel, error) to the handler', async () => {
+        const mockError = new Error('Quota Exceeded');
+        mockHandler.mockResolvedValue('retry');
+        await handleFallback(mockConfig, MOCK_PRO_MODEL, AUTH_OAUTH, mockError);
+        expect(mockHandler).toHaveBeenCalledWith(MOCK_PRO_MODEL, FALLBACK_MODEL, mockError);
+    });
+    it('should not call setFallbackMode or log telemetry if already in fallback mode', async () => {
+        // Setup config where fallback mode is already active
+        const activeFallbackConfig = createMockConfig({
+            fallbackModelHandler: mockHandler,
+            isInFallbackMode: vi.fn(() => true), // Already active
+            setFallbackMode: vi.fn(),
+        });
+        mockHandler.mockResolvedValue('retry');
+        const result = await handleFallback(activeFallbackConfig, MOCK_PRO_MODEL, AUTH_OAUTH);
+        // Should still return true to allow the retry (which will use the active fallback mode)
+        expect(result).toBe(true);
+        // Should still consult the handler
+        expect(mockHandler).toHaveBeenCalled();
+        // But should not mutate state or log telemetry again
+        expect(activeFallbackConfig.setFallbackMode).not.toHaveBeenCalled();
+        expect(logFlashFallback).not.toHaveBeenCalled();
+    });
+    it('should catch errors from the handler, log an error, and return null', async () => {
+        const handlerError = new Error('UI interaction failed');
+        mockHandler.mockRejectedValue(handlerError);
+        const result = await handleFallback(mockConfig, MOCK_PRO_MODEL, AUTH_OAUTH);
+        expect(result).toBeNull();
+        expect(consoleErrorSpy).toHaveBeenCalledWith('Fallback UI handler failed:', handlerError);
+        expect(mockConfig.setFallbackMode).not.toHaveBeenCalled();
+    });
+});
+//# sourceMappingURL=handler.test.js.map

package/dist/src/fallback/handler.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handler.test.js","sourceRoot":"","sources":["../../../src/fallback/handler.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,QAAQ,EACR,EAAE,EACF,MAAM,EACN,EAAE,EACF,UAAU,EAGV,SAAS,GACV,MAAM,QAAQ,CAAC;AAChB,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAE9C,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AACvD,OAAO,EACL,0BAA0B,EAC1B,oBAAoB,GACrB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAGzD,4CAA4C;AAC5C,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;IACzB,kBAAkB,EAAE;KAAQ;CAC7B,CAAC,CAAC,CAAC;AAEJ,MAAM,cAAc,GAAG,oBAAoB,CAAC;AAC5C,MAAM,cAAc,GAAG,0BAA0B,CAAC;AAClD,MAAM,UAAU,GAAG,QAAQ,CAAC,iBAAiB,CAAC;AAC9C,MAAM,YAAY,GAAG,QAAQ,CAAC,UAAU,CAAC;AAEzC,MAAM,gBAAgB,GAAG,CAAC,YAA6B,EAAE,EAAU,EAAE,CACnE,CAAC;IACC,gBAAgB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;IACpC,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;IACxB,eAAe,EAAE,SAAS;IAC1B,GAAG,SAAS;CACb,CAAsB,CAAC;AAE1B,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,UAAkB,CAAC;IACvB,IAAI,WAAuC,CAAC;IAC5C,IAAI,eAA6B,CAAC;IAElC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACtB,gEAAgE;QAChE,UAAU,GAAG,gBAAgB,CAAC;YAC5B,oBAAoB,EAAE,WAAW;SAClC,CAAC,CAAC;QACH,eAAe,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,eAAe,CAAC,WAAW,EAAE,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,MAAM,GAAG,MAAM,cAAc,CACjC,UAAU,EACV,cAAc,EACd,YAAY,CACb,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC1B,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC3C,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACpF,MAAM,MAAM,GAAG,MAAM,cAAc,CACjC,UAAU,EACV,cAAc,EAAE,wBAAwB;QACxC,UAAU,CACX,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC1B,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC9E,MAAM,oBAAoB,GAAG,gBAAgB,CAAC;YAC5C,oBAAoB,EAAE,SAAS;SAChC,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,cAAc,CACjC,oBAAoB,EACpB,cAAc,EACd,UAAU,CACX,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;QAC5C,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,WAAW,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAEvC,MAAM,MAAM,GAAG,MAAM,cAAc,CACjC,UAAU,EACV,cAAc,EACd,UAAU,CACX,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1B,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;YAC9D,MAAM,CAAC,gBAAgB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;QAC3C,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,WAAW,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAEtC,MAAM,MAAM,GAAG,MAAM,cAAc,CACjC,UAAU,EACV,cAAc,EACd,UAAU,CACX,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC3B,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;YAC9D,MAAM,CAAC,gBAAgB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;QAC3C,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,WAAW,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAEtC,MAAM,MAAM,GAAG,MAAM,cAAc,CACjC,UAAU,EACV,cAAc,EACd,UAAU,CACX,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC3B,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAC1D,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0CAA0C,EAAE,GAAG,EAAE;QACxD,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,WAAW,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAEpC,MAAM,MAAM,GAAG,MAAM,cAAc,CACjC,UAAU,EACV,cAAc,EACd,UAAU,CACX,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;YAC1B,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAA6B,EAC7B,IAAI,KAAK,CACP,uEAAuE,CACxE,CACF,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC5D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oFAAoF,EAAE,KAAK,IAAI,EAAE;QAClG,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAC9C,WAAW,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAEvC,MAAM,cAAc,CAAC,UAAU,EAAE,cAAc,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;QAExE,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CACtC,cAAc,EACd,cAAc,EACd,SAAS,CACV,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;QAC5F,qDAAqD;QACrD,MAAM,oBAAoB,GAAG,gBAAgB,CAAC;YAC5C,oBAAoB,EAAE,WAAW;YACjC,gBAAgB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,iBAAiB;YACtD,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;SACzB,CAAC,CAAC;QAEH,WAAW,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAEvC,MAAM,MAAM,GAAG,MAAM,cAAc,CACjC,oBAAoB,EACpB,cAAc,EACd,UAAU,CACX,CAAC;QAEF,wFAAwF;QACxF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1B,mCAAmC;QACnC,MAAM,CAAC,WAAW,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvC,qDAAqD;QACrD,MAAM,CAAC,oBAAoB,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACpE,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,MAAM,YAAY,GAAG,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QACxD,WAAW,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAE5C,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,UAAU,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;QAE5E,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC1B,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAA6B,EAC7B,YAAY,CACb,CAAC;QACF,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC5D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/fallback/types.d.ts

@@ -0,0 +1,14 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Defines the intent returned by the UI layer during a fallback scenario.
+ */
+export type FallbackIntent = 'retry' | 'stop' | 'auth';
+/**
+ * The interface for the handler provided by the UI layer (e.g., the CLI)
+ * to interact with the user during a fallback scenario.
+ */
+export type FallbackModelHandler = (failedModel: string, fallbackModel: string, error?: unknown) => Promise<FallbackIntent | null>;

package/dist/src/fallback/types.js

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};
+//# sourceMappingURL=types.js.map

package/dist/src/fallback/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/fallback/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG"}

package/dist/src/generated/git-commit.d.ts

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare const GIT_COMMIT_INFO = "db5b49b2";
+export declare const CLI_VERSION = "0.0.1";

package/dist/src/generated/git-commit.js

@@ -0,0 +1,10 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+// This file is auto-generated by the build script (scripts/generate-git-commit-info.js)
+// Do not edit this file manually.
+export const GIT_COMMIT_INFO = 'db5b49b2';
+export const CLI_VERSION = '0.0.1';
+//# sourceMappingURL=git-commit.js.map

package/dist/src/generated/git-commit.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"git-commit.js","sourceRoot":"","sources":["../../../src/generated/git-commit.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,wFAAwF;AACxF,kCAAkC;AAClC,MAAM,CAAC,MAAM,eAAe,GAAG,UAAU,CAAC;AAC1C,MAAM,CAAC,MAAM,WAAW,GAAG,OAAO,CAAC"}

package/dist/src/ide/constants.d.ts

@@ -0,0 +1,9 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare const GEMINI_CLI_COMPANION_EXTENSION_NAME = "Gemini CLI Companion";
+export declare const IDE_MAX_OPEN_FILES = 10;
+export declare const IDE_MAX_SELECTED_TEXT_LENGTH = 16384;
+export declare const IDE_REQUEST_TIMEOUT_MS: number;

package/dist/src/ide/constants.js

@@ -0,0 +1,10 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export const GEMINI_CLI_COMPANION_EXTENSION_NAME = 'Gemini CLI Companion';
+export const IDE_MAX_OPEN_FILES = 10;
+export const IDE_MAX_SELECTED_TEXT_LENGTH = 16384; // 16 KiB limit
+export const IDE_REQUEST_TIMEOUT_MS = 10 * 60 * 1000; // 10 minutes
+//# sourceMappingURL=constants.js.map

package/dist/src/ide/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../src/ide/constants.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,MAAM,CAAC,MAAM,mCAAmC,GAAG,sBAAsB,CAAC;AAC1E,MAAM,CAAC,MAAM,kBAAkB,GAAG,EAAE,CAAC;AACrC,MAAM,CAAC,MAAM,4BAA4B,GAAG,KAAK,CAAC,CAAC,eAAe;AAClE,MAAM,CAAC,MAAM,sBAAsB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,aAAa"}

package/dist/src/ide/detect-ide.d.ts

@@ -0,0 +1,53 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare const IDE_DEFINITIONS: {
+    readonly devin: {
+        readonly name: "devin";
+        readonly displayName: "Devin";
+    };
+    readonly replit: {
+        readonly name: "replit";
+        readonly displayName: "Replit";
+    };
+    readonly cursor: {
+        readonly name: "cursor";
+        readonly displayName: "Cursor";
+    };
+    readonly cloudshell: {
+        readonly name: "cloudshell";
+        readonly displayName: "Cloud Shell";
+    };
+    readonly codespaces: {
+        readonly name: "codespaces";
+        readonly displayName: "GitHub Codespaces";
+    };
+    readonly firebasestudio: {
+        readonly name: "firebasestudio";
+        readonly displayName: "Firebase Studio";
+    };
+    readonly trae: {
+        readonly name: "trae";
+        readonly displayName: "Trae";
+    };
+    readonly vscode: {
+        readonly name: "vscode";
+        readonly displayName: "VS Code";
+    };
+    readonly vscodefork: {
+        readonly name: "vscodefork";
+        readonly displayName: "IDE";
+    };
+};
+export type IdeName = keyof typeof IDE_DEFINITIONS;
+export interface IdeInfo {
+    name: IdeName;
+    displayName: string;
+}
+export declare function detectIdeFromEnv(): IdeInfo;
+export declare function detectIde(ideProcessInfo: {
+    pid: number;
+    command: string;
+}): IdeInfo | undefined;

package/dist/src/ide/detect-ide.js

@@ -0,0 +1,58 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export const IDE_DEFINITIONS = {
+    devin: { name: 'devin', displayName: 'Devin' },
+    replit: { name: 'replit', displayName: 'Replit' },
+    cursor: { name: 'cursor', displayName: 'Cursor' },
+    cloudshell: { name: 'cloudshell', displayName: 'Cloud Shell' },
+    codespaces: { name: 'codespaces', displayName: 'GitHub Codespaces' },
+    firebasestudio: { name: 'firebasestudio', displayName: 'Firebase Studio' },
+    trae: { name: 'trae', displayName: 'Trae' },
+    vscode: { name: 'vscode', displayName: 'VS Code' },
+    vscodefork: { name: 'vscodefork', displayName: 'IDE' },
+};
+export function detectIdeFromEnv() {
+    if (process.env['__COG_BASHRC_SOURCED']) {
+        return IDE_DEFINITIONS.devin;
+    }
+    if (process.env['REPLIT_USER']) {
+        return IDE_DEFINITIONS.replit;
+    }
+    if (process.env['CURSOR_TRACE_ID']) {
+        return IDE_DEFINITIONS.cursor;
+    }
+    if (process.env['CODESPACES']) {
+        return IDE_DEFINITIONS.codespaces;
+    }
+    if (process.env['EDITOR_IN_CLOUD_SHELL'] || process.env['CLOUD_SHELL']) {
+        return IDE_DEFINITIONS.cloudshell;
+    }
+    if (process.env['TERM_PRODUCT'] === 'Trae') {
+        return IDE_DEFINITIONS.trae;
+    }
+    if (process.env['MONOSPACE_ENV']) {
+        return IDE_DEFINITIONS.firebasestudio;
+    }
+    return IDE_DEFINITIONS.vscode;
+}
+function verifyVSCode(ide, ideProcessInfo) {
+    if (ide.name !== IDE_DEFINITIONS.vscode.name) {
+        return ide;
+    }
+    if (ideProcessInfo.command.toLowerCase().includes('code')) {
+        return IDE_DEFINITIONS.vscode;
+    }
+    return IDE_DEFINITIONS.vscodefork;
+}
+export function detectIde(ideProcessInfo) {
+    // Only VSCode-based integrations are currently supported.
+    if (process.env['TERM_PROGRAM'] !== 'vscode') {
+        return undefined;
+    }
+    const ide = detectIdeFromEnv();
+    return verifyVSCode(ide, ideProcessInfo);
+}
+//# sourceMappingURL=detect-ide.js.map

package/dist/src/ide/detect-ide.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"detect-ide.js","sourceRoot":"","sources":["../../../src/ide/detect-ide.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE;IAC9C,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE;IACjD,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE;IACjD,UAAU,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,aAAa,EAAE;IAC9D,UAAU,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,mBAAmB,EAAE;IACpE,cAAc,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,WAAW,EAAE,iBAAiB,EAAE;IAC1E,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE;IAC3C,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE;IAClD,UAAU,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,KAAK,EAAE;CAC9C,CAAC;AASX,MAAM,UAAU,gBAAgB;IAC9B,IAAI,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,EAAE,CAAC;QACxC,OAAO,eAAe,CAAC,KAAK,CAAC;IAC/B,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;QAC/B,OAAO,eAAe,CAAC,MAAM,CAAC;IAChC,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACnC,OAAO,eAAe,CAAC,MAAM,CAAC;IAChC,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;QAC9B,OAAO,eAAe,CAAC,UAAU,CAAC;IACpC,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;QACvE,OAAO,eAAe,CAAC,UAAU,CAAC;IACpC,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,KAAK,MAAM,EAAE,CAAC;QAC3C,OAAO,eAAe,CAAC,IAAI,CAAC;IAC9B,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,EAAE,CAAC;QACjC,OAAO,eAAe,CAAC,cAAc,CAAC;IACxC,CAAC;IACD,OAAO,eAAe,CAAC,MAAM,CAAC;AAChC,CAAC;AAED,SAAS,YAAY,CACnB,GAAY,EACZ,cAGC;IAED,IAAI,GAAG,CAAC,IAAI,KAAK,eAAe,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QAC7C,OAAO,GAAG,CAAC;IACb,CAAC;IACD,IAAI,cAAc,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1D,OAAO,eAAe,CAAC,MAAM,CAAC;IAChC,CAAC;IACD,OAAO,eAAe,CAAC,UAAU,CAAC;AACpC,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,cAGzB;IACC,0DAA0D;IAC1D,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,KAAK,QAAQ,EAAE,CAAC;QAC7C,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,GAAG,GAAG,gBAAgB,EAAE,CAAC;IAC/B,OAAO,YAAY,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;AAC3C,CAAC"}

package/dist/src/ide/detect-ide.test.d.ts

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

package/dist/src/ide/detect-ide.test.js

@@ -0,0 +1,69 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, afterEach } from 'vitest';
+import { detectIde, IDE_DEFINITIONS } from './detect-ide.js';
+describe('detectIde', () => {
+    const ideProcessInfo = { pid: 123, command: 'some/path/to/code' };
+    const ideProcessInfoNoCode = { pid: 123, command: 'some/path/to/fork' };
+    afterEach(() => {
+        vi.unstubAllEnvs();
+    });
+    it('should return undefined if TERM_PROGRAM is not vscode', () => {
+        vi.stubEnv('TERM_PROGRAM', '');
+        expect(detectIde(ideProcessInfo)).toBeUndefined();
+    });
+    it('should detect Devin', () => {
+        vi.stubEnv('TERM_PROGRAM', 'vscode');
+        vi.stubEnv('__COG_BASHRC_SOURCED', '1');
+        expect(detectIde(ideProcessInfo)).toBe(IDE_DEFINITIONS.devin);
+    });
+    it('should detect Replit', () => {
+        vi.stubEnv('TERM_PROGRAM', 'vscode');
+        vi.stubEnv('REPLIT_USER', 'testuser');
+        expect(detectIde(ideProcessInfo)).toBe(IDE_DEFINITIONS.replit);
+    });
+    it('should detect Cursor', () => {
+        vi.stubEnv('TERM_PROGRAM', 'vscode');
+        vi.stubEnv('CURSOR_TRACE_ID', 'some-id');
+        expect(detectIde(ideProcessInfo)).toBe(IDE_DEFINITIONS.cursor);
+    });
+    it('should detect Codespaces', () => {
+        vi.stubEnv('TERM_PROGRAM', 'vscode');
+        vi.stubEnv('CODESPACES', 'true');
+        expect(detectIde(ideProcessInfo)).toBe(IDE_DEFINITIONS.codespaces);
+    });
+    it('should detect Cloud Shell via EDITOR_IN_CLOUD_SHELL', () => {
+        vi.stubEnv('TERM_PROGRAM', 'vscode');
+        vi.stubEnv('EDITOR_IN_CLOUD_SHELL', 'true');
+        expect(detectIde(ideProcessInfo)).toBe(IDE_DEFINITIONS.cloudshell);
+    });
+    it('should detect Cloud Shell via CLOUD_SHELL', () => {
+        vi.stubEnv('TERM_PROGRAM', 'vscode');
+        vi.stubEnv('CLOUD_SHELL', 'true');
+        expect(detectIde(ideProcessInfo)).toBe(IDE_DEFINITIONS.cloudshell);
+    });
+    it('should detect Trae', () => {
+        vi.stubEnv('TERM_PROGRAM', 'vscode');
+        vi.stubEnv('TERM_PRODUCT', 'Trae');
+        expect(detectIde(ideProcessInfo)).toBe(IDE_DEFINITIONS.trae);
+    });
+    it('should detect Firebase Studio via MONOSPACE_ENV', () => {
+        vi.stubEnv('TERM_PROGRAM', 'vscode');
+        vi.stubEnv('MONOSPACE_ENV', 'true');
+        expect(detectIde(ideProcessInfo)).toBe(IDE_DEFINITIONS.firebasestudio);
+    });
+    it('should detect VSCode when no other IDE is detected and command includes "code"', () => {
+        vi.stubEnv('TERM_PROGRAM', 'vscode');
+        vi.stubEnv('MONOSPACE_ENV', '');
+        expect(detectIde(ideProcessInfo)).toBe(IDE_DEFINITIONS.vscode);
+    });
+    it('should detect VSCodeFork when no other IDE is detected and command does not include "code"', () => {
+        vi.stubEnv('TERM_PROGRAM', 'vscode');
+        vi.stubEnv('MONOSPACE_ENV', '');
+        expect(detectIde(ideProcessInfoNoCode)).toBe(IDE_DEFINITIONS.vscodefork);
+    });
+});
+//# sourceMappingURL=detect-ide.test.js.map

package/dist/src/ide/detect-ide.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"detect-ide.test.js","sourceRoot":"","sources":["../../../src/ide/detect-ide.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAC7D,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAE7D,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,MAAM,cAAc,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,mBAAmB,EAAE,CAAC;IAClE,MAAM,oBAAoB,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,mBAAmB,EAAE,CAAC;IAExE,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;QAC/B,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qBAAqB,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QACrC,EAAE,CAAC,OAAO,CAAC,sBAAsB,EAAE,GAAG,CAAC,CAAC;QACxC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QACrC,EAAE,CAAC,OAAO,CAAC,aAAa,EAAE,UAAU,CAAC,CAAC;QACtC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QACrC,EAAE,CAAC,OAAO,CAAC,iBAAiB,EAAE,SAAS,CAAC,CAAC;QACzC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QACrC,EAAE,CAAC,OAAO,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QACjC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QACrC,EAAE,CAAC,OAAO,CAAC,uBAAuB,EAAE,MAAM,CAAC,CAAC;QAC5C,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QACrC,EAAE,CAAC,OAAO,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;QAClC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QACrC,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;QACnC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QACrC,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;QACpC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;QACxF,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QACrC,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;QAChC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4FAA4F,EAAE,GAAG,EAAE;QACpG,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QACrC,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;QAChC,MAAM,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/ide/ide-client.d.ts

@@ -0,0 +1,109 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type IdeInfo } from '../ide/detect-ide.js';
+export type DiffUpdateResult = {
+    status: 'accepted';
+    content?: string;
+} | {
+    status: 'rejected';
+    content: undefined;
+};
+export type IDEConnectionState = {
+    status: IDEConnectionStatus;
+    details?: string;
+};
+export declare enum IDEConnectionStatus {
+    Connected = "connected",
+    Disconnected = "disconnected",
+    Connecting = "connecting"
+}
+/**
+ * Manages the connection to and interaction with the IDE server.
+ */
+export declare class IdeClient {
+    private static instancePromise;
+    private client;
+    private state;
+    private currentIde;
+    private ideProcessInfo;
+    private authToken;
+    private diffResponses;
+    private statusListeners;
+    private trustChangeListeners;
+    private availableTools;
+    /**
+     * A mutex to ensure that only one diff view is open in the IDE at a time.
+     * This prevents race conditions and UI issues in IDEs like VSCode that
+     * can't handle multiple diff views being opened simultaneously.
+     */
+    private diffMutex;
+    private constructor();
+    static getInstance(): Promise<IdeClient>;
+    addStatusChangeListener(listener: (state: IDEConnectionState) => void): void;
+    removeStatusChangeListener(listener: (state: IDEConnectionState) => void): void;
+    addTrustChangeListener(listener: (isTrusted: boolean) => void): void;
+    removeTrustChangeListener(listener: (isTrusted: boolean) => void): void;
+    connect(): Promise<void>;
+    /**
+     * Opens a diff view in the IDE, allowing the user to review and accept or
+     * reject changes.
+     *
+     * This method sends a request to the IDE to display a diff between the
+     * current content of a file and the new content provided. It then waits for
+     * a notification from the IDE indicating that the user has either accepted
+     * (potentially with manual edits) or rejected the diff.
+     *
+     * A mutex ensures that only one diff view can be open at a time to prevent
+     * race conditions.
+     *
+     * @param filePath The absolute path to the file to be diffed.
+     * @param newContent The proposed new content for the file.
+     * @returns A promise that resolves with a `DiffUpdateResult`, indicating
+     *   whether the diff was 'accepted' or 'rejected' and including the final
+     *   content if accepted.
+     */
+    openDiff(filePath: string, newContent: string): Promise<DiffUpdateResult>;
+    /**
+     * Acquires a lock to ensure sequential execution of critical sections.
+     *
+     * This method implements a promise-based mutex. It works by chaining promises.
+     * Each call to `acquireMutex` gets the current `diffMutex` promise. It then
+     * creates a *new* promise (`newMutex`) that will be resolved when the caller
+     * invokes the returned `release` function. The `diffMutex` is immediately
+     * updated to this `newMutex`.
+     *
+     * The method returns a promise that resolves with the `release` function only
+     * *after* the *previous* `diffMutex` promise has resolved. This creates a
+     * queue where each subsequent operation must wait for the previous one to release
+     * the lock.
+     *
+     * @returns A promise that resolves to a function that must be called to
+     *   release the lock.
+     */
+    private acquireMutex;
+    closeDiff(filePath: string, options?: {
+        suppressNotification?: boolean;
+    }): Promise<string | undefined>;
+    resolveDiffFromCli(filePath: string, outcome: 'accepted' | 'rejected'): Promise<void>;
+    disconnect(): Promise<void>;
+    getCurrentIde(): IdeInfo | undefined;
+    getConnectionStatus(): IDEConnectionState;
+    getDetectedIdeDisplayName(): string | undefined;
+    isDiffingEnabled(): boolean;
+    private discoverTools;
+    private setState;
+    static validateWorkspacePath(ideWorkspacePath: string | undefined, currentIdeDisplayName: string | undefined, cwd: string): {
+        isValid: boolean;
+        error?: string;
+    };
+    private getPortFromEnv;
+    private getStdioConfigFromEnv;
+    private getConnectionConfigFromFile;
+    private createProxyAwareFetch;
+    private registerClientHandlers;
+    private establishHttpConnection;
+    private establishStdioConnection;
+}