Support mode cli args

Author: jahooma

cli/src/app.tsx

@@ -21,6 +21,7 @@ import { openFileAtPath } from './utils/open-file'
 
 import type { MultilineInputHandle } from './components/multiline-input'
 import type { AuthStatus } from './utils/status-indicator-state'
+import type { AgentMode } from './utils/constants'
 import type { FileTreeNode } from '@codebuff/common/util/file'
 
 interface AppProps {
@@ -31,6 +32,7 @@ interface AppProps {
   fileTree: FileTreeNode[]
   continueChat: boolean
   continueChatId?: string
+  initialMode?: AgentMode
 }
 
 export const App = ({
@@ -41,6 +43,7 @@ export const App = ({
   fileTree,
   continueChat,
   continueChatId,
+  initialMode,
 }: AppProps) => {
   const { contentMaxWidth, terminalWidth } = useTerminalDimensions()
   const theme = useTheme()
@@ -194,6 +197,7 @@ export const App = ({
       continueChat={continueChat}
       continueChatId={continueChatId}
       authStatus={authStatus}
+      initialMode={initialMode}
     />
   )
 }

cli/src/chat.tsx

@@ -92,6 +92,7 @@ export const Chat = ({
   continueChat,
   continueChatId,
   authStatus,
+  initialMode,
 }: {
   headerContent: React.ReactNode
   initialPrompt: string | null
@@ -104,6 +105,7 @@ export const Chat = ({
   continueChat: boolean
   continueChatId?: string
   authStatus: AuthStatus
+  initialMode?: AgentMode
 }) => {
   const scrollRef = useRef<ScrollBoxRenderable | null>(null)
   const [hasOverflow, setHasOverflow] = useState(false)
@@ -229,6 +231,13 @@ export const Chat = ({
   const mainAgentTimer = useElapsedTime()
   const timerStartTime = mainAgentTimer.startTime
 
+  // Set initial mode from CLI flag on mount
+  useEffect(() => {
+    if (initialMode) {
+      setAgentMode(initialMode)
+    }
+  }, [initialMode, setAgentMode])
+
   // Sync refs with state
   useEffect(() => {
     isChainInProgressRef.current = isChainInProgress

cli/src/index.tsx

@@ -28,6 +28,7 @@ import { detectTerminalTheme } from './utils/terminal-color-detection'
 import { setOscDetectedTheme } from './utils/theme-system'
 
 import type { FileTreeNode } from '@codebuff/common/util/file'
+import type { AgentMode } from './utils/constants'
 
 const require = createRequire(import.meta.url)
 
@@ -82,6 +83,7 @@ type ParsedArgs = {
   continue: boolean
   continueId?: string | null
   cwd?: string
+  initialMode?: AgentMode
 }
 
 function parseArgs(): ParsedArgs {
@@ -104,6 +106,9 @@ function parseArgs(): ParsedArgs {
       '--cwd <directory>',
       'Set the working directory (default: current directory)',
     )
+    .option('--lite', 'Start in LITE mode')
+    .option('--max', 'Start in MAX mode')
+    .option('--plan', 'Start in PLAN mode')
     .helpOption('-h, --help', 'Show this help message')
     .argument('[prompt...]', 'Initial prompt to send to the agent')
     .allowExcessArguments(true)
@@ -114,6 +119,12 @@ function parseArgs(): ParsedArgs {
 
   const continueFlag = options.continue
 
+  // Determine initial mode from flags (last flag wins if multiple specified)
+  let initialMode: AgentMode | undefined
+  if (options.lite) initialMode = 'LITE'
+  if (options.max) initialMode = 'MAX'
+  if (options.plan) initialMode = 'PLAN'
+
   return {
     initialPrompt: args.length > 0 ? args.join(' ') : null,
     agent: options.agent,
@@ -124,6 +135,7 @@ function parseArgs(): ParsedArgs {
         ? continueFlag.trim()
         : null,
     cwd: options.cwd,
+    initialMode,
   }
 }
 
@@ -149,6 +161,7 @@ async function main(): Promise<void> {
     continue: continueChat,
     continueId,
     cwd,
+    initialMode,
   } = parseArgs()
 
   await initializeApp({ cwd })
@@ -245,6 +258,7 @@ async function main(): Promise<void> {
         fileTree={fileTree}
         continueChat={continueChat}
         continueChatId={continueId ?? undefined}
+        initialMode={initialMode}
       />
     )
   }