mirror of
https://github.com/google-gemini/gemini-cli.git
synced 2026-03-10 22:21:22 -07:00
2409 lines
73 KiB
TypeScript
2409 lines
73 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2026 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
import {
|
|
useMemo,
|
|
useState,
|
|
useCallback,
|
|
useEffect,
|
|
useRef,
|
|
useLayoutEffect,
|
|
} from 'react';
|
|
import {
|
|
type DOMElement,
|
|
measureElement,
|
|
useApp,
|
|
useStdout,
|
|
useStdin,
|
|
type AppProps,
|
|
} from 'ink';
|
|
import { App } from './App.js';
|
|
import { AppContext } from './contexts/AppContext.js';
|
|
import { UIStateContext, type UIState } from './contexts/UIStateContext.js';
|
|
import {
|
|
UIActionsContext,
|
|
type UIActions,
|
|
} from './contexts/UIActionsContext.js';
|
|
import { ConfigContext } from './contexts/ConfigContext.js';
|
|
import {
|
|
type HistoryItem,
|
|
ToolCallStatus,
|
|
type HistoryItemWithoutId,
|
|
type HistoryItemToolGroup,
|
|
AuthState,
|
|
type ConfirmationRequest,
|
|
type PermissionConfirmationRequest,
|
|
type QuotaStats,
|
|
} from './types.js';
|
|
import { checkPermissions } from './hooks/atCommandProcessor.js';
|
|
import { MessageType, StreamingState } from './types.js';
|
|
import { ToolActionsProvider } from './contexts/ToolActionsContext.js';
|
|
import {
|
|
type EditorType,
|
|
type Config,
|
|
type IdeInfo,
|
|
type IdeContext,
|
|
type UserTierId,
|
|
type UserFeedbackPayload,
|
|
type AgentDefinition,
|
|
type ApprovalMode,
|
|
IdeClient,
|
|
ideContextStore,
|
|
getErrorMessage,
|
|
getAllGeminiMdFilenames,
|
|
AuthType,
|
|
clearCachedCredentialFile,
|
|
type ResumedSessionData,
|
|
recordExitFail,
|
|
ShellExecutionService,
|
|
saveApiKey,
|
|
debugLogger,
|
|
coreEvents,
|
|
CoreEvent,
|
|
refreshServerHierarchicalMemory,
|
|
flattenMemory,
|
|
type MemoryChangedPayload,
|
|
writeToStdout,
|
|
disableMouseEvents,
|
|
enterAlternateScreen,
|
|
enableMouseEvents,
|
|
disableLineWrapping,
|
|
shouldEnterAlternateScreen,
|
|
startupProfiler,
|
|
SessionStartSource,
|
|
SessionEndReason,
|
|
generateSummary,
|
|
type ConsentRequestPayload,
|
|
type AgentsDiscoveredPayload,
|
|
ChangeAuthRequestedError,
|
|
} from '@google/gemini-cli-core';
|
|
import { validateAuthMethod } from '../config/auth.js';
|
|
import process from 'node:process';
|
|
import { useHistory } from './hooks/useHistoryManager.js';
|
|
import { useMemoryMonitor } from './hooks/useMemoryMonitor.js';
|
|
import { useThemeCommand } from './hooks/useThemeCommand.js';
|
|
import { useAuthCommand } from './auth/useAuth.js';
|
|
import { useQuotaAndFallback } from './hooks/useQuotaAndFallback.js';
|
|
import { useEditorSettings } from './hooks/useEditorSettings.js';
|
|
import { useSettingsCommand } from './hooks/useSettingsCommand.js';
|
|
import { useModelCommand } from './hooks/useModelCommand.js';
|
|
import { useSlashCommandProcessor } from './hooks/slashCommandProcessor.js';
|
|
import { useVimMode } from './contexts/VimModeContext.js';
|
|
import { useConsoleMessages } from './hooks/useConsoleMessages.js';
|
|
import { useTerminalSize } from './hooks/useTerminalSize.js';
|
|
import { calculatePromptWidths } from './components/InputPrompt.js';
|
|
import { calculateMainAreaWidth } from './utils/ui-sizing.js';
|
|
import ansiEscapes from 'ansi-escapes';
|
|
import { basename } from 'node:path';
|
|
import { computeTerminalTitle } from '../utils/windowTitle.js';
|
|
import { useTextBuffer } from './components/shared/text-buffer.js';
|
|
import { useLogger } from './hooks/useLogger.js';
|
|
import { useGeminiStream } from './hooks/useGeminiStream.js';
|
|
import { type BackgroundShell } from './hooks/shellCommandProcessor.js';
|
|
import { useVim } from './hooks/vim.js';
|
|
import { type LoadableSettingScope, SettingScope } from '../config/settings.js';
|
|
import { type InitializationResult } from '../core/initializer.js';
|
|
import { useFocus } from './hooks/useFocus.js';
|
|
import { useKeypress, type Key } from './hooks/useKeypress.js';
|
|
import { KeypressPriority } from './contexts/KeypressContext.js';
|
|
import { keyMatchers, Command } from './keyMatchers.js';
|
|
import { useLoadingIndicator } from './hooks/useLoadingIndicator.js';
|
|
import { useShellInactivityStatus } from './hooks/useShellInactivityStatus.js';
|
|
import { useFolderTrust } from './hooks/useFolderTrust.js';
|
|
import { useIdeTrustListener } from './hooks/useIdeTrustListener.js';
|
|
import { type IdeIntegrationNudgeResult } from './IdeIntegrationNudge.js';
|
|
import { appEvents, AppEvent, TransientMessageType } from '../utils/events.js';
|
|
import { type UpdateObject } from './utils/updateCheck.js';
|
|
import { setUpdateHandler } from '../utils/handleAutoUpdate.js';
|
|
import { registerCleanup, runExitCleanup } from '../utils/cleanup.js';
|
|
import { RELAUNCH_EXIT_CODE } from '../utils/processUtils.js';
|
|
import type { SessionInfo } from '../utils/sessionUtils.js';
|
|
import { useMessageQueue } from './hooks/useMessageQueue.js';
|
|
import { useMcpStatus } from './hooks/useMcpStatus.js';
|
|
import { useApprovalModeIndicator } from './hooks/useApprovalModeIndicator.js';
|
|
import { useSessionStats } from './contexts/SessionContext.js';
|
|
import { useGitBranchName } from './hooks/useGitBranchName.js';
|
|
import {
|
|
useConfirmUpdateRequests,
|
|
useExtensionUpdates,
|
|
} from './hooks/useExtensionUpdates.js';
|
|
import { ShellFocusContext } from './contexts/ShellFocusContext.js';
|
|
import { type ExtensionManager } from '../config/extension-manager.js';
|
|
import { requestConsentInteractive } from '../config/extensions/consent.js';
|
|
import { useSessionBrowser } from './hooks/useSessionBrowser.js';
|
|
import { persistentState } from '../utils/persistentState.js';
|
|
import { useSessionResume } from './hooks/useSessionResume.js';
|
|
import { useIncludeDirsTrust } from './hooks/useIncludeDirsTrust.js';
|
|
import { isWorkspaceTrusted } from '../config/trustedFolders.js';
|
|
import { useAlternateBuffer } from './hooks/useAlternateBuffer.js';
|
|
import { useSettings } from './contexts/SettingsContext.js';
|
|
import { terminalCapabilityManager } from './utils/terminalCapabilityManager.js';
|
|
import { useInputHistoryStore } from './hooks/useInputHistoryStore.js';
|
|
import { useBanner } from './hooks/useBanner.js';
|
|
import { useHookDisplayState } from './hooks/useHookDisplayState.js';
|
|
import { useBackgroundShellManager } from './hooks/useBackgroundShellManager.js';
|
|
import {
|
|
WARNING_PROMPT_DURATION_MS,
|
|
QUEUE_ERROR_DISPLAY_DURATION_MS,
|
|
} from './constants.js';
|
|
import { LoginWithGoogleRestartDialog } from './auth/LoginWithGoogleRestartDialog.js';
|
|
import { NewAgentsChoice } from './components/NewAgentsNotification.js';
|
|
import { isSlashCommand } from './utils/commandUtils.js';
|
|
import { useTerminalTheme } from './hooks/useTerminalTheme.js';
|
|
import { useTimedMessage } from './hooks/useTimedMessage.js';
|
|
import { shouldDismissShortcutsHelpOnHotkey } from './utils/shortcutsHelp.js';
|
|
import { useSuspend } from './hooks/useSuspend.js';
|
|
|
|
function isToolExecuting(pendingHistoryItems: HistoryItemWithoutId[]) {
|
|
return pendingHistoryItems.some((item) => {
|
|
if (item && item.type === 'tool_group') {
|
|
return item.tools.some(
|
|
(tool) => ToolCallStatus.Executing === tool.status,
|
|
);
|
|
}
|
|
return false;
|
|
});
|
|
}
|
|
|
|
function isToolAwaitingConfirmation(
|
|
pendingHistoryItems: HistoryItemWithoutId[],
|
|
) {
|
|
return pendingHistoryItems
|
|
.filter((item): item is HistoryItemToolGroup => item.type === 'tool_group')
|
|
.some((item) =>
|
|
item.tools.some((tool) => ToolCallStatus.Confirming === tool.status),
|
|
);
|
|
}
|
|
|
|
interface AppContainerProps {
|
|
config: Config;
|
|
startupWarnings?: string[];
|
|
version: string;
|
|
initializationResult: InitializationResult;
|
|
resumedSessionData?: ResumedSessionData;
|
|
}
|
|
|
|
const APPROVAL_MODE_REVEAL_DURATION_MS = 1200;
|
|
const FOCUS_UI_ENABLED_STATE_KEY = 'focusUiEnabled';
|
|
|
|
/**
|
|
* The fraction of the terminal width to allocate to the shell.
|
|
* This provides horizontal padding.
|
|
*/
|
|
const SHELL_WIDTH_FRACTION = 0.89;
|
|
|
|
/**
|
|
* The number of lines to subtract from the available terminal height
|
|
* for the shell. This provides vertical padding and space for other UI elements.
|
|
*/
|
|
const SHELL_HEIGHT_PADDING = 10;
|
|
|
|
export const AppContainer = (props: AppContainerProps) => {
|
|
const { config, initializationResult, resumedSessionData } = props;
|
|
const settings = useSettings();
|
|
|
|
const historyManager = useHistory({
|
|
chatRecordingService: config.getGeminiClient()?.getChatRecordingService(),
|
|
});
|
|
|
|
useMemoryMonitor(historyManager);
|
|
const isAlternateBuffer = useAlternateBuffer();
|
|
const [corgiMode, setCorgiMode] = useState(false);
|
|
const [forceRerenderKey, setForceRerenderKey] = useState(0);
|
|
const [debugMessage, setDebugMessage] = useState<string>('');
|
|
const [quittingMessages, setQuittingMessages] = useState<
|
|
HistoryItem[] | null
|
|
>(null);
|
|
const [showPrivacyNotice, setShowPrivacyNotice] = useState<boolean>(false);
|
|
const [themeError, setThemeError] = useState<string | null>(
|
|
initializationResult.themeError,
|
|
);
|
|
const [isProcessing, setIsProcessing] = useState<boolean>(false);
|
|
const [embeddedShellFocused, setEmbeddedShellFocused] = useState(false);
|
|
const [showDebugProfiler, setShowDebugProfiler] = useState(false);
|
|
const [customDialog, setCustomDialog] = useState<React.ReactNode | null>(
|
|
null,
|
|
);
|
|
const [copyModeEnabled, setCopyModeEnabled] = useState(false);
|
|
const [pendingRestorePrompt, setPendingRestorePrompt] = useState(false);
|
|
const toggleBackgroundShellRef = useRef<() => void>(() => {});
|
|
const isBackgroundShellVisibleRef = useRef<boolean>(false);
|
|
const backgroundShellsRef = useRef<Map<number, BackgroundShell>>(new Map());
|
|
|
|
const [adminSettingsChanged, setAdminSettingsChanged] = useState(false);
|
|
|
|
const [shellModeActive, setShellModeActive] = useState(false);
|
|
const [modelSwitchedFromQuotaError, setModelSwitchedFromQuotaError] =
|
|
useState<boolean>(false);
|
|
const [historyRemountKey, setHistoryRemountKey] = useState(0);
|
|
const [settingsNonce, setSettingsNonce] = useState(0);
|
|
const activeHooks = useHookDisplayState();
|
|
const [updateInfo, setUpdateInfo] = useState<UpdateObject | null>(null);
|
|
const [isTrustedFolder, setIsTrustedFolder] = useState<boolean | undefined>(
|
|
() => isWorkspaceTrusted(settings.merged).isTrusted,
|
|
);
|
|
|
|
const [queueErrorMessage, setQueueErrorMessage] = useState<string | null>(
|
|
null,
|
|
);
|
|
|
|
const [newAgents, setNewAgents] = useState<AgentDefinition[] | null>(null);
|
|
|
|
const [defaultBannerText, setDefaultBannerText] = useState('');
|
|
const [warningBannerText, setWarningBannerText] = useState('');
|
|
const [bannerVisible, setBannerVisible] = useState(true);
|
|
|
|
const bannerData = useMemo(
|
|
() => ({
|
|
defaultText: defaultBannerText,
|
|
warningText: warningBannerText,
|
|
}),
|
|
[defaultBannerText, warningBannerText],
|
|
);
|
|
|
|
const { bannerText } = useBanner(bannerData);
|
|
|
|
// eslint-disable-next-line @typescript-eslint/no-unsafe-type-assertion
|
|
const extensionManager = config.getExtensionLoader() as ExtensionManager;
|
|
// We are in the interactive CLI, update how we request consent and settings.
|
|
extensionManager.setRequestConsent((description) =>
|
|
requestConsentInteractive(description, addConfirmUpdateExtensionRequest),
|
|
);
|
|
extensionManager.setRequestSetting();
|
|
|
|
const { addConfirmUpdateExtensionRequest, confirmUpdateExtensionRequests } =
|
|
useConfirmUpdateRequests();
|
|
const {
|
|
extensionsUpdateState,
|
|
extensionsUpdateStateInternal,
|
|
dispatchExtensionStateUpdate,
|
|
} = useExtensionUpdates(
|
|
extensionManager,
|
|
historyManager.addItem,
|
|
config.getEnableExtensionReloading(),
|
|
);
|
|
|
|
const [isPermissionsDialogOpen, setPermissionsDialogOpen] = useState(false);
|
|
const [permissionsDialogProps, setPermissionsDialogProps] = useState<{
|
|
targetDirectory?: string;
|
|
} | null>(null);
|
|
const openPermissionsDialog = useCallback(
|
|
(props?: { targetDirectory?: string }) => {
|
|
setPermissionsDialogOpen(true);
|
|
setPermissionsDialogProps(props ?? null);
|
|
},
|
|
[],
|
|
);
|
|
const closePermissionsDialog = useCallback(() => {
|
|
setPermissionsDialogOpen(false);
|
|
setPermissionsDialogProps(null);
|
|
}, []);
|
|
|
|
const [isAgentConfigDialogOpen, setIsAgentConfigDialogOpen] = useState(false);
|
|
const [selectedAgentName, setSelectedAgentName] = useState<
|
|
string | undefined
|
|
>();
|
|
const [selectedAgentDisplayName, setSelectedAgentDisplayName] = useState<
|
|
string | undefined
|
|
>();
|
|
const [selectedAgentDefinition, setSelectedAgentDefinition] = useState<
|
|
AgentDefinition | undefined
|
|
>();
|
|
|
|
const openAgentConfigDialog = useCallback(
|
|
(name: string, displayName: string, definition: AgentDefinition) => {
|
|
setSelectedAgentName(name);
|
|
setSelectedAgentDisplayName(displayName);
|
|
setSelectedAgentDefinition(definition);
|
|
setIsAgentConfigDialogOpen(true);
|
|
},
|
|
[],
|
|
);
|
|
|
|
const closeAgentConfigDialog = useCallback(() => {
|
|
setIsAgentConfigDialogOpen(false);
|
|
setSelectedAgentName(undefined);
|
|
setSelectedAgentDisplayName(undefined);
|
|
setSelectedAgentDefinition(undefined);
|
|
}, []);
|
|
|
|
const toggleDebugProfiler = useCallback(
|
|
() => setShowDebugProfiler((prev) => !prev),
|
|
[],
|
|
);
|
|
|
|
const [currentModel, setCurrentModel] = useState(config.getModel());
|
|
|
|
const [userTier, setUserTier] = useState<UserTierId | undefined>(undefined);
|
|
const [quotaStats, setQuotaStats] = useState<QuotaStats | undefined>(() => {
|
|
const remaining = config.getQuotaRemaining();
|
|
const limit = config.getQuotaLimit();
|
|
const resetTime = config.getQuotaResetTime();
|
|
return remaining !== undefined ||
|
|
limit !== undefined ||
|
|
resetTime !== undefined
|
|
? { remaining, limit, resetTime }
|
|
: undefined;
|
|
});
|
|
|
|
const [isConfigInitialized, setConfigInitialized] = useState(false);
|
|
|
|
const logger = useLogger(config.storage);
|
|
const { inputHistory, addInput, initializeFromLogger } =
|
|
useInputHistoryStore();
|
|
|
|
// Terminal and layout hooks
|
|
const { columns: terminalWidth, rows: terminalHeight } = useTerminalSize();
|
|
const { stdin, setRawMode } = useStdin();
|
|
const { stdout } = useStdout();
|
|
const app: AppProps = useApp();
|
|
|
|
// Additional hooks moved from App.tsx
|
|
const { stats: sessionStats } = useSessionStats();
|
|
const branchName = useGitBranchName(config.getTargetDir());
|
|
|
|
// Layout measurements
|
|
const mainControlsRef = useRef<DOMElement>(null);
|
|
// For performance profiling only
|
|
const rootUiRef = useRef<DOMElement>(null);
|
|
const lastTitleRef = useRef<string | null>(null);
|
|
const staticExtraHeight = 3;
|
|
|
|
useEffect(() => {
|
|
// eslint-disable-next-line @typescript-eslint/no-floating-promises
|
|
(async () => {
|
|
// Note: the program will not work if this fails so let errors be
|
|
// handled by the global catch.
|
|
if (!config.isInitialized()) {
|
|
await config.initialize();
|
|
}
|
|
setConfigInitialized(true);
|
|
startupProfiler.flush(config);
|
|
|
|
const sessionStartSource = resumedSessionData
|
|
? SessionStartSource.Resume
|
|
: SessionStartSource.Startup;
|
|
const result = await config
|
|
.getHookSystem()
|
|
?.fireSessionStartEvent(sessionStartSource);
|
|
|
|
if (result) {
|
|
if (result.systemMessage) {
|
|
historyManager.addItem(
|
|
{
|
|
type: MessageType.INFO,
|
|
text: result.systemMessage,
|
|
},
|
|
Date.now(),
|
|
);
|
|
}
|
|
|
|
const additionalContext = result.getAdditionalContext();
|
|
const geminiClient = config.getGeminiClient();
|
|
if (additionalContext && geminiClient) {
|
|
await geminiClient.addHistory({
|
|
role: 'user',
|
|
parts: [
|
|
{ text: `<hook_context>${additionalContext}</hook_context>` },
|
|
],
|
|
});
|
|
}
|
|
}
|
|
|
|
// Fire-and-forget: generate summary for previous session in background
|
|
generateSummary(config).catch((e) => {
|
|
debugLogger.warn('Background summary generation failed:', e);
|
|
});
|
|
})();
|
|
registerCleanup(async () => {
|
|
// Turn off mouse scroll.
|
|
disableMouseEvents();
|
|
|
|
// Kill all background shells
|
|
for (const pid of backgroundShellsRef.current.keys()) {
|
|
ShellExecutionService.kill(pid);
|
|
}
|
|
|
|
const ideClient = await IdeClient.getInstance();
|
|
await ideClient.disconnect();
|
|
|
|
// Fire SessionEnd hook on cleanup (only if hooks are enabled)
|
|
await config?.getHookSystem()?.fireSessionEndEvent(SessionEndReason.Exit);
|
|
});
|
|
// Disable the dependencies check here. historyManager gets flagged
|
|
// but we don't want to react to changes to it because each new history
|
|
// item, including the ones from the start session hook will cause a
|
|
// re-render and an error when we try to reload config.
|
|
//
|
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
|
}, [config, resumedSessionData]);
|
|
|
|
useEffect(
|
|
() => setUpdateHandler(historyManager.addItem, setUpdateInfo),
|
|
[historyManager.addItem],
|
|
);
|
|
|
|
// Subscribe to fallback mode and model changes from core
|
|
useEffect(() => {
|
|
const handleModelChanged = () => {
|
|
setCurrentModel(config.getModel());
|
|
};
|
|
|
|
const handleQuotaChanged = (payload: {
|
|
remaining: number | undefined;
|
|
limit: number | undefined;
|
|
resetTime?: string;
|
|
}) => {
|
|
setQuotaStats({
|
|
remaining: payload.remaining,
|
|
limit: payload.limit,
|
|
resetTime: payload.resetTime,
|
|
});
|
|
};
|
|
|
|
coreEvents.on(CoreEvent.ModelChanged, handleModelChanged);
|
|
coreEvents.on(CoreEvent.QuotaChanged, handleQuotaChanged);
|
|
return () => {
|
|
coreEvents.off(CoreEvent.ModelChanged, handleModelChanged);
|
|
coreEvents.off(CoreEvent.QuotaChanged, handleQuotaChanged);
|
|
};
|
|
}, [config]);
|
|
|
|
useEffect(() => {
|
|
const handleSettingsChanged = () => {
|
|
setSettingsNonce((prev) => prev + 1);
|
|
};
|
|
|
|
const handleAdminSettingsChanged = () => {
|
|
setAdminSettingsChanged(true);
|
|
};
|
|
|
|
const handleAgentsDiscovered = (payload: AgentsDiscoveredPayload) => {
|
|
setNewAgents(payload.agents);
|
|
};
|
|
|
|
coreEvents.on(CoreEvent.SettingsChanged, handleSettingsChanged);
|
|
coreEvents.on(CoreEvent.AdminSettingsChanged, handleAdminSettingsChanged);
|
|
coreEvents.on(CoreEvent.AgentsDiscovered, handleAgentsDiscovered);
|
|
return () => {
|
|
coreEvents.off(CoreEvent.SettingsChanged, handleSettingsChanged);
|
|
coreEvents.off(
|
|
CoreEvent.AdminSettingsChanged,
|
|
handleAdminSettingsChanged,
|
|
);
|
|
coreEvents.off(CoreEvent.AgentsDiscovered, handleAgentsDiscovered);
|
|
};
|
|
}, [settings]);
|
|
|
|
const { consoleMessages, clearConsoleMessages: clearConsoleMessagesState } =
|
|
useConsoleMessages();
|
|
|
|
const mainAreaWidth = calculateMainAreaWidth(terminalWidth, settings);
|
|
// Derive widths for InputPrompt using shared helper
|
|
const { inputWidth, suggestionsWidth } = useMemo(() => {
|
|
const { inputWidth, suggestionsWidth } =
|
|
calculatePromptWidths(mainAreaWidth);
|
|
return { inputWidth, suggestionsWidth };
|
|
}, [mainAreaWidth]);
|
|
|
|
const staticAreaMaxItemHeight = Math.max(terminalHeight * 4, 100);
|
|
|
|
const getPreferredEditor = useCallback(
|
|
// eslint-disable-next-line @typescript-eslint/no-unsafe-type-assertion
|
|
() => settings.merged.general.preferredEditor as EditorType,
|
|
[settings.merged.general.preferredEditor],
|
|
);
|
|
|
|
const buffer = useTextBuffer({
|
|
initialText: '',
|
|
viewport: { height: 10, width: inputWidth },
|
|
stdin,
|
|
setRawMode,
|
|
escapePastedPaths: true,
|
|
shellModeActive,
|
|
getPreferredEditor,
|
|
});
|
|
const bufferRef = useRef(buffer);
|
|
useEffect(() => {
|
|
bufferRef.current = buffer;
|
|
}, [buffer]);
|
|
|
|
const stableSetText = useCallback((text: string) => {
|
|
bufferRef.current.setText(text);
|
|
}, []);
|
|
|
|
// Initialize input history from logger (past sessions)
|
|
useEffect(() => {
|
|
// eslint-disable-next-line @typescript-eslint/no-floating-promises
|
|
initializeFromLogger(logger);
|
|
}, [logger, initializeFromLogger]);
|
|
|
|
const refreshStatic = useCallback(() => {
|
|
if (!isAlternateBuffer) {
|
|
stdout.write(ansiEscapes.clearTerminal);
|
|
}
|
|
setHistoryRemountKey((prev) => prev + 1);
|
|
}, [setHistoryRemountKey, isAlternateBuffer, stdout]);
|
|
|
|
const shouldUseAlternateScreen = shouldEnterAlternateScreen(
|
|
isAlternateBuffer,
|
|
config.getScreenReader(),
|
|
);
|
|
|
|
const handleEditorClose = useCallback(() => {
|
|
if (shouldUseAlternateScreen) {
|
|
// The editor may have exited alternate buffer mode so we need to
|
|
// enter it again to be safe.
|
|
enterAlternateScreen();
|
|
enableMouseEvents();
|
|
disableLineWrapping();
|
|
app.rerender();
|
|
}
|
|
terminalCapabilityManager.enableSupportedModes();
|
|
refreshStatic();
|
|
}, [refreshStatic, shouldUseAlternateScreen, app]);
|
|
|
|
const [editorError, setEditorError] = useState<string | null>(null);
|
|
const {
|
|
isEditorDialogOpen,
|
|
openEditorDialog,
|
|
handleEditorSelect,
|
|
exitEditorDialog,
|
|
} = useEditorSettings(settings, setEditorError, historyManager.addItem);
|
|
|
|
useEffect(() => {
|
|
coreEvents.on(CoreEvent.ExternalEditorClosed, handleEditorClose);
|
|
coreEvents.on(CoreEvent.RequestEditorSelection, openEditorDialog);
|
|
return () => {
|
|
coreEvents.off(CoreEvent.ExternalEditorClosed, handleEditorClose);
|
|
coreEvents.off(CoreEvent.RequestEditorSelection, openEditorDialog);
|
|
};
|
|
}, [handleEditorClose, openEditorDialog]);
|
|
|
|
useEffect(() => {
|
|
if (
|
|
!(settings.merged.ui.hideBanner || config.getScreenReader()) &&
|
|
bannerVisible &&
|
|
bannerText
|
|
) {
|
|
// The header should show a banner but the Header is rendered in static
|
|
// so we must trigger a static refresh for it to be visible.
|
|
refreshStatic();
|
|
}
|
|
}, [bannerVisible, bannerText, settings, config, refreshStatic]);
|
|
|
|
const { isSettingsDialogOpen, openSettingsDialog, closeSettingsDialog } =
|
|
useSettingsCommand();
|
|
|
|
const {
|
|
isThemeDialogOpen,
|
|
openThemeDialog,
|
|
closeThemeDialog,
|
|
handleThemeSelect,
|
|
handleThemeHighlight,
|
|
} = useThemeCommand(
|
|
settings,
|
|
setThemeError,
|
|
historyManager.addItem,
|
|
initializationResult.themeError,
|
|
);
|
|
|
|
// Poll for terminal background color changes to auto-switch theme
|
|
useTerminalTheme(handleThemeSelect, config, refreshStatic);
|
|
|
|
const {
|
|
authState,
|
|
setAuthState,
|
|
authError,
|
|
onAuthError,
|
|
apiKeyDefaultValue,
|
|
reloadApiKey,
|
|
} = useAuthCommand(settings, config, initializationResult.authError);
|
|
const [authContext, setAuthContext] = useState<{ requiresRestart?: boolean }>(
|
|
{},
|
|
);
|
|
|
|
useEffect(() => {
|
|
if (authState === AuthState.Authenticated && authContext.requiresRestart) {
|
|
setAuthState(AuthState.AwaitingGoogleLoginRestart);
|
|
setAuthContext({});
|
|
}
|
|
}, [authState, authContext, setAuthState]);
|
|
|
|
const {
|
|
proQuotaRequest,
|
|
handleProQuotaChoice,
|
|
validationRequest,
|
|
handleValidationChoice,
|
|
} = useQuotaAndFallback({
|
|
config,
|
|
historyManager,
|
|
userTier,
|
|
setModelSwitchedFromQuotaError,
|
|
onShowAuthSelection: () => setAuthState(AuthState.Updating),
|
|
});
|
|
|
|
// Derive auth state variables for backward compatibility with UIStateContext
|
|
const isAuthDialogOpen = authState === AuthState.Updating;
|
|
const isAuthenticating = authState === AuthState.Unauthenticated;
|
|
|
|
// Session browser and resume functionality
|
|
const isGeminiClientInitialized = config.getGeminiClient()?.isInitialized();
|
|
|
|
const { loadHistoryForResume, isResuming } = useSessionResume({
|
|
config,
|
|
historyManager,
|
|
refreshStatic,
|
|
isGeminiClientInitialized,
|
|
setQuittingMessages,
|
|
resumedSessionData,
|
|
isAuthenticating,
|
|
});
|
|
const {
|
|
isSessionBrowserOpen,
|
|
openSessionBrowser,
|
|
closeSessionBrowser,
|
|
handleResumeSession,
|
|
handleDeleteSession: handleDeleteSessionSync,
|
|
} = useSessionBrowser(config, loadHistoryForResume);
|
|
// Wrap handleDeleteSession to return a Promise for UIActions interface
|
|
const handleDeleteSession = useCallback(
|
|
async (session: SessionInfo): Promise<void> => {
|
|
handleDeleteSessionSync(session);
|
|
},
|
|
[handleDeleteSessionSync],
|
|
);
|
|
|
|
// Create handleAuthSelect wrapper for backward compatibility
|
|
const handleAuthSelect = useCallback(
|
|
async (authType: AuthType | undefined, scope: LoadableSettingScope) => {
|
|
if (authType) {
|
|
if (authType === AuthType.LOGIN_WITH_GOOGLE) {
|
|
setAuthContext({ requiresRestart: true });
|
|
} else {
|
|
setAuthContext({});
|
|
}
|
|
await clearCachedCredentialFile();
|
|
settings.setValue(scope, 'security.auth.selectedType', authType);
|
|
|
|
try {
|
|
await config.refreshAuth(authType);
|
|
setAuthState(AuthState.Authenticated);
|
|
} catch (e) {
|
|
if (e instanceof ChangeAuthRequestedError) {
|
|
return;
|
|
}
|
|
onAuthError(
|
|
`Failed to authenticate: ${e instanceof Error ? e.message : String(e)}`,
|
|
);
|
|
return;
|
|
}
|
|
|
|
if (
|
|
authType === AuthType.LOGIN_WITH_GOOGLE &&
|
|
config.isBrowserLaunchSuppressed()
|
|
) {
|
|
await runExitCleanup();
|
|
writeToStdout(`
|
|
----------------------------------------------------------------
|
|
Logging in with Google... Restarting Gemini CLI to continue.
|
|
----------------------------------------------------------------
|
|
`);
|
|
process.exit(RELAUNCH_EXIT_CODE);
|
|
}
|
|
}
|
|
setAuthState(AuthState.Authenticated);
|
|
},
|
|
[settings, config, setAuthState, onAuthError, setAuthContext],
|
|
);
|
|
|
|
const handleApiKeySubmit = useCallback(
|
|
async (apiKey: string) => {
|
|
try {
|
|
onAuthError(null);
|
|
if (!apiKey.trim() && apiKey.length > 1) {
|
|
onAuthError(
|
|
'API key cannot be empty string with length greater than 1.',
|
|
);
|
|
return;
|
|
}
|
|
|
|
await saveApiKey(apiKey);
|
|
await reloadApiKey();
|
|
await config.refreshAuth(AuthType.USE_GEMINI);
|
|
setAuthState(AuthState.Authenticated);
|
|
} catch (e) {
|
|
onAuthError(
|
|
`Failed to save API key: ${e instanceof Error ? e.message : String(e)}`,
|
|
);
|
|
}
|
|
},
|
|
[setAuthState, onAuthError, reloadApiKey, config],
|
|
);
|
|
|
|
const handleApiKeyCancel = useCallback(() => {
|
|
// Go back to auth method selection
|
|
setAuthState(AuthState.Updating);
|
|
}, [setAuthState]);
|
|
|
|
// Sync user tier from config when authentication changes
|
|
useEffect(() => {
|
|
// Only sync when not currently authenticating
|
|
if (authState === AuthState.Authenticated) {
|
|
setUserTier(config.getUserTier());
|
|
}
|
|
}, [config, authState]);
|
|
|
|
// Check for enforced auth type mismatch
|
|
useEffect(() => {
|
|
if (
|
|
settings.merged.security.auth.enforcedType &&
|
|
settings.merged.security.auth.selectedType &&
|
|
settings.merged.security.auth.enforcedType !==
|
|
settings.merged.security.auth.selectedType
|
|
) {
|
|
onAuthError(
|
|
`Authentication is enforced to be ${settings.merged.security.auth.enforcedType}, but you are currently using ${settings.merged.security.auth.selectedType}.`,
|
|
);
|
|
} else if (
|
|
settings.merged.security.auth.selectedType &&
|
|
!settings.merged.security.auth.useExternal
|
|
) {
|
|
// We skip validation for Gemini API key here because it might be stored
|
|
// in the keychain, which we can't check synchronously.
|
|
// The useAuth hook handles validation for this case.
|
|
if (settings.merged.security.auth.selectedType === AuthType.USE_GEMINI) {
|
|
return;
|
|
}
|
|
|
|
const error = validateAuthMethod(
|
|
settings.merged.security.auth.selectedType,
|
|
);
|
|
if (error) {
|
|
onAuthError(error);
|
|
}
|
|
}
|
|
}, [
|
|
settings.merged.security.auth.selectedType,
|
|
settings.merged.security.auth.enforcedType,
|
|
settings.merged.security.auth.useExternal,
|
|
onAuthError,
|
|
]);
|
|
|
|
const { isModelDialogOpen, openModelDialog, closeModelDialog } =
|
|
useModelCommand();
|
|
|
|
const { toggleVimEnabled } = useVimMode();
|
|
|
|
const setIsBackgroundShellListOpenRef = useRef<(open: boolean) => void>(
|
|
() => {},
|
|
);
|
|
const [focusUiEnabledByDefault] = useState(
|
|
() => persistentState.get(FOCUS_UI_ENABLED_STATE_KEY) === true,
|
|
);
|
|
const [shortcutsHelpVisible, setShortcutsHelpVisible] = useState(false);
|
|
const [cleanUiDetailsVisible, setCleanUiDetailsVisibleState] = useState(
|
|
!focusUiEnabledByDefault,
|
|
);
|
|
const modeRevealTimeoutRef = useRef<NodeJS.Timeout | null>(null);
|
|
const cleanUiDetailsPinnedRef = useRef(!focusUiEnabledByDefault);
|
|
|
|
const clearModeRevealTimeout = useCallback(() => {
|
|
if (modeRevealTimeoutRef.current) {
|
|
clearTimeout(modeRevealTimeoutRef.current);
|
|
modeRevealTimeoutRef.current = null;
|
|
}
|
|
}, []);
|
|
|
|
const persistFocusUiPreference = useCallback((isFullUiVisible: boolean) => {
|
|
persistentState.set(FOCUS_UI_ENABLED_STATE_KEY, !isFullUiVisible);
|
|
}, []);
|
|
|
|
const setCleanUiDetailsVisible = useCallback(
|
|
(visible: boolean) => {
|
|
clearModeRevealTimeout();
|
|
cleanUiDetailsPinnedRef.current = visible;
|
|
setCleanUiDetailsVisibleState(visible);
|
|
persistFocusUiPreference(visible);
|
|
},
|
|
[clearModeRevealTimeout, persistFocusUiPreference],
|
|
);
|
|
|
|
const toggleCleanUiDetailsVisible = useCallback(() => {
|
|
clearModeRevealTimeout();
|
|
setCleanUiDetailsVisibleState((visible) => {
|
|
const nextVisible = !visible;
|
|
cleanUiDetailsPinnedRef.current = nextVisible;
|
|
persistFocusUiPreference(nextVisible);
|
|
return nextVisible;
|
|
});
|
|
}, [clearModeRevealTimeout, persistFocusUiPreference]);
|
|
|
|
const revealCleanUiDetailsTemporarily = useCallback(
|
|
(durationMs: number = APPROVAL_MODE_REVEAL_DURATION_MS) => {
|
|
if (cleanUiDetailsPinnedRef.current) {
|
|
return;
|
|
}
|
|
clearModeRevealTimeout();
|
|
setCleanUiDetailsVisibleState(true);
|
|
modeRevealTimeoutRef.current = setTimeout(() => {
|
|
if (!cleanUiDetailsPinnedRef.current) {
|
|
setCleanUiDetailsVisibleState(false);
|
|
}
|
|
modeRevealTimeoutRef.current = null;
|
|
}, durationMs);
|
|
},
|
|
[clearModeRevealTimeout],
|
|
);
|
|
|
|
useEffect(() => () => clearModeRevealTimeout(), [clearModeRevealTimeout]);
|
|
|
|
const slashCommandActions = useMemo(
|
|
() => ({
|
|
openAuthDialog: () => setAuthState(AuthState.Updating),
|
|
openThemeDialog,
|
|
openEditorDialog,
|
|
openPrivacyNotice: () => setShowPrivacyNotice(true),
|
|
openSettingsDialog,
|
|
openSessionBrowser,
|
|
openModelDialog,
|
|
openAgentConfigDialog,
|
|
openPermissionsDialog,
|
|
quit: (messages: HistoryItem[]) => {
|
|
setQuittingMessages(messages);
|
|
setTimeout(async () => {
|
|
await runExitCleanup();
|
|
process.exit(0);
|
|
}, 100);
|
|
},
|
|
setDebugMessage,
|
|
toggleCorgiMode: () => setCorgiMode((prev) => !prev),
|
|
toggleDebugProfiler,
|
|
dispatchExtensionStateUpdate,
|
|
addConfirmUpdateExtensionRequest,
|
|
toggleBackgroundShell: () => {
|
|
toggleBackgroundShellRef.current();
|
|
if (!isBackgroundShellVisibleRef.current) {
|
|
setEmbeddedShellFocused(true);
|
|
if (backgroundShellsRef.current.size > 1) {
|
|
setIsBackgroundShellListOpenRef.current(true);
|
|
} else {
|
|
setIsBackgroundShellListOpenRef.current(false);
|
|
}
|
|
}
|
|
},
|
|
toggleShortcutsHelp: () => setShortcutsHelpVisible((visible) => !visible),
|
|
setText: stableSetText,
|
|
}),
|
|
[
|
|
setAuthState,
|
|
openThemeDialog,
|
|
openEditorDialog,
|
|
openSettingsDialog,
|
|
openSessionBrowser,
|
|
openModelDialog,
|
|
openAgentConfigDialog,
|
|
setQuittingMessages,
|
|
setDebugMessage,
|
|
setShowPrivacyNotice,
|
|
setCorgiMode,
|
|
dispatchExtensionStateUpdate,
|
|
openPermissionsDialog,
|
|
addConfirmUpdateExtensionRequest,
|
|
toggleDebugProfiler,
|
|
setShortcutsHelpVisible,
|
|
stableSetText,
|
|
],
|
|
);
|
|
|
|
const {
|
|
handleSlashCommand,
|
|
slashCommands,
|
|
pendingHistoryItems: pendingSlashCommandHistoryItems,
|
|
commandContext,
|
|
confirmationRequest: commandConfirmationRequest,
|
|
} = useSlashCommandProcessor(
|
|
config,
|
|
settings,
|
|
historyManager.addItem,
|
|
historyManager.clearItems,
|
|
historyManager.loadHistory,
|
|
refreshStatic,
|
|
toggleVimEnabled,
|
|
setIsProcessing,
|
|
slashCommandActions,
|
|
extensionsUpdateStateInternal,
|
|
isConfigInitialized,
|
|
setBannerVisible,
|
|
setCustomDialog,
|
|
);
|
|
|
|
const [authConsentRequest, setAuthConsentRequest] =
|
|
useState<ConfirmationRequest | null>(null);
|
|
const [permissionConfirmationRequest, setPermissionConfirmationRequest] =
|
|
useState<PermissionConfirmationRequest | null>(null);
|
|
|
|
useEffect(() => {
|
|
const handleConsentRequest = (payload: ConsentRequestPayload) => {
|
|
setAuthConsentRequest({
|
|
prompt: payload.prompt,
|
|
onConfirm: (confirmed: boolean) => {
|
|
setAuthConsentRequest(null);
|
|
payload.onConfirm(confirmed);
|
|
},
|
|
});
|
|
};
|
|
|
|
coreEvents.on(CoreEvent.ConsentRequest, handleConsentRequest);
|
|
return () => {
|
|
coreEvents.off(CoreEvent.ConsentRequest, handleConsentRequest);
|
|
};
|
|
}, []);
|
|
|
|
const performMemoryRefresh = useCallback(async () => {
|
|
historyManager.addItem(
|
|
{
|
|
type: MessageType.INFO,
|
|
text: 'Refreshing hierarchical memory (GEMINI.md or other context files)...',
|
|
},
|
|
Date.now(),
|
|
);
|
|
try {
|
|
const { memoryContent, fileCount } =
|
|
await refreshServerHierarchicalMemory(config);
|
|
|
|
const flattenedMemory = flattenMemory(memoryContent);
|
|
|
|
historyManager.addItem(
|
|
{
|
|
type: MessageType.INFO,
|
|
text: `Memory refreshed successfully. ${
|
|
flattenedMemory.length > 0
|
|
? `Loaded ${flattenedMemory.length} characters from ${fileCount} file(s).`
|
|
: 'No memory content found.'
|
|
}`,
|
|
},
|
|
Date.now(),
|
|
);
|
|
if (config.getDebugMode()) {
|
|
debugLogger.log(
|
|
`[DEBUG] Refreshed memory content in config: ${flattenedMemory.substring(
|
|
0,
|
|
200,
|
|
)}...`,
|
|
);
|
|
}
|
|
} catch (error) {
|
|
const errorMessage = getErrorMessage(error);
|
|
historyManager.addItem(
|
|
{
|
|
type: MessageType.ERROR,
|
|
text: `Error refreshing memory: ${errorMessage}`,
|
|
},
|
|
Date.now(),
|
|
);
|
|
debugLogger.warn('Error refreshing memory:', error);
|
|
}
|
|
}, [config, historyManager]);
|
|
|
|
const cancelHandlerRef = useRef<(shouldRestorePrompt?: boolean) => void>(
|
|
() => {},
|
|
);
|
|
|
|
const onCancelSubmit = useCallback((shouldRestorePrompt?: boolean) => {
|
|
if (shouldRestorePrompt) {
|
|
setPendingRestorePrompt(true);
|
|
} else {
|
|
setPendingRestorePrompt(false);
|
|
cancelHandlerRef.current(false);
|
|
}
|
|
}, []);
|
|
|
|
useEffect(() => {
|
|
if (pendingRestorePrompt) {
|
|
const lastHistoryUserMsg = historyManager.history.findLast(
|
|
(h) => h.type === 'user',
|
|
);
|
|
const lastUserMsg = inputHistory.at(-1);
|
|
|
|
if (
|
|
!lastHistoryUserMsg ||
|
|
(typeof lastHistoryUserMsg.text === 'string' &&
|
|
lastHistoryUserMsg.text === lastUserMsg)
|
|
) {
|
|
cancelHandlerRef.current(true);
|
|
setPendingRestorePrompt(false);
|
|
}
|
|
}
|
|
}, [pendingRestorePrompt, inputHistory, historyManager.history]);
|
|
|
|
const {
|
|
streamingState,
|
|
submitQuery,
|
|
initError,
|
|
pendingHistoryItems: pendingGeminiHistoryItems,
|
|
thought,
|
|
cancelOngoingRequest,
|
|
pendingToolCalls,
|
|
handleApprovalModeChange,
|
|
activePtyId,
|
|
loopDetectionConfirmationRequest,
|
|
lastOutputTime,
|
|
backgroundShellCount,
|
|
isBackgroundShellVisible,
|
|
toggleBackgroundShell,
|
|
backgroundCurrentShell,
|
|
backgroundShells,
|
|
dismissBackgroundShell,
|
|
retryStatus,
|
|
} = useGeminiStream(
|
|
config.getGeminiClient(),
|
|
historyManager.history,
|
|
historyManager.addItem,
|
|
config,
|
|
settings,
|
|
setDebugMessage,
|
|
handleSlashCommand,
|
|
shellModeActive,
|
|
getPreferredEditor,
|
|
onAuthError,
|
|
performMemoryRefresh,
|
|
modelSwitchedFromQuotaError,
|
|
setModelSwitchedFromQuotaError,
|
|
onCancelSubmit,
|
|
setEmbeddedShellFocused,
|
|
terminalWidth,
|
|
terminalHeight,
|
|
embeddedShellFocused,
|
|
);
|
|
|
|
toggleBackgroundShellRef.current = toggleBackgroundShell;
|
|
isBackgroundShellVisibleRef.current = isBackgroundShellVisible;
|
|
backgroundShellsRef.current = backgroundShells;
|
|
|
|
const {
|
|
activeBackgroundShellPid,
|
|
setIsBackgroundShellListOpen,
|
|
isBackgroundShellListOpen,
|
|
setActiveBackgroundShellPid,
|
|
backgroundShellHeight,
|
|
} = useBackgroundShellManager({
|
|
backgroundShells,
|
|
backgroundShellCount,
|
|
isBackgroundShellVisible,
|
|
activePtyId,
|
|
embeddedShellFocused,
|
|
setEmbeddedShellFocused,
|
|
terminalHeight,
|
|
});
|
|
|
|
setIsBackgroundShellListOpenRef.current = setIsBackgroundShellListOpen;
|
|
|
|
const lastOutputTimeRef = useRef(0);
|
|
|
|
useEffect(() => {
|
|
lastOutputTimeRef.current = lastOutputTime;
|
|
}, [lastOutputTime]);
|
|
|
|
const { shouldShowFocusHint, inactivityStatus } = useShellInactivityStatus({
|
|
activePtyId,
|
|
lastOutputTime,
|
|
streamingState,
|
|
pendingToolCalls,
|
|
embeddedShellFocused,
|
|
isInteractiveShellEnabled: config.isInteractiveShellEnabled(),
|
|
});
|
|
|
|
const shouldShowActionRequiredTitle = inactivityStatus === 'action_required';
|
|
const shouldShowSilentWorkingTitle = inactivityStatus === 'silent_working';
|
|
|
|
const handleApprovalModeChangeWithUiReveal = useCallback(
|
|
(mode: ApprovalMode) => {
|
|
void handleApprovalModeChange(mode);
|
|
if (!cleanUiDetailsVisible) {
|
|
revealCleanUiDetailsTemporarily(APPROVAL_MODE_REVEAL_DURATION_MS);
|
|
}
|
|
},
|
|
[
|
|
handleApprovalModeChange,
|
|
cleanUiDetailsVisible,
|
|
revealCleanUiDetailsTemporarily,
|
|
],
|
|
);
|
|
|
|
// Auto-accept indicator
|
|
const showApprovalModeIndicator = useApprovalModeIndicator({
|
|
config,
|
|
addItem: historyManager.addItem,
|
|
onApprovalModeChange: handleApprovalModeChangeWithUiReveal,
|
|
isActive: !embeddedShellFocused,
|
|
});
|
|
|
|
const { isMcpReady } = useMcpStatus(config);
|
|
|
|
const {
|
|
messageQueue,
|
|
addMessage,
|
|
clearQueue,
|
|
getQueuedMessagesText,
|
|
popAllMessages,
|
|
} = useMessageQueue({
|
|
isConfigInitialized,
|
|
streamingState,
|
|
submitQuery,
|
|
isMcpReady,
|
|
});
|
|
|
|
cancelHandlerRef.current = useCallback(
|
|
(shouldRestorePrompt: boolean = true) => {
|
|
const pendingHistoryItems = [
|
|
...pendingSlashCommandHistoryItems,
|
|
...pendingGeminiHistoryItems,
|
|
];
|
|
if (isToolAwaitingConfirmation(pendingHistoryItems)) {
|
|
return; // Don't clear - user may be composing a follow-up message
|
|
}
|
|
if (isToolExecuting(pendingHistoryItems)) {
|
|
buffer.setText(''); // Clear for Ctrl+C cancellation
|
|
return;
|
|
}
|
|
|
|
const lastUserMessage = inputHistory.at(-1);
|
|
let textToSet = shouldRestorePrompt ? lastUserMessage || '' : '';
|
|
|
|
const queuedText = getQueuedMessagesText();
|
|
if (queuedText) {
|
|
textToSet = textToSet ? `${textToSet}\n\n${queuedText}` : queuedText;
|
|
clearQueue();
|
|
}
|
|
|
|
if (textToSet || !shouldRestorePrompt) {
|
|
buffer.setText(textToSet);
|
|
}
|
|
},
|
|
[
|
|
buffer,
|
|
inputHistory,
|
|
getQueuedMessagesText,
|
|
clearQueue,
|
|
pendingSlashCommandHistoryItems,
|
|
pendingGeminiHistoryItems,
|
|
],
|
|
);
|
|
|
|
const handleFinalSubmit = useCallback(
|
|
async (submittedValue: string) => {
|
|
const isSlash = isSlashCommand(submittedValue.trim());
|
|
const isIdle = streamingState === StreamingState.Idle;
|
|
|
|
if (isSlash || (isIdle && isMcpReady)) {
|
|
if (!isSlash) {
|
|
const permissions = await checkPermissions(submittedValue, config);
|
|
if (permissions.length > 0) {
|
|
setPermissionConfirmationRequest({
|
|
files: permissions,
|
|
onComplete: (result) => {
|
|
setPermissionConfirmationRequest(null);
|
|
if (result.allowed) {
|
|
permissions.forEach((p) =>
|
|
config.getWorkspaceContext().addReadOnlyPath(p),
|
|
);
|
|
}
|
|
void submitQuery(submittedValue);
|
|
},
|
|
});
|
|
addInput(submittedValue);
|
|
return;
|
|
}
|
|
}
|
|
void submitQuery(submittedValue);
|
|
} else {
|
|
// Check messageQueue.length === 0 to only notify on the first queued item
|
|
if (isIdle && !isMcpReady && messageQueue.length === 0) {
|
|
coreEvents.emitFeedback(
|
|
'info',
|
|
'Waiting for MCP servers to initialize... Slash commands are still available and prompts will be queued.',
|
|
);
|
|
}
|
|
addMessage(submittedValue);
|
|
}
|
|
addInput(submittedValue); // Track input for up-arrow history
|
|
},
|
|
[
|
|
addMessage,
|
|
addInput,
|
|
submitQuery,
|
|
isMcpReady,
|
|
streamingState,
|
|
messageQueue.length,
|
|
config,
|
|
],
|
|
);
|
|
|
|
const handleClearScreen = useCallback(() => {
|
|
historyManager.clearItems();
|
|
clearConsoleMessagesState();
|
|
refreshStatic();
|
|
}, [historyManager, clearConsoleMessagesState, refreshStatic]);
|
|
|
|
const { handleInput: vimHandleInput } = useVim(buffer, handleFinalSubmit);
|
|
|
|
/**
|
|
* Determines if the input prompt should be active and accept user input.
|
|
* Input is disabled during:
|
|
* - Initialization errors
|
|
* - Slash command processing
|
|
* - Tool confirmations (WaitingForConfirmation state)
|
|
* - Any future streaming states not explicitly allowed
|
|
*/
|
|
const isInputActive =
|
|
isConfigInitialized &&
|
|
!initError &&
|
|
!isProcessing &&
|
|
!isResuming &&
|
|
!!slashCommands &&
|
|
(streamingState === StreamingState.Idle ||
|
|
streamingState === StreamingState.Responding) &&
|
|
!proQuotaRequest;
|
|
|
|
const [controlsHeight, setControlsHeight] = useState(0);
|
|
|
|
useLayoutEffect(() => {
|
|
if (mainControlsRef.current) {
|
|
const fullFooterMeasurement = measureElement(mainControlsRef.current);
|
|
const roundedHeight = Math.round(fullFooterMeasurement.height);
|
|
if (roundedHeight > 0 && roundedHeight !== controlsHeight) {
|
|
setControlsHeight(roundedHeight);
|
|
}
|
|
}
|
|
}, [buffer, terminalWidth, terminalHeight, controlsHeight]);
|
|
|
|
// Compute available terminal height based on controls measurement
|
|
const availableTerminalHeight = Math.max(
|
|
0,
|
|
terminalHeight -
|
|
controlsHeight -
|
|
staticExtraHeight -
|
|
2 -
|
|
backgroundShellHeight,
|
|
);
|
|
|
|
config.setShellExecutionConfig({
|
|
terminalWidth: Math.floor(terminalWidth * SHELL_WIDTH_FRACTION),
|
|
terminalHeight: Math.max(
|
|
Math.floor(availableTerminalHeight - SHELL_HEIGHT_PADDING),
|
|
1,
|
|
),
|
|
pager: settings.merged.tools.shell.pager,
|
|
showColor: settings.merged.tools.shell.showColor,
|
|
sanitizationConfig: config.sanitizationConfig,
|
|
});
|
|
|
|
const isFocused = useFocus();
|
|
|
|
// Context file names computation
|
|
const contextFileNames = useMemo(() => {
|
|
const fromSettings = settings.merged.context.fileName;
|
|
return fromSettings
|
|
? Array.isArray(fromSettings)
|
|
? fromSettings
|
|
: [fromSettings]
|
|
: getAllGeminiMdFilenames();
|
|
}, [settings.merged.context.fileName]);
|
|
// Initial prompt handling
|
|
const initialPrompt = useMemo(() => config.getQuestion(), [config]);
|
|
const initialPromptSubmitted = useRef(false);
|
|
const geminiClient = config.getGeminiClient();
|
|
|
|
useEffect(() => {
|
|
if (activePtyId) {
|
|
try {
|
|
ShellExecutionService.resizePty(
|
|
activePtyId,
|
|
Math.floor(terminalWidth * SHELL_WIDTH_FRACTION),
|
|
Math.max(
|
|
Math.floor(availableTerminalHeight - SHELL_HEIGHT_PADDING),
|
|
1,
|
|
),
|
|
);
|
|
} catch (e) {
|
|
// This can happen in a race condition where the pty exits
|
|
// right before we try to resize it.
|
|
if (
|
|
!(
|
|
e instanceof Error &&
|
|
e.message.includes('Cannot resize a pty that has already exited')
|
|
)
|
|
) {
|
|
throw e;
|
|
}
|
|
}
|
|
}
|
|
}, [terminalWidth, availableTerminalHeight, activePtyId]);
|
|
|
|
useEffect(() => {
|
|
if (
|
|
initialPrompt &&
|
|
isConfigInitialized &&
|
|
!initialPromptSubmitted.current &&
|
|
!isAuthenticating &&
|
|
!isAuthDialogOpen &&
|
|
!isThemeDialogOpen &&
|
|
!isEditorDialogOpen &&
|
|
!showPrivacyNotice &&
|
|
geminiClient?.isInitialized?.()
|
|
) {
|
|
void handleFinalSubmit(initialPrompt);
|
|
initialPromptSubmitted.current = true;
|
|
}
|
|
}, [
|
|
initialPrompt,
|
|
isConfigInitialized,
|
|
handleFinalSubmit,
|
|
isAuthenticating,
|
|
isAuthDialogOpen,
|
|
isThemeDialogOpen,
|
|
isEditorDialogOpen,
|
|
showPrivacyNotice,
|
|
geminiClient,
|
|
]);
|
|
|
|
const [idePromptAnswered, setIdePromptAnswered] = useState(false);
|
|
const [currentIDE, setCurrentIDE] = useState<IdeInfo | null>(null);
|
|
|
|
useEffect(() => {
|
|
const getIde = async () => {
|
|
const ideClient = await IdeClient.getInstance();
|
|
const currentIde = ideClient.getCurrentIde();
|
|
setCurrentIDE(currentIde || null);
|
|
};
|
|
// eslint-disable-next-line @typescript-eslint/no-floating-promises
|
|
getIde();
|
|
}, []);
|
|
const shouldShowIdePrompt = Boolean(
|
|
currentIDE &&
|
|
!config.getIdeMode() &&
|
|
!settings.merged.ide.hasSeenNudge &&
|
|
!idePromptAnswered,
|
|
);
|
|
|
|
const [showErrorDetails, setShowErrorDetails] = useState<boolean>(false);
|
|
const [showFullTodos, setShowFullTodos] = useState<boolean>(false);
|
|
const [renderMarkdown, setRenderMarkdown] = useState<boolean>(true);
|
|
|
|
const [ctrlCPressCount, setCtrlCPressCount] = useState(0);
|
|
const ctrlCTimerRef = useRef<NodeJS.Timeout | null>(null);
|
|
const [ctrlDPressCount, setCtrlDPressCount] = useState(0);
|
|
const ctrlDTimerRef = useRef<NodeJS.Timeout | null>(null);
|
|
const [constrainHeight, setConstrainHeight] = useState<boolean>(true);
|
|
const [ideContextState, setIdeContextState] = useState<
|
|
IdeContext | undefined
|
|
>();
|
|
const [showEscapePrompt, setShowEscapePrompt] = useState(false);
|
|
const [showIdeRestartPrompt, setShowIdeRestartPrompt] = useState(false);
|
|
|
|
const [transientMessage, showTransientMessage] = useTimedMessage<{
|
|
text: string;
|
|
type: TransientMessageType;
|
|
}>(WARNING_PROMPT_DURATION_MS);
|
|
|
|
const { isFolderTrustDialogOpen, handleFolderTrustSelect, isRestarting } =
|
|
useFolderTrust(settings, setIsTrustedFolder, historyManager.addItem);
|
|
const {
|
|
needsRestart: ideNeedsRestart,
|
|
restartReason: ideTrustRestartReason,
|
|
} = useIdeTrustListener();
|
|
const isInitialMount = useRef(true);
|
|
|
|
useIncludeDirsTrust(config, isTrustedFolder, historyManager, setCustomDialog);
|
|
|
|
const tabFocusTimeoutRef = useRef<NodeJS.Timeout | null>(null);
|
|
|
|
useEffect(() => {
|
|
const handleTransientMessage = (payload: {
|
|
message: string;
|
|
type: TransientMessageType;
|
|
}) => {
|
|
showTransientMessage({ text: payload.message, type: payload.type });
|
|
};
|
|
|
|
const handleSelectionWarning = () => {
|
|
showTransientMessage({
|
|
text: 'Press Ctrl-S to enter selection mode to copy text.',
|
|
type: TransientMessageType.Warning,
|
|
});
|
|
};
|
|
const handlePasteTimeout = () => {
|
|
showTransientMessage({
|
|
text: 'Paste Timed out. Possibly due to slow connection.',
|
|
type: TransientMessageType.Warning,
|
|
});
|
|
};
|
|
|
|
appEvents.on(AppEvent.TransientMessage, handleTransientMessage);
|
|
appEvents.on(AppEvent.SelectionWarning, handleSelectionWarning);
|
|
appEvents.on(AppEvent.PasteTimeout, handlePasteTimeout);
|
|
|
|
return () => {
|
|
appEvents.off(AppEvent.TransientMessage, handleTransientMessage);
|
|
appEvents.off(AppEvent.SelectionWarning, handleSelectionWarning);
|
|
appEvents.off(AppEvent.PasteTimeout, handlePasteTimeout);
|
|
if (tabFocusTimeoutRef.current) {
|
|
clearTimeout(tabFocusTimeoutRef.current);
|
|
}
|
|
if (modeRevealTimeoutRef.current) {
|
|
clearTimeout(modeRevealTimeoutRef.current);
|
|
}
|
|
};
|
|
}, [showTransientMessage]);
|
|
|
|
const handleWarning = useCallback(
|
|
(message: string) => {
|
|
showTransientMessage({
|
|
text: message,
|
|
type: TransientMessageType.Warning,
|
|
});
|
|
},
|
|
[showTransientMessage],
|
|
);
|
|
|
|
const { handleSuspend } = useSuspend({
|
|
handleWarning,
|
|
setRawMode,
|
|
refreshStatic,
|
|
setForceRerenderKey,
|
|
shouldUseAlternateScreen,
|
|
});
|
|
|
|
useEffect(() => {
|
|
if (ideNeedsRestart) {
|
|
// IDE trust changed, force a restart.
|
|
setShowIdeRestartPrompt(true);
|
|
}
|
|
}, [ideNeedsRestart]);
|
|
|
|
useEffect(() => {
|
|
if (queueErrorMessage) {
|
|
const timer = setTimeout(() => {
|
|
setQueueErrorMessage(null);
|
|
}, QUEUE_ERROR_DISPLAY_DURATION_MS);
|
|
|
|
return () => clearTimeout(timer);
|
|
}
|
|
return undefined;
|
|
}, [queueErrorMessage, setQueueErrorMessage]);
|
|
|
|
useEffect(() => {
|
|
if (isInitialMount.current) {
|
|
isInitialMount.current = false;
|
|
return;
|
|
}
|
|
|
|
const handler = setTimeout(() => {
|
|
refreshStatic();
|
|
}, 300);
|
|
|
|
return () => {
|
|
clearTimeout(handler);
|
|
};
|
|
}, [terminalWidth, refreshStatic]);
|
|
|
|
useEffect(() => {
|
|
const unsubscribe = ideContextStore.subscribe(setIdeContextState);
|
|
setIdeContextState(ideContextStore.get());
|
|
return unsubscribe;
|
|
}, []);
|
|
|
|
useEffect(() => {
|
|
const openDebugConsole = () => {
|
|
setShowErrorDetails(true);
|
|
setConstrainHeight(false);
|
|
};
|
|
appEvents.on(AppEvent.OpenDebugConsole, openDebugConsole);
|
|
|
|
return () => {
|
|
appEvents.off(AppEvent.OpenDebugConsole, openDebugConsole);
|
|
};
|
|
}, [config]);
|
|
|
|
useEffect(() => {
|
|
if (ctrlCTimerRef.current) {
|
|
clearTimeout(ctrlCTimerRef.current);
|
|
ctrlCTimerRef.current = null;
|
|
}
|
|
if (ctrlCPressCount > 2) {
|
|
recordExitFail(config);
|
|
}
|
|
if (ctrlCPressCount > 1) {
|
|
// eslint-disable-next-line @typescript-eslint/no-floating-promises
|
|
handleSlashCommand('/quit', undefined, undefined, false);
|
|
} else if (ctrlCPressCount > 0) {
|
|
ctrlCTimerRef.current = setTimeout(() => {
|
|
setCtrlCPressCount(0);
|
|
ctrlCTimerRef.current = null;
|
|
}, WARNING_PROMPT_DURATION_MS);
|
|
}
|
|
}, [ctrlCPressCount, config, setCtrlCPressCount, handleSlashCommand]);
|
|
|
|
useEffect(() => {
|
|
if (ctrlDTimerRef.current) {
|
|
clearTimeout(ctrlDTimerRef.current);
|
|
ctrlCTimerRef.current = null;
|
|
}
|
|
if (ctrlDPressCount > 2) {
|
|
recordExitFail(config);
|
|
}
|
|
if (ctrlDPressCount > 1) {
|
|
// eslint-disable-next-line @typescript-eslint/no-floating-promises
|
|
handleSlashCommand('/quit', undefined, undefined, false);
|
|
} else if (ctrlDPressCount > 0) {
|
|
ctrlDTimerRef.current = setTimeout(() => {
|
|
setCtrlDPressCount(0);
|
|
ctrlDTimerRef.current = null;
|
|
}, WARNING_PROMPT_DURATION_MS);
|
|
}
|
|
}, [ctrlDPressCount, config, setCtrlDPressCount, handleSlashCommand]);
|
|
|
|
const handleEscapePromptChange = useCallback((showPrompt: boolean) => {
|
|
setShowEscapePrompt(showPrompt);
|
|
}, []);
|
|
|
|
const handleIdePromptComplete = useCallback(
|
|
(result: IdeIntegrationNudgeResult) => {
|
|
if (result.userSelection === 'yes') {
|
|
// eslint-disable-next-line @typescript-eslint/no-floating-promises
|
|
handleSlashCommand('/ide install');
|
|
settings.setValue(SettingScope.User, 'ide.hasSeenNudge', true);
|
|
} else if (result.userSelection === 'dismiss') {
|
|
settings.setValue(SettingScope.User, 'ide.hasSeenNudge', true);
|
|
}
|
|
setIdePromptAnswered(true);
|
|
},
|
|
[handleSlashCommand, settings],
|
|
);
|
|
|
|
const { elapsedTime, currentLoadingPhrase } = useLoadingIndicator({
|
|
streamingState,
|
|
shouldShowFocusHint,
|
|
retryStatus,
|
|
});
|
|
|
|
const handleGlobalKeypress = useCallback(
|
|
(key: Key): boolean => {
|
|
// Debug log keystrokes if enabled
|
|
if (settings.merged.general.debugKeystrokeLogging) {
|
|
debugLogger.log('[DEBUG] Keystroke:', JSON.stringify(key));
|
|
}
|
|
|
|
if (shortcutsHelpVisible && shouldDismissShortcutsHelpOnHotkey(key)) {
|
|
setShortcutsHelpVisible(false);
|
|
}
|
|
|
|
if (isAlternateBuffer && keyMatchers[Command.TOGGLE_COPY_MODE](key)) {
|
|
setCopyModeEnabled(true);
|
|
disableMouseEvents();
|
|
return true;
|
|
}
|
|
|
|
if (keyMatchers[Command.QUIT](key)) {
|
|
// If the user presses Ctrl+C, we want to cancel any ongoing requests.
|
|
// This should happen regardless of the count.
|
|
cancelOngoingRequest?.();
|
|
|
|
setCtrlCPressCount((prev) => prev + 1);
|
|
return true;
|
|
} else if (keyMatchers[Command.EXIT](key)) {
|
|
setCtrlDPressCount((prev) => prev + 1);
|
|
return true;
|
|
} else if (keyMatchers[Command.SUSPEND_APP](key)) {
|
|
handleSuspend();
|
|
} else if (
|
|
keyMatchers[Command.TOGGLE_COPY_MODE](key) &&
|
|
!isAlternateBuffer
|
|
) {
|
|
showTransientMessage({
|
|
text: 'Use Ctrl+O to expand and collapse blocks of content.',
|
|
type: TransientMessageType.Warning,
|
|
});
|
|
return true;
|
|
}
|
|
|
|
let enteringConstrainHeightMode = false;
|
|
if (!constrainHeight) {
|
|
enteringConstrainHeightMode = true;
|
|
setConstrainHeight(true);
|
|
}
|
|
|
|
if (keyMatchers[Command.SHOW_ERROR_DETAILS](key)) {
|
|
if (settings.merged.general.devtools) {
|
|
void (async () => {
|
|
const { toggleDevToolsPanel } = await import(
|
|
'../utils/devtoolsService.js'
|
|
);
|
|
await toggleDevToolsPanel(
|
|
config,
|
|
showErrorDetails,
|
|
() => setShowErrorDetails((prev) => !prev),
|
|
() => setShowErrorDetails(true),
|
|
);
|
|
})();
|
|
} else {
|
|
setShowErrorDetails((prev) => !prev);
|
|
}
|
|
return true;
|
|
} else if (keyMatchers[Command.SHOW_FULL_TODOS](key)) {
|
|
setShowFullTodos((prev) => !prev);
|
|
return true;
|
|
} else if (keyMatchers[Command.TOGGLE_MARKDOWN](key)) {
|
|
setRenderMarkdown((prev) => {
|
|
const newValue = !prev;
|
|
// Force re-render of static content
|
|
refreshStatic();
|
|
return newValue;
|
|
});
|
|
return true;
|
|
} else if (
|
|
keyMatchers[Command.SHOW_IDE_CONTEXT_DETAIL](key) &&
|
|
config.getIdeMode() &&
|
|
ideContextState
|
|
) {
|
|
// eslint-disable-next-line @typescript-eslint/no-floating-promises
|
|
handleSlashCommand('/ide status');
|
|
return true;
|
|
} else if (
|
|
keyMatchers[Command.SHOW_MORE_LINES](key) &&
|
|
!enteringConstrainHeightMode
|
|
) {
|
|
setConstrainHeight(false);
|
|
return true;
|
|
} else if (
|
|
(keyMatchers[Command.FOCUS_SHELL_INPUT](key) ||
|
|
keyMatchers[Command.UNFOCUS_BACKGROUND_SHELL_LIST](key)) &&
|
|
(activePtyId || (isBackgroundShellVisible && backgroundShells.size > 0))
|
|
) {
|
|
if (embeddedShellFocused) {
|
|
const capturedTime = lastOutputTimeRef.current;
|
|
if (tabFocusTimeoutRef.current)
|
|
clearTimeout(tabFocusTimeoutRef.current);
|
|
tabFocusTimeoutRef.current = setTimeout(() => {
|
|
if (lastOutputTimeRef.current === capturedTime) {
|
|
setEmbeddedShellFocused(false);
|
|
} else {
|
|
showTransientMessage({
|
|
text: 'Use Shift+Tab to unfocus',
|
|
type: TransientMessageType.Warning,
|
|
});
|
|
}
|
|
}, 150);
|
|
return false;
|
|
}
|
|
|
|
const isIdle = Date.now() - lastOutputTimeRef.current >= 100;
|
|
|
|
if (isIdle && !activePtyId && !isBackgroundShellVisible) {
|
|
if (tabFocusTimeoutRef.current)
|
|
clearTimeout(tabFocusTimeoutRef.current);
|
|
toggleBackgroundShell();
|
|
setEmbeddedShellFocused(true);
|
|
if (backgroundShells.size > 1) setIsBackgroundShellListOpen(true);
|
|
return true;
|
|
}
|
|
|
|
setEmbeddedShellFocused(true);
|
|
return true;
|
|
} else if (
|
|
keyMatchers[Command.UNFOCUS_SHELL_INPUT](key) ||
|
|
keyMatchers[Command.UNFOCUS_BACKGROUND_SHELL](key)
|
|
) {
|
|
if (embeddedShellFocused) {
|
|
setEmbeddedShellFocused(false);
|
|
return true;
|
|
}
|
|
return false;
|
|
} else if (keyMatchers[Command.TOGGLE_BACKGROUND_SHELL](key)) {
|
|
if (activePtyId) {
|
|
backgroundCurrentShell();
|
|
// After backgrounding, we explicitly do NOT show or focus the background UI.
|
|
} else {
|
|
toggleBackgroundShell();
|
|
// Toggle focus based on intent: if we were hiding, unfocus; if showing, focus.
|
|
if (!isBackgroundShellVisible && backgroundShells.size > 0) {
|
|
setEmbeddedShellFocused(true);
|
|
if (backgroundShells.size > 1) {
|
|
setIsBackgroundShellListOpen(true);
|
|
}
|
|
} else {
|
|
setEmbeddedShellFocused(false);
|
|
}
|
|
}
|
|
return true;
|
|
} else if (keyMatchers[Command.TOGGLE_BACKGROUND_SHELL_LIST](key)) {
|
|
if (backgroundShells.size > 0 && isBackgroundShellVisible) {
|
|
if (!embeddedShellFocused) {
|
|
setEmbeddedShellFocused(true);
|
|
}
|
|
setIsBackgroundShellListOpen(true);
|
|
}
|
|
return true;
|
|
}
|
|
return false;
|
|
},
|
|
[
|
|
constrainHeight,
|
|
setConstrainHeight,
|
|
setShowErrorDetails,
|
|
config,
|
|
ideContextState,
|
|
setCtrlCPressCount,
|
|
setCtrlDPressCount,
|
|
handleSlashCommand,
|
|
cancelOngoingRequest,
|
|
activePtyId,
|
|
handleSuspend,
|
|
embeddedShellFocused,
|
|
settings.merged.general.debugKeystrokeLogging,
|
|
refreshStatic,
|
|
setCopyModeEnabled,
|
|
tabFocusTimeoutRef,
|
|
isAlternateBuffer,
|
|
shortcutsHelpVisible,
|
|
backgroundCurrentShell,
|
|
toggleBackgroundShell,
|
|
backgroundShells,
|
|
isBackgroundShellVisible,
|
|
setIsBackgroundShellListOpen,
|
|
lastOutputTimeRef,
|
|
showTransientMessage,
|
|
settings.merged.general.devtools,
|
|
showErrorDetails,
|
|
],
|
|
);
|
|
|
|
useKeypress(handleGlobalKeypress, { isActive: true, priority: true });
|
|
|
|
useKeypress(
|
|
() => {
|
|
setCopyModeEnabled(false);
|
|
enableMouseEvents();
|
|
return true;
|
|
},
|
|
{
|
|
isActive: copyModeEnabled,
|
|
// We need to receive keypresses first so they do not bubble to other
|
|
// handlers.
|
|
priority: KeypressPriority.Critical,
|
|
},
|
|
);
|
|
|
|
useEffect(() => {
|
|
// Respect hideWindowTitle settings
|
|
if (settings.merged.ui.hideWindowTitle) return;
|
|
|
|
const paddedTitle = computeTerminalTitle({
|
|
streamingState,
|
|
thoughtSubject: thought?.subject,
|
|
isConfirming:
|
|
!!commandConfirmationRequest || shouldShowActionRequiredTitle,
|
|
isSilentWorking: shouldShowSilentWorkingTitle,
|
|
folderName: basename(config.getTargetDir()),
|
|
showThoughts: !!settings.merged.ui.showStatusInTitle,
|
|
useDynamicTitle: settings.merged.ui.dynamicWindowTitle,
|
|
});
|
|
|
|
// Only update the title if it's different from the last value we set
|
|
if (lastTitleRef.current !== paddedTitle) {
|
|
lastTitleRef.current = paddedTitle;
|
|
stdout.write(`\x1b]0;${paddedTitle}\x07`);
|
|
}
|
|
// Note: We don't need to reset the window title on exit because Gemini CLI is already doing that elsewhere
|
|
}, [
|
|
streamingState,
|
|
thought,
|
|
commandConfirmationRequest,
|
|
shouldShowActionRequiredTitle,
|
|
shouldShowSilentWorkingTitle,
|
|
settings.merged.ui.showStatusInTitle,
|
|
settings.merged.ui.dynamicWindowTitle,
|
|
settings.merged.ui.hideWindowTitle,
|
|
config,
|
|
stdout,
|
|
]);
|
|
|
|
useEffect(() => {
|
|
const handleUserFeedback = (payload: UserFeedbackPayload) => {
|
|
let type: MessageType;
|
|
switch (payload.severity) {
|
|
case 'error':
|
|
type = MessageType.ERROR;
|
|
break;
|
|
case 'warning':
|
|
type = MessageType.WARNING;
|
|
break;
|
|
case 'info':
|
|
type = MessageType.INFO;
|
|
break;
|
|
default:
|
|
throw new Error(
|
|
`Unexpected severity for user feedback: ${payload.severity}`,
|
|
);
|
|
}
|
|
|
|
historyManager.addItem(
|
|
{
|
|
type,
|
|
text: payload.message,
|
|
},
|
|
Date.now(),
|
|
);
|
|
|
|
// If there is an attached error object, log it to the debug drawer.
|
|
if (payload.error) {
|
|
debugLogger.warn(
|
|
`[Feedback Details for "${payload.message}"]`,
|
|
payload.error,
|
|
);
|
|
}
|
|
};
|
|
|
|
coreEvents.on(CoreEvent.UserFeedback, handleUserFeedback);
|
|
|
|
// Flush any messages that happened during startup before this component
|
|
// mounted.
|
|
coreEvents.drainBacklogs();
|
|
|
|
return () => {
|
|
coreEvents.off(CoreEvent.UserFeedback, handleUserFeedback);
|
|
};
|
|
}, [historyManager]);
|
|
|
|
const filteredConsoleMessages = useMemo(() => {
|
|
if (config.getDebugMode()) {
|
|
return consoleMessages;
|
|
}
|
|
return consoleMessages.filter((msg) => msg.type !== 'debug');
|
|
}, [consoleMessages, config]);
|
|
|
|
// Computed values
|
|
const errorCount = useMemo(
|
|
() =>
|
|
filteredConsoleMessages
|
|
.filter((msg) => msg.type === 'error')
|
|
.reduce((total, msg) => total + msg.count, 0),
|
|
[filteredConsoleMessages],
|
|
);
|
|
|
|
const nightly = props.version.includes('nightly');
|
|
|
|
const dialogsVisible =
|
|
shouldShowIdePrompt ||
|
|
isFolderTrustDialogOpen ||
|
|
adminSettingsChanged ||
|
|
!!commandConfirmationRequest ||
|
|
!!authConsentRequest ||
|
|
!!permissionConfirmationRequest ||
|
|
!!customDialog ||
|
|
confirmUpdateExtensionRequests.length > 0 ||
|
|
!!loopDetectionConfirmationRequest ||
|
|
isThemeDialogOpen ||
|
|
isSettingsDialogOpen ||
|
|
isModelDialogOpen ||
|
|
isAgentConfigDialogOpen ||
|
|
isPermissionsDialogOpen ||
|
|
isAuthenticating ||
|
|
isAuthDialogOpen ||
|
|
isEditorDialogOpen ||
|
|
showPrivacyNotice ||
|
|
showIdeRestartPrompt ||
|
|
!!proQuotaRequest ||
|
|
!!validationRequest ||
|
|
isSessionBrowserOpen ||
|
|
authState === AuthState.AwaitingApiKeyInput ||
|
|
!!newAgents;
|
|
|
|
const pendingHistoryItems = useMemo(
|
|
() => [...pendingSlashCommandHistoryItems, ...pendingGeminiHistoryItems],
|
|
[pendingSlashCommandHistoryItems, pendingGeminiHistoryItems],
|
|
);
|
|
|
|
const hasPendingToolConfirmation = useMemo(
|
|
() => isToolAwaitingConfirmation(pendingHistoryItems),
|
|
[pendingHistoryItems],
|
|
);
|
|
|
|
const hasPendingActionRequired =
|
|
hasPendingToolConfirmation ||
|
|
!!commandConfirmationRequest ||
|
|
!!authConsentRequest ||
|
|
confirmUpdateExtensionRequests.length > 0 ||
|
|
!!loopDetectionConfirmationRequest ||
|
|
!!proQuotaRequest ||
|
|
!!validationRequest ||
|
|
!!customDialog;
|
|
|
|
const isPassiveShortcutsHelpState =
|
|
isInputActive &&
|
|
streamingState === StreamingState.Idle &&
|
|
!hasPendingActionRequired;
|
|
|
|
useEffect(() => {
|
|
if (shortcutsHelpVisible && !isPassiveShortcutsHelpState) {
|
|
setShortcutsHelpVisible(false);
|
|
}
|
|
}, [
|
|
shortcutsHelpVisible,
|
|
isPassiveShortcutsHelpState,
|
|
setShortcutsHelpVisible,
|
|
]);
|
|
|
|
const allToolCalls = useMemo(
|
|
() =>
|
|
pendingHistoryItems
|
|
.filter(
|
|
(item): item is HistoryItemToolGroup => item.type === 'tool_group',
|
|
)
|
|
.flatMap((item) => item.tools),
|
|
[pendingHistoryItems],
|
|
);
|
|
|
|
const [geminiMdFileCount, setGeminiMdFileCount] = useState<number>(
|
|
config.getGeminiMdFileCount(),
|
|
);
|
|
useEffect(() => {
|
|
const handleMemoryChanged = (result: MemoryChangedPayload) => {
|
|
setGeminiMdFileCount(result.fileCount);
|
|
};
|
|
coreEvents.on(CoreEvent.MemoryChanged, handleMemoryChanged);
|
|
return () => {
|
|
coreEvents.off(CoreEvent.MemoryChanged, handleMemoryChanged);
|
|
};
|
|
}, []);
|
|
|
|
useEffect(() => {
|
|
let isMounted = true;
|
|
|
|
const fetchBannerTexts = async () => {
|
|
const [defaultBanner, warningBanner] = await Promise.all([
|
|
// TODO: temporarily disabling the banner, it will be re-added.
|
|
'',
|
|
config.getBannerTextCapacityIssues(),
|
|
]);
|
|
|
|
if (isMounted) {
|
|
setDefaultBannerText(defaultBanner);
|
|
setWarningBannerText(warningBanner);
|
|
setBannerVisible(true);
|
|
}
|
|
};
|
|
// eslint-disable-next-line @typescript-eslint/no-floating-promises
|
|
fetchBannerTexts();
|
|
|
|
return () => {
|
|
isMounted = false;
|
|
};
|
|
}, [config, refreshStatic]);
|
|
|
|
const uiState: UIState = useMemo(
|
|
() => ({
|
|
history: historyManager.history,
|
|
historyManager,
|
|
isThemeDialogOpen,
|
|
themeError,
|
|
isAuthenticating,
|
|
isConfigInitialized,
|
|
authError,
|
|
isAuthDialogOpen,
|
|
isAwaitingApiKeyInput: authState === AuthState.AwaitingApiKeyInput,
|
|
apiKeyDefaultValue,
|
|
editorError,
|
|
isEditorDialogOpen,
|
|
showPrivacyNotice,
|
|
corgiMode,
|
|
debugMessage,
|
|
quittingMessages,
|
|
isSettingsDialogOpen,
|
|
isSessionBrowserOpen,
|
|
isModelDialogOpen,
|
|
isAgentConfigDialogOpen,
|
|
selectedAgentName,
|
|
selectedAgentDisplayName,
|
|
selectedAgentDefinition,
|
|
isPermissionsDialogOpen,
|
|
permissionsDialogProps,
|
|
slashCommands,
|
|
pendingSlashCommandHistoryItems,
|
|
commandContext,
|
|
commandConfirmationRequest,
|
|
authConsentRequest,
|
|
confirmUpdateExtensionRequests,
|
|
loopDetectionConfirmationRequest,
|
|
permissionConfirmationRequest,
|
|
geminiMdFileCount,
|
|
streamingState,
|
|
initError,
|
|
pendingGeminiHistoryItems,
|
|
thought,
|
|
shellModeActive,
|
|
userMessages: inputHistory,
|
|
buffer,
|
|
inputWidth,
|
|
suggestionsWidth,
|
|
isInputActive,
|
|
isResuming,
|
|
shouldShowIdePrompt,
|
|
isFolderTrustDialogOpen: isFolderTrustDialogOpen ?? false,
|
|
isTrustedFolder,
|
|
constrainHeight,
|
|
showErrorDetails,
|
|
showFullTodos,
|
|
filteredConsoleMessages,
|
|
ideContextState,
|
|
renderMarkdown,
|
|
ctrlCPressedOnce: ctrlCPressCount >= 1,
|
|
ctrlDPressedOnce: ctrlDPressCount >= 1,
|
|
showEscapePrompt,
|
|
shortcutsHelpVisible,
|
|
cleanUiDetailsVisible,
|
|
isFocused,
|
|
elapsedTime,
|
|
currentLoadingPhrase,
|
|
historyRemountKey,
|
|
activeHooks,
|
|
messageQueue,
|
|
queueErrorMessage,
|
|
showApprovalModeIndicator,
|
|
currentModel,
|
|
quota: {
|
|
userTier,
|
|
stats: quotaStats,
|
|
proQuotaRequest,
|
|
validationRequest,
|
|
},
|
|
contextFileNames,
|
|
errorCount,
|
|
availableTerminalHeight,
|
|
mainAreaWidth,
|
|
staticAreaMaxItemHeight,
|
|
staticExtraHeight,
|
|
dialogsVisible,
|
|
pendingHistoryItems,
|
|
nightly,
|
|
branchName,
|
|
sessionStats,
|
|
terminalWidth,
|
|
terminalHeight,
|
|
mainControlsRef,
|
|
rootUiRef,
|
|
currentIDE,
|
|
updateInfo,
|
|
showIdeRestartPrompt,
|
|
ideTrustRestartReason,
|
|
isRestarting,
|
|
extensionsUpdateState,
|
|
activePtyId,
|
|
backgroundShellCount,
|
|
isBackgroundShellVisible,
|
|
embeddedShellFocused,
|
|
showDebugProfiler,
|
|
customDialog,
|
|
copyModeEnabled,
|
|
transientMessage,
|
|
bannerData,
|
|
bannerVisible,
|
|
terminalBackgroundColor: config.getTerminalBackground(),
|
|
settingsNonce,
|
|
backgroundShells,
|
|
activeBackgroundShellPid,
|
|
backgroundShellHeight,
|
|
isBackgroundShellListOpen,
|
|
adminSettingsChanged,
|
|
newAgents,
|
|
}),
|
|
[
|
|
isThemeDialogOpen,
|
|
themeError,
|
|
isAuthenticating,
|
|
isConfigInitialized,
|
|
authError,
|
|
isAuthDialogOpen,
|
|
editorError,
|
|
isEditorDialogOpen,
|
|
showPrivacyNotice,
|
|
corgiMode,
|
|
debugMessage,
|
|
quittingMessages,
|
|
isSettingsDialogOpen,
|
|
isSessionBrowserOpen,
|
|
isModelDialogOpen,
|
|
isAgentConfigDialogOpen,
|
|
selectedAgentName,
|
|
selectedAgentDisplayName,
|
|
selectedAgentDefinition,
|
|
isPermissionsDialogOpen,
|
|
permissionsDialogProps,
|
|
slashCommands,
|
|
pendingSlashCommandHistoryItems,
|
|
commandContext,
|
|
commandConfirmationRequest,
|
|
authConsentRequest,
|
|
confirmUpdateExtensionRequests,
|
|
loopDetectionConfirmationRequest,
|
|
permissionConfirmationRequest,
|
|
geminiMdFileCount,
|
|
streamingState,
|
|
initError,
|
|
pendingGeminiHistoryItems,
|
|
thought,
|
|
shellModeActive,
|
|
inputHistory,
|
|
buffer,
|
|
inputWidth,
|
|
suggestionsWidth,
|
|
isInputActive,
|
|
isResuming,
|
|
shouldShowIdePrompt,
|
|
isFolderTrustDialogOpen,
|
|
isTrustedFolder,
|
|
constrainHeight,
|
|
showErrorDetails,
|
|
showFullTodos,
|
|
filteredConsoleMessages,
|
|
ideContextState,
|
|
renderMarkdown,
|
|
ctrlCPressCount,
|
|
ctrlDPressCount,
|
|
showEscapePrompt,
|
|
shortcutsHelpVisible,
|
|
cleanUiDetailsVisible,
|
|
isFocused,
|
|
elapsedTime,
|
|
currentLoadingPhrase,
|
|
historyRemountKey,
|
|
activeHooks,
|
|
messageQueue,
|
|
queueErrorMessage,
|
|
showApprovalModeIndicator,
|
|
userTier,
|
|
quotaStats,
|
|
proQuotaRequest,
|
|
validationRequest,
|
|
contextFileNames,
|
|
errorCount,
|
|
availableTerminalHeight,
|
|
mainAreaWidth,
|
|
staticAreaMaxItemHeight,
|
|
staticExtraHeight,
|
|
dialogsVisible,
|
|
pendingHistoryItems,
|
|
nightly,
|
|
branchName,
|
|
sessionStats,
|
|
terminalWidth,
|
|
terminalHeight,
|
|
mainControlsRef,
|
|
rootUiRef,
|
|
currentIDE,
|
|
updateInfo,
|
|
showIdeRestartPrompt,
|
|
ideTrustRestartReason,
|
|
isRestarting,
|
|
currentModel,
|
|
extensionsUpdateState,
|
|
activePtyId,
|
|
backgroundShellCount,
|
|
isBackgroundShellVisible,
|
|
historyManager,
|
|
embeddedShellFocused,
|
|
showDebugProfiler,
|
|
customDialog,
|
|
apiKeyDefaultValue,
|
|
authState,
|
|
copyModeEnabled,
|
|
transientMessage,
|
|
bannerData,
|
|
bannerVisible,
|
|
config,
|
|
settingsNonce,
|
|
backgroundShellHeight,
|
|
isBackgroundShellListOpen,
|
|
activeBackgroundShellPid,
|
|
backgroundShells,
|
|
adminSettingsChanged,
|
|
newAgents,
|
|
],
|
|
);
|
|
|
|
const exitPrivacyNotice = useCallback(
|
|
() => setShowPrivacyNotice(false),
|
|
[setShowPrivacyNotice],
|
|
);
|
|
|
|
const uiActions: UIActions = useMemo(
|
|
() => ({
|
|
handleThemeSelect,
|
|
closeThemeDialog,
|
|
handleThemeHighlight,
|
|
handleAuthSelect,
|
|
setAuthState,
|
|
onAuthError,
|
|
handleEditorSelect,
|
|
exitEditorDialog,
|
|
exitPrivacyNotice,
|
|
closeSettingsDialog,
|
|
closeModelDialog,
|
|
openAgentConfigDialog,
|
|
closeAgentConfigDialog,
|
|
openPermissionsDialog,
|
|
closePermissionsDialog,
|
|
setShellModeActive,
|
|
vimHandleInput,
|
|
handleIdePromptComplete,
|
|
handleFolderTrustSelect,
|
|
setConstrainHeight,
|
|
onEscapePromptChange: handleEscapePromptChange,
|
|
refreshStatic,
|
|
handleFinalSubmit,
|
|
handleClearScreen,
|
|
handleProQuotaChoice,
|
|
handleValidationChoice,
|
|
openSessionBrowser,
|
|
closeSessionBrowser,
|
|
handleResumeSession,
|
|
handleDeleteSession,
|
|
setQueueErrorMessage,
|
|
popAllMessages,
|
|
handleApiKeySubmit,
|
|
handleApiKeyCancel,
|
|
setBannerVisible,
|
|
setShortcutsHelpVisible,
|
|
setCleanUiDetailsVisible,
|
|
toggleCleanUiDetailsVisible,
|
|
revealCleanUiDetailsTemporarily,
|
|
handleWarning,
|
|
setEmbeddedShellFocused,
|
|
dismissBackgroundShell,
|
|
setActiveBackgroundShellPid,
|
|
setIsBackgroundShellListOpen,
|
|
setAuthContext,
|
|
handleRestart: async () => {
|
|
if (process.send) {
|
|
const remoteSettings = config.getRemoteAdminSettings();
|
|
if (remoteSettings) {
|
|
process.send({
|
|
type: 'admin-settings-update',
|
|
settings: remoteSettings,
|
|
});
|
|
}
|
|
}
|
|
await runExitCleanup();
|
|
process.exit(RELAUNCH_EXIT_CODE);
|
|
},
|
|
handleNewAgentsSelect: async (choice: NewAgentsChoice) => {
|
|
if (newAgents && choice === NewAgentsChoice.ACKNOWLEDGE) {
|
|
const registry = config.getAgentRegistry();
|
|
try {
|
|
await Promise.all(
|
|
newAgents.map((agent) => registry.acknowledgeAgent(agent)),
|
|
);
|
|
} catch (error) {
|
|
debugLogger.error('Failed to acknowledge agents:', error);
|
|
historyManager.addItem(
|
|
{
|
|
type: MessageType.ERROR,
|
|
text: `Failed to acknowledge agents: ${getErrorMessage(error)}`,
|
|
},
|
|
Date.now(),
|
|
);
|
|
}
|
|
}
|
|
setNewAgents(null);
|
|
},
|
|
}),
|
|
[
|
|
handleThemeSelect,
|
|
closeThemeDialog,
|
|
handleThemeHighlight,
|
|
handleAuthSelect,
|
|
setAuthState,
|
|
onAuthError,
|
|
handleEditorSelect,
|
|
exitEditorDialog,
|
|
exitPrivacyNotice,
|
|
closeSettingsDialog,
|
|
closeModelDialog,
|
|
openAgentConfigDialog,
|
|
closeAgentConfigDialog,
|
|
openPermissionsDialog,
|
|
closePermissionsDialog,
|
|
setShellModeActive,
|
|
vimHandleInput,
|
|
handleIdePromptComplete,
|
|
handleFolderTrustSelect,
|
|
setConstrainHeight,
|
|
handleEscapePromptChange,
|
|
refreshStatic,
|
|
handleFinalSubmit,
|
|
handleClearScreen,
|
|
handleProQuotaChoice,
|
|
handleValidationChoice,
|
|
openSessionBrowser,
|
|
closeSessionBrowser,
|
|
handleResumeSession,
|
|
handleDeleteSession,
|
|
setQueueErrorMessage,
|
|
popAllMessages,
|
|
handleApiKeySubmit,
|
|
handleApiKeyCancel,
|
|
setBannerVisible,
|
|
setShortcutsHelpVisible,
|
|
setCleanUiDetailsVisible,
|
|
toggleCleanUiDetailsVisible,
|
|
revealCleanUiDetailsTemporarily,
|
|
handleWarning,
|
|
setEmbeddedShellFocused,
|
|
dismissBackgroundShell,
|
|
setActiveBackgroundShellPid,
|
|
setIsBackgroundShellListOpen,
|
|
setAuthContext,
|
|
newAgents,
|
|
config,
|
|
historyManager,
|
|
],
|
|
);
|
|
|
|
if (authState === AuthState.AwaitingGoogleLoginRestart) {
|
|
return (
|
|
<LoginWithGoogleRestartDialog
|
|
onDismiss={() => {
|
|
setAuthContext({});
|
|
setAuthState(AuthState.Updating);
|
|
}}
|
|
config={config}
|
|
/>
|
|
);
|
|
}
|
|
|
|
return (
|
|
<UIStateContext.Provider value={uiState}>
|
|
<UIActionsContext.Provider value={uiActions}>
|
|
<ConfigContext.Provider value={config}>
|
|
<AppContext.Provider
|
|
value={{
|
|
version: props.version,
|
|
startupWarnings: props.startupWarnings || [],
|
|
}}
|
|
>
|
|
<ToolActionsProvider config={config} toolCalls={allToolCalls}>
|
|
<ShellFocusContext.Provider value={isFocused}>
|
|
<App key={`app-${forceRerenderKey}`} />
|
|
</ShellFocusContext.Provider>
|
|
</ToolActionsProvider>
|
|
</AppContext.Provider>
|
|
</ConfigContext.Provider>
|
|
</UIActionsContext.Provider>
|
|
</UIStateContext.Provider>
|
|
);
|
|
};
|