feat(codex): implement CLI argument passthrough for codex

Collect unknown CLI arguments from index.ts and pass them through the execution chain (runCodex → loop → CodexSession → codexLocalLauncher → codexLocal), similar to how claude already works. This enables users to pass CLI arguments like --model and --sandbox to the underlying codex process.

Filter out the 'resume' subcommand which is managed internally by hapi, while allowing other CLI arguments to pass through. Add warning log in remote mode when CLI args are ignored since remote mode uses message-based configuration instead. Include unit tests for the resume filtering logic.
This commit is contained in:
weishu
2025-12-23 09:06:08 +08:00
parent 0128792495
commit f132a7dc79
8 changed files with 92 additions and 8 deletions
+36
View File
@@ -0,0 +1,36 @@
import { describe, it, expect } from 'vitest';
import { filterResumeSubcommand } from './codexLocal';
describe('filterResumeSubcommand', () => {
it('returns empty array unchanged', () => {
expect(filterResumeSubcommand([])).toEqual([]);
});
it('passes through args when first arg is not resume', () => {
expect(filterResumeSubcommand(['--model', 'gpt-4'])).toEqual(['--model', 'gpt-4']);
expect(filterResumeSubcommand(['--sandbox', 'read-only'])).toEqual(['--sandbox', 'read-only']);
});
it('filters resume subcommand with session ID', () => {
expect(filterResumeSubcommand(['resume', 'abc-123'])).toEqual([]);
expect(filterResumeSubcommand(['resume', 'abc-123', '--model', 'gpt-4']))
.toEqual(['--model', 'gpt-4']);
});
it('filters resume subcommand without session ID', () => {
expect(filterResumeSubcommand(['resume'])).toEqual([]);
expect(filterResumeSubcommand(['resume', '--model', 'gpt-4']))
.toEqual(['--model', 'gpt-4']);
});
it('does not filter resume when it appears as flag value', () => {
// --name resume should pass through (resume is value, not subcommand)
expect(filterResumeSubcommand(['--name', 'resume'])).toEqual(['--name', 'resume']);
});
it('does not filter resume in middle of args', () => {
// If resume appears after flags, it's not the subcommand position
expect(filterResumeSubcommand(['--model', 'gpt-4', 'resume', '123']))
.toEqual(['--model', 'gpt-4', 'resume', '123']);
});
});
+25
View File
@@ -2,6 +2,25 @@ import { spawn } from 'node:child_process';
import { logger } from '@/ui/logger';
import { restoreTerminalState } from '@/ui/terminalState';
/**
* Filter out 'resume' subcommand which is managed internally by hapi.
* Codex CLI format is `codex resume <session-id>`, so subcommand is always first.
*/
export function filterResumeSubcommand(args: string[]): string[] {
if (args.length === 0 || args[0] !== 'resume') {
return args;
}
// First arg is 'resume', filter it and optional session ID
if (args.length > 1 && !args[1].startsWith('-')) {
logger.debug(`[CodexLocal] Filtered 'resume ${args[1]}' - session managed by hapi`);
return args.slice(2);
}
logger.debug(`[CodexLocal] Filtered 'resume' - session managed by hapi`);
return args.slice(1);
}
export async function codexLocal(opts: {
abort: AbortSignal;
sessionId: string | null;
@@ -9,6 +28,7 @@ export async function codexLocal(opts: {
model?: string;
sandbox?: 'read-only' | 'workspace-write' | 'danger-full-access';
onSessionFound: (id: string) => void;
codexArgs?: string[];
}): Promise<void> {
const args: string[] = [];
@@ -25,6 +45,11 @@ export async function codexLocal(opts: {
args.push('--sandbox', opts.sandbox);
}
if (opts.codexArgs) {
const safeArgs = filterResumeSubcommand(opts.codexArgs);
args.push(...safeArgs);
}
logger.debug(`[CodexLocal] Spawning codex with args: ${JSON.stringify(args)}`);
process.stdin.pause();
+2 -1
View File
@@ -78,7 +78,8 @@ export async function codexLocalLauncher(session: CodexSession): Promise<'switch
path: session.path,
sessionId: session.sessionId,
onSessionFound: handleSessionFound,
abort: processAbortController.signal
abort: processAbortController.signal,
codexArgs: session.codexArgs
});
if (!exitReason) {
+6
View File
@@ -22,6 +22,12 @@ import type { EnhancedMode } from './loop';
import { restoreTerminalState } from '@/ui/terminalState';
export async function codexRemoteLauncher(session: CodexSession): Promise<'switch' | 'exit'> {
// Warn if CLI args were passed that won't apply in remote mode
if (session.codexArgs && session.codexArgs.length > 0) {
logger.debug(`[codex-remote] Warning: CLI args [${session.codexArgs.join(', ')}] are ignored in remote mode. ` +
`Remote mode uses message-based configuration (model/sandbox set via web interface).`);
}
const hasTTY = process.stdout.isTTY && process.stdin.isTTY;
const messageBuffer = new MessageBuffer();
let inkInstance: any = null;
+3 -1
View File
@@ -20,6 +20,7 @@ interface LoopOptions {
messageQueue: MessageQueue2<EnhancedMode>;
session: ApiSessionClient;
api: ApiClient;
codexArgs?: string[];
onSessionReady?: (session: CodexSession) => void;
}
@@ -33,7 +34,8 @@ export async function loop(opts: LoopOptions): Promise<void> {
logPath,
messageQueue: opts.messageQueue,
onModeChange: opts.onModeChange,
mode: opts.startingMode ?? 'local'
mode: opts.startingMode ?? 'local',
codexArgs: opts.codexArgs
});
if (opts.onSessionReady) {
+2
View File
@@ -22,6 +22,7 @@ export { emitReadyIfIdle } from './utils/emitReadyIfIdle';
export async function runCodex(opts: {
startedBy?: 'daemon' | 'terminal';
codexArgs?: string[];
}): Promise<void> {
const workingDirectory = process.cwd();
const sessionTag = randomUUID();
@@ -188,6 +189,7 @@ export async function runCodex(opts: {
messageQueue,
api,
session,
codexArgs: opts.codexArgs,
onModeChange: (newMode) => {
session.sendSessionEvent({ type: 'switch', mode: newMode });
session.updateAgentState((currentState) => ({
+5
View File
@@ -4,6 +4,8 @@ import { AgentSessionBase } from '@/agent/sessionBase';
import type { EnhancedMode } from './loop';
export class CodexSession extends AgentSessionBase<EnhancedMode> {
readonly codexArgs?: string[];
constructor(opts: {
api: ApiClient;
client: ApiSessionClient;
@@ -13,6 +15,7 @@ export class CodexSession extends AgentSessionBase<EnhancedMode> {
messageQueue: MessageQueue2<EnhancedMode>;
onModeChange: (mode: 'local' | 'remote') => void;
mode?: 'local' | 'remote';
codexArgs?: string[];
}) {
super({
api: opts.api,
@@ -30,6 +33,8 @@ export class CodexSession extends AgentSessionBase<EnhancedMode> {
codexSessionId: sessionId
})
});
this.codexArgs = opts.codexArgs;
}
sendCodexMessage = (message: unknown): void => {
+13 -6
View File
@@ -125,18 +125,25 @@ import { withBunRuntimeEnv } from './utils/bunRuntime'
// Handle codex command
try {
const { runCodex } = await import('@/codex/runCodex');
// Parse startedBy argument
let startedBy: 'daemon' | 'terminal' | undefined = undefined;
// Parse known arguments and collect unknown ones for passthrough
const options: { startedBy?: 'daemon' | 'terminal'; codexArgs?: string[] } = {};
const unknownArgs: string[] = [];
for (let i = 1; i < args.length; i++) {
if (args[i] === '--started-by') {
startedBy = args[++i] as 'daemon' | 'terminal';
const arg = args[i];
if (arg === '--started-by') {
options.startedBy = args[++i] as 'daemon' | 'terminal';
} else {
unknownArgs.push(arg);
}
}
if (unknownArgs.length > 0) {
options.codexArgs = unknownArgs;
}
await initializeToken();
await authAndSetupMachineIfNeeded();
await runCodex({ startedBy });
await runCodex(options);
// Do not force exit here; allow instrumentation to show lingering handles
} catch (error) {
console.error(chalk.red('Error:'), error instanceof Error ? error.message : 'Unknown error')