/** * @license * Copyright 2025 Google LLC * SPDX-License-Identifier: Apache-2.0 */ import React, { useState, useMemo, useEffect } from 'react'; // Added useEffect import { Box, Text } from 'ink'; import { StreamingState, type HistoryItem } from './types.js'; import { useGeminiStream } from './hooks/useGeminiStream.js'; import { useLoadingIndicator } from './hooks/useLoadingIndicator.js'; import { useInputHistory } from './hooks/useInputHistory.js'; import { Header } from './components/Header.js'; import { HistoryDisplay } from './components/HistoryDisplay.js'; import { LoadingIndicator } from './components/LoadingIndicator.js'; import { InputPrompt } from './components/InputPrompt.js'; import { Footer } from './components/Footer.js'; import { ITermDetectionWarning } from './utils/itermDetection.js'; import { useStartupWarnings, useInitializationErrorEffect, } from './hooks/useAppEffects.js'; import { shortenPath, type Config } from '@gemini-code/server'; import { Colors } from './colors.js'; interface AppProps { config: Config; initialInput?: string; // Added optional prop } export const App = ({ config, initialInput }: AppProps) => { // Destructured prop const [history, setHistory] = useState([]); const [startupWarnings, setStartupWarnings] = useState([]); const { streamingState, submitQuery, initError, debugMessage } = useGeminiStream(setHistory, config); const { elapsedTime, currentLoadingPhrase } = useLoadingIndicator(streamingState); useStartupWarnings(setStartupWarnings); useInitializationErrorEffect(initError, history, setHistory); // Effect to handle initial piped input useEffect(() => { if (initialInput && initialInput.trim() !== '') { submitQuery(initialInput); } // Run only once on mount // eslint-disable-next-line react-hooks/exhaustive-deps }, []); const userMessages = useMemo( () => history .filter( (item): item is HistoryItem & { type: 'user'; text: string } => item.type === 'user' && typeof item.text === 'string' && item.text.trim() !== '', ) .map((item) => item.text), [history], ); const isInputActive = streamingState === StreamingState.Idle && !initError; const { query, handleSubmit: handleHistorySubmit } = useInputHistory({ userMessages, onSubmit: submitQuery, isActive: isInputActive, }); return (
{startupWarnings.length > 0 && ( {startupWarnings.map((warning, index) => ( {warning} ))} )} {initError && streamingState !== StreamingState.Responding && ( {history.find( (item) => item.type === 'error' && item.text?.includes(initError), )?.text ? ( { history.find( (item) => item.type === 'error' && item.text?.includes(initError), )?.text } ) : ( <> Initialization Error: {initError} {' '} Please check API key and configuration. )} )} {isInputActive && ( <> cwd: {shortenPath(config.getTargetDir(), /*maxLength*/ 70)} )}