diff options
Diffstat (limited to 'packages/cli/src/ui/App.tsx')
| -rw-r--r-- | packages/cli/src/ui/App.tsx | 90 |
1 files changed, 90 insertions, 0 deletions
diff --git a/packages/cli/src/ui/App.tsx b/packages/cli/src/ui/App.tsx new file mode 100644 index 00000000..32dcaac0 --- /dev/null +++ b/packages/cli/src/ui/App.tsx @@ -0,0 +1,90 @@ +import React, { useState, useEffect } from 'react'; +import { Box, Text } from 'ink'; +import type { HistoryItem } from './types.js'; +import { useGeminiStream } from './hooks/useGeminiStream.js'; +import { useLoadingIndicator } from './hooks/useLoadingIndicator.js'; +import Header from './components/Header.js'; +import Tips from './components/Tips.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 { StreamingState } from '../core/StreamingState.js'; +import { PartListUnion } from '@google/genai'; + +interface AppProps { + directory: string; +} + +const App = ({ directory }: AppProps) => { + const [query, setQuery] = useState(''); + const [history, setHistory] = useState<HistoryItem[]>([]); + const { streamingState, submitQuery, initError } = useGeminiStream(setHistory); + const { elapsedTime, currentLoadingPhrase } = useLoadingIndicator(streamingState); + + const handleInputSubmit = (value: PartListUnion) => { + submitQuery(value).then(() => { + setQuery(''); + }).catch(() => { + setQuery(''); + }); + }; + + useEffect(() => { + if (initError && !history.some(item => item.type === 'error' && item.text?.includes(initError))) { + setHistory(prev => [ + ...prev, + { id: Date.now(), type: 'error', text: `Initialization Error: ${initError}. Please check API key and configuration.` } as HistoryItem + ]); + } + }, [initError, history]); + + const isWaitingForToolConfirmation = history.some(item => + item.type === 'tool_group' && item.tools.some(tool => tool.confirmationDetails !== undefined) + ); + const isInputActive = streamingState === StreamingState.Idle && !initError; + + + return ( + <Box flexDirection="column" padding={1} marginBottom={1} width="100%"> + <Header cwd={directory} /> + + <Tips /> + + {initError && streamingState !== StreamingState.Responding && !isWaitingForToolConfirmation && ( + <Box borderStyle="round" borderColor="red" paddingX={1} marginBottom={1}> + {history.find(item => item.type === 'error' && item.text?.includes(initError))?.text ? ( + <Text color="red">{history.find(item => item.type === 'error' && item.text?.includes(initError))?.text}</Text> + ) : ( + <> + <Text color="red">Initialization Error: {initError}</Text> + <Text color="red"> Please check API key and configuration.</Text> + </> + )} + </Box> + )} + + <Box flexDirection="column"> + <HistoryDisplay history={history} onSubmit={handleInputSubmit} /> + <LoadingIndicator + isLoading={streamingState === StreamingState.Responding} + currentLoadingPhrase={currentLoadingPhrase} + elapsedTime={elapsedTime} + /> + </Box> + + {!isWaitingForToolConfirmation && isInputActive && ( + <InputPrompt + query={query} + setQuery={setQuery} + onSubmit={handleInputSubmit} + isActive={isInputActive} + /> + )} + + <Footer queryLength={query.length} /> + </Box> + ); +}; + +export default App;
\ No newline at end of file |
