@zgsm/costrict-cli 0.0.1

package/dist/src/config/trustedFolders.js

@@ -0,0 +1,176 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import * as fs from 'node:fs';
+import * as path from 'node:path';
+import { homedir } from 'node:os';
+import { FatalConfigError, getErrorMessage, isWithinRoot, ideContextStore, GEMINI_DIR, } from '@google/gemini-cli-core';
+import stripJsonComments from 'strip-json-comments';
+export const TRUSTED_FOLDERS_FILENAME = 'trustedFolders.json';
+export function getUserSettingsDir() {
+    return path.join(homedir(), GEMINI_DIR);
+}
+export function getTrustedFoldersPath() {
+    if (process.env['GEMINI_CLI_TRUSTED_FOLDERS_PATH']) {
+        return process.env['GEMINI_CLI_TRUSTED_FOLDERS_PATH'];
+    }
+    return path.join(getUserSettingsDir(), TRUSTED_FOLDERS_FILENAME);
+}
+export var TrustLevel;
+(function (TrustLevel) {
+    TrustLevel["TRUST_FOLDER"] = "TRUST_FOLDER";
+    TrustLevel["TRUST_PARENT"] = "TRUST_PARENT";
+    TrustLevel["DO_NOT_TRUST"] = "DO_NOT_TRUST";
+})(TrustLevel || (TrustLevel = {}));
+export class LoadedTrustedFolders {
+    user;
+    errors;
+    constructor(user, errors) {
+        this.user = user;
+        this.errors = errors;
+    }
+    get rules() {
+        return Object.entries(this.user.config).map(([path, trustLevel]) => ({
+            path,
+            trustLevel,
+        }));
+    }
+    /**
+     * Returns true or false if the path should be "trusted". This function
+     * should only be invoked when the folder trust setting is active.
+     *
+     * @param location path
+     * @returns
+     */
+    isPathTrusted(location, config) {
+        const configToUse = config ?? this.user.config;
+        const trustedPaths = [];
+        const untrustedPaths = [];
+        for (const rule of Object.entries(configToUse).map(([path, trustLevel]) => ({ path, trustLevel }))) {
+            switch (rule.trustLevel) {
+                case TrustLevel.TRUST_FOLDER:
+                    trustedPaths.push(rule.path);
+                    break;
+                case TrustLevel.TRUST_PARENT:
+                    trustedPaths.push(path.dirname(rule.path));
+                    break;
+                case TrustLevel.DO_NOT_TRUST:
+                    untrustedPaths.push(rule.path);
+                    break;
+                default:
+                    // Do nothing for unknown trust levels.
+                    break;
+            }
+        }
+        for (const trustedPath of trustedPaths) {
+            if (isWithinRoot(location, trustedPath)) {
+                return true;
+            }
+        }
+        for (const untrustedPath of untrustedPaths) {
+            if (path.normalize(location) === path.normalize(untrustedPath)) {
+                return false;
+            }
+        }
+        return undefined;
+    }
+    setValue(path, trustLevel) {
+        const originalTrustLevel = this.user.config[path];
+        this.user.config[path] = trustLevel;
+        try {
+            saveTrustedFolders(this.user);
+        }
+        catch (e) {
+            // Revert the in-memory change if the save failed.
+            if (originalTrustLevel === undefined) {
+                delete this.user.config[path];
+            }
+            else {
+                this.user.config[path] = originalTrustLevel;
+            }
+            throw e;
+        }
+    }
+}
+let loadedTrustedFolders;
+/**
+ * FOR TESTING PURPOSES ONLY.
+ * Resets the in-memory cache of the trusted folders configuration.
+ */
+export function resetTrustedFoldersForTesting() {
+    loadedTrustedFolders = undefined;
+}
+export function loadTrustedFolders() {
+    if (loadedTrustedFolders) {
+        return loadedTrustedFolders;
+    }
+    const errors = [];
+    let userConfig = {};
+    const userPath = getTrustedFoldersPath();
+    // Load user trusted folders
+    try {
+        if (fs.existsSync(userPath)) {
+            const content = fs.readFileSync(userPath, 'utf-8');
+            const parsed = JSON.parse(stripJsonComments(content));
+            if (typeof parsed !== 'object' ||
+                parsed === null ||
+                Array.isArray(parsed)) {
+                errors.push({
+                    message: 'Trusted folders file is not a valid JSON object.',
+                    path: userPath,
+                });
+            }
+            else {
+                userConfig = parsed;
+            }
+        }
+    }
+    catch (error) {
+        errors.push({
+            message: getErrorMessage(error),
+            path: userPath,
+        });
+    }
+    loadedTrustedFolders = new LoadedTrustedFolders({ path: userPath, config: userConfig }, errors);
+    return loadedTrustedFolders;
+}
+export function saveTrustedFolders(trustedFoldersFile) {
+    // Ensure the directory exists
+    const dirPath = path.dirname(trustedFoldersFile.path);
+    if (!fs.existsSync(dirPath)) {
+        fs.mkdirSync(dirPath, { recursive: true });
+    }
+    fs.writeFileSync(trustedFoldersFile.path, JSON.stringify(trustedFoldersFile.config, null, 2), { encoding: 'utf-8', mode: 0o600 });
+}
+/** Is folder trust feature enabled per the current applied settings */
+export function isFolderTrustEnabled(settings) {
+    const folderTrustSetting = settings.security?.folderTrust?.enabled ?? false;
+    return folderTrustSetting;
+}
+function getWorkspaceTrustFromLocalConfig(trustConfig) {
+    const folders = loadTrustedFolders();
+    const configToUse = trustConfig ?? folders.user.config;
+    if (folders.errors.length > 0) {
+        const errorMessages = folders.errors.map((error) => `Error in ${error.path}: ${error.message}`);
+        throw new FatalConfigError(`${errorMessages.join('\n')}\nPlease fix the configuration file and try again.`);
+    }
+    const isTrusted = folders.isPathTrusted(process.cwd(), configToUse);
+    return {
+        isTrusted,
+        source: isTrusted !== undefined ? 'file' : undefined,
+    };
+}
+export function isWorkspaceTrusted(settings, trustConfig) {
+    if (!isFolderTrustEnabled(settings)) {
+        return { isTrusted: true, source: undefined };
+    }
+    const ideTrust = ideContextStore.get()?.workspaceState?.isTrusted;
+    if (ideTrust !== undefined) {
+        return { isTrusted: ideTrust, source: 'ide' };
+    }
+    // Fall back to the local user configuration
+    return getWorkspaceTrustFromLocalConfig(trustConfig);
+}
+//# sourceMappingURL=trustedFolders.js.map

package/dist/src/config/trustedFolders.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"trustedFolders.js","sourceRoot":"","sources":["../../../src/config/trustedFolders.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EACL,gBAAgB,EAChB,eAAe,EACf,YAAY,EACZ,eAAe,EACf,UAAU,GACX,MAAM,yBAAyB,CAAC;AAEjC,OAAO,iBAAiB,MAAM,qBAAqB,CAAC;AAEpD,MAAM,CAAC,MAAM,wBAAwB,GAAG,qBAAqB,CAAC;AAE9D,MAAM,UAAU,kBAAkB;IAChC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,UAAU,CAAC,CAAC;AAC1C,CAAC;AAED,MAAM,UAAU,qBAAqB;IACnC,IAAI,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,EAAE,CAAC;QACnD,OAAO,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;IACxD,CAAC;IACD,OAAO,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,wBAAwB,CAAC,CAAC;AACnE,CAAC;AAED,MAAM,CAAN,IAAY,UAIX;AAJD,WAAY,UAAU;IACpB,2CAA6B,CAAA;IAC7B,2CAA6B,CAAA;IAC7B,2CAA6B,CAAA;AAC/B,CAAC,EAJW,UAAU,KAAV,UAAU,QAIrB;AAsBD,MAAM,OAAO,oBAAoB;IAEpB;IACA;IAFX,YACW,IAAwB,EACxB,MAA6B;QAD7B,SAAI,GAAJ,IAAI,CAAoB;QACxB,WAAM,GAAN,MAAM,CAAuB;IACrC,CAAC;IAEJ,IAAI,KAAK;QACP,OAAO,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,CAAC,CAAC;YACnE,IAAI;YACJ,UAAU;SACX,CAAC,CAAC,CAAC;IACN,CAAC;IAED;;;;;;OAMG;IACH,aAAa,CACX,QAAgB,EAChB,MAAmC;QAEnC,MAAM,WAAW,GAAG,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;QAC/C,MAAM,YAAY,GAAa,EAAE,CAAC;QAClC,MAAM,cAAc,GAAa,EAAE,CAAC;QAEpC,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,GAAG,CAChD,CAAC,CAAC,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAC/C,EAAE,CAAC;YACF,QAAQ,IAAI,CAAC,UAAU,EAAE,CAAC;gBACxB,KAAK,UAAU,CAAC,YAAY;oBAC1B,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBAC7B,MAAM;gBACR,KAAK,UAAU,CAAC,YAAY;oBAC1B,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;oBAC3C,MAAM;gBACR,KAAK,UAAU,CAAC,YAAY;oBAC1B,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBAC/B,MAAM;gBACR;oBACE,uCAAuC;oBACvC,MAAM;YACV,CAAC;QACH,CAAC;QAED,KAAK,MAAM,WAAW,IAAI,YAAY,EAAE,CAAC;YACvC,IAAI,YAAY,CAAC,QAAQ,EAAE,WAAW,CAAC,EAAE,CAAC;gBACxC,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QAED,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE,CAAC;YAC3C,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,EAAE,CAAC;gBAC/D,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,QAAQ,CAAC,IAAY,EAAE,UAAsB;QAC3C,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAClD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC;QACpC,IAAI,CAAC;YACH,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,kDAAkD;YAClD,IAAI,kBAAkB,KAAK,SAAS,EAAE,CAAC;gBACrC,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAChC,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC;YAC9C,CAAC;YACD,MAAM,CAAC,CAAC;QACV,CAAC;IACH,CAAC;CACF;AAED,IAAI,oBAAsD,CAAC;AAE3D;;;GAGG;AACH,MAAM,UAAU,6BAA6B;IAC3C,oBAAoB,GAAG,SAAS,CAAC;AACnC,CAAC;AAED,MAAM,UAAU,kBAAkB;IAChC,IAAI,oBAAoB,EAAE,CAAC;QACzB,OAAO,oBAAoB,CAAC;IAC9B,CAAC;IAED,MAAM,MAAM,GAA0B,EAAE,CAAC;IACzC,IAAI,UAAU,GAA+B,EAAE,CAAC;IAEhD,MAAM,QAAQ,GAAG,qBAAqB,EAAE,CAAC;IAEzC,4BAA4B;IAC5B,IAAI,CAAC;QACH,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC5B,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACnD,MAAM,MAAM,GAAY,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC;YAE/D,IACE,OAAO,MAAM,KAAK,QAAQ;gBAC1B,MAAM,KAAK,IAAI;gBACf,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EACrB,CAAC;gBACD,MAAM,CAAC,IAAI,CAAC;oBACV,OAAO,EAAE,kDAAkD;oBAC3D,IAAI,EAAE,QAAQ;iBACf,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,UAAU,GAAG,MAAoC,CAAC;YACpD,CAAC;QACH,CAAC;IACH,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,MAAM,CAAC,IAAI,CAAC;YACV,OAAO,EAAE,eAAe,CAAC,KAAK,CAAC;YAC/B,IAAI,EAAE,QAAQ;SACf,CAAC,CAAC;IACL,CAAC;IAED,oBAAoB,GAAG,IAAI,oBAAoB,CAC7C,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,EACtC,MAAM,CACP,CAAC;IACF,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,kBAAsC;IAEtC,8BAA8B;IAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;IACtD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5B,EAAE,CAAC,SAAS,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED,EAAE,CAAC,aAAa,CACd,kBAAkB,CAAC,IAAI,EACvB,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAClD,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CACnC,CAAC;AACJ,CAAC;AAED,uEAAuE;AACvE,MAAM,UAAU,oBAAoB,CAAC,QAAkB;IACrD,MAAM,kBAAkB,GAAG,QAAQ,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,IAAI,KAAK,CAAC;IAC5E,OAAO,kBAAkB,CAAC;AAC5B,CAAC;AAED,SAAS,gCAAgC,CACvC,WAAwC;IAExC,MAAM,OAAO,GAAG,kBAAkB,EAAE,CAAC;IACrC,MAAM,WAAW,GAAG,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC;IAEvD,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9B,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC,GAAG,CACtC,CAAC,KAAK,EAAE,EAAE,CAAC,YAAY,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,CACtD,CAAC;QACF,MAAM,IAAI,gBAAgB,CACxB,GAAG,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,oDAAoD,CAChF,CAAC;IACJ,CAAC;IAED,MAAM,SAAS,GAAG,OAAO,CAAC,aAAa,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,WAAW,CAAC,CAAC;IACpE,OAAO;QACL,SAAS;QACT,MAAM,EAAE,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;KACrD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,QAAkB,EAClB,WAAwC;IAExC,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,EAAE,CAAC;QACpC,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;IAChD,CAAC;IAED,MAAM,QAAQ,GAAG,eAAe,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,SAAS,CAAC;IAClE,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;IAChD,CAAC;IAED,4CAA4C;IAC5C,OAAO,gCAAgC,CAAC,WAAW,CAAC,CAAC;AACvD,CAAC"}

package/dist/src/config/trustedFolders.test.d.ts

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

package/dist/src/config/trustedFolders.test.js

@@ -0,0 +1,340 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import * as osActual from 'node:os';
+import { FatalConfigError, ideContextStore } from '@google/gemini-cli-core';
+import { describe, it, expect, vi, beforeEach, afterEach, } from 'vitest';
+import * as fs from 'node:fs';
+import stripJsonComments from 'strip-json-comments';
+import * as path from 'node:path';
+import { loadTrustedFolders, getTrustedFoldersPath, TrustLevel, isWorkspaceTrusted, resetTrustedFoldersForTesting, } from './trustedFolders.js';
+vi.mock('os', async (importOriginal) => {
+    const actualOs = await importOriginal();
+    return {
+        ...actualOs,
+        homedir: vi.fn(() => '/mock/home/user'),
+        platform: vi.fn(() => 'linux'),
+    };
+});
+vi.mock('fs', async (importOriginal) => {
+    const actualFs = await importOriginal();
+    return {
+        ...actualFs,
+        existsSync: vi.fn(),
+        readFileSync: vi.fn(),
+        writeFileSync: vi.fn(),
+        mkdirSync: vi.fn(),
+    };
+});
+vi.mock('strip-json-comments', () => ({
+    default: vi.fn((content) => content),
+}));
+describe('Trusted Folders Loading', () => {
+    let mockFsExistsSync;
+    let mockStripJsonComments;
+    let mockFsWriteFileSync;
+    beforeEach(() => {
+        resetTrustedFoldersForTesting();
+        vi.resetAllMocks();
+        mockFsExistsSync = vi.mocked(fs.existsSync);
+        mockStripJsonComments = vi.mocked(stripJsonComments);
+        mockFsWriteFileSync = vi.mocked(fs.writeFileSync);
+        vi.mocked(osActual.homedir).mockReturnValue('/mock/home/user');
+        mockStripJsonComments.mockImplementation((jsonString) => jsonString);
+        mockFsExistsSync.mockReturnValue(false);
+        fs.readFileSync.mockReturnValue('{}');
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    it('should load empty rules if no files exist', () => {
+        const { rules, errors } = loadTrustedFolders();
+        expect(rules).toEqual([]);
+        expect(errors).toEqual([]);
+    });
+    describe('isPathTrusted', () => {
+        function setup({ config = {} } = {}) {
+            mockFsExistsSync.mockImplementation((p) => p === getTrustedFoldersPath());
+            fs.readFileSync.mockImplementation((p) => {
+                if (p === getTrustedFoldersPath())
+                    return JSON.stringify(config);
+                return '{}';
+            });
+            const folders = loadTrustedFolders();
+            return { folders };
+        }
+        it('provides a method to determine if a path is trusted', () => {
+            const { folders } = setup({
+                config: {
+                    './myfolder': TrustLevel.TRUST_FOLDER,
+                    '/trustedparent/trustme': TrustLevel.TRUST_PARENT,
+                    '/user/folder': TrustLevel.TRUST_FOLDER,
+                    '/secret': TrustLevel.DO_NOT_TRUST,
+                    '/secret/publickeys': TrustLevel.TRUST_FOLDER,
+                },
+            });
+            expect(folders.isPathTrusted('/secret')).toBe(false);
+            expect(folders.isPathTrusted('/user/folder')).toBe(true);
+            expect(folders.isPathTrusted('/secret/publickeys/public.pem')).toBe(true);
+            expect(folders.isPathTrusted('/user/folder/harhar')).toBe(true);
+            expect(folders.isPathTrusted('myfolder/somefile.jpg')).toBe(true);
+            expect(folders.isPathTrusted('/trustedparent/someotherfolder')).toBe(true);
+            expect(folders.isPathTrusted('/trustedparent/trustme')).toBe(true);
+            // No explicit rule covers this file
+            expect(folders.isPathTrusted('/secret/bankaccounts.json')).toBe(undefined);
+            expect(folders.isPathTrusted('/secret/mine/privatekey.pem')).toBe(undefined);
+            expect(folders.isPathTrusted('/user/someotherfolder')).toBe(undefined);
+        });
+    });
+    it('should load user rules if only user file exists', () => {
+        const userPath = getTrustedFoldersPath();
+        mockFsExistsSync.mockImplementation((p) => p === userPath);
+        const userContent = {
+            '/user/folder': TrustLevel.TRUST_FOLDER,
+        };
+        fs.readFileSync.mockImplementation((p) => {
+            if (p === userPath)
+                return JSON.stringify(userContent);
+            return '{}';
+        });
+        const { rules, errors } = loadTrustedFolders();
+        expect(rules).toEqual([
+            { path: '/user/folder', trustLevel: TrustLevel.TRUST_FOLDER },
+        ]);
+        expect(errors).toEqual([]);
+    });
+    it('should handle JSON parsing errors gracefully', () => {
+        const userPath = getTrustedFoldersPath();
+        mockFsExistsSync.mockImplementation((p) => p === userPath);
+        fs.readFileSync.mockImplementation((p) => {
+            if (p === userPath)
+                return 'invalid json';
+            return '{}';
+        });
+        const { rules, errors } = loadTrustedFolders();
+        expect(rules).toEqual([]);
+        expect(errors.length).toBe(1);
+        expect(errors[0].path).toBe(userPath);
+        expect(errors[0].message).toContain('Unexpected token');
+    });
+    it('should use GEMINI_CLI_TRUSTED_FOLDERS_PATH env var if set', () => {
+        const customPath = '/custom/path/to/trusted_folders.json';
+        process.env['GEMINI_CLI_TRUSTED_FOLDERS_PATH'] = customPath;
+        mockFsExistsSync.mockImplementation((p) => p === customPath);
+        const userContent = {
+            '/user/folder/from/env': TrustLevel.TRUST_FOLDER,
+        };
+        fs.readFileSync.mockImplementation((p) => {
+            if (p === customPath)
+                return JSON.stringify(userContent);
+            return '{}';
+        });
+        const { rules, errors } = loadTrustedFolders();
+        expect(rules).toEqual([
+            {
+                path: '/user/folder/from/env',
+                trustLevel: TrustLevel.TRUST_FOLDER,
+            },
+        ]);
+        expect(errors).toEqual([]);
+        delete process.env['GEMINI_CLI_TRUSTED_FOLDERS_PATH'];
+    });
+    it('setValue should update the user config and save it', () => {
+        const loadedFolders = loadTrustedFolders();
+        loadedFolders.setValue('/new/path', TrustLevel.TRUST_FOLDER);
+        expect(loadedFolders.user.config['/new/path']).toBe(TrustLevel.TRUST_FOLDER);
+        expect(mockFsWriteFileSync).toHaveBeenCalledWith(getTrustedFoldersPath(), JSON.stringify({ '/new/path': TrustLevel.TRUST_FOLDER }, null, 2), { encoding: 'utf-8', mode: 0o600 });
+    });
+});
+describe('isWorkspaceTrusted', () => {
+    let mockCwd;
+    const mockRules = {};
+    const mockSettings = {
+        security: {
+            folderTrust: {
+                enabled: true,
+            },
+        },
+    };
+    beforeEach(() => {
+        resetTrustedFoldersForTesting();
+        vi.spyOn(process, 'cwd').mockImplementation(() => mockCwd);
+        vi.spyOn(fs, 'readFileSync').mockImplementation((p) => {
+            if (p === getTrustedFoldersPath()) {
+                return JSON.stringify(mockRules);
+            }
+            return '{}';
+        });
+        vi.spyOn(fs, 'existsSync').mockImplementation((p) => p === getTrustedFoldersPath());
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+        // Clear the object
+        Object.keys(mockRules).forEach((key) => delete mockRules[key]);
+    });
+    it('should throw a fatal error if the config is malformed', () => {
+        mockCwd = '/home/user/projectA';
+        // This mock needs to be specific to this test to override the one in beforeEach
+        vi.spyOn(fs, 'readFileSync').mockImplementation((p) => {
+            if (p === getTrustedFoldersPath()) {
+                return '{"foo": "bar",}'; // Malformed JSON with trailing comma
+            }
+            return '{}';
+        });
+        expect(() => isWorkspaceTrusted(mockSettings)).toThrow(FatalConfigError);
+        expect(() => isWorkspaceTrusted(mockSettings)).toThrow(/Please fix the configuration file/);
+    });
+    it('should throw a fatal error if the config is not a JSON object', () => {
+        mockCwd = '/home/user/projectA';
+        vi.spyOn(fs, 'readFileSync').mockImplementation((p) => {
+            if (p === getTrustedFoldersPath()) {
+                return 'null';
+            }
+            return '{}';
+        });
+        expect(() => isWorkspaceTrusted(mockSettings)).toThrow(FatalConfigError);
+        expect(() => isWorkspaceTrusted(mockSettings)).toThrow(/not a valid JSON object/);
+    });
+    it('should return true for a directly trusted folder', () => {
+        mockCwd = '/home/user/projectA';
+        mockRules['/home/user/projectA'] = TrustLevel.TRUST_FOLDER;
+        expect(isWorkspaceTrusted(mockSettings)).toEqual({
+            isTrusted: true,
+            source: 'file',
+        });
+    });
+    it('should return true for a child of a trusted folder', () => {
+        mockCwd = '/home/user/projectA/src';
+        mockRules['/home/user/projectA'] = TrustLevel.TRUST_FOLDER;
+        expect(isWorkspaceTrusted(mockSettings)).toEqual({
+            isTrusted: true,
+            source: 'file',
+        });
+    });
+    it('should return true for a child of a trusted parent folder', () => {
+        mockCwd = '/home/user/projectB';
+        mockRules['/home/user/projectB/somefile.txt'] = TrustLevel.TRUST_PARENT;
+        expect(isWorkspaceTrusted(mockSettings)).toEqual({
+            isTrusted: true,
+            source: 'file',
+        });
+    });
+    it('should return false for a directly untrusted folder', () => {
+        mockCwd = '/home/user/untrusted';
+        mockRules['/home/user/untrusted'] = TrustLevel.DO_NOT_TRUST;
+        expect(isWorkspaceTrusted(mockSettings)).toEqual({
+            isTrusted: false,
+            source: 'file',
+        });
+    });
+    it('should return undefined for a child of an untrusted folder', () => {
+        mockCwd = '/home/user/untrusted/src';
+        mockRules['/home/user/untrusted'] = TrustLevel.DO_NOT_TRUST;
+        expect(isWorkspaceTrusted(mockSettings).isTrusted).toBeUndefined();
+    });
+    it('should return undefined when no rules match', () => {
+        mockCwd = '/home/user/other';
+        mockRules['/home/user/projectA'] = TrustLevel.TRUST_FOLDER;
+        mockRules['/home/user/untrusted'] = TrustLevel.DO_NOT_TRUST;
+        expect(isWorkspaceTrusted(mockSettings).isTrusted).toBeUndefined();
+    });
+    it('should prioritize trust over distrust', () => {
+        mockCwd = '/home/user/projectA/untrusted';
+        mockRules['/home/user/projectA'] = TrustLevel.TRUST_FOLDER;
+        mockRules['/home/user/projectA/untrusted'] = TrustLevel.DO_NOT_TRUST;
+        expect(isWorkspaceTrusted(mockSettings)).toEqual({
+            isTrusted: true,
+            source: 'file',
+        });
+    });
+    it('should handle path normalization', () => {
+        mockCwd = '/home/user/projectA';
+        mockRules[`/home/user/../user/${path.basename('/home/user/projectA')}`] =
+            TrustLevel.TRUST_FOLDER;
+        expect(isWorkspaceTrusted(mockSettings)).toEqual({
+            isTrusted: true,
+            source: 'file',
+        });
+    });
+});
+describe('isWorkspaceTrusted with IDE override', () => {
+    afterEach(() => {
+        vi.clearAllMocks();
+        ideContextStore.clear();
+        resetTrustedFoldersForTesting();
+    });
+    const mockSettings = {
+        security: {
+            folderTrust: {
+                enabled: true,
+            },
+        },
+    };
+    it('should return true when ideTrust is true, ignoring config', () => {
+        ideContextStore.set({ workspaceState: { isTrusted: true } });
+        // Even if config says don't trust, ideTrust should win.
+        vi.spyOn(fs, 'readFileSync').mockReturnValue(JSON.stringify({ [process.cwd()]: TrustLevel.DO_NOT_TRUST }));
+        expect(isWorkspaceTrusted(mockSettings)).toEqual({
+            isTrusted: true,
+            source: 'ide',
+        });
+    });
+    it('should return false when ideTrust is false, ignoring config', () => {
+        ideContextStore.set({ workspaceState: { isTrusted: false } });
+        // Even if config says trust, ideTrust should win.
+        vi.spyOn(fs, 'readFileSync').mockReturnValue(JSON.stringify({ [process.cwd()]: TrustLevel.TRUST_FOLDER }));
+        expect(isWorkspaceTrusted(mockSettings)).toEqual({
+            isTrusted: false,
+            source: 'ide',
+        });
+    });
+    it('should fall back to config when ideTrust is undefined', () => {
+        vi.spyOn(fs, 'existsSync').mockReturnValue(true);
+        vi.spyOn(fs, 'readFileSync').mockReturnValue(JSON.stringify({ [process.cwd()]: TrustLevel.TRUST_FOLDER }));
+        expect(isWorkspaceTrusted(mockSettings)).toEqual({
+            isTrusted: true,
+            source: 'file',
+        });
+    });
+    it('should always return true if folderTrust setting is disabled', () => {
+        const settings = {
+            security: {
+                folderTrust: {
+                    enabled: false,
+                },
+            },
+        };
+        ideContextStore.set({ workspaceState: { isTrusted: false } });
+        expect(isWorkspaceTrusted(settings)).toEqual({
+            isTrusted: true,
+            source: undefined,
+        });
+    });
+});
+describe('Trusted Folders Caching', () => {
+    beforeEach(() => {
+        resetTrustedFoldersForTesting();
+        vi.mocked(fs.existsSync).mockReturnValue(true);
+        vi.mocked(fs.readFileSync).mockReturnValue('{}');
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    it('should cache the loaded folders object', () => {
+        const readSpy = vi.spyOn(fs, 'readFileSync');
+        // First call should read the file
+        loadTrustedFolders();
+        expect(readSpy).toHaveBeenCalledTimes(1);
+        // Second call should use the cache
+        loadTrustedFolders();
+        expect(readSpy).toHaveBeenCalledTimes(1);
+        // Resetting should clear the cache
+        resetTrustedFoldersForTesting();
+        // Third call should read the file again
+        loadTrustedFolders();
+        expect(readSpy).toHaveBeenCalledTimes(2);
+    });
+});
+//# sourceMappingURL=trustedFolders.test.js.map

package/dist/src/config/trustedFolders.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"trustedFolders.test.js","sourceRoot":"","sources":["../../../src/config/trustedFolders.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,QAAQ,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC5E,OAAO,EACL,QAAQ,EACR,EAAE,EACF,MAAM,EACN,EAAE,EACF,UAAU,EACV,SAAS,GAGV,MAAM,QAAQ,CAAC;AAChB,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,iBAAiB,MAAM,qBAAqB,CAAC;AACpD,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EACL,kBAAkB,EAClB,qBAAqB,EACrB,UAAU,EACV,kBAAkB,EAClB,6BAA6B,GAC9B,MAAM,qBAAqB,CAAC;AAG7B,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IACrC,MAAM,QAAQ,GAAG,MAAM,cAAc,EAAmB,CAAC;IACzD,OAAO;QACL,GAAG,QAAQ;QACX,OAAO,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC;QACvC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC;KAC/B,CAAC;AACJ,CAAC,CAAC,CAAC;AACH,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IACrC,MAAM,QAAQ,GAAG,MAAM,cAAc,EAAa,CAAC;IACnD,OAAO;QACL,GAAG,QAAQ;QACX,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;QACnB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;QACrB,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE;QACtB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;KACnB,CAAC;AACJ,CAAC,CAAC,CAAC;AACH,EAAE,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;IACpC,OAAO,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC;CACrC,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,IAAI,gBAA8C,CAAC;IACnD,IAAI,qBAAuD,CAAC;IAC5D,IAAI,mBAAoD,CAAC;IAEzD,UAAU,CAAC,GAAG,EAAE;QACd,6BAA6B,EAAE,CAAC;QAChC,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,gBAAgB,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC;QAC5C,qBAAqB,GAAG,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;QACrD,mBAAmB,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC;QAClD,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,iBAAiB,CAAC,CAAC;QAC9D,qBAAyC,CAAC,kBAAkB,CAC3D,CAAC,UAAkB,EAAE,EAAE,CAAC,UAAU,CACnC,CAAC;QACD,gBAAyB,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QACjD,EAAE,CAAC,YAAqB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,kBAAkB,EAAE,CAAC;QAC/C,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC1B,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,SAAS,KAAK,CAAC,EAAE,MAAM,GAAG,EAAgC,KAAK,EAAE;YAC9D,gBAAyB,CAAC,kBAAkB,CAC3C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,qBAAqB,EAAE,CACrC,CAAC;YACD,EAAE,CAAC,YAAqB,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,EAAE;gBACjD,IAAI,CAAC,KAAK,qBAAqB,EAAE;oBAAE,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;gBACjE,OAAO,IAAI,CAAC;YACd,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,kBAAkB,EAAE,CAAC;YAErC,OAAO,EAAE,OAAO,EAAE,CAAC;QACrB,CAAC;QAED,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;gBACxB,MAAM,EAAE;oBACN,YAAY,EAAE,UAAU,CAAC,YAAY;oBACrC,wBAAwB,EAAE,UAAU,CAAC,YAAY;oBACjD,cAAc,EAAE,UAAU,CAAC,YAAY;oBACvC,SAAS,EAAE,UAAU,CAAC,YAAY;oBAClC,oBAAoB,EAAE,UAAU,CAAC,YAAY;iBAC9C;aACF,CAAC,CAAC;YACH,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrD,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzD,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,+BAA+B,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1E,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChE,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClE,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,gCAAgC,CAAC,CAAC,CAAC,IAAI,CAClE,IAAI,CACL,CAAC;YACF,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,wBAAwB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEnE,oCAAoC;YACpC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,2BAA2B,CAAC,CAAC,CAAC,IAAI,CAC7D,SAAS,CACV,CAAC;YACF,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,6BAA6B,CAAC,CAAC,CAAC,IAAI,CAC/D,SAAS,CACV,CAAC;YACF,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,QAAQ,GAAG,qBAAqB,EAAE,CAAC;QACxC,gBAAyB,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC;QACrE,MAAM,WAAW,GAAG;YAClB,cAAc,EAAE,UAAU,CAAC,YAAY;SACxC,CAAC;QACD,EAAE,CAAC,YAAqB,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,EAAE;YACjD,IAAI,CAAC,KAAK,QAAQ;gBAAE,OAAO,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YACvD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,kBAAkB,EAAE,CAAC;QAC/C,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB,EAAE,IAAI,EAAE,cAAc,EAAE,UAAU,EAAE,UAAU,CAAC,YAAY,EAAE;SAC9D,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,QAAQ,GAAG,qBAAqB,EAAE,CAAC;QACxC,gBAAyB,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC;QACpE,EAAE,CAAC,YAAqB,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,EAAE;YACjD,IAAI,CAAC,KAAK,QAAQ;gBAAE,OAAO,cAAc,CAAC;YAC1C,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,kBAAkB,EAAE,CAAC;QAC/C,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC1B,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,MAAM,UAAU,GAAG,sCAAsC,CAAC;QAC1D,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,GAAG,UAAU,CAAC;QAE3D,gBAAyB,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,UAAU,CAAC,CAAC;QACvE,MAAM,WAAW,GAAG;YAClB,uBAAuB,EAAE,UAAU,CAAC,YAAY;SACjD,CAAC;QACD,EAAE,CAAC,YAAqB,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,EAAE;YACjD,IAAI,CAAC,KAAK,UAAU;gBAAE,OAAO,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YACzD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,kBAAkB,EAAE,CAAC;QAC/C,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB;gBACE,IAAI,EAAE,uBAAuB;gBAC7B,UAAU,EAAE,UAAU,CAAC,YAAY;aACpC;SACF,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAE3B,OAAO,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,aAAa,GAAG,kBAAkB,EAAE,CAAC;QAC3C,aAAa,CAAC,QAAQ,CAAC,WAAW,EAAE,UAAU,CAAC,YAAY,CAAC,CAAC;QAE7D,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CACjD,UAAU,CAAC,YAAY,CACxB,CAAC;QACF,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAC9C,qBAAqB,EAAE,EACvB,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,UAAU,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,EACjE,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CACnC,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,IAAI,OAAe,CAAC;IACpB,MAAM,SAAS,GAA+B,EAAE,CAAC;IACjD,MAAM,YAAY,GAAa;QAC7B,QAAQ,EAAE;YACR,WAAW,EAAE;gBACX,OAAO,EAAE,IAAI;aACd;SACF;KACF,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,6BAA6B,EAAE,CAAC;QAChC,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,CAAC;QAC3D,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,EAAE;YACpD,IAAI,CAAC,KAAK,qBAAqB,EAAE,EAAE,CAAC;gBAClC,OAAO,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YACnC,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,YAAY,CAAC,CAAC,kBAAkB,CAC3C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,qBAAqB,EAAE,CACrC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;QACrB,mBAAmB;QACnB,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,OAAO,GAAG,qBAAqB,CAAC;QAChC,gFAAgF;QAChF,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,EAAE;YACpD,IAAI,CAAC,KAAK,qBAAqB,EAAE,EAAE,CAAC;gBAClC,OAAO,iBAAiB,CAAC,CAAC,qCAAqC;YACjE,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QACzE,MAAM,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CACpD,mCAAmC,CACpC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,OAAO,GAAG,qBAAqB,CAAC;QAChC,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,EAAE;YACpD,IAAI,CAAC,KAAK,qBAAqB,EAAE,EAAE,CAAC;gBAClC,OAAO,MAAM,CAAC;YAChB,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QACzE,MAAM,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CACpD,yBAAyB,CAC1B,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,OAAO,GAAG,qBAAqB,CAAC;QAChC,SAAS,CAAC,qBAAqB,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;QAC3D,MAAM,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC;YAC/C,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,OAAO,GAAG,yBAAyB,CAAC;QACpC,SAAS,CAAC,qBAAqB,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;QAC3D,MAAM,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC;YAC/C,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,OAAO,GAAG,qBAAqB,CAAC;QAChC,SAAS,CAAC,kCAAkC,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;QACxE,MAAM,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC;YAC/C,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,OAAO,GAAG,sBAAsB,CAAC;QACjC,SAAS,CAAC,sBAAsB,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;QAC5D,MAAM,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC;YAC/C,SAAS,EAAE,KAAK;YAChB,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,OAAO,GAAG,0BAA0B,CAAC;QACrC,SAAS,CAAC,sBAAsB,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;QAC5D,MAAM,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,CAAC,aAAa,EAAE,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,OAAO,GAAG,kBAAkB,CAAC;QAC7B,SAAS,CAAC,qBAAqB,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;QAC3D,SAAS,CAAC,sBAAsB,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;QAC5D,MAAM,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,CAAC,aAAa,EAAE,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,OAAO,GAAG,+BAA+B,CAAC;QAC1C,SAAS,CAAC,qBAAqB,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;QAC3D,SAAS,CAAC,+BAA+B,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;QACrE,MAAM,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC;YAC/C,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,OAAO,GAAG,qBAAqB,CAAC;QAChC,SAAS,CAAC,sBAAsB,IAAI,CAAC,QAAQ,CAAC,qBAAqB,CAAC,EAAE,CAAC;YACrE,UAAU,CAAC,YAAY,CAAC;QAC1B,MAAM,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC;YAC/C,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACpD,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,eAAe,CAAC,KAAK,EAAE,CAAC;QACxB,6BAA6B,EAAE,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,MAAM,YAAY,GAAa;QAC7B,QAAQ,EAAE;YACR,WAAW,EAAE;gBACX,OAAO,EAAE,IAAI;aACd;SACF;KACF,CAAC;IAEF,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,eAAe,CAAC,GAAG,CAAC,EAAE,cAAc,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;QAC7D,wDAAwD;QACxD,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC,eAAe,CAC1C,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,UAAU,CAAC,YAAY,EAAE,CAAC,CAC7D,CAAC;QACF,MAAM,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC;YAC/C,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,KAAK;SACd,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,eAAe,CAAC,GAAG,CAAC,EAAE,cAAc,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;QAC9D,kDAAkD;QAClD,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC,eAAe,CAC1C,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,UAAU,CAAC,YAAY,EAAE,CAAC,CAC7D,CAAC;QACF,MAAM,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC;YAC/C,SAAS,EAAE,KAAK;YAChB,MAAM,EAAE,KAAK;SACd,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,YAAY,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QACjD,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC,eAAe,CAC1C,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,UAAU,CAAC,YAAY,EAAE,CAAC,CAC7D,CAAC;QACF,MAAM,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC;YAC/C,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,QAAQ,GAAa;YACzB,QAAQ,EAAE;gBACR,WAAW,EAAE;oBACX,OAAO,EAAE,KAAK;iBACf;aACF;SACF,CAAC;QACF,eAAe,CAAC,GAAG,CAAC,EAAE,cAAc,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;QAC9D,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;YAC3C,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,SAAS;SAClB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,UAAU,CAAC,GAAG,EAAE;QACd,6BAA6B,EAAE,CAAC;QAChC,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;QAE7C,kCAAkC;QAClC,kBAAkB,EAAE,CAAC;QACrB,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAEzC,mCAAmC;QACnC,kBAAkB,EAAE,CAAC;QACrB,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAEzC,mCAAmC;QACnC,6BAA6B,EAAE,CAAC;QAEhC,wCAAwC;QACxC,kBAAkB,EAAE,CAAC;QACrB,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/core/auth.d.ts

@@ -0,0 +1,13 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type AuthType, type Config } from '@google/gemini-cli-core';
+/**
+ * Handles the initial authentication flow.
+ * @param config The application config.
+ * @param authType The selected auth type.
+ * @returns An error message if authentication fails, otherwise null.
+ */
+export declare function performInitialAuth(config: Config, authType: AuthType | undefined): Promise<string | null>;

package/dist/src/core/auth.js

@@ -0,0 +1,27 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { getErrorMessage, } from '@google/gemini-cli-core';
+/**
+ * Handles the initial authentication flow.
+ * @param config The application config.
+ * @param authType The selected auth type.
+ * @returns An error message if authentication fails, otherwise null.
+ */
+export async function performInitialAuth(config, authType) {
+    if (!authType) {
+        return null;
+    }
+    try {
+        await config.refreshAuth(authType);
+        // The console.log is intentionally left out here.
+        // We can add a dedicated startup message later if needed.
+    }
+    catch (e) {
+        return `Failed to login. Message: ${getErrorMessage(e)}`;
+    }
+    return null;
+}
+//# sourceMappingURL=auth.js.map

package/dist/src/core/auth.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.js","sourceRoot":"","sources":["../../../src/core/auth.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAGL,eAAe,GAChB,MAAM,yBAAyB,CAAC;AAEjC;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,MAAc,EACd,QAA8B;IAE9B,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QACnC,kDAAkD;QAClD,0DAA0D;IAC5D,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,OAAO,6BAA6B,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3D,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/src/core/auth.test.d.ts

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

package/dist/src/core/auth.test.js

@@ -0,0 +1,43 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { performInitialAuth } from './auth.js';
+import {} from '@google/gemini-cli-core';
+vi.mock('@google/gemini-cli-core', () => ({
+    AuthType: {
+        OAUTH: 'oauth',
+    },
+    getErrorMessage: (e) => e.message,
+}));
+const AuthType = {
+    OAUTH: 'oauth',
+};
+describe('auth', () => {
+    let mockConfig;
+    beforeEach(() => {
+        mockConfig = {
+            refreshAuth: vi.fn(),
+        };
+    });
+    it('should return null if authType is undefined', async () => {
+        const result = await performInitialAuth(mockConfig, undefined);
+        expect(result).toBeNull();
+        expect(mockConfig.refreshAuth).not.toHaveBeenCalled();
+    });
+    it('should return null on successful auth', async () => {
+        const result = await performInitialAuth(mockConfig, AuthType.OAUTH);
+        expect(result).toBeNull();
+        expect(mockConfig.refreshAuth).toHaveBeenCalledWith(AuthType.OAUTH);
+    });
+    it('should return error message on failed auth', async () => {
+        const error = new Error('Auth failed');
+        vi.mocked(mockConfig.refreshAuth).mockRejectedValue(error);
+        const result = await performInitialAuth(mockConfig, AuthType.OAUTH);
+        expect(result).toBe('Failed to login. Message: Auth failed');
+        expect(mockConfig.refreshAuth).toHaveBeenCalledWith(AuthType.OAUTH);
+    });
+});
+//# sourceMappingURL=auth.test.js.map

package/dist/src/core/auth.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.test.js","sourceRoot":"","sources":["../../../src/core/auth.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,kBAAkB,EAAE,MAAM,WAAW,CAAC;AAC/C,OAAO,EAAe,MAAM,yBAAyB,CAAC;AAEtD,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,QAAQ,EAAE;QACR,KAAK,EAAE,OAAO;KACf;IACD,eAAe,EAAE,CAAC,CAAU,EAAE,EAAE,CAAE,CAAW,CAAC,OAAO;CACtD,CAAC,CAAC,CAAC;AAEJ,MAAM,QAAQ,GAAG;IACf,KAAK,EAAE,OAAO;CACN,CAAC;AAEX,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;IACpB,IAAI,UAAkB,CAAC;IAEvB,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG;YACX,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;SACA,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC1B,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,UAAU,EACV,QAAQ,CAAC,KAA4D,CACtE,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC1B,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC;QACvC,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAC3D,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,UAAU,EACV,QAAQ,CAAC,KAA4D,CACtE,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;QAC7D,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}