abbyswag/qwen-code
1
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2025-12-21 09:17:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Ignore workspace settings for untrusted folders (#6606)

Browse Source
This commit is contained in:
shrutip90
2025-08-19 21:20:41 -07:00
committed by GitHub
parent 179f1414da
commit d250293c2e
16 changed files with 386 additions and 122 deletions
Download Patch File Download Diff File Expand all files Collapse all files

90
packages/cli/src/ui/App.test.tsx
View File

@@ -278,6 +278,7 @@ describe('App UI', () => {
user?: Partial<Settings>;
workspace?: Partial<Settings>;
} = {},
isTrusted = true,
): LoadedSettings => {
const systemSettingsFile: SettingsFile = {
path: '/system/settings.json',
@@ -296,6 +297,7 @@ describe('App UI', () => {
userSettingsFile,
workspaceSettingsFile,
[],
isTrusted,
);
};
@@ -377,9 +379,9 @@ describe('App UI', () => {
const { unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
@@ -403,9 +405,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
@@ -433,9 +435,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
@@ -463,9 +465,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
@@ -497,9 +499,9 @@ describe('App UI', () => {
const { unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
@@ -526,9 +528,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -545,9 +547,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -581,9 +583,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -609,9 +611,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -630,9 +632,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve(); // Wait for any async updates
@@ -651,9 +653,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -672,9 +674,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -699,9 +701,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -724,9 +726,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -745,9 +747,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -769,9 +771,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -791,9 +793,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -804,9 +806,9 @@ describe('App UI', () => {
const { unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -814,18 +816,21 @@ describe('App UI', () => {
});
it('should not display Tips component when hideTips is true', async () => {
mockSettings = createMockSettings({
workspace: {
hideTips: true,
mockSettings = createMockSettings(
{
workspace: {
hideTips: true,
},
},
});
true,
);
const { unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -837,9 +842,9 @@ describe('App UI', () => {
const { unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -855,9 +860,9 @@ describe('App UI', () => {
const { unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -868,9 +873,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -906,9 +911,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -926,9 +931,9 @@ describe('App UI', () => {
const { unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -956,9 +961,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
@@ -971,9 +976,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
@@ -986,9 +991,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
expect(lastFrame()).toMatchSnapshot();
@@ -1006,9 +1011,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
expect(lastFrame()).toMatchSnapshot();
@@ -1036,9 +1041,9 @@ describe('App UI', () => {
const { unmount, rerender } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
@@ -1046,7 +1051,6 @@ describe('App UI', () => {
rerender(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
);
@@ -1078,9 +1082,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -1104,9 +1108,9 @@ describe('App UI', () => {
const { unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
@@ -1126,9 +1130,9 @@ describe('App UI', () => {
const { unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
@@ -1146,9 +1150,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
expect(lastFrame()).toMatchSnapshot();
@@ -1172,9 +1176,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
@@ -1194,9 +1198,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -1214,9 +1218,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -1234,9 +1238,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;
await Promise.resolve();
@@ -1483,9 +1487,9 @@ describe('App UI', () => {
const { lastFrame, unmount } = renderWithProviders(
<App
config={mockConfig as unknown as ServerConfig}
settings={mockSettings}
version={mockVersion}
/>,
mockSettings,
);
currentUnmount = unmount;

34
packages/cli/src/ui/App.tsx
View File

@@ -4,7 +4,14 @@
* SPDX-License-Identifier: Apache-2.0
*/
import { useCallback, useEffect, useMemo, useState, useRef } from 'react';
import {
useCallback,
useEffect,
useMemo,
useState,
useRef,
useContext,
} from 'react';
import {
Box,
DOMElement,
@@ -41,7 +48,7 @@ import { ShellConfirmationDialog } from './components/ShellConfirmationDialog.js
import { RadioButtonSelect } from './components/shared/RadioButtonSelect.js';
import { Colors } from './colors.js';
import { loadHierarchicalGeminiMemory } from '../config/config.js';
import { LoadedSettings, SettingScope } from '../config/settings.js';
import { SettingScope } from '../config/settings.js';
import { Tips } from './components/Tips.js';
import { ConsolePatcher } from './utils/ConsolePatcher.js';
import { registerCleanup } from '../utils/cleanup.js';
@@ -100,6 +107,7 @@ import { useSettingsCommand } from './hooks/useSettingsCommand.js';
import { SettingsDialog } from './components/SettingsDialog.js';
import { setUpdateHandler } from '../utils/handleAutoUpdate.js';
import { appEvents, AppEvent } from '../utils/events.js';
import { SettingsContext } from './contexts/SettingsContext.js';
import { isNarrowWidth } from './utils/isNarrowWidth.js';
const CTRL_EXIT_PROMPT_DURATION_MS = 1000;
@@ -108,20 +116,26 @@ const MAX_DISPLAYED_QUEUED_MESSAGES = 3;
interface AppProps {
config: Config;
settings: LoadedSettings;
startupWarnings?: string[];
version: string;
}
export const AppWrapper = (props: AppProps) => {
const kittyProtocolStatus = useKittyKeyboardProtocol();
const settingsContext = useContext(SettingsContext);
if (!settingsContext) {
// This should not happen as AppWrapper is always rendered within the provider.
throw new Error('SettingsContext is not available');
}
const { settings } = settingsContext;
return (
<KeypressProvider
kittyProtocolEnabled={kittyProtocolStatus.enabled}
config={props.config}
>
<SessionStatsProvider>
<VimModeProvider settings={props.settings}>
<VimModeProvider settings={settings}>
<App {...props} />
</VimModeProvider>
</SessionStatsProvider>
@@ -129,13 +143,19 @@ export const AppWrapper = (props: AppProps) => {
);
};
const App = ({ config, settings, startupWarnings = [], version }: AppProps) => {
const App = ({ config, startupWarnings = [], version }: AppProps) => {
const isFocused = useFocus();
useBracketedPaste();
const [updateInfo, setUpdateInfo] = useState<UpdateObject | null>(null);
const { stdout } = useStdout();
const nightly = version.includes('nightly');
const { history, addItem, clearItems, loadHistory } = useHistory();
const settingsContext = useContext(SettingsContext);
if (!settingsContext) {
// This should not happen as App is always rendered within the provider.
throw new Error('SettingsContext is not available');
}
const { settings } = settingsContext;
const [idePromptAnswered, setIdePromptAnswered] = useState(false);
const currentIDE = config.getIdeClient().getCurrentIde();
@@ -262,7 +282,7 @@ const App = ({ config, settings, startupWarnings = [], version }: AppProps) => {
openThemeDialog,
handleThemeSelect,
handleThemeHighlight,
} = useThemeCommand(settings, setThemeError, addItem);
} = useThemeCommand(setThemeError, addItem);
const { isSettingsDialogOpen, openSettingsDialog, closeSettingsDialog } =
useSettingsCommand();
@@ -308,7 +328,7 @@ const App = ({ config, settings, startupWarnings = [], version }: AppProps) => {
openEditorDialog,
handleEditorSelect,
exitEditorDialog,
} = useEditorSettings(settings, setEditorError, addItem);
} = useEditorSettings(setEditorError, addItem);
const toggleCorgiMode = useCallback(() => {
setCorgiMode((prev) => !prev);

95
packages/cli/src/ui/MainComponent.tsx Normal file
View File

@@ -0,0 +1,95 @@
/**
* @license
* Copyright 2025 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import React, { useState } from 'react';
import { Config, sessionId } from '@google/gemini-cli-core';
import { loadSettings, LoadedSettings } from '../config/settings.js';
import { themeManager } from './themes/theme-manager.js';
import { SettingsContext } from './contexts/SettingsContext.js';
import { AppWrapper } from './App.js';
import { loadCliConfig, CliArgs } from '../config/config.js';
import { Extension } from '../config/extension.js';
interface MainComponentProps {
initialConfig: Config;
settings: LoadedSettings;
startupWarnings: string[];
version: string;
workspaceRoot: string;
extensions: Extension[];
argv: CliArgs;
}
export const MainComponent = ({
initialConfig,
settings,
startupWarnings,
version,
workspaceRoot,
extensions,
argv,
}: MainComponentProps) => {
const [currentSettings, setCurrentSettings] =
useState<LoadedSettings>(settings);
const [config, setConfig] = useState<Config>(initialConfig);
const recomputeSettings = () => {
const newSettings = loadSettings(workspaceRoot);
setCurrentSettings(newSettings);
};
React.useEffect(() => {
const recomputeConfigAndTheme = async () => {
// Don't run on initial mount, since the initial config is correct.
if (currentSettings === settings) {
return;
}
// Reload config
const newConfig = await loadCliConfig(
currentSettings.merged,
extensions,
sessionId,
argv,
);
await newConfig.initialize();
if (newConfig.getIdeMode()) {
await newConfig.getIdeClient().connect();
}
// Reload themes
themeManager.loadCustomThemes(currentSettings.merged.customThemes);
if (currentSettings.merged.theme) {
if (!themeManager.setActiveTheme(currentSettings.merged.theme)) {
console.warn(
`Warning: Theme "${currentSettings.merged.theme}" not found.`,
);
}
}
setConfig(newConfig);
};
recomputeConfigAndTheme();
}, [currentSettings, settings, extensions, argv, workspaceRoot]);
const contextValue = {
settings: currentSettings,
recomputeSettings,
};
return (
<React.StrictMode>
<SettingsContext.Provider value={contextValue}>
<AppWrapper
config={config}
startupWarnings={startupWarnings}
version={version}
/>
</SettingsContext.Provider>
</React.StrictMode>
);
};

24
packages/cli/src/ui/contexts/SettingsContext.ts Normal file
View File

@@ -0,0 +1,24 @@
/**
* @license
* Copyright 2025 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import { createContext, useContext } from 'react';
import { LoadedSettings } from '../../config/settings.js';
export interface SettingsContextType {
settings: LoadedSettings;
recomputeSettings: () => void;
}
// This context is initialized in gemini.tsx with the loaded settings.
export const SettingsContext = createContext<SettingsContextType | null>(null);
export function useSettings(): LoadedSettings {
const context = useContext(SettingsContext);
if (!context) {
throw new Error('useSettings must be used within a SettingsProvider');
}
return context.settings;
}

5
packages/cli/src/ui/contexts/VimModeContext.tsx
View File

@@ -12,6 +12,7 @@ import {
useState,
} from 'react';
import { LoadedSettings, SettingScope } from '../../config/settings.js';
import { SettingsContext } from './SettingsContext.js';
export type VimMode = 'NORMAL' | 'INSERT';
@@ -26,11 +27,13 @@ const VimModeContext = createContext<VimModeContextType | undefined>(undefined);
export const VimModeProvider = ({
children,
settings,
settings: initialSettings,
}: {
children: React.ReactNode;
settings: LoadedSettings;
}) => {
const settingsContext = useContext(SettingsContext);
const settings = settingsContext?.settings || initialSettings;
const initialVimEnabled = settings.merged.vimMode ?? false;
const [vimEnabled, setVimEnabled] = useState(initialVimEnabled);
const [vimMode, setVimMode] = useState<VimMode>(

8
packages/cli/src/ui/hooks/useAuthCommand.ts
View File

@@ -4,7 +4,7 @@
* SPDX-License-Identifier: Apache-2.0
*/
import { useState, useCallback, useEffect } from 'react';
import { useState, useCallback, useEffect, useContext } from 'react';
import { LoadedSettings, SettingScope } from '../../config/settings.js';
import {
AuthType,
@@ -13,6 +13,7 @@ import {
getErrorMessage,
} from '@google/gemini-cli-core';
import { runExitCleanup } from '../../utils/cleanup.js';
import { SettingsContext } from '../contexts/SettingsContext.js';
export const useAuthCommand = (
settings: LoadedSettings,
@@ -22,6 +23,7 @@ export const useAuthCommand = (
const [isAuthDialogOpen, setIsAuthDialogOpen] = useState(
settings.merged.selectedAuthType === undefined,
);
const settingsContext = useContext(SettingsContext);
const openAuthDialog = useCallback(() => {
setIsAuthDialogOpen(true);
@@ -56,7 +58,7 @@ export const useAuthCommand = (
if (authType) {
await clearCachedCredentialFile();
settings.setValue(scope, 'selectedAuthType', authType);
settingsContext?.settings.setValue(scope, 'selectedAuthType', authType);
if (
authType === AuthType.LOGIN_WITH_GOOGLE &&
config.isBrowserLaunchSuppressed()
@@ -75,7 +77,7 @@ Logging in with Google... Please restart Gemini CLI to continue.
setIsAuthDialogOpen(false);
setAuthError(null);
},
[settings, setAuthError, config],
[settingsContext, setAuthError, config],
);
const cancelAuthentication = useCallback(() => {

72
packages/cli/src/ui/hooks/useEditorSettings.test.ts → packages/cli/src/ui/hooks/useEditorSettings.test.tsx
View File

@@ -23,6 +23,8 @@ import {
checkHasEditorType,
allowEditorTypeInSandbox,
} from '@google/gemini-cli-core';
import { SettingsContext } from '../contexts/SettingsContext.js';
import { type ReactNode } from 'react';
vi.mock('@google/gemini-cli-core', async () => {
const actual = await vi.importActual('@google/gemini-cli-core');
@@ -43,13 +45,23 @@ describe('useEditorSettings', () => {
(item: Omit<HistoryItem, 'id'>, timestamp: number) => void
>;
const wrapper = ({ children }: { children: ReactNode }) => (
<SettingsContext.Provider
value={{ settings: mockLoadedSettings, recomputeSettings: () => {} }}
>
{children}
</SettingsContext.Provider>
);
beforeEach(() => {
vi.resetAllMocks();
mockLoadedSettings = {
setValue: vi.fn(),
} as unknown as LoadedSettings;
mockLoadedSettings = new LoadedSettings(
{ path: '', settings: {} },
{ path: '', settings: {} },
{ path: '', settings: {} },
[],
);
mockLoadedSettings.setValue = vi.fn();
mockSetEditorError = vi.fn();
mockAddItem = vi.fn();
@@ -63,16 +75,18 @@ describe('useEditorSettings', () => {
});
it('should initialize with dialog closed', () => {
const { result } = renderHook(() =>
useEditorSettings(mockLoadedSettings, mockSetEditorError, mockAddItem),
const { result } = renderHook(
() => useEditorSettings(mockSetEditorError, mockAddItem),
{ wrapper },
);
expect(result.current.isEditorDialogOpen).toBe(false);
});
it('should open editor dialog when openEditorDialog is called', () => {
const { result } = renderHook(() =>
useEditorSettings(mockLoadedSettings, mockSetEditorError, mockAddItem),
const { result } = renderHook(
() => useEditorSettings(mockSetEditorError, mockAddItem),
{ wrapper },
);
act(() => {
@@ -83,8 +97,9 @@ describe('useEditorSettings', () => {
});
it('should close editor dialog when exitEditorDialog is called', () => {
const { result } = renderHook(() =>
useEditorSettings(mockLoadedSettings, mockSetEditorError, mockAddItem),
const { result } = renderHook(
() => useEditorSettings(mockSetEditorError, mockAddItem),
{ wrapper },
);
act(() => {
result.current.openEditorDialog();
@@ -94,8 +109,9 @@ describe('useEditorSettings', () => {
});
it('should handle editor selection successfully', () => {
const { result } = renderHook(() =>
useEditorSettings(mockLoadedSettings, mockSetEditorError, mockAddItem),
const { result } = renderHook(
() => useEditorSettings(mockSetEditorError, mockAddItem),
{ wrapper },
);
const editorType: EditorType = 'vscode';
@@ -125,8 +141,9 @@ describe('useEditorSettings', () => {
});
it('should handle clearing editor preference (undefined editor)', () => {
const { result } = renderHook(() =>
useEditorSettings(mockLoadedSettings, mockSetEditorError, mockAddItem),
const { result } = renderHook(
() => useEditorSettings(mockSetEditorError, mockAddItem),
{ wrapper },
);
const scope = SettingScope.Workspace;
@@ -155,8 +172,9 @@ describe('useEditorSettings', () => {
});
it('should handle different editor types', () => {
const { result } = renderHook(() =>
useEditorSettings(mockLoadedSettings, mockSetEditorError, mockAddItem),
const { result } = renderHook(
() => useEditorSettings(mockSetEditorError, mockAddItem),
{ wrapper },
);
const editorTypes: EditorType[] = ['cursor', 'windsurf', 'vim'];
@@ -184,8 +202,9 @@ describe('useEditorSettings', () => {
});
it('should handle different setting scopes', () => {
const { result } = renderHook(() =>
useEditorSettings(mockLoadedSettings, mockSetEditorError, mockAddItem),
const { result } = renderHook(
() => useEditorSettings(mockSetEditorError, mockAddItem),
{ wrapper },
);
const editorType: EditorType = 'vscode';
@@ -213,8 +232,9 @@ describe('useEditorSettings', () => {
});
it('should not set preference for unavailable editors', () => {
const { result } = renderHook(() =>
useEditorSettings(mockLoadedSettings, mockSetEditorError, mockAddItem),
const { result } = renderHook(
() => useEditorSettings(mockSetEditorError, mockAddItem),
{ wrapper },
);
mockCheckHasEditorType.mockReturnValue(false);
@@ -233,8 +253,9 @@ describe('useEditorSettings', () => {
});
it('should not set preference for editors not allowed in sandbox', () => {
const { result } = renderHook(() =>
useEditorSettings(mockLoadedSettings, mockSetEditorError, mockAddItem),
const { result } = renderHook(
() => useEditorSettings(mockSetEditorError, mockAddItem),
{ wrapper },
);
mockAllowEditorTypeInSandbox.mockReturnValue(false);
@@ -253,8 +274,9 @@ describe('useEditorSettings', () => {
});
it('should handle errors during editor selection', () => {
const { result } = renderHook(() =>
useEditorSettings(mockLoadedSettings, mockSetEditorError, mockAddItem),
const { result } = renderHook(
() => useEditorSettings(mockSetEditorError, mockAddItem),
{ wrapper },
);
const errorMessage = 'Failed to save settings';

15
packages/cli/src/ui/hooks/useEditorSettings.ts
View File

@@ -4,14 +4,15 @@
* SPDX-License-Identifier: Apache-2.0
*/
import { useState, useCallback } from 'react';
import { LoadedSettings, SettingScope } from '../../config/settings.js';
import { useState, useCallback, useContext } from 'react';
import { SettingScope } from '../../config/settings.js';
import { type HistoryItem, MessageType } from '../types.js';
import {
allowEditorTypeInSandbox,
checkHasEditorType,
EditorType,
} from '@google/gemini-cli-core';
import { SettingsContext } from '../contexts/SettingsContext.js';
interface UseEditorSettingsReturn {
isEditorDialogOpen: boolean;
@@ -24,11 +25,11 @@ interface UseEditorSettingsReturn {
}
export const useEditorSettings = (
loadedSettings: LoadedSettings,
setEditorError: (error: string | null) => void,
addItem: (item: Omit<HistoryItem, 'id'>, timestamp: number) => void,
): UseEditorSettingsReturn => {
const [isEditorDialogOpen, setIsEditorDialogOpen] = useState(false);
const settingsContext = useContext(SettingsContext);
const openEditorDialog = useCallback(() => {
setIsEditorDialogOpen(true);
@@ -45,7 +46,11 @@ export const useEditorSettings = (
}
try {
loadedSettings.setValue(scope, 'preferredEditor', editorType);
settingsContext?.settings.setValue(
scope,
'preferredEditor',
editorType,
);
addItem(
{
type: MessageType.INFO,
@@ -59,7 +64,7 @@ export const useEditorSettings = (
setEditorError(`Failed to set editor preference: ${error}`);
}
},
[loadedSettings, setEditorError, addItem],
[settingsContext, setEditorError, addItem],
);
const exitEditorDialog = useCallback(() => {

7
packages/cli/src/ui/hooks/useFolderTrust.ts
View File

@@ -4,7 +4,7 @@
* SPDX-License-Identifier: Apache-2.0
*/
import { useState, useCallback, useEffect } from 'react';
import { useState, useCallback, useEffect, useContext } from 'react';
import { Settings, LoadedSettings } from '../../config/settings.js';
import { FolderTrustChoice } from '../components/FolderTrustDialog.js';
import {
@@ -13,6 +13,7 @@ import {
isWorkspaceTrusted,
} from '../../config/trustedFolders.js';
import * as process from 'process';
import { SettingsContext } from '../contexts/SettingsContext.js';
export const useFolderTrust = (
settings: LoadedSettings,
@@ -20,6 +21,7 @@ export const useFolderTrust = (
) => {
const [isTrusted, setIsTrusted] = useState<boolean | undefined>(undefined);
const [isFolderTrustDialogOpen, setIsFolderTrustDialogOpen] = useState(false);
const settingsContext = useContext(SettingsContext);
const { folderTrust, folderTrustFeature } = settings.merged;
useEffect(() => {
@@ -60,8 +62,9 @@ export const useFolderTrust = (
setIsTrusted(trusted);
setIsFolderTrustDialogOpen(false);
onTrustChange(trusted);
settingsContext?.recomputeSettings();
},
[onTrustChange, folderTrust, folderTrustFeature],
[onTrustChange, folderTrust, folderTrustFeature, settingsContext],
);
return {

10
packages/cli/src/ui/hooks/useThemeCommand.ts
View File

@@ -4,10 +4,11 @@
* SPDX-License-Identifier: Apache-2.0
*/
import { useState, useCallback, useEffect } from 'react';
import { useState, useCallback, useEffect, useContext } from 'react';
import { themeManager } from '../themes/theme-manager.js';
import { LoadedSettings, SettingScope } from '../../config/settings.js'; // Import LoadedSettings, AppSettings, MergedSetting
import { type HistoryItem, MessageType } from '../types.js';
import { HistoryItem, MessageType } from '../types.js';
import { SettingScope } from '../../config/settings.js';
import { SettingsContext } from '../contexts/SettingsContext.js';
import process from 'node:process';
interface UseThemeCommandReturn {
@@ -21,11 +22,12 @@ interface UseThemeCommandReturn {
}
export const useThemeCommand = (
loadedSettings: LoadedSettings,
setThemeError: (error: string | null) => void,
addItem: (item: Omit<HistoryItem, 'id'>, timestamp: number) => void,
): UseThemeCommandReturn => {
const [isThemeDialogOpen, setIsThemeDialogOpen] = useState(false);
const settingsContext = useContext(SettingsContext);
const loadedSettings = settingsContext!.settings;
// Check for invalid theme configuration on startup
useEffect(() => {

66
packages/cli/src/ui/utils/MarkdownDisplay.test.tsx
View File

@@ -30,7 +30,9 @@ describe('<MarkdownDisplay />', () => {
it('renders nothing for empty text', () => {
const { lastFrame } = render(
<SettingsContext.Provider value={mockSettings}>
<SettingsContext.Provider
value={{ settings: mockSettings, recomputeSettings: () => {} }}
>
<MarkdownDisplay {...baseProps} text="" />
</SettingsContext.Provider>,
);
@@ -40,7 +42,9 @@ describe('<MarkdownDisplay />', () => {
it('renders a simple paragraph', () => {
const text = 'Hello, world.';
const { lastFrame } = render(
<SettingsContext.Provider value={mockSettings}>
<SettingsContext.Provider
value={{ settings: mockSettings, recomputeSettings: () => {} }}
>
<MarkdownDisplay {...baseProps} text={text} />
</SettingsContext.Provider>,
);
@@ -55,7 +59,9 @@ describe('<MarkdownDisplay />', () => {
#### Header 4
`;
const { lastFrame } = render(
<SettingsContext.Provider value={mockSettings}>
<SettingsContext.Provider
value={{ settings: mockSettings, recomputeSettings: () => {} }}
>
<MarkdownDisplay {...baseProps} text={text} />
</SettingsContext.Provider>,
);
@@ -65,7 +71,9 @@ describe('<MarkdownDisplay />', () => {
it('renders a fenced code block with a language', () => {
const text = '```javascript\nconst x = 1;\nconsole.log(x);\n```';
const { lastFrame } = render(
<SettingsContext.Provider value={mockSettings}>
<SettingsContext.Provider
value={{ settings: mockSettings, recomputeSettings: () => {} }}
>
<MarkdownDisplay {...baseProps} text={text} />
</SettingsContext.Provider>,
);
@@ -75,7 +83,9 @@ describe('<MarkdownDisplay />', () => {
it('renders a fenced code block without a language', () => {
const text = '```\nplain text\n```';
const { lastFrame } = render(
<SettingsContext.Provider value={mockSettings}>
<SettingsContext.Provider
value={{ settings: mockSettings, recomputeSettings: () => {} }}
>
<MarkdownDisplay {...baseProps} text={text} />
</SettingsContext.Provider>,
);
@@ -85,7 +95,9 @@ describe('<MarkdownDisplay />', () => {
it('handles unclosed (pending) code blocks', () => {
const text = '```typescript\nlet y = 2;';
const { lastFrame } = render(
<SettingsContext.Provider value={mockSettings}>
<SettingsContext.Provider
value={{ settings: mockSettings, recomputeSettings: () => {} }}
>
<MarkdownDisplay {...baseProps} text={text} isPending={true} />
</SettingsContext.Provider>,
);
@@ -99,7 +111,9 @@ describe('<MarkdownDisplay />', () => {
+ item C
`;
const { lastFrame } = render(
<SettingsContext.Provider value={mockSettings}>
<SettingsContext.Provider
value={{ settings: mockSettings, recomputeSettings: () => {} }}
>
<MarkdownDisplay {...baseProps} text={text} />
</SettingsContext.Provider>,
);
@@ -113,7 +127,9 @@ describe('<MarkdownDisplay />', () => {
* Level 3
`;
const { lastFrame } = render(
<SettingsContext.Provider value={mockSettings}>
<SettingsContext.Provider
value={{ settings: mockSettings, recomputeSettings: () => {} }}
>
<MarkdownDisplay {...baseProps} text={text} />
</SettingsContext.Provider>,
);
@@ -126,7 +142,9 @@ describe('<MarkdownDisplay />', () => {
2. Second item
`;
const { lastFrame } = render(
<SettingsContext.Provider value={mockSettings}>
<SettingsContext.Provider
value={{ settings: mockSettings, recomputeSettings: () => {} }}
>
<MarkdownDisplay {...baseProps} text={text} />
</SettingsContext.Provider>,
);
@@ -142,7 +160,9 @@ World
Test
`;
const { lastFrame } = render(
<SettingsContext.Provider value={mockSettings}>
<SettingsContext.Provider
value={{ settings: mockSettings, recomputeSettings: () => {} }}
>
<MarkdownDisplay {...baseProps} text={text} />
</SettingsContext.Provider>,
);
@@ -157,7 +177,9 @@ Test
| Cell 3 | Cell 4 |
`;
const { lastFrame } = render(
<SettingsContext.Provider value={mockSettings}>
<SettingsContext.Provider
value={{ settings: mockSettings, recomputeSettings: () => {} }}
>
<MarkdownDisplay {...baseProps} text={text} />
</SettingsContext.Provider>,
);
@@ -168,10 +190,12 @@ Test
const text = `
Some text before.
| A | B |
|---|
|---|
| 1 | 2 |`;
const { lastFrame } = render(
<SettingsContext.Provider value={mockSettings}>
<SettingsContext.Provider
value={{ settings: mockSettings, recomputeSettings: () => {} }}
>
<MarkdownDisplay {...baseProps} text={text} />
</SettingsContext.Provider>,
);
@@ -183,7 +207,9 @@ Some text before.
Paragraph 2.`;
const { lastFrame } = render(
<SettingsContext.Provider value={mockSettings}>
<SettingsContext.Provider
value={{ settings: mockSettings, recomputeSettings: () => {} }}
>
<MarkdownDisplay {...baseProps} text={text} />
</SettingsContext.Provider>,
);
@@ -206,7 +232,9 @@ some code
Another paragraph.
`;
const { lastFrame } = render(
<SettingsContext.Provider value={mockSettings}>
<SettingsContext.Provider
value={{ settings: mockSettings, recomputeSettings: () => {} }}
>
<MarkdownDisplay {...baseProps} text={text} />
</SettingsContext.Provider>,
);
@@ -223,7 +251,9 @@ Another paragraph.
);
const { lastFrame } = render(
<SettingsContext.Provider value={settings}>
<SettingsContext.Provider
value={{ settings, recomputeSettings: () => {} }}
>
<MarkdownDisplay {...baseProps} text={text} />
</SettingsContext.Provider>,
);
@@ -234,7 +264,9 @@ Another paragraph.
it('shows line numbers in code blocks by default', () => {
const text = '```javascript\nconst x = 1;\n```';
const { lastFrame } = render(
<SettingsContext.Provider value={mockSettings}>
<SettingsContext.Provider
value={{ settings: mockSettings, recomputeSettings: () => {} }}
>
<MarkdownDisplay {...baseProps} text={text} />
</SettingsContext.Provider>,
);