feat: Patch console.debug and display only in debug mode

- Patches `console.debug` in `ConsolePatcher.tsx` to capture debug messages.
- Updates `ConsoleOutput` to only display debug messages when `debugMode` is enabled.
- Passes `debugMode` prop from `App.tsx` to `ConsoleOutput`.

Fixes https://github.com/google-gemini/gemini-cli/issues/397
This commit is contained in:
Taylor Mullen 2025-05-16 22:26:03 -07:00 committed by N. Taylor Mullen
parent e0b88dc8da
commit e486d84d6a
2 changed files with 22 additions and 4 deletions

View File

@ -409,7 +409,7 @@ export const App = ({
cliVersion={cliVersion}
geminiMdFileCount={geminiMdFileCount}
/>
<ConsoleOutput />
<ConsoleOutput debugMode={config.getDebugMode()} />
</Box>
</Box>
);

View File

@ -10,7 +10,7 @@ import util from 'util';
interface ConsoleMessage {
id: Key;
type: 'log' | 'warn' | 'error';
type: 'log' | 'warn' | 'error' | 'debug';
content: string;
}
@ -18,16 +18,24 @@ interface ConsoleMessage {
// This ensures IDs are unique across messages.
let messageIdCounter = 0;
export const ConsoleOutput: React.FC = () => {
interface ConsoleOutputProps {
debugMode: boolean;
}
export const ConsoleOutput: React.FC<ConsoleOutputProps> = ({ debugMode }) => {
const [messages, setMessages] = useState<ConsoleMessage[]>([]);
useEffect(() => {
const originalConsoleLog = console.log;
const originalConsoleWarn = console.warn;
const originalConsoleError = console.error;
const originalConsoleDebug = console.debug;
const formatArgs = (args: unknown[]): string => util.format(...args);
const addMessage = (type: 'log' | 'warn' | 'error', args: unknown[]) => {
const addMessage = (
type: 'log' | 'warn' | 'error' | 'debug',
args: unknown[],
) => {
setMessages((prevMessages) => [
...prevMessages,
{
@ -42,17 +50,23 @@ export const ConsoleOutput: React.FC = () => {
console.log = (...args: unknown[]) => addMessage('log', args);
console.warn = (...args: unknown[]) => addMessage('warn', args);
console.error = (...args: unknown[]) => addMessage('error', args);
console.debug = (...args: unknown[]) => addMessage('debug', args);
return () => {
console.log = originalConsoleLog;
console.warn = originalConsoleWarn;
console.error = originalConsoleError;
console.debug = originalConsoleDebug;
};
}, []);
return (
<Box flexDirection="column">
{messages.map((msg) => {
if (msg.type === 'debug' && !debugMode) {
return null;
}
const textProps: { color?: string } = {};
let prefix = '';
@ -65,6 +79,10 @@ export const ConsoleOutput: React.FC = () => {
textProps.color = 'red';
prefix = 'ERROR: ';
break;
case 'debug':
textProps.color = 'gray';
prefix = 'DEBUG: ';
break;
case 'log':
default:
prefix = 'LOG: ';