@terminai/core 0.21.0

package/dist/src/code_assist/experiments/experiments.test.d.ts

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

package/dist/src/code_assist/experiments/experiments.test.js

@@ -0,0 +1,93 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { getClientMetadata } from './client_metadata.js';
+// Mock dependencies before importing the module under test
+vi.mock('../server.js');
+vi.mock('./client_metadata.js');
+describe('experiments', () => {
+    let mockServer;
+    beforeEach(() => {
+        // Reset modules to clear the cached `experimentsPromise`
+        vi.resetModules();
+        // Mock the dependencies that `getExperiments` relies on
+        vi.mocked(getClientMetadata).mockResolvedValue({
+            ideName: 'GEMINI_CLI',
+            ideVersion: '1.0.0',
+            platform: 'LINUX_AMD64',
+            updateChannel: 'stable',
+        });
+        // Create a mock instance of the server for each test
+        mockServer = {
+            listExperiments: vi.fn(),
+        };
+    });
+    afterEach(() => {
+        vi.clearAllMocks();
+    });
+    it('should fetch and parse experiments from the server', async () => {
+        const { getExperiments } = await import('./experiments.js');
+        const mockApiResponse = {
+            flags: [
+                { flagId: 234, boolValue: true },
+                { flagId: 345, stringValue: 'value' },
+            ],
+            experimentIds: [123, 456],
+        };
+        vi.mocked(mockServer.listExperiments).mockResolvedValue(mockApiResponse);
+        const experiments = await getExperiments(mockServer);
+        // Verify that the dependencies were called
+        expect(getClientMetadata).toHaveBeenCalled();
+        expect(mockServer.listExperiments).toHaveBeenCalledWith(await getClientMetadata());
+        // Verify that the response was parsed correctly
+        expect(experiments.flags[234]).toEqual({
+            flagId: 234,
+            boolValue: true,
+        });
+        expect(experiments.flags[345]).toEqual({
+            flagId: 345,
+            stringValue: 'value',
+        });
+        expect(experiments.experimentIds).toEqual([123, 456]);
+    });
+    it('should handle an empty or partial response from the server', async () => {
+        const { getExperiments } = await import('./experiments.js');
+        const mockApiResponse = {}; // No flags or experimentIds
+        vi.mocked(mockServer.listExperiments).mockResolvedValue(mockApiResponse);
+        const experiments = await getExperiments(mockServer);
+        expect(experiments.flags).toEqual({});
+        expect(experiments.experimentIds).toEqual([]);
+    });
+    it('should ignore flags that are missing a name', async () => {
+        const { getExperiments } = await import('./experiments.js');
+        const mockApiResponse = {
+            flags: [
+                { boolValue: true }, // No name
+                { flagId: 256, stringValue: 'value' },
+            ],
+        };
+        vi.mocked(mockServer.listExperiments).mockResolvedValue(mockApiResponse);
+        const experiments = await getExperiments(mockServer);
+        expect(Object.keys(experiments.flags)).toHaveLength(1);
+        expect(experiments.flags[256]).toBeDefined();
+        expect(experiments.flags['undefined']).toBeUndefined();
+    });
+    it('should cache the experiments promise to avoid multiple fetches', async () => {
+        const { getExperiments } = await import('./experiments.js');
+        const mockApiResponse = {
+            experimentIds: [1, 2, 3],
+        };
+        vi.mocked(mockServer.listExperiments).mockResolvedValue(mockApiResponse);
+        const firstCall = await getExperiments(mockServer);
+        const secondCall = await getExperiments(mockServer);
+        expect(firstCall).toBe(secondCall); // Should be the exact same promise object
+        // Verify the underlying functions were only called once
+        expect(getClientMetadata).toHaveBeenCalledTimes(1);
+        expect(mockServer.listExperiments).toHaveBeenCalledTimes(1);
+    });
+});
+//# sourceMappingURL=experiments.test.js.map

package/dist/src/code_assist/experiments/experiments.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"experiments.test.js","sourceRoot":"","sources":["../../../../src/code_assist/experiments/experiments.test.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAEzE,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAGzD,2DAA2D;AAC3D,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;AACxB,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;AAEhC,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,IAAI,UAA4B,CAAC;IAEjC,UAAU,CAAC,GAAG,EAAE;QACd,yDAAyD;QACzD,EAAE,CAAC,YAAY,EAAE,CAAC;QAElB,wDAAwD;QACxD,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,iBAAiB,CAAC;YAC7C,OAAO,EAAE,YAAY;YACrB,UAAU,EAAE,OAAO;YACnB,QAAQ,EAAE,aAAa;YACvB,aAAa,EAAE,QAAQ;SACxB,CAAC,CAAC;QAEH,qDAAqD;QACrD,UAAU,GAAG;YACX,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;SACM,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAC5D,MAAM,eAAe,GAA4B;YAC/C,KAAK,EAAE;gBACL,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE;gBAChC,EAAE,MAAM,EAAE,GAAG,EAAE,WAAW,EAAE,OAAO,EAAE;aACtC;YACD,aAAa,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC;SAC1B,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QAEzE,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,CAAC;QAErD,2CAA2C;QAC3C,MAAM,CAAC,iBAAiB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC7C,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,oBAAoB,CACrD,MAAM,iBAAiB,EAAE,CAC1B,CAAC;QAEF,gDAAgD;QAChD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC;YACrC,MAAM,EAAE,GAAG;YACX,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC;YACrC,MAAM,EAAE,GAAG;YACX,WAAW,EAAE,OAAO;SACrB,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAC5D,MAAM,eAAe,GAA4B,EAAE,CAAC,CAAC,4BAA4B;QACjF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QAEzE,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,CAAC;QAErD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACtC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAC5D,MAAM,eAAe,GAA4B;YAC/C,KAAK,EAAE;gBACL,EAAE,SAAS,EAAE,IAAI,EAAU,EAAE,UAAU;gBACvC,EAAE,MAAM,EAAE,GAAG,EAAE,WAAW,EAAE,OAAO,EAAE;aACtC;SACF,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QAEzE,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,CAAC;QAErD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACvD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7C,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC9E,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,CAAC;QAC5D,MAAM,eAAe,GAA4B;YAC/C,aAAa,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;SACzB,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QAEzE,MAAM,SAAS,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,CAAC;QACnD,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,CAAC;QAEpD,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,0CAA0C;QAC9E,wDAAwD;QACxD,MAAM,CAAC,iBAAiB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/code_assist/experiments/flagNames.d.ts

@@ -0,0 +1,14 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare const ExperimentFlags: {
+    readonly CONTEXT_COMPRESSION_THRESHOLD: 45740197;
+    readonly USER_CACHING: 45740198;
+    readonly BANNER_TEXT_NO_CAPACITY_ISSUES: 45740199;
+    readonly BANNER_TEXT_CAPACITY_ISSUES: 45740200;
+    readonly ENABLE_PREVIEW: 45740196;
+};
+export type ExperimentFlagName = (typeof ExperimentFlags)[keyof typeof ExperimentFlags];

package/dist/src/code_assist/experiments/flagNames.js

@@ -0,0 +1,14 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export const ExperimentFlags = {
+    CONTEXT_COMPRESSION_THRESHOLD: 45740197,
+    USER_CACHING: 45740198,
+    BANNER_TEXT_NO_CAPACITY_ISSUES: 45740199,
+    BANNER_TEXT_CAPACITY_ISSUES: 45740200,
+    ENABLE_PREVIEW: 45740196,
+};
+//# sourceMappingURL=flagNames.js.map

package/dist/src/code_assist/experiments/flagNames.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"flagNames.js","sourceRoot":"","sources":["../../../../src/code_assist/experiments/flagNames.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,6BAA6B,EAAE,QAAQ;IACvC,YAAY,EAAE,QAAQ;IACtB,8BAA8B,EAAE,QAAQ;IACxC,2BAA2B,EAAE,QAAQ;IACrC,cAAc,EAAE,QAAQ;CAChB,CAAC"}

package/dist/src/code_assist/experiments/types.d.ts

@@ -0,0 +1,36 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { ClientMetadata } from '../types.js';
+export interface ListExperimentsRequest {
+    project: string;
+    metadata?: ClientMetadata;
+}
+export interface ListExperimentsResponse {
+    experimentIds?: number[];
+    flags?: Flag[];
+    filteredFlags?: FilteredFlag[];
+    debugString?: string;
+}
+export interface Flag {
+    flagId?: number;
+    boolValue?: boolean;
+    floatValue?: number;
+    intValue?: string;
+    stringValue?: string;
+    int32ListValue?: Int32List;
+    stringListValue?: StringList;
+}
+export interface Int32List {
+    values?: number[];
+}
+export interface StringList {
+    values?: string[];
+}
+export interface FilteredFlag {
+    name?: string;
+    reason?: string;
+}

package/dist/src/code_assist/experiments/types.js

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

package/dist/src/code_assist/experiments/types.js.map

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

package/dist/src/code_assist/oauth-credential-storage.d.ts

@@ -0,0 +1,26 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type Credentials } from 'google-auth-library';
+export declare class OAuthCredentialStorage {
+    private static storage;
+    /**
+     * Load cached OAuth credentials
+     */
+    static loadCredentials(): Promise<Credentials | null>;
+    /**
+     * Save OAuth credentials
+     */
+    static saveCredentials(credentials: Credentials): Promise<void>;
+    /**
+     * Clear cached OAuth credentials
+     */
+    static clearCredentials(): Promise<void>;
+    /**
+     * Migrate credentials from old file-based storage to keychain
+     */
+    private static migrateFromFileStorage;
+}

package/dist/src/code_assist/oauth-credential-storage.js

@@ -0,0 +1,111 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import {} from 'google-auth-library';
+import { HybridTokenStorage } from '../mcp/token-storage/hybrid-token-storage.js';
+import { OAUTH_FILE } from '../config/storage.js';
+import * as path from 'node:path';
+import * as os from 'node:os';
+import { promises as fs } from 'node:fs';
+import { GEMINI_DIR } from '../utils/paths.js';
+import { coreEvents } from '../utils/events.js';
+const KEYCHAIN_SERVICE_NAME = 'gemini-cli-oauth';
+const MAIN_ACCOUNT_KEY = 'main-account';
+export class OAuthCredentialStorage {
+    static storage = new HybridTokenStorage(KEYCHAIN_SERVICE_NAME);
+    /**
+     * Load cached OAuth credentials
+     */
+    static async loadCredentials() {
+        try {
+            const credentials = await this.storage.getCredentials(MAIN_ACCOUNT_KEY);
+            if (credentials?.token) {
+                const { accessToken, refreshToken, expiresAt, tokenType, scope } = credentials.token;
+                // Convert from OAuthCredentials format to Google Credentials format
+                const googleCreds = {
+                    access_token: accessToken,
+                    refresh_token: refreshToken || undefined,
+                    token_type: tokenType || undefined,
+                    scope: scope || undefined,
+                };
+                if (expiresAt) {
+                    googleCreds.expiry_date = expiresAt;
+                }
+                return googleCreds;
+            }
+            // Fallback: Try to migrate from old file-based storage
+            return await this.migrateFromFileStorage();
+        }
+        catch (error) {
+            coreEvents.emitFeedback('error', 'Failed to load OAuth credentials', error);
+            throw new Error('Failed to load OAuth credentials', { cause: error });
+        }
+    }
+    /**
+     * Save OAuth credentials
+     */
+    static async saveCredentials(credentials) {
+        if (!credentials.access_token) {
+            throw new Error('Attempted to save credentials without an access token.');
+        }
+        // Convert Google Credentials to OAuthCredentials format
+        const mcpCredentials = {
+            serverName: MAIN_ACCOUNT_KEY,
+            token: {
+                accessToken: credentials.access_token,
+                refreshToken: credentials.refresh_token || undefined,
+                tokenType: credentials.token_type || 'Bearer',
+                scope: credentials.scope || undefined,
+                expiresAt: credentials.expiry_date || undefined,
+            },
+            updatedAt: Date.now(),
+        };
+        await this.storage.setCredentials(mcpCredentials);
+    }
+    /**
+     * Clear cached OAuth credentials
+     */
+    static async clearCredentials() {
+        try {
+            await this.storage.deleteCredentials(MAIN_ACCOUNT_KEY);
+            // Also try to remove the old file if it exists
+            const oldFilePath = path.join(os.homedir(), GEMINI_DIR, OAUTH_FILE);
+            await fs.rm(oldFilePath, { force: true }).catch(() => { });
+        }
+        catch (error) {
+            coreEvents.emitFeedback('error', 'Failed to clear OAuth credentials', error);
+            throw new Error('Failed to clear OAuth credentials', { cause: error });
+        }
+    }
+    /**
+     * Migrate credentials from old file-based storage to keychain
+     */
+    static async migrateFromFileStorage() {
+        const oldFilePath = path.join(os.homedir(), GEMINI_DIR, OAUTH_FILE);
+        let credsJson;
+        try {
+            credsJson = await fs.readFile(oldFilePath, 'utf-8');
+        }
+        catch (error) {
+            if (typeof error === 'object' &&
+                error !== null &&
+                'code' in error &&
+                error.code === 'ENOENT') {
+                // File doesn't exist, so no migration.
+                return null;
+            }
+            // Other read errors should propagate.
+            throw error;
+        }
+        const credentials = JSON.parse(credsJson);
+        // Save to new storage
+        await this.saveCredentials(credentials);
+        // Remove old file after successful migration
+        await fs.rm(oldFilePath, { force: true }).catch(() => { });
+        return credentials;
+    }
+}
+//# sourceMappingURL=oauth-credential-storage.js.map

package/dist/src/code_assist/oauth-credential-storage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"oauth-credential-storage.js","sourceRoot":"","sources":["../../../src/code_assist/oauth-credential-storage.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAoB,MAAM,qBAAqB,CAAC;AACvD,OAAO,EAAE,kBAAkB,EAAE,MAAM,8CAA8C,CAAC;AAClF,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAElD,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC/C,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAEhD,MAAM,qBAAqB,GAAG,kBAAkB,CAAC;AACjD,MAAM,gBAAgB,GAAG,cAAc,CAAC;AAExC,MAAM,OAAO,sBAAsB;IACzB,MAAM,CAAC,OAAO,GAAuB,IAAI,kBAAkB,CACjE,qBAAqB,CACtB,CAAC;IAEF;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,eAAe;QAC1B,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;YAExE,IAAI,WAAW,EAAE,KAAK,EAAE,CAAC;gBACvB,MAAM,EAAE,WAAW,EAAE,YAAY,EAAE,SAAS,EAAE,SAAS,EAAE,KAAK,EAAE,GAC9D,WAAW,CAAC,KAAK,CAAC;gBACpB,oEAAoE;gBACpE,MAAM,WAAW,GAAgB;oBAC/B,YAAY,EAAE,WAAW;oBACzB,aAAa,EAAE,YAAY,IAAI,SAAS;oBACxC,UAAU,EAAE,SAAS,IAAI,SAAS;oBAClC,KAAK,EAAE,KAAK,IAAI,SAAS;iBAC1B,CAAC;gBAEF,IAAI,SAAS,EAAE,CAAC;oBACd,WAAW,CAAC,WAAW,GAAG,SAAS,CAAC;gBACtC,CAAC;gBAED,OAAO,WAAW,CAAC;YACrB,CAAC;YAED,uDAAuD;YACvD,OAAO,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAC7C,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,UAAU,CAAC,YAAY,CACrB,OAAO,EACP,kCAAkC,EAClC,KAAK,CACN,CAAC;YACF,MAAM,IAAI,KAAK,CAAC,kCAAkC,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QACxE,CAAC;IACH,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,WAAwB;QACnD,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;QAC5E,CAAC;QAED,wDAAwD;QACxD,MAAM,cAAc,GAAqB;YACvC,UAAU,EAAE,gBAAgB;YAC5B,KAAK,EAAE;gBACL,WAAW,EAAE,WAAW,CAAC,YAAY;gBACrC,YAAY,EAAE,WAAW,CAAC,aAAa,IAAI,SAAS;gBACpD,SAAS,EAAE,WAAW,CAAC,UAAU,IAAI,QAAQ;gBAC7C,KAAK,EAAE,WAAW,CAAC,KAAK,IAAI,SAAS;gBACrC,SAAS,EAAE,WAAW,CAAC,WAAW,IAAI,SAAS;aAChD;YACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;QAEF,MAAM,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;IACpD,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,gBAAgB;QAC3B,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;YAEvD,+CAA+C;YAC/C,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;YACpE,MAAM,EAAE,CAAC,EAAE,CAAC,WAAW,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAC5D,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,UAAU,CAAC,YAAY,CACrB,OAAO,EACP,mCAAmC,EACnC,KAAK,CACN,CAAC;YACF,MAAM,IAAI,KAAK,CAAC,mCAAmC,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QACzE,CAAC;IACH,CAAC;IAED;;OAEG;IACK,MAAM,CAAC,KAAK,CAAC,sBAAsB;QACzC,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;QAEpE,IAAI,SAAiB,CAAC;QACtB,IAAI,CAAC;YACH,SAAS,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QACtD,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,IACE,OAAO,KAAK,KAAK,QAAQ;gBACzB,KAAK,KAAK,IAAI;gBACd,MAAM,IAAI,KAAK;gBACf,KAAK,CAAC,IAAI,KAAK,QAAQ,EACvB,CAAC;gBACD,uCAAuC;gBACvC,OAAO,IAAI,CAAC;YACd,CAAC;YACD,sCAAsC;YACtC,MAAM,KAAK,CAAC;QACd,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAgB,CAAC;QAEzD,sBAAsB;QACtB,MAAM,IAAI,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;QAExC,6CAA6C;QAC7C,MAAM,EAAE,CAAC,EAAE,CAAC,WAAW,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAE1D,OAAO,WAAW,CAAC;IACrB,CAAC"}

package/dist/src/code_assist/oauth-credential-storage.test.d.ts

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

package/dist/src/code_assist/oauth-credential-storage.test.js

@@ -0,0 +1,199 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import {} from 'google-auth-library';
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { OAuthCredentialStorage } from './oauth-credential-storage.js';
+import { coreEvents } from '../utils/events.js';
+import * as path from 'node:path';
+import * as os from 'node:os';
+import { promises as fs } from 'node:fs';
+// Mock external dependencies
+const mockHybridTokenStorage = vi.hoisted(() => ({
+    getCredentials: vi.fn(),
+    setCredentials: vi.fn(),
+    deleteCredentials: vi.fn(),
+}));
+vi.mock('../mcp/token-storage/hybrid-token-storage.js', () => ({
+    HybridTokenStorage: vi.fn(() => mockHybridTokenStorage),
+}));
+vi.mock('node:fs', () => ({
+    promises: {
+        readFile: vi.fn(),
+        rm: vi.fn(),
+    },
+}));
+vi.mock('node:os');
+vi.mock('node:path');
+vi.mock('../utils/events.js', () => ({
+    coreEvents: {
+        emitFeedback: vi.fn(),
+        emitConsoleLog: vi.fn(),
+    },
+}));
+describe('OAuthCredentialStorage', () => {
+    const mockCredentials = {
+        access_token: 'mock_access_token',
+        refresh_token: 'mock_refresh_token',
+        expiry_date: Date.now() + 3600 * 1000,
+        token_type: 'Bearer',
+        scope: 'email profile',
+    };
+    const mockMcpCredentials = {
+        serverName: 'main-account',
+        token: {
+            accessToken: 'mock_access_token',
+            refreshToken: 'mock_refresh_token',
+            tokenType: 'Bearer',
+            scope: 'email profile',
+            expiresAt: mockCredentials.expiry_date,
+        },
+        updatedAt: expect.any(Number),
+    };
+    const oldFilePath = '/mock/home/.gemini/oauth.json';
+    beforeEach(() => {
+        vi.spyOn(mockHybridTokenStorage, 'getCredentials').mockResolvedValue(null);
+        vi.spyOn(mockHybridTokenStorage, 'setCredentials').mockResolvedValue(undefined);
+        vi.spyOn(mockHybridTokenStorage, 'deleteCredentials').mockResolvedValue(undefined);
+        vi.spyOn(fs, 'readFile').mockRejectedValue(new Error('File not found'));
+        vi.spyOn(fs, 'rm').mockResolvedValue(undefined);
+        vi.spyOn(os, 'homedir').mockReturnValue('/mock/home');
+        vi.spyOn(path, 'join').mockReturnValue(oldFilePath);
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    describe('loadCredentials', () => {
+        it('should load credentials from HybridTokenStorage if available', async () => {
+            vi.spyOn(mockHybridTokenStorage, 'getCredentials').mockResolvedValue(mockMcpCredentials);
+            const result = await OAuthCredentialStorage.loadCredentials();
+            expect(mockHybridTokenStorage.getCredentials).toHaveBeenCalledWith('main-account');
+            expect(result).toEqual(mockCredentials);
+        });
+        it('should fallback to migrateFromFileStorage if no credentials in HybridTokenStorage', async () => {
+            vi.spyOn(mockHybridTokenStorage, 'getCredentials').mockResolvedValue(null);
+            vi.spyOn(fs, 'readFile').mockResolvedValue(JSON.stringify(mockCredentials));
+            const result = await OAuthCredentialStorage.loadCredentials();
+            expect(mockHybridTokenStorage.getCredentials).toHaveBeenCalledWith('main-account');
+            expect(fs.readFile).toHaveBeenCalledWith(oldFilePath, 'utf-8');
+            expect(mockHybridTokenStorage.setCredentials).toHaveBeenCalled(); // Verify credentials were saved
+            expect(fs.rm).toHaveBeenCalledWith(oldFilePath, { force: true }); // Verify old file was removed
+            expect(result).toEqual(mockCredentials);
+        });
+        it('should return null if no credentials found and no old file to migrate', async () => {
+            vi.spyOn(fs, 'readFile').mockRejectedValue({
+                message: 'File not found',
+                code: 'ENOENT',
+            });
+            const result = await OAuthCredentialStorage.loadCredentials();
+            expect(result).toBeNull();
+        });
+        it('should throw an error if loading fails', async () => {
+            const mockError = new Error('HybridTokenStorage error');
+            vi.spyOn(mockHybridTokenStorage, 'getCredentials').mockRejectedValue(mockError);
+            await expect(OAuthCredentialStorage.loadCredentials()).rejects.toThrow('Failed to load OAuth credentials');
+            expect(coreEvents.emitFeedback).toHaveBeenCalledWith('error', 'Failed to load OAuth credentials', mockError);
+        });
+        it('should throw an error if read file fails', async () => {
+            const mockError = new Error('Permission denied');
+            vi.spyOn(mockHybridTokenStorage, 'getCredentials').mockResolvedValue(null);
+            vi.spyOn(fs, 'readFile').mockRejectedValue(mockError);
+            await expect(OAuthCredentialStorage.loadCredentials()).rejects.toThrow('Failed to load OAuth credentials');
+            expect(coreEvents.emitFeedback).toHaveBeenCalledWith('error', 'Failed to load OAuth credentials', mockError);
+        });
+        it('should not throw error if migration file removal failed', async () => {
+            vi.spyOn(mockHybridTokenStorage, 'getCredentials').mockResolvedValue(null);
+            vi.spyOn(fs, 'readFile').mockResolvedValue(JSON.stringify(mockCredentials));
+            vi.spyOn(OAuthCredentialStorage, 'saveCredentials').mockResolvedValue(undefined);
+            vi.spyOn(fs, 'rm').mockRejectedValue(new Error('Deletion failed'));
+            const result = await OAuthCredentialStorage.loadCredentials();
+            expect(result).toEqual(mockCredentials);
+        });
+        it('should throw an error if the migration file contains invalid JSON', async () => {
+            vi.spyOn(mockHybridTokenStorage, 'getCredentials').mockResolvedValue(null);
+            vi.spyOn(fs, 'readFile').mockResolvedValue('invalid json');
+            await expect(OAuthCredentialStorage.loadCredentials()).rejects.toThrow('Failed to load OAuth credentials');
+        });
+        it('should not delete the old file if saving migrated credentials fails', async () => {
+            vi.spyOn(mockHybridTokenStorage, 'getCredentials').mockResolvedValue(null);
+            vi.spyOn(fs, 'readFile').mockResolvedValue(JSON.stringify(mockCredentials));
+            vi.spyOn(mockHybridTokenStorage, 'setCredentials').mockRejectedValue(new Error('Save failed'));
+            await expect(OAuthCredentialStorage.loadCredentials()).rejects.toThrow('Failed to load OAuth credentials');
+            expect(fs.rm).not.toHaveBeenCalled();
+        });
+        it('should return credentials even if access_token is missing from storage', async () => {
+            const partialMcpCredentials = {
+                ...mockMcpCredentials,
+                token: {
+                    ...mockMcpCredentials.token,
+                    accessToken: undefined,
+                },
+            };
+            vi.spyOn(mockHybridTokenStorage, 'getCredentials').mockResolvedValue(partialMcpCredentials);
+            const result = await OAuthCredentialStorage.loadCredentials();
+            expect(result).toEqual({
+                access_token: undefined,
+                refresh_token: mockCredentials.refresh_token,
+                token_type: mockCredentials.token_type,
+                scope: mockCredentials.scope,
+                expiry_date: mockCredentials.expiry_date,
+            });
+        });
+    });
+    describe('saveCredentials', () => {
+        it('should save credentials to HybridTokenStorage', async () => {
+            await OAuthCredentialStorage.saveCredentials(mockCredentials);
+            expect(mockHybridTokenStorage.setCredentials).toHaveBeenCalledWith(mockMcpCredentials);
+        });
+        it('should throw an error if access_token is missing', async () => {
+            const invalidCredentials = {
+                ...mockCredentials,
+                access_token: undefined,
+            };
+            await expect(OAuthCredentialStorage.saveCredentials(invalidCredentials)).rejects.toThrow('Attempted to save credentials without an access token.');
+        });
+        it('should handle saving credentials with null or undefined optional fields', async () => {
+            const partialCredentials = {
+                access_token: 'only_access_token',
+                refresh_token: null, // test null
+                scope: undefined, // test undefined
+            };
+            await OAuthCredentialStorage.saveCredentials(partialCredentials);
+            expect(mockHybridTokenStorage.setCredentials).toHaveBeenCalledWith({
+                serverName: 'main-account',
+                token: {
+                    accessToken: 'only_access_token',
+                    refreshToken: undefined,
+                    tokenType: 'Bearer', // default
+                    scope: undefined,
+                    expiresAt: undefined,
+                },
+                updatedAt: expect.any(Number),
+            });
+        });
+    });
+    describe('clearCredentials', () => {
+        it('should delete credentials from HybridTokenStorage', async () => {
+            await OAuthCredentialStorage.clearCredentials();
+            expect(mockHybridTokenStorage.deleteCredentials).toHaveBeenCalledWith('main-account');
+        });
+        it('should attempt to remove the old file-based storage', async () => {
+            await OAuthCredentialStorage.clearCredentials();
+            expect(fs.rm).toHaveBeenCalledWith(oldFilePath, { force: true });
+        });
+        it('should not throw an error if deleting old file fails', async () => {
+            vi.spyOn(fs, 'rm').mockRejectedValue(new Error('File deletion failed'));
+            await expect(OAuthCredentialStorage.clearCredentials()).resolves.toBeUndefined();
+        });
+        it('should throw an error if clearing from HybridTokenStorage fails', async () => {
+            const mockError = new Error('Deletion error');
+            vi.spyOn(mockHybridTokenStorage, 'deleteCredentials').mockRejectedValue(mockError);
+            await expect(OAuthCredentialStorage.clearCredentials()).rejects.toThrow('Failed to clear OAuth credentials');
+            expect(coreEvents.emitFeedback).toHaveBeenCalledWith('error', 'Failed to clear OAuth credentials', mockError);
+        });
+    });
+});
+//# sourceMappingURL=oauth-credential-storage.test.js.map

package/dist/src/code_assist/oauth-credential-storage.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"oauth-credential-storage.test.js","sourceRoot":"","sources":["../../../src/code_assist/oauth-credential-storage.test.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAoB,MAAM,qBAAqB,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AAEvE,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAEhD,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,SAAS,CAAC;AAEzC,6BAA6B;AAC7B,MAAM,sBAAsB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;IAC/C,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;IACvB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;IACvB,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC3B,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,8CAA8C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC7D,kBAAkB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,sBAAsB,CAAC;CACxD,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;IACxB,QAAQ,EAAE;QACR,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;QACjB,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE;KACZ;CACF,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AACnB,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;AACrB,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,UAAU,EAAE;QACV,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;QACrB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;KACxB;CACF,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,MAAM,eAAe,GAAgB;QACnC,YAAY,EAAE,mBAAmB;QACjC,aAAa,EAAE,oBAAoB;QACnC,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI;QACrC,UAAU,EAAE,QAAQ;QACpB,KAAK,EAAE,eAAe;KACvB,CAAC;IAEF,MAAM,kBAAkB,GAAqB;QAC3C,UAAU,EAAE,cAAc;QAC1B,KAAK,EAAE;YACL,WAAW,EAAE,mBAAmB;YAChC,YAAY,EAAE,oBAAoB;YAClC,SAAS,EAAE,QAAQ;YACnB,KAAK,EAAE,eAAe;YACtB,SAAS,EAAE,eAAe,CAAC,WAAY;SACxC;QACD,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;KAC9B,CAAC;IAEF,MAAM,WAAW,GAAG,+BAA+B,CAAC;IAEpD,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,KAAK,CAAC,sBAAsB,EAAE,gBAAgB,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAC3E,EAAE,CAAC,KAAK,CAAC,sBAAsB,EAAE,gBAAgB,CAAC,CAAC,iBAAiB,CAClE,SAAS,CACV,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,sBAAsB,EAAE,mBAAmB,CAAC,CAAC,iBAAiB,CACrE,SAAS,CACV,CAAC;QAEF,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC;QACxE,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAEhD,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC;QACtD,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,EAAE,CAAC,KAAK,CAAC,sBAAsB,EAAE,gBAAgB,CAAC,CAAC,iBAAiB,CAClE,kBAAkB,CACnB,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,sBAAsB,CAAC,eAAe,EAAE,CAAC;YAE9D,MAAM,CAAC,sBAAsB,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAChE,cAAc,CACf,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mFAAmF,EAAE,KAAK,IAAI,EAAE;YACjG,EAAE,CAAC,KAAK,CAAC,sBAAsB,EAAE,gBAAgB,CAAC,CAAC,iBAAiB,CAClE,IAAI,CACL,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC,iBAAiB,CACxC,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,CAChC,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,sBAAsB,CAAC,eAAe,EAAE,CAAC;YAE9D,MAAM,CAAC,sBAAsB,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAChE,cAAc,CACf,CAAC;YACF,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YAC/D,MAAM,CAAC,sBAAsB,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC,gCAAgC;YAClG,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,oBAAoB,CAAC,WAAW,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,8BAA8B;YAChG,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;YACrF,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC,iBAAiB,CAAC;gBACzC,OAAO,EAAE,gBAAgB;gBACzB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,sBAAsB,CAAC,eAAe,EAAE,CAAC;YAE9D,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;YACxD,EAAE,CAAC,KAAK,CAAC,sBAAsB,EAAE,gBAAgB,CAAC,CAAC,iBAAiB,CAClE,SAAS,CACV,CAAC;YAEF,MAAM,MAAM,CAAC,sBAAsB,CAAC,eAAe,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CACpE,kCAAkC,CACnC,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAClD,OAAO,EACP,kCAAkC,EAClC,SAAS,CACV,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;YACjD,EAAE,CAAC,KAAK,CAAC,sBAAsB,EAAE,gBAAgB,CAAC,CAAC,iBAAiB,CAClE,IAAI,CACL,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEtD,MAAM,MAAM,CAAC,sBAAsB,CAAC,eAAe,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CACpE,kCAAkC,CACnC,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAClD,OAAO,EACP,kCAAkC,EAClC,SAAS,CACV,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,EAAE,CAAC,KAAK,CAAC,sBAAsB,EAAE,gBAAgB,CAAC,CAAC,iBAAiB,CAClE,IAAI,CACL,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC,iBAAiB,CACxC,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,CAChC,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,sBAAsB,EAAE,iBAAiB,CAAC,CAAC,iBAAiB,CACnE,SAAS,CACV,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC;YAEnE,MAAM,MAAM,GAAG,MAAM,sBAAsB,CAAC,eAAe,EAAE,CAAC;YAE9D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YACjF,EAAE,CAAC,KAAK,CAAC,sBAAsB,EAAE,gBAAgB,CAAC,CAAC,iBAAiB,CAClE,IAAI,CACL,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;YAE3D,MAAM,MAAM,CAAC,sBAAsB,CAAC,eAAe,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CACpE,kCAAkC,CACnC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,EAAE,CAAC,KAAK,CAAC,sBAAsB,EAAE,gBAAgB,CAAC,CAAC,iBAAiB,CAClE,IAAI,CACL,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC,iBAAiB,CACxC,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,CAChC,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,sBAAsB,EAAE,gBAAgB,CAAC,CAAC,iBAAiB,CAClE,IAAI,KAAK,CAAC,aAAa,CAAC,CACzB,CAAC;YAEF,MAAM,MAAM,CAAC,sBAAsB,CAAC,eAAe,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CACpE,kCAAkC,CACnC,CAAC;YAEF,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,MAAM,qBAAqB,GAAG;gBAC5B,GAAG,kBAAkB;gBACrB,KAAK,EAAE;oBACL,GAAG,kBAAkB,CAAC,KAAK;oBAC3B,WAAW,EAAE,SAAS;iBACvB;aACF,CAAC;YACF,EAAE,CAAC,KAAK,CAAC,sBAAsB,EAAE,gBAAgB,CAAC,CAAC,iBAAiB,CAClE,qBAAqB,CACtB,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,sBAAsB,CAAC,eAAe,EAAE,CAAC;YAE9D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,YAAY,EAAE,SAAS;gBACvB,aAAa,EAAE,eAAe,CAAC,aAAa;gBAC5C,UAAU,EAAE,eAAe,CAAC,UAAU;gBACtC,KAAK,EAAE,eAAe,CAAC,KAAK;gBAC5B,WAAW,EAAE,eAAe,CAAC,WAAW;aACzC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,sBAAsB,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;YAE9D,MAAM,CAAC,sBAAsB,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAChE,kBAAkB,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,kBAAkB,GAAgB;gBACtC,GAAG,eAAe;gBAClB,YAAY,EAAE,SAAS;aACxB,CAAC;YACF,MAAM,MAAM,CACV,sBAAsB,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAC3D,CAAC,OAAO,CAAC,OAAO,CACf,wDAAwD,CACzD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;YACvF,MAAM,kBAAkB,GAAgB;gBACtC,YAAY,EAAE,mBAAmB;gBACjC,aAAa,EAAE,IAAI,EAAE,YAAY;gBACjC,KAAK,EAAE,SAAS,EAAE,iBAAiB;aACpC,CAAC;YAEF,MAAM,sBAAsB,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;YAEjE,MAAM,CAAC,sBAAsB,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC;gBACjE,UAAU,EAAE,cAAc;gBAC1B,KAAK,EAAE;oBACL,WAAW,EAAE,mBAAmB;oBAChC,YAAY,EAAE,SAAS;oBACvB,SAAS,EAAE,QAAQ,EAAE,UAAU;oBAC/B,KAAK,EAAE,SAAS;oBAChB,SAAS,EAAE,SAAS;iBACrB;gBACD,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aAC9B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,sBAAsB,CAAC,gBAAgB,EAAE,CAAC;YAEhD,MAAM,CAAC,sBAAsB,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CACnE,cAAc,CACf,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,sBAAsB,CAAC,gBAAgB,EAAE,CAAC;YAEhD,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,oBAAoB,CAAC,WAAW,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC,CAAC;YAExE,MAAM,MAAM,CACV,sBAAsB,CAAC,gBAAgB,EAAE,CAC1C,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;YAC/E,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;YAC9C,EAAE,CAAC,KAAK,CAAC,sBAAsB,EAAE,mBAAmB,CAAC,CAAC,iBAAiB,CACrE,SAAS,CACV,CAAC;YAEF,MAAM,MAAM,CAAC,sBAAsB,CAAC,gBAAgB,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CACrE,mCAAmC,CACpC,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAClD,OAAO,EACP,mCAAmC,EACnC,SAAS,CACV,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/code_assist/oauth2.begin.test.d.ts

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

package/dist/src/code_assist/oauth2.begin.test.js

@@ -0,0 +1,57 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
+import { beginGeminiOAuthLoopbackFlow } from './oauth2.js';
+import * as http from 'node:http';
+// Mock http server to prevent actual port binding
+vi.mock('node:http', async () => {
+    const actual = await vi.importActual('node:http');
+    return {
+        ...actual,
+        createServer: vi.fn(),
+    };
+});
+describe('beginGeminiOAuthLoopbackFlow', () => {
+    let mockConfig;
+    let mockServer;
+    beforeEach(() => {
+        mockConfig = {
+            getProxy: () => undefined,
+        };
+        mockServer = {
+            listen: vi.fn(),
+            close: vi.fn(),
+            on: vi.fn(),
+        };
+        vi.mocked(http.createServer).mockReturnValue(mockServer);
+    });
+    afterEach(() => {
+        vi.resetAllMocks();
+    });
+    it('should return authUrl and controls', async () => {
+        const result = await beginGeminiOAuthLoopbackFlow(mockConfig);
+        expect(result.authUrl).toContain('https://accounts.google.com/o/oauth2/v2/auth');
+        expect(result.authUrl).toContain('client_id=');
+        expect(result.authUrl).toContain('state=');
+        expect(result.authUrl).toContain('redirect_uri=http%3A%2F%2Flocalhost');
+        expect(result.waitForCompletion).toBeInstanceOf(Promise);
+        expect(typeof result.cancel).toBe('function');
+    });
+    it('should call server.close() when cancelled', async () => {
+        const result = await beginGeminiOAuthLoopbackFlow(mockConfig);
+        // Trigger cancel
+        try {
+            result.cancel();
+        }
+        catch (e) {
+            // It might reject the promise, but cancel itself is sync-ish or void
+        }
+        expect(mockServer.close).toHaveBeenCalled();
+        await expect(result.waitForCompletion).rejects.toThrow('User cancelled authentication');
+    });
+});
+//# sourceMappingURL=oauth2.begin.test.js.map

package/dist/src/code_assist/oauth2.begin.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"oauth2.begin.test.js","sourceRoot":"","sources":["../../../src/code_assist/oauth2.begin.test.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,4BAA4B,EAAE,MAAM,aAAa,CAAC;AAC3D,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAGlC,kDAAkD;AAClD,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,IAAI,EAAE;IAC9B,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,YAAY,CAA6B,WAAW,CAAC,CAAC;IAC9E,OAAO;QACL,GAAG,MAAM;QACT,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;KACtB,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;IAC5C,IAAI,UAAkB,CAAC;IACvB,IAAI,UAAe,CAAC;IAEpB,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG;YACX,QAAQ,EAAE,GAAG,EAAE,CAAC,SAAS;SACL,CAAC;QAEvB,UAAU,GAAG;YACX,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;YACd,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE;SACZ,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,MAAM,GAAG,MAAM,4BAA4B,CAAC,UAAU,CAAC,CAAC;QAE9D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAC9B,8CAA8C,CAC/C,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,qCAAqC,CAAC,CAAC;QACxE,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACzD,MAAM,CAAC,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,MAAM,GAAG,MAAM,4BAA4B,CAAC,UAAU,CAAC,CAAC;QAE9D,iBAAiB;QACjB,IAAI,CAAC;YACH,MAAM,CAAC,MAAM,EAAE,CAAC;QAClB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,qEAAqE;QACvE,CAAC;QAED,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC5C,MAAM,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,OAAO,CACpD,+BAA+B,CAChC,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}