/**
 * @license
 * Copyright 2025 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */
import { vi } from 'vitest';
import { act } from 'react';
import { renderHook } from '../../test-utils/render.js';
import { waitFor } from '../../test-utils/async.js';
import { useFolderTrust } from './useFolderTrust.js';
import { FolderTrustChoice } from '../components/FolderTrustDialog.js';
import { TrustLevel } from '../../config/trustedFolders.js';
import * as trustedFolders from '../../config/trustedFolders.js';
import { coreEvents } from '@google/gemini-cli-core';
const mockedCwd = vi.hoisted(() => vi.fn());
const mockedExit = vi.hoisted(() => vi.fn());
vi.mock('node:process', async () => {
    const actual = await vi.importActual('node:process');
    return {
        ...actual,
        cwd: mockedCwd,
        exit: mockedExit,
        platform: 'linux',
    };
});
describe('useFolderTrust', () => {
    let mockSettings;
    let mockTrustedFolders;
    let isWorkspaceTrustedSpy;
    let onTrustChange;
    let addItem;
    beforeEach(() => {
        vi.useFakeTimers();
        mockSettings = {
            merged: {
                security: {
                    folderTrust: {
                        enabled: true,
                    },
                },
            },
            setValue: vi.fn(),
        };
        mockTrustedFolders = {
            setValue: vi.fn(),
        };
        vi.spyOn(trustedFolders, 'loadTrustedFolders').mockReturnValue(mockTrustedFolders);
        isWorkspaceTrustedSpy = vi.spyOn(trustedFolders, 'isWorkspaceTrusted');
        mockedCwd.mockReturnValue('/test/path');
        onTrustChange = vi.fn();
        addItem = vi.fn();
    });
    afterEach(() => {
        vi.useRealTimers();
        vi.clearAllMocks();
    });
    it('should not open dialog when folder is already trusted', () => {
        isWorkspaceTrustedSpy.mockReturnValue({ isTrusted: true, source: 'file' });
        const { result } = renderHook(() => useFolderTrust(mockSettings, onTrustChange, addItem));
        expect(result.current.isFolderTrustDialogOpen).toBe(false);
        expect(onTrustChange).toHaveBeenCalledWith(true);
    });
    it('should not open dialog when folder is already untrusted', () => {
        isWorkspaceTrustedSpy.mockReturnValue({ isTrusted: false, source: 'file' });
        const { result } = renderHook(() => useFolderTrust(mockSettings, onTrustChange, addItem));
        expect(result.current.isFolderTrustDialogOpen).toBe(false);
        expect(onTrustChange).toHaveBeenCalledWith(false);
    });
    it('should open dialog when folder trust is undefined', async () => {
        isWorkspaceTrustedSpy.mockReturnValue({
            isTrusted: undefined,
            source: undefined,
        });
        const { result } = renderHook(() => useFolderTrust(mockSettings, onTrustChange, addItem));
        await waitFor(() => {
            expect(result.current.isFolderTrustDialogOpen).toBe(true);
        });
        expect(onTrustChange).toHaveBeenCalledWith(undefined);
    });
    it('should send a message if the folder is untrusted', () => {
        isWorkspaceTrustedSpy.mockReturnValue({ isTrusted: false, source: 'file' });
        renderHook(() => useFolderTrust(mockSettings, onTrustChange, addItem));
        expect(addItem).toHaveBeenCalledWith({
            text: 'This folder is not trusted. Some features may be disabled. Use the `/permissions` command to change the trust level.',
            type: 'info',
        }, expect.any(Number));
    });
    it('should not send a message if the folder is trusted', () => {
        isWorkspaceTrustedSpy.mockReturnValue({ isTrusted: true, source: 'file' });
        renderHook(() => useFolderTrust(mockSettings, onTrustChange, addItem));
        expect(addItem).not.toHaveBeenCalled();
    });
    it('should handle TRUST_FOLDER choice', async () => {
        isWorkspaceTrustedSpy.mockReturnValue({
            isTrusted: undefined,
            source: undefined,
        });
        mockTrustedFolders.setValue.mockImplementation(() => {
            isWorkspaceTrustedSpy.mockReturnValue({
                isTrusted: true,
                source: 'file',
            });
        });
        const { result } = renderHook(() => useFolderTrust(mockSettings, onTrustChange, addItem));
        await waitFor(() => {
            expect(result.current.isTrusted).toBeUndefined();
        });
        await act(async () => {
            await result.current.handleFolderTrustSelect(FolderTrustChoice.TRUST_FOLDER);
        });
        await waitFor(() => {
            expect(mockTrustedFolders.setValue).toHaveBeenCalledWith('/test/path', TrustLevel.TRUST_FOLDER);
            expect(result.current.isFolderTrustDialogOpen).toBe(false);
            expect(onTrustChange).toHaveBeenLastCalledWith(true);
        });
    });
    it('should handle TRUST_PARENT choice', () => {
        isWorkspaceTrustedSpy.mockReturnValue({
            isTrusted: undefined,
            source: undefined,
        });
        const { result } = renderHook(() => useFolderTrust(mockSettings, onTrustChange, addItem));
        act(() => {
            result.current.handleFolderTrustSelect(FolderTrustChoice.TRUST_PARENT);
        });
        expect(mockTrustedFolders.setValue).toHaveBeenCalledWith('/test/path', TrustLevel.TRUST_PARENT);
        expect(result.current.isFolderTrustDialogOpen).toBe(false);
        expect(onTrustChange).toHaveBeenLastCalledWith(true);
    });
    it('should handle DO_NOT_TRUST choice and trigger restart', () => {
        isWorkspaceTrustedSpy.mockReturnValue({
            isTrusted: undefined,
            source: undefined,
        });
        const { result } = renderHook(() => useFolderTrust(mockSettings, onTrustChange, addItem));
        act(() => {
            result.current.handleFolderTrustSelect(FolderTrustChoice.DO_NOT_TRUST);
        });
        expect(mockTrustedFolders.setValue).toHaveBeenCalledWith('/test/path', TrustLevel.DO_NOT_TRUST);
        expect(onTrustChange).toHaveBeenLastCalledWith(false);
        expect(result.current.isRestarting).toBe(true);
        expect(result.current.isFolderTrustDialogOpen).toBe(true);
    });
    it('should do nothing for default choice', async () => {
        isWorkspaceTrustedSpy.mockReturnValue({
            isTrusted: undefined,
            source: undefined,
        });
        const { result } = renderHook(() => useFolderTrust(mockSettings, onTrustChange, addItem));
        act(() => {
            result.current.handleFolderTrustSelect('invalid_choice');
        });
        await waitFor(() => {
            expect(mockTrustedFolders.setValue).not.toHaveBeenCalled();
            expect(mockSettings.setValue).not.toHaveBeenCalled();
            expect(result.current.isFolderTrustDialogOpen).toBe(true);
            expect(onTrustChange).toHaveBeenCalledWith(undefined);
        });
    });
    it('should set isRestarting to true when trust status changes from false to true', async () => {
        isWorkspaceTrustedSpy.mockReturnValue({ isTrusted: false, source: 'file' }); // Initially untrusted
        mockTrustedFolders.setValue.mockImplementation(() => {
            isWorkspaceTrustedSpy.mockReturnValue({
                isTrusted: true,
                source: 'file',
            });
        });
        const { result } = renderHook(() => useFolderTrust(mockSettings, onTrustChange, addItem));
        await waitFor(() => {
            expect(result.current.isTrusted).toBe(false);
        });
        act(() => {
            result.current.handleFolderTrustSelect(FolderTrustChoice.TRUST_FOLDER);
        });
        await waitFor(() => {
            expect(result.current.isRestarting).toBe(true);
            expect(result.current.isFolderTrustDialogOpen).toBe(true); // Dialog should stay open
        });
    });
    it('should not set isRestarting to true when trust status does not change', () => {
        isWorkspaceTrustedSpy.mockReturnValue({
            isTrusted: undefined,
            source: undefined,
        });
        const { result } = renderHook(() => useFolderTrust(mockSettings, onTrustChange, addItem));
        act(() => {
            result.current.handleFolderTrustSelect(FolderTrustChoice.TRUST_FOLDER);
        });
        expect(result.current.isRestarting).toBe(false);
        expect(result.current.isFolderTrustDialogOpen).toBe(false); // Dialog should close
    });
    it('should emit feedback on failure to set value', () => {
        isWorkspaceTrustedSpy.mockReturnValue({
            isTrusted: undefined,
            source: undefined,
        });
        mockTrustedFolders.setValue.mockImplementation(() => {
            throw new Error('test error');
        });
        const emitFeedbackSpy = vi.spyOn(coreEvents, 'emitFeedback');
        const { result } = renderHook(() => useFolderTrust(mockSettings, onTrustChange, addItem));
        act(() => {
            result.current.handleFolderTrustSelect(FolderTrustChoice.TRUST_FOLDER);
        });
        vi.runAllTimers();
        expect(emitFeedbackSpy).toHaveBeenCalledWith('error', 'Failed to save trust settings. Exiting Gemini CLI.');
        expect(mockedExit).toHaveBeenCalledWith(1);
    });
});
//# sourceMappingURL=useFolderTrust.test.js.map