From a66ad2e2af3d4ffd0a7ab125215e7ed3c3325452 Mon Sep 17 00:00:00 2001
From: Juliette Love <juliette29love@gmail.com>
Date: Sun, 20 Apr 2025 20:20:40 +0100
Subject: [PATCH] Simple debug mode for CLI side (#66)

* Adds debug mode config flag.

* Wire through debug lines

* Add debug mode logging example

* Run format

* Run format again
---
 packages/cli/src/config/config.ts            |  8 ++++++++
 packages/cli/src/ui/App.tsx                  | 13 +++++++------
 packages/cli/src/ui/components/Footer.tsx    | 11 ++++++++++-
 packages/cli/src/ui/hooks/useGeminiStream.ts |  7 ++++++-
 packages/server/src/config/config.ts         | 16 ++++++++++++++--
 5 files changed, 45 insertions(+), 10 deletions(-)

diff --git a/packages/cli/src/config/config.ts b/packages/cli/src/config/config.ts
index 48cc96a0..6d8c10f6 100644
--- a/packages/cli/src/config/config.ts
+++ b/packages/cli/src/config/config.ts
@@ -20,6 +20,7 @@ const DEFAULT_GEMINI_MODEL = 'gemini-2.5-flash-preview-04-17';
 interface CliArgs {
   target_dir: string | undefined;
   model: string | undefined;
+  debug_mode: boolean | undefined;
 }
 
 function parseArguments(): CliArgs {
@@ -36,6 +37,12 @@ function parseArguments(): CliArgs {
       description: `The Gemini model to use. Defaults to ${DEFAULT_GEMINI_MODEL}.`,
       default: DEFAULT_GEMINI_MODEL,
     })
+    .option('debug_mode', {
+      alias: 'z',
+      type: 'boolean',
+      description: 'Whether to run in debug mode. Defaults to false.',
+      default: false,
+    })
     .help()
     .alias('h', 'help')
     .strict().argv;
@@ -64,6 +71,7 @@ export function loadCliConfig(): Config {
     process.env.GEMINI_API_KEY,
     argv.model || DEFAULT_GEMINI_MODEL,
     argv.target_dir || process.cwd(),
+    argv.debug_mode || false,
   );
 }
 
diff --git a/packages/cli/src/ui/App.tsx b/packages/cli/src/ui/App.tsx
index 14b76c8f..cfbc024e 100644
--- a/packages/cli/src/ui/App.tsx
+++ b/packages/cli/src/ui/App.tsx
@@ -31,11 +31,8 @@ interface AppProps {
 export const App = ({ config }: AppProps) => {
   const [history, setHistory] = useState<HistoryItem[]>([]);
   const [startupWarnings, setStartupWarnings] = useState<string[]>([]);
-  const { streamingState, submitQuery, initError } = useGeminiStream(
-    setHistory,
-    config.getApiKey(),
-    config.getModel(),
-  );
+  const { streamingState, submitQuery, initError, debugMessage } =
+    useGeminiStream(setHistory, config.getApiKey(), config.getModel());
   const { elapsedTime, currentLoadingPhrase } =
     useLoadingIndicator(streamingState);
 
@@ -147,7 +144,11 @@ export const App = ({ config }: AppProps) => {
         </>
       )}
 
-      <Footer queryLength={query.length} />
+      <Footer
+        queryLength={query.length}
+        debugMode={config.getDebugMode()}
+        debugMessage={debugMessage}
+      />
       <ITermDetectionWarning />
     </Box>
   );
diff --git a/packages/cli/src/ui/components/Footer.tsx b/packages/cli/src/ui/components/Footer.tsx
index 3b4e197b..35ed6910 100644
--- a/packages/cli/src/ui/components/Footer.tsx
+++ b/packages/cli/src/ui/components/Footer.tsx
@@ -10,13 +10,22 @@ import { Colors } from '../colors.js';
 
 interface FooterProps {
   queryLength: number;
+  debugMode: boolean;
+  debugMessage: string;
 }
 
-export const Footer: React.FC<FooterProps> = ({ queryLength }) => (
+export const Footer: React.FC<FooterProps> = ({
+  queryLength,
+  debugMode,
+  debugMessage,
+}) => (
   <Box marginTop={1} justifyContent="space-between">
     <Box minWidth={15}>
       <Text color={Colors.SubtleComment}>
         {queryLength === 0 ? '? for shortcuts' : ''}
+        {debugMode && (
+          <Text color="red"> {debugMessage || 'Running in debug mode.'}</Text>
+        )}
       </Text>
     </Box>
     <Text color={Colors.AccentBlue}>Gemini</Text>
diff --git a/packages/cli/src/ui/hooks/useGeminiStream.ts b/packages/cli/src/ui/hooks/useGeminiStream.ts
index 56203179..1d839998 100644
--- a/packages/cli/src/ui/hooks/useGeminiStream.ts
+++ b/packages/cli/src/ui/hooks/useGeminiStream.ts
@@ -49,6 +49,7 @@ export const useGeminiStream = (
   const [streamingState, setStreamingState] = useState<StreamingState>(
     StreamingState.Idle,
   );
+  const [debugMessage, setDebugMessage] = useState<string>('');
   const [initError, setInitError] = useState<string | null>(null);
   const abortControllerRef = useRef<AbortController | null>(null);
   const chatSessionRef = useRef<Chat | null>(null);
@@ -104,6 +105,10 @@ export const useGeminiStream = (
       if (streamingState === StreamingState.Responding) return;
       if (typeof query === 'string' && query.trim().length === 0) return;
 
+      if (typeof query === 'string') {
+        setDebugMessage(`User query: ${query}`);
+      }
+
       const userMessageTimestamp = Date.now();
       const client = geminiClientRef.current;
       if (!client) {
@@ -403,7 +408,7 @@ export const useGeminiStream = (
     ],
   );
 
-  return { streamingState, submitQuery, initError };
+  return { streamingState, submitQuery, initError, debugMessage };
 };
 
 // Define ServerTool interface here if not importing from server (circular dep issue?)
diff --git a/packages/server/src/config/config.ts b/packages/server/src/config/config.ts
index 86cd7a6c..bd698cf6 100644
--- a/packages/server/src/config/config.ts
+++ b/packages/server/src/config/config.ts
@@ -13,11 +13,18 @@ export class Config {
   private apiKey: string;
   private model: string;
   private targetDir: string;
+  private debugMode: boolean;
 
-  constructor(apiKey: string, model: string, targetDir: string) {
+  constructor(
+    apiKey: string,
+    model: string,
+    targetDir: string,
+    debugMode: boolean,
+  ) {
     this.apiKey = apiKey;
     this.model = model;
     this.targetDir = targetDir;
+    this.debugMode = debugMode;
   }
 
   getApiKey(): string {
@@ -31,6 +38,10 @@ export class Config {
   getTargetDir(): string {
     return this.targetDir;
   }
+
+  getDebugMode(): boolean {
+    return this.debugMode;
+  }
 }
 
 function findEnvFile(startDir: string): string | null {
@@ -60,6 +71,7 @@ export function createServerConfig(
   apiKey: string,
   model: string,
   targetDir: string,
+  debugMode: boolean,
 ): Config {
-  return new Config(apiKey, model, path.resolve(targetDir));
+  return new Config(apiKey, model, path.resolve(targetDir), debugMode);
 }