@machina.ai/cell-cli 1.16.0-rc2 → 1.19.4-rc1

package/dist/src/ui/AppContainer.js

@@ -13,7 +13,7 @@ import { UIActionsContext, } from './contexts/UIActionsContext.js';
 import { ConfigContext } from './contexts/ConfigContext.js';
 import { ToolCallStatus, AuthState, } from './types.js';
 import { MessageType, StreamingState } from './types.js';
-import { DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_MODEL, PREVIEW_GEMINI_MODEL, resolveModel, IdeClient, ideContextStore, getErrorMessage, getAllGeminiMdFilenames, AuthType, clearCachedCredentialFile, recordExitFail, ShellExecutionService, saveApiKey, debugLogger, coreEvents, CoreEvent, refreshServerHierarchicalMemory, } from '@google/gemini-cli-core';
+import { DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_MODEL, PREVIEW_GEMINI_MODEL, resolveModel, IdeClient, ideContextStore, getErrorMessage, getAllGeminiMdFilenames, AuthType, clearCachedCredentialFile, recordExitFail, ShellExecutionService, saveApiKey, debugLogger, coreEvents, CoreEvent, refreshServerHierarchicalMemory, writeToStdout, disableMouseEvents, enterAlternateScreen, enableMouseEvents, disableLineWrapping, shouldEnterAlternateScreen, } from '@google/gemini-cli-core';
 import { validateAuthMethod } from '../config/auth.js';
 import process from 'node:process';
 import { useHistory } from './hooks/useHistoryManager.js';
@@ -29,7 +29,7 @@ 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 { useStdout, useStdin } from 'ink';
+import { useApp, useStdout, useStdin } from 'ink';
 import { calculateMainAreaWidth } from './utils/ui-sizing.js';
 import ansiEscapes from 'ansi-escapes';
 import * as fs from 'node:fs';
@@ -52,8 +52,8 @@ import {} from './IdeIntegrationNudge.js';
 import { appEvents, AppEvent } from '../utils/events.js';
 import {} from './utils/updateCheck.js';
 import { setUpdateHandler } from '../utils/handleAutoUpdate.js';
-import { ConsolePatcher } from './utils/ConsolePatcher.js';
 import { registerCleanup, runExitCleanup } from '../utils/cleanup.js';
+import { RELAUNCH_EXIT_CODE } from '../utils/processUtils.js';
 import { useMessageQueue } from './hooks/useMessageQueue.js';
 import { useAutoAcceptIndicator } from './hooks/useAutoAcceptIndicator.js';
 import { useSessionStats } from './contexts/SessionContext.js';
@@ -61,12 +61,16 @@ import { useGitBranchName } from './hooks/useGitBranchName.js';
 import { FeedbackService } from '../services/FeedbackService.js';
 import { useConfirmUpdateRequests, useExtensionUpdates, } from './hooks/useExtensionUpdates.js';
 import { ShellFocusContext } from './contexts/ShellFocusContext.js';
-import { useSessionResume } from './hooks/useSessionResume.js';
 import {} from '../config/extension-manager.js';
 import { requestConsentInteractive } from '../config/extensions/consent.js';
-import { disableMouseEvents, enableMouseEvents } from './utils/mouse.js';
+import { useSessionBrowser } from './hooks/useSessionBrowser.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 { enableSupportedProtocol } from './utils/kittyProtocolDetector.js';
+import { enableBracketedPaste } from './utils/bracketedPaste.js';
 const WARNING_PROMPT_DURATION_MS = 1000;
 const QUEUE_ERROR_DISPLAY_DURATION_MS = 3000;
 function isToolExecuting(pendingHistoryItems) {
@@ -89,7 +93,9 @@ const SHELL_WIDTH_FRACTION = 0.89;
 const SHELL_HEIGHT_PADDING = 10;
 export const AppContainer = (props) => {
     const { config, initializationResult, resumedSessionData } = props;
-    const historyManager = useHistory();
+    const historyManager = useHistory({
+        chatRecordingService: config.getGeminiClient()?.getChatRecordingService(),
+    });
     useMemoryMonitor(historyManager);
     const settings = useSettings();
     const isAlternateBuffer = useAlternateBuffer();
@@ -102,12 +108,14 @@ export const AppContainer = (props) => {
     const [isProcessing, setIsProcessing] = useState(false);
     const [embeddedShellFocused, setEmbeddedShellFocused] = useState(false);
     const [showDebugProfiler, setShowDebugProfiler] = useState(false);
+    const [customDialog, setCustomDialog] = useState(null);
     const [copyModeEnabled, setCopyModeEnabled] = useState(false);
+    const [pendingRestorePrompt, setPendingRestorePrompt] = useState(false);
     const [shellModeActive, setShellModeActive] = useState(false);
     const [modelSwitchedFromQuotaError, setModelSwitchedFromQuotaError] = useState(false);
     const [historyRemountKey, setHistoryRemountKey] = useState(0);
     const [updateInfo, setUpdateInfo] = useState(null);
-    const [isTrustedFolder, setIsTrustedFolder] = useState(config.isTrustedFolder());
+    const [isTrustedFolder, setIsTrustedFolder] = useState(isWorkspaceTrusted(settings.merged).isTrusted);
     const [queueErrorMessage, setQueueErrorMessage] = useState(null);
     const [defaultBannerText, setDefaultBannerText] = useState('');
     const [warningBannerText, setWarningBannerText] = useState('');
@@ -119,8 +127,15 @@ export const AppContainer = (props) => {
     const { addConfirmUpdateExtensionRequest, confirmUpdateExtensionRequests } = useConfirmUpdateRequests();
     const { extensionsUpdateState, extensionsUpdateStateInternal, dispatchExtensionStateUpdate, } = useExtensionUpdates(extensionManager, historyManager.addItem, config.getEnableExtensionReloading());
     const [isPermissionsDialogOpen, setPermissionsDialogOpen] = useState(false);
-    const openPermissionsDialog = useCallback(() => setPermissionsDialogOpen(true), []);
-    const closePermissionsDialog = useCallback(() => setPermissionsDialogOpen(false), []);
+    const [permissionsDialogProps, setPermissionsDialogProps] = useState(null);
+    const openPermissionsDialog = useCallback((props) => {
+        setPermissionsDialogOpen(true);
+        setPermissionsDialogProps(props ?? null);
+    }, []);
+    const closePermissionsDialog = useCallback(() => {
+        setPermissionsDialogOpen(false);
+        setPermissionsDialogProps(null);
+    }, []);
     const toggleDebugProfiler = useCallback(() => setShowDebugProfiler((prev) => !prev), []);
     // Helper to determine the effective model, considering the fallback state.
     const getEffectiveModel = useCallback(() => {
@@ -146,6 +161,7 @@ export const AppContainer = (props) => {
     const { columns: terminalWidth, rows: terminalHeight } = useTerminalSize();
     const { stdin, setRawMode } = useStdin();
     const { stdout } = useStdout();
+    const app = useApp();
     // Additional hooks moved from App.tsx
     const { stats: sessionStats } = useSessionStats();
     const branchName = useGitBranchName(config.getTargetDir());
@@ -189,15 +205,7 @@ export const AppContainer = (props) => {
             coreEvents.off(CoreEvent.ModelChanged, handleModelChanged);
         };
     }, [getEffectiveModel]);
-    const { consoleMessages, handleNewMessage, clearConsoleMessages: clearConsoleMessagesState, } = useConsoleMessages();
-    useEffect(() => {
-        const consolePatcher = new ConsolePatcher({
-            onNewMessage: handleNewMessage,
-            debugMode: config.getDebugMode(),
-        });
-        consolePatcher.patch();
-        registerCleanup(consolePatcher.cleanup);
-    }, [handleNewMessage, config]);
+    const { consoleMessages, clearConsoleMessages: clearConsoleMessagesState } = useConsoleMessages();
     const mainAreaWidth = calculateMainAreaWidth(terminalWidth, settings);
     // Derive widths for InputPrompt using shared helper
     const { inputWidth, suggestionsWidth } = useMemo(() => {
@@ -252,9 +260,28 @@ export const AppContainer = (props) => {
             stdout.write(ansiEscapes.clearTerminal);
         }
         setHistoryRemountKey((prev) => prev + 1);
-    }, [setHistoryRemountKey, stdout, isAlternateBuffer]);
+    }, [setHistoryRemountKey, isAlternateBuffer, stdout]);
+    const handleEditorClose = useCallback(() => {
+        if (shouldEnterAlternateScreen(isAlternateBuffer, config.getScreenReader())) {
+            // The editor may have exited alternate buffer mode so we need to
+            // enter it again to be safe.
+            enterAlternateScreen();
+            enableMouseEvents();
+            disableLineWrapping();
+            app.rerender();
+        }
+        enableBracketedPaste();
+        enableSupportedProtocol();
+        refreshStatic();
+    }, [refreshStatic, isAlternateBuffer, app, config]);
+    useEffect(() => {
+        coreEvents.on(CoreEvent.ExternalEditorClosed, handleEditorClose);
+        return () => {
+            coreEvents.off(CoreEvent.ExternalEditorClosed, handleEditorClose);
+        };
+    }, [handleEditorClose]);
     const { isThemeDialogOpen, openThemeDialog, closeThemeDialog, handleThemeSelect, handleThemeHighlight, } = useThemeCommand(settings, setThemeError, historyManager.addItem, initializationResult.themeError);
-    const { authState, setAuthState, authError, onAuthError, apiKeyDefaultValue, reloadApiKey, } = useAuthCommand(settings, config);
+    const { authState, setAuthState, authError, authInfo, setAuthInfo, onAuthError, apiKeyDefaultValue, reloadApiKey, } = useAuthCommand(settings, config);
     const { proQuotaRequest, handleProQuotaChoice } = useQuotaAndFallback({
         config,
         historyManager,
@@ -266,7 +293,7 @@ export const AppContainer = (props) => {
     const isAuthenticating = authState === AuthState.Unauthenticated;
     // Session browser and resume functionality
     const isGeminiClientInitialized = config.getGeminiClient()?.isInitialized();
-    useSessionResume({
+    const { loadHistoryForResume } = useSessionResume({
         config,
         historyManager,
         refreshStatic,
@@ -275,13 +302,20 @@ export const AppContainer = (props) => {
         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) => {
+        handleDeleteSessionSync(session);
+    }, [handleDeleteSessionSync]);
     // Create handleAuthSelect wrapper for backward compatibility
     const handleAuthSelect = useCallback(async (authType, scope) => {
         if (authType) {
             await clearCachedCredentialFile();
             settings.setValue(scope, 'security.auth.selectedType', authType);
             try {
-                await config.refreshAuth(authType);
+                setAuthInfo(null);
+                await config.refreshAuth(authType, (msg) => setAuthInfo(msg));
+                setAuthInfo(null);
                 setAuthState(AuthState.Authenticated);
             }
             catch (e) {
@@ -291,16 +325,16 @@ export const AppContainer = (props) => {
             if (authType === AuthType.LOGIN_WITH_GOOGLE &&
                 config.isBrowserLaunchSuppressed()) {
                 await runExitCleanup();
-                debugLogger.log(`
+                writeToStdout(`
 ----------------------------------------------------------------
-Logging in with Google... Please restart Cell CLI to continue.
+Logging in with Google... Restarting Cell CLI to continue.
 ----------------------------------------------------------------
           `);
-                process.exit(0);
+                process.exit(RELAUNCH_EXIT_CODE);
             }
         }
         setAuthState(AuthState.Authenticated);
-    }, [settings, config, setAuthState, onAuthError]);
+    }, [settings, config, setAuthState, onAuthError, setAuthInfo]);
     const handleApiKeySubmit = useCallback(async (apiKey) => {
         try {
             onAuthError(null);
@@ -360,6 +394,7 @@ Logging in with Google... Please restart Cell CLI to continue.
         openEditorDialog,
         openPrivacyNotice: () => setShowPrivacyNotice(true),
         openSettingsDialog,
+        openSessionBrowser,
         openModelDialog,
         openPermissionsDialog,
         quit: (messages) => {
@@ -401,6 +436,7 @@ Logging in with Google... Please restart Cell CLI to continue.
         openThemeDialog,
         openEditorDialog,
         openSettingsDialog,
+        openSessionBrowser,
         openModelDialog,
         setQuittingMessages,
         setDebugMessage,
@@ -413,7 +449,7 @@ Logging in with Google... Please restart Cell CLI to continue.
         feedbackService,
         toggleDebugProfiler,
     ]);
-    const { handleSlashCommand, slashCommands, pendingHistoryItems: pendingSlashCommandHistoryItems, commandContext, shellConfirmationRequest, confirmationRequest, } = useSlashCommandProcessor(config, settings, historyManager.addItem, historyManager.clearItems, historyManager.loadHistory, refreshStatic, toggleVimEnabled, setIsProcessing, slashCommandActions, extensionsUpdateStateInternal, isConfigInitialized, setBannerVisible);
+    const { handleSlashCommand, slashCommands, pendingHistoryItems: pendingSlashCommandHistoryItems, commandContext, shellConfirmationRequest, confirmationRequest, } = useSlashCommandProcessor(config, settings, historyManager.addItem, historyManager.clearItems, historyManager.loadHistory, refreshStatic, toggleVimEnabled, setIsProcessing, slashCommandActions, extensionsUpdateStateInternal, isConfigInitialized, setBannerVisible, setCustomDialog);
     const performMemoryRefresh = useCallback(async () => {
         historyManager.addItem({
             type: MessageType.INFO,
@@ -442,10 +478,28 @@ Logging in with Google... Please restart Cell CLI to continue.
     }, [config, historyManager]);
     const cancelHandlerRef = useRef(() => { });
     const getPreferredEditor = useCallback(() => settings.merged.general?.preferredEditor, [settings.merged.general?.preferredEditor]);
-    const onCancelSubmit = useCallback(() => {
-        cancelHandlerRef.current();
+    const onCancelSubmit = useCallback((shouldRestorePrompt) => {
+        if (shouldRestorePrompt) {
+            setPendingRestorePrompt(true);
+        }
+        else {
+            setPendingRestorePrompt(false);
+            cancelHandlerRef.current(false);
+        }
     }, []);
-    const { streamingState, submitQuery, initError, pendingHistoryItems: pendingGeminiHistoryItems, thought, cancelOngoingRequest, handleApprovalModeChange, activePtyId, loopDetectionConfirmationRequest, } = useGeminiStream(config.getGeminiClient(), historyManager.history, historyManager.addItem, config, settings, setDebugMessage, handleSlashCommand, shellModeActive, getPreferredEditor, onAuthError, performMemoryRefresh, modelSwitchedFromQuotaError, setModelSwitchedFromQuotaError, refreshStatic, onCancelSubmit, setEmbeddedShellFocused, terminalWidth, terminalHeight, embeddedShellFocused);
+    useEffect(() => {
+        if (pendingRestorePrompt) {
+            const lastHistoryUserMsg = historyManager.history.findLast((h) => h.type === 'user');
+            const lastUserMsg = userMessages.at(-1);
+            if (!lastHistoryUserMsg ||
+                (typeof lastHistoryUserMsg.text === 'string' &&
+                    lastHistoryUserMsg.text === lastUserMsg)) {
+                cancelHandlerRef.current(true);
+                setPendingRestorePrompt(false);
+            }
+        }
+    }, [pendingRestorePrompt, userMessages, historyManager.history]);
+    const { streamingState, submitQuery, initError, pendingHistoryItems: pendingGeminiHistoryItems, thought, cancelOngoingRequest, handleApprovalModeChange, activePtyId, loopDetectionConfirmationRequest, lastOutputTime, } = useGeminiStream(config.getGeminiClient(), historyManager.history, historyManager.addItem, config, settings, setDebugMessage, handleSlashCommand, shellModeActive, getPreferredEditor, onAuthError, performMemoryRefresh, modelSwitchedFromQuotaError, setModelSwitchedFromQuotaError, onCancelSubmit, setEmbeddedShellFocused, terminalWidth, terminalHeight, embeddedShellFocused);
     // Auto-accept indicator
     const showAutoAcceptIndicator = useAutoAcceptIndicator({
         config,
@@ -457,7 +511,7 @@ Logging in with Google... Please restart Cell CLI to continue.
         streamingState,
         submitQuery,
     });
-    cancelHandlerRef.current = useCallback(() => {
+    cancelHandlerRef.current = useCallback((shouldRestorePrompt = true) => {
         const pendingHistoryItems = [
             ...pendingSlashCommandHistoryItems,
             ...pendingGeminiHistoryItems,
@@ -467,13 +521,13 @@ Logging in with Google... Please restart Cell CLI to continue.
             return;
         }
         const lastUserMessage = userMessages.at(-1);
-        let textToSet = lastUserMessage || '';
+        let textToSet = shouldRestorePrompt ? lastUserMessage || '' : '';
         const queuedText = getQueuedMessagesText();
         if (queuedText) {
             textToSet = textToSet ? `${textToSet}\n\n${queuedText}` : queuedText;
             clearQueue();
         }
-        if (textToSet) {
+        if (textToSet || !shouldRestorePrompt) {
             buffer.setText(textToSet);
         }
     }, [
@@ -615,6 +669,34 @@ Logging in with Google... Please restart Cell CLI to continue.
     const { isFolderTrustDialogOpen, handleFolderTrustSelect, isRestarting } = useFolderTrust(settings, setIsTrustedFolder, historyManager.addItem);
     const { needsRestart: ideNeedsRestart, restartReason: ideTrustRestartReason, } = useIdeTrustListener();
     const isInitialMount = useRef(true);
+    useIncludeDirsTrust(config, isTrustedFolder, historyManager, setCustomDialog);
+    useEffect(() => {
+        let timeoutId;
+        const handleWarning = (message) => {
+            setWarningMessage(message);
+            if (timeoutId) {
+                clearTimeout(timeoutId);
+            }
+            timeoutId = setTimeout(() => {
+                setWarningMessage(null);
+            }, WARNING_PROMPT_DURATION_MS);
+        };
+        const handleSelectionWarning = () => {
+            handleWarning('Press Ctrl-S to enter selection mode to copy text.');
+        };
+        const handlePasteTimeout = () => {
+            handleWarning('Paste Timed out. Possibly due to slow connection.');
+        };
+        appEvents.on(AppEvent.SelectionWarning, handleSelectionWarning);
+        appEvents.on(AppEvent.PasteTimeout, handlePasteTimeout);
+        return () => {
+            appEvents.off(AppEvent.SelectionWarning, handleSelectionWarning);
+            appEvents.off(AppEvent.PasteTimeout, handlePasteTimeout);
+            if (timeoutId) {
+                clearTimeout(timeoutId);
+            }
+        };
+    }, []);
     useEffect(() => {
         let timeoutId;
         const handleWarning = (message) => {
@@ -680,19 +762,10 @@ Logging in with Google... Please restart Cell CLI to continue.
             setConstrainHeight(false);
         };
         appEvents.on(AppEvent.OpenDebugConsole, openDebugConsole);
-        const logErrorHandler = (errorMessage) => {
-            handleNewMessage({
-                type: 'error',
-                content: String(errorMessage),
-                count: 1,
-            });
-        };
-        appEvents.on(AppEvent.LogError, logErrorHandler);
         return () => {
             appEvents.off(AppEvent.OpenDebugConsole, openDebugConsole);
-            appEvents.off(AppEvent.LogError, logErrorHandler);
         };
-    }, [handleNewMessage, config]);
+    }, [config]);
     useEffect(() => {
         if (ctrlCTimerRef.current) {
             clearTimeout(ctrlCTimerRef.current);
@@ -702,7 +775,7 @@ Logging in with Google... Please restart Cell CLI to continue.
             recordExitFail(config);
         }
         if (ctrlCPressCount > 1) {
-            handleSlashCommand('/quit');
+            handleSlashCommand('/quit', undefined, undefined, false);
         }
         else {
             ctrlCTimerRef.current = setTimeout(() => {
@@ -720,7 +793,7 @@ Logging in with Google... Please restart Cell CLI to continue.
             recordExitFail(config);
         }
         if (ctrlDPressCount > 1) {
-            handleSlashCommand('/quit');
+            handleSlashCommand('/quit', undefined, undefined, false);
         }
         else {
             ctrlDTimerRef.current = setTimeout(() => {
@@ -742,7 +815,7 @@ Logging in with Google... Please restart Cell CLI to continue.
         }
         setIdePromptAnswered(true);
     }, [handleSlashCommand, settings]);
-    const { elapsedTime, currentLoadingPhrase } = useLoadingIndicator(streamingState, settings.merged.ui?.customWittyPhrases);
+    const { elapsedTime, currentLoadingPhrase } = useLoadingIndicator(streamingState, settings.merged.ui?.customWittyPhrases, !!activePtyId && !embeddedShellFocused, lastOutputTime);
     const handleGlobalKeypress = useCallback((key) => {
         if (copyModeEnabled) {
             setCopyModeEnabled(false);
@@ -885,7 +958,7 @@ Logging in with Google... Please restart Cell CLI to continue.
         coreEvents.on(CoreEvent.UserFeedback, handleUserFeedback);
         // Flush any messages that happened during startup before this component
         // mounted.
-        coreEvents.drainFeedbackBacklog();
+        coreEvents.drainBacklogs();
         return () => {
             coreEvents.off(CoreEvent.UserFeedback, handleUserFeedback);
         };
@@ -905,6 +978,7 @@ Logging in with Google... Please restart Cell CLI to continue.
         isFolderTrustDialogOpen ||
         !!shellConfirmationRequest ||
         !!confirmationRequest ||
+        !!customDialog ||
         confirmUpdateExtensionRequests.length > 0 ||
         !!loopDetectionConfirmationRequest ||
         isThemeDialogOpen ||
@@ -917,6 +991,7 @@ Logging in with Google... Please restart Cell CLI to continue.
         showPrivacyNotice ||
         showIdeRestartPrompt ||
         !!proQuotaRequest ||
+        isSessionBrowserOpen ||
         isAuthDialogOpen ||
         authState === AuthState.AwaitingApiKeyInput;
     const pendingHistoryItems = useMemo(() => [...pendingSlashCommandHistoryItems, ...pendingGeminiHistoryItems], [pendingSlashCommandHistoryItems, pendingGeminiHistoryItems]);
@@ -962,7 +1037,7 @@ Logging in with Google... Please restart Cell CLI to continue.
         isAuthenticating,
         isConfigInitialized,
         authError,
-        authMessage: authError ?? '',
+        authMessage: authError ?? authInfo ?? '',
         isAuthDialogOpen,
         isAwaitingApiKeyInput: authState === AuthState.AwaitingApiKeyInput,
         apiKeyDefaultValue,
@@ -978,8 +1053,10 @@ Logging in with Google... Please restart Cell CLI to continue.
         },
         quittingMessages,
         isSettingsDialogOpen,
+        isSessionBrowserOpen,
         isModelDialogOpen,
         isPermissionsDialogOpen,
+        permissionsDialogProps,
         slashCommands,
         pendingSlashCommandHistoryItems,
         commandContext,
@@ -1044,6 +1121,7 @@ Logging in with Google... Please restart Cell CLI to continue.
         activePtyId,
         embeddedShellFocused,
         showDebugProfiler,
+        customDialog,
         copyModeEnabled,
         warningMessage,
         bannerData: {
@@ -1057,6 +1135,7 @@ Logging in with Google... Please restart Cell CLI to continue.
         isAuthenticating,
         isConfigInitialized,
         authError,
+        authInfo,
         isAuthDialogOpen,
         editorError,
         isEditorDialogOpen,
@@ -1065,8 +1144,10 @@ Logging in with Google... Please restart Cell CLI to continue.
         debugMessage,
         quittingMessages,
         isSettingsDialogOpen,
+        isSessionBrowserOpen,
         isModelDialogOpen,
         isPermissionsDialogOpen,
+        permissionsDialogProps,
         slashCommands,
         pendingSlashCommandHistoryItems,
         commandContext,
@@ -1133,6 +1214,7 @@ Logging in with Google... Please restart Cell CLI to continue.
         embeddedShellFocused,
         awaitingFeedback,
         showDebugProfiler,
+        customDialog,
         apiKeyDefaultValue,
         authState,
         copyModeEnabled,
@@ -1154,6 +1236,7 @@ Logging in with Google... Please restart Cell CLI to continue.
         exitPrivacyNotice,
         closeSettingsDialog,
         closeModelDialog,
+        openPermissionsDialog,
         closePermissionsDialog,
         setShellModeActive,
         vimHandleInput,
@@ -1165,11 +1248,16 @@ Logging in with Google... Please restart Cell CLI to continue.
         handleFinalSubmit,
         handleClearScreen,
         handleProQuotaChoice,
+        openSessionBrowser,
+        closeSessionBrowser,
+        handleResumeSession,
+        handleDeleteSession,
         setQueueErrorMessage,
         popAllMessages,
         handleApiKeySubmit,
         handleApiKeyCancel,
         setBannerVisible,
+        setEmbeddedShellFocused,
     }), [
         handleThemeSelect,
         closeThemeDialog,
@@ -1182,6 +1270,7 @@ Logging in with Google... Please restart Cell CLI to continue.
         exitPrivacyNotice,
         closeSettingsDialog,
         closeModelDialog,
+        openPermissionsDialog,
         closePermissionsDialog,
         setShellModeActive,
         vimHandleInput,
@@ -1193,11 +1282,16 @@ Logging in with Google... Please restart Cell CLI to continue.
         handleFinalSubmit,
         handleClearScreen,
         handleProQuotaChoice,
+        openSessionBrowser,
+        closeSessionBrowser,
+        handleResumeSession,
+        handleDeleteSession,
         setQueueErrorMessage,
         popAllMessages,
         handleApiKeySubmit,
         handleApiKeyCancel,
         setBannerVisible,
+        setEmbeddedShellFocused,
     ]);
     return (_jsx(UIStateContext.Provider, { value: uiState, children: _jsx(UIActionsContext.Provider, { value: uiActions, children: _jsx(ConfigContext.Provider, { value: config, children: _jsx(AppContext.Provider, { value: {
                         version: props.version,