refactor(delegation): modularize headless-executor into executor/ directory

- extract types, stream-parser, result-aggregator

- consolidate ExecutionResult to executor/types.ts (DRY)

- slim headless-executor.ts from 729 to 391 lines (46% reduction)

- add barrel exports at executor/index.ts and delegation/index.ts
This commit is contained in:
kaitranntt
2025-12-19 12:35:59 -05:00
parent 5e4fa200df
commit c3baaa8251
7 changed files with 480 additions and 464 deletions
+7
View File
@@ -0,0 +1,7 @@
/**
* Barrel export for executor module
*/
export * from './types';
export { StreamBuffer, formatToolVerbose } from './stream-parser';
export { buildExecutionResult, extractSessionInfo } from './result-aggregator';
@@ -0,0 +1,80 @@
/**
* Result aggregation utilities for headless executor
*/
import type { ExecutionResult, StreamMessage } from './types';
import { warn } from '../../utils/ui';
/**
* Build execution result from stream messages
* @param params - Parameters for building result
* @returns ExecutionResult with all fields populated
*/
export function buildExecutionResult(params: {
exitCode: number;
stdout: string;
stderr: string;
cwd: string;
profile: string;
duration: number;
timedOut: boolean;
messages: StreamMessage[];
}): ExecutionResult {
const { exitCode, stdout, stderr, cwd, profile, duration, timedOut, messages } = params;
const result: ExecutionResult = {
exitCode,
stdout,
stderr,
cwd,
profile,
duration,
timedOut,
success: exitCode === 0 && !timedOut,
messages,
};
// Extract metadata from final 'result' message in stream-json
const resultMessage = messages.find((m) => m.type === 'result');
if (resultMessage) {
result.sessionId = resultMessage.session_id || undefined;
result.totalCost = resultMessage.total_cost_usd || 0;
result.numTurns = resultMessage.num_turns || 0;
result.isError = resultMessage.is_error || false;
result.type = resultMessage.type || null;
result.subtype = resultMessage.subtype || undefined;
result.durationApi = resultMessage.duration_api_ms || 0;
result.permissionDenials = resultMessage.permission_denials || [];
result.errors = resultMessage.errors || [];
result.content = resultMessage.result || '';
} else {
// Fallback: no result message found (shouldn't happen)
result.content = stdout;
if (process.env.CCS_DEBUG) {
console.error(warn('No result message found in stream-json output'));
}
}
return result;
}
/**
* Extract session info from result for session management
* @param result - Execution result
* @returns Session info or null
*/
export function extractSessionInfo(result: ExecutionResult): {
sessionId: string;
totalCost?: number;
cwd: string;
} | null {
if (!result.sessionId) {
return null;
}
return {
sessionId: result.sessionId,
totalCost: result.totalCost,
cwd: result.cwd,
};
}
+151
View File
@@ -0,0 +1,151 @@
/**
* Stream parsing utilities for Claude CLI stream-json output
*/
import type { StreamMessage, ToolInput } from './types';
import { warn } from '../../utils/ui';
/**
* Buffer for incomplete JSON lines during streaming
*/
export class StreamBuffer {
private partialLine = '';
/**
* Process incoming data chunk and extract complete JSON messages
* @param dataStr - Raw data string from stream
* @returns Array of parsed StreamMessage objects
*/
parseChunk(dataStr: string): StreamMessage[] {
const messages: StreamMessage[] = [];
const chunk = this.partialLine + dataStr;
const lines = chunk.split('\n');
this.partialLine = lines.pop() || ''; // Save incomplete line for next chunk
for (const line of lines) {
if (!line.trim()) continue;
try {
const msg: StreamMessage = JSON.parse(line);
messages.push(msg);
} catch (parseError) {
// Skip malformed JSON lines (shouldn't happen with stream-json)
if (process.env.CCS_DEBUG) {
console.error(warn(`Failed to parse stream-json line: ${(parseError as Error).message}`));
}
}
}
return messages;
}
/**
* Reset buffer state
*/
reset(): void {
this.partialLine = '';
}
}
/**
* Format tool use message for verbose logging
* @param toolName - Name of the tool
* @param toolInput - Tool input parameters
* @returns Formatted verbose message
*/
export function formatToolVerbose(toolName: string, toolInput: ToolInput): string {
let verboseMsg = `[Tool] ${toolName}`;
switch (toolName) {
case 'Bash':
if (toolInput.command) {
const command = toolInput.command as string;
const cmd = command.length > 80 ? command.substring(0, 77) + '...' : command;
verboseMsg += `: ${cmd}`;
}
break;
case 'Edit':
case 'Write':
case 'Read':
if (toolInput.file_path) {
verboseMsg += `: ${toolInput.file_path}`;
}
break;
case 'NotebookEdit':
case 'NotebookRead':
if (toolInput.notebook_path) {
verboseMsg += `: ${toolInput.notebook_path}`;
}
break;
case 'Grep':
if (toolInput.pattern) {
verboseMsg += `: searching for "${toolInput.pattern}"`;
if (toolInput.path) {
verboseMsg += ` in ${toolInput.path}`;
}
}
break;
case 'Glob':
if (toolInput.pattern) {
verboseMsg += `: ${toolInput.pattern}`;
}
break;
case 'SlashCommand':
if (toolInput.command) {
verboseMsg += `: ${toolInput.command}`;
}
break;
case 'Task':
if (toolInput.description) {
verboseMsg += `: ${toolInput.description}`;
} else if (toolInput.prompt) {
const promptText = toolInput.prompt as string;
const prompt = promptText.length > 60 ? promptText.substring(0, 57) + '...' : promptText;
verboseMsg += `: ${prompt}`;
}
break;
case 'TodoWrite':
if (toolInput.todos && Array.isArray(toolInput.todos)) {
const inProgressTask = toolInput.todos.find(
(t: { status: string; activeForm?: string }) => t.status === 'in_progress'
);
if (inProgressTask && inProgressTask.activeForm) {
verboseMsg += `: ${inProgressTask.activeForm}`;
} else {
verboseMsg += `: ${toolInput.todos.length} task(s)`;
}
}
break;
case 'WebFetch':
if (toolInput.url) {
verboseMsg += `: ${toolInput.url}`;
}
break;
case 'WebSearch':
if (toolInput.query) {
verboseMsg += `: "${toolInput.query}"`;
}
break;
default:
// For unknown tools, show first meaningful parameter
if (Object.keys(toolInput).length > 0) {
const firstKey = Object.keys(toolInput)[0];
const firstValue = toolInput[firstKey];
if (typeof firstValue === 'string' && firstValue.length < 60) {
verboseMsg += `: ${firstValue}`;
}
}
}
return verboseMsg;
}
+124
View File
@@ -0,0 +1,124 @@
/**
* Type definitions for headless executor
*/
/**
* Claude message from stream-json output
*/
export interface ClaudeMessage {
type: string;
content?: string;
thinking?: string;
tool_use?: {
id: string;
name: string;
input: Record<string, unknown>;
};
tool_result?: {
tool_use_id: string;
content: string;
};
}
/**
* Permission denial information
*/
export interface PermissionDenial {
tool_name?: string;
reason?: string;
tool_input?: {
command?: string;
description?: string;
[key: string]: unknown;
};
}
/**
* Execution error information
*/
export interface ExecutionError {
message?: string;
error?: string;
type?: string;
tool_name?: string;
[key: string]: unknown;
}
/**
* Options for headless execution
*/
export interface ExecutionOptions {
cwd?: string;
timeout?: number;
outputFormat?: string;
permissionMode?: string;
resumeSession?: boolean;
sessionId?: string;
maxRetries?: number;
extraArgs?: string[]; // Passthrough args for Claude CLI
}
/**
* Result of headless execution
*/
export interface ExecutionResult {
exitCode: number;
stdout: string;
stderr: string;
cwd: string;
profile: string;
duration: number;
timedOut: boolean;
success: boolean;
messages: StreamMessage[];
sessionId?: string;
totalCost?: number;
numTurns?: number;
isError?: boolean;
type?: string | null;
subtype?: string;
durationApi?: number;
permissionDenials?: PermissionDenial[];
errors?: ExecutionError[];
content?: string;
}
/**
* Stream message from Claude CLI stream-json output
*/
export interface StreamMessage {
type: string;
message?: {
content?: Array<{
type: string;
name?: string;
input?: Record<string, unknown>;
}>;
};
session_id?: string;
total_cost_usd?: number;
num_turns?: number;
is_error?: boolean;
result?: string;
duration_api_ms?: number;
permission_denials?: PermissionDenial[];
errors?: ExecutionError[];
subtype?: string;
}
/**
* Tool input types for verbose logging
*/
export interface ToolInput {
command?: string;
file_path?: string;
notebook_path?: string;
pattern?: string;
path?: string;
description?: string;
prompt?: string;
todos?: Array<{ status: string; activeForm?: string }>;
url?: string;
query?: string;
[key: string]: unknown;
}
+102 -431
View File
@@ -1,5 +1,10 @@
#!/usr/bin/env node
/**
* Headless executor for Claude CLI delegation
* Spawns claude with -p flag for single-turn execution
*/
import { spawn } from 'child_process';
import * as path from 'path';
import * as os from 'os';
@@ -7,92 +12,15 @@ import * as fs from 'fs';
import { SessionManager } from './session-manager';
import { SettingsParser } from './settings-parser';
import { ui, warn, info } from '../utils/ui';
import { type ExecutionOptions, type ExecutionResult, type StreamMessage } from './executor/types';
import { StreamBuffer, formatToolVerbose } from './executor/stream-parser';
import { buildExecutionResult } from './executor/result-aggregator';
// Type definitions for delegation responses
interface ClaudeMessage {
type: string;
content?: string;
thinking?: string;
tool_use?: {
id: string;
name: string;
input: Record<string, unknown>;
};
tool_result?: {
tool_use_id: string;
content: string;
};
}
interface PermissionDenial {
tool_name: string;
reason: string;
}
interface ExecutionError {
message?: string;
error?: string;
type?: string;
tool_name?: string;
[key: string]: unknown;
}
interface ExecutionOptions {
cwd?: string;
timeout?: number;
outputFormat?: string;
permissionMode?: string;
resumeSession?: boolean;
sessionId?: string;
maxRetries?: number;
extraArgs?: string[]; // Passthrough args for Claude CLI
}
interface ExecutionResult {
exitCode: number;
stdout: string;
stderr: string;
cwd: string;
profile: string;
duration: number;
timedOut: boolean;
success: boolean;
messages: ClaudeMessage[];
sessionId?: string;
totalCost?: number;
numTurns?: number;
isError?: boolean;
type?: string | null;
subtype?: string;
durationApi?: number;
permissionDenials?: PermissionDenial[];
errors?: ExecutionError[];
content?: string;
}
interface StreamMessage {
type: string;
message?: {
content?: Array<{
type: string;
name?: string;
input?: Record<string, unknown>;
}>;
};
session_id?: string;
total_cost_usd?: number;
num_turns?: number;
is_error?: boolean;
result?: string;
duration_api_ms?: number;
permission_denials?: PermissionDenial[];
errors?: ExecutionError[];
subtype?: string;
}
// Re-export types for consumers
export type { ExecutionOptions, ExecutionResult, StreamMessage } from './executor/types';
/**
* Headless executor for Claude CLI delegation
* Spawns claude with -p flag for single-turn execution
*/
export class HeadlessExecutor {
/**
@@ -141,26 +69,20 @@ export class HeadlessExecutor {
}
// Smart slash command detection and preservation
// Detects if prompt contains slash command and restructures for proper execution
const processedPrompt = this._processSlashCommand(enhancedPrompt);
// Prepare arguments
const args: string[] = ['-p', processedPrompt, '--settings', settingsPath];
// Always use stream-json for real-time progress visibility
// Note: --verbose is required when using --print with stream-json
args.push('--output-format', 'stream-json', '--verbose');
// Add permission mode
if (permissionMode && permissionMode !== 'default') {
if (permissionMode === 'bypassPermissions') {
args.push('--dangerously-skip-permissions');
// Warn about dangerous mode
if (process.env.CCS_DEBUG) {
console.warn(warn('WARNING: Using --dangerously-skip-permissions mode'));
console.warn(
warn('This bypasses ALL permission checks. Use only in trusted environments.')
);
}
} else {
args.push('--permission-mode', permissionMode);
@@ -170,56 +92,35 @@ export class HeadlessExecutor {
// Add resume flag for multi-turn sessions
if (resumeSession) {
const lastSession = sessionMgr.getLastSession(profile);
if (lastSession) {
args.push('--resume', lastSession.sessionId);
if (process.env.CCS_DEBUG) {
const cost =
lastSession.totalCost !== undefined && lastSession.totalCost !== null
? lastSession.totalCost.toFixed(4)
: '0.0000';
console.error(
info(
`Resuming session: ${lastSession.sessionId} (${lastSession.turns} turns, $${cost})`
)
);
const cost = lastSession.totalCost?.toFixed(4) || '0.0000';
console.error(info(`Resuming session: ${lastSession.sessionId} ($${cost})`));
}
} else if (sessionId) {
args.push('--resume', sessionId);
if (process.env.CCS_DEBUG) {
console.error(info(`Resuming specific session: ${sessionId}`));
}
} else {
console.warn(warn('No previous session found, starting new session'));
}
} else if (sessionId) {
args.push('--resume', sessionId);
if (process.env.CCS_DEBUG) {
console.error(info(`Resuming specific session: ${sessionId}`));
}
}
// Add tool restrictions from settings
const toolRestrictions = SettingsParser.parseToolRestrictions(cwd);
if (toolRestrictions.allowedTools.length > 0) {
args.push('--allowedTools');
toolRestrictions.allowedTools.forEach((tool) => args.push(tool));
args.push('--allowedTools', ...toolRestrictions.allowedTools);
}
if (toolRestrictions.disallowedTools.length > 0) {
args.push('--disallowedTools');
toolRestrictions.disallowedTools.forEach((tool) => args.push(tool));
args.push('--disallowedTools', ...toolRestrictions.disallowedTools);
}
// Note: No max-turns limit - using time-based limits instead (default 10min timeout)
// Passthrough extra args (from Claude CLI flags like --agent, --system-prompt-file, etc.)
// Passthrough extra args
if (extraArgs.length > 0) {
args.push(...extraArgs);
}
// Debug log args
if (process.env.CCS_DEBUG) {
console.error(info(`Claude CLI args: ${args.join(' ')}`));
}
@@ -228,13 +129,38 @@ export class HeadlessExecutor {
await ui.init();
// Execute with spawn
return this._spawnAndExecute(claudeCli, args, {
cwd,
profile,
timeout,
resumeSession,
sessionId,
sessionMgr,
});
}
/**
* Spawn Claude CLI and handle execution
*/
private static _spawnAndExecute(
claudeCli: string,
args: string[],
ctx: {
cwd: string;
profile: string;
timeout: number;
resumeSession: boolean;
sessionId: string | null;
sessionMgr: SessionManager;
}
): Promise<ExecutionResult> {
const { cwd, profile, timeout, resumeSession, sessionId, sessionMgr } = ctx;
return new Promise((resolve, reject) => {
const startTime = Date.now();
// Show progress unless explicitly disabled with CCS_QUIET
const showProgress = !process.env.CCS_QUIET;
const streamBuffer = new StreamBuffer();
// Show initial progress message
if (showProgress) {
const modelName =
profile === 'glm' ? 'GLM-4.6' : profile === 'kimi' ? 'Kimi' : profile.toUpperCase();
@@ -250,40 +176,28 @@ export class HeadlessExecutor {
let stdout = '';
let stderr = '';
let progressInterval: NodeJS.Timeout | undefined;
const messages: StreamMessage[] = []; // Accumulate stream-json messages
let partialLine = ''; // Buffer for incomplete JSON lines
const messages: StreamMessage[] = [];
let timedOut = false;
// Handle parent process termination (Ctrl+C or Esc in Claude)
// When main Claude session is killed, cleanup spawned child process
// Setup signal handlers for cleanup
const cleanupHandler = () => {
if (!proc.killed) {
if (process.env.CCS_DEBUG) {
console.error(warn('Parent process terminating, killing delegated session...'));
}
proc.kill('SIGTERM');
// Force kill if not dead after 2s
setTimeout(() => {
if (!proc.killed) {
proc.kill('SIGKILL');
}
if (!proc.killed) proc.kill('SIGKILL');
}, 2000);
}
};
// Register signal handlers for parent process termination
process.once('SIGINT', cleanupHandler);
process.once('SIGTERM', cleanupHandler);
// Cleanup signal handlers when child process exits
const removeSignalHandlers = () => {
process.removeListener('SIGINT', cleanupHandler);
process.removeListener('SIGTERM', cleanupHandler);
};
proc.on('close', removeSignalHandlers);
proc.on('error', removeSignalHandlers);
// Progress indicator (show elapsed time every 5 seconds)
// Progress indicator
if (showProgress) {
progressInterval = setInterval(() => {
const elapsed = ((Date.now() - startTime) / 1000).toFixed(1);
@@ -291,155 +205,34 @@ export class HeadlessExecutor {
}, 5000);
}
// Capture stdout (stream-json format - jsonl)
// Capture stdout (stream-json format)
proc.stdout?.on('data', (data: Buffer) => {
const dataStr = data.toString();
stdout += dataStr;
// Parse stream-json messages (jsonl format - one JSON per line)
const chunk = partialLine + dataStr;
const lines = chunk.split('\n');
partialLine = lines.pop() || ''; // Save incomplete line for next chunk
const parsedMessages = streamBuffer.parseChunk(dataStr);
for (const msg of parsedMessages) {
messages.push(msg);
for (const line of lines) {
if (!line.trim()) continue;
try {
const msg: StreamMessage = JSON.parse(line);
messages.push(msg);
// Show real-time tool use with verbose details
if (showProgress && msg.type === 'assistant') {
const toolUses = msg.message?.content?.filter((c) => c.type === 'tool_use') || [];
for (const tool of toolUses) {
process.stderr.write('\r\x1b[K'); // Clear line
// Show verbose tool use with description/input if available
const toolInput = tool.input || {};
let verboseMsg = `[Tool] ${tool.name}`;
// Add context based on tool type (all Claude Code tools)
switch (tool.name) {
case 'Bash':
if (toolInput.command) {
// Truncate long commands
const command = toolInput.command as string;
const cmd = command.length > 80 ? command.substring(0, 77) + '...' : command;
verboseMsg += `: ${cmd}`;
}
break;
case 'Edit':
case 'Write':
case 'Read':
if (toolInput.file_path) {
verboseMsg += `: ${toolInput.file_path}`;
}
break;
case 'NotebookEdit':
case 'NotebookRead':
if (toolInput.notebook_path) {
verboseMsg += `: ${toolInput.notebook_path}`;
}
break;
case 'Grep':
if (toolInput.pattern) {
verboseMsg += `: searching for "${toolInput.pattern}"`;
if (toolInput.path) {
verboseMsg += ` in ${toolInput.path}`;
}
}
break;
case 'Glob':
if (toolInput.pattern) {
verboseMsg += `: ${toolInput.pattern}`;
}
break;
case 'SlashCommand':
if (toolInput.command) {
verboseMsg += `: ${toolInput.command}`;
}
break;
case 'Task':
if (toolInput.description) {
verboseMsg += `: ${toolInput.description}`;
} else if (toolInput.prompt) {
const promptText = toolInput.prompt as string;
const prompt =
promptText.length > 60 ? promptText.substring(0, 57) + '...' : promptText;
verboseMsg += `: ${prompt}`;
}
break;
case 'TodoWrite':
if (toolInput.todos && Array.isArray(toolInput.todos)) {
// Show in_progress task instead of just count
const inProgressTask = toolInput.todos.find(
(t: { status: string; activeForm?: string }) => t.status === 'in_progress'
);
if (inProgressTask && inProgressTask.activeForm) {
verboseMsg += `: ${inProgressTask.activeForm}`;
} else {
// Fallback to count if no in_progress task
verboseMsg += `: ${toolInput.todos.length} task(s)`;
}
}
break;
case 'WebFetch':
if (toolInput.url) {
verboseMsg += `: ${toolInput.url}`;
}
break;
case 'WebSearch':
if (toolInput.query) {
verboseMsg += `: "${toolInput.query}"`;
}
break;
default:
// For unknown tools, show first meaningful parameter
if (Object.keys(toolInput).length > 0) {
const firstKey = Object.keys(toolInput)[0];
const firstValue = toolInput[firstKey];
if (typeof firstValue === 'string' && firstValue.length < 60) {
verboseMsg += `: ${firstValue}`;
}
}
}
process.stderr.write(`${verboseMsg}\n`);
}
}
} catch (parseError) {
// Skip malformed JSON lines (shouldn't happen with stream-json)
if (process.env.CCS_DEBUG) {
console.error(
warn(`Failed to parse stream-json line: ${(parseError as Error).message}`)
);
// Show real-time tool use
if (showProgress && msg.type === 'assistant') {
const toolUses = msg.message?.content?.filter((c) => c.type === 'tool_use') || [];
for (const tool of toolUses) {
process.stderr.write('\r\x1b[K');
const toolInput = tool.input || {};
const verboseMsg = formatToolVerbose(tool.name || 'Unknown', toolInput);
process.stderr.write(`${verboseMsg}\n`);
}
}
}
});
// Stream stderr in real-time (progress messages from Claude CLI)
// Stream stderr in real-time
proc.stderr?.on('data', (data: Buffer) => {
const stderrText = data.toString();
stderr += stderrText;
// Show stderr in real-time if in TTY
if (showProgress) {
// Clear progress line before showing stderr
if (progressInterval) {
process.stderr.write('\r\x1b[K'); // Clear line
}
if (progressInterval) process.stderr.write('\r\x1b[K');
process.stderr.write(stderrText);
}
});
@@ -448,80 +241,44 @@ export class HeadlessExecutor {
proc.on('close', (exitCode: number | null) => {
const duration = Date.now() - startTime;
// Clear progress indicator
if (progressInterval) {
clearInterval(progressInterval);
process.stderr.write('\r\x1b[K'); // Clear line
process.stderr.write('\r\x1b[K');
}
// Show completion message
if (showProgress) {
const durationSec = (duration / 1000).toFixed(1);
if (timedOut) {
console.error(ui.warn(`Execution timed out after ${durationSec}s`));
} else {
console.error(ui.info(`Execution completed in ${durationSec}s`));
}
console.error(''); // Blank line before formatted output
console.error(
timedOut
? ui.warn(`Timed out after ${durationSec}s`)
: ui.info(`Completed in ${durationSec}s`)
);
console.error('');
}
const result: ExecutionResult = {
const result = buildExecutionResult({
exitCode: exitCode || 0,
stdout,
stderr,
cwd,
profile,
duration,
timedOut: false,
success: exitCode === 0 && !timedOut,
messages, // Include all stream-json messages
};
timedOut,
messages,
});
// Extract metadata from final 'result' message in stream-json
const resultMessage = messages.find((m) => m.type === 'result');
if (resultMessage) {
// Add parsed fields from result message
result.sessionId = resultMessage.session_id || undefined;
result.totalCost = resultMessage.total_cost_usd || 0;
result.numTurns = resultMessage.num_turns || 0;
result.isError = resultMessage.is_error || false;
result.type = resultMessage.type || null;
result.subtype = resultMessage.subtype || undefined;
result.durationApi = resultMessage.duration_api_ms || 0;
result.permissionDenials = resultMessage.permission_denials || [];
result.errors = resultMessage.errors || [];
// Extract content from result message
result.content = resultMessage.result || '';
} else {
// Fallback: no result message found (shouldn't happen)
result.content = stdout;
if (process.env.CCS_DEBUG) {
console.error(warn('No result message found in stream-json output'));
}
}
// Store or update session if we have session ID (even on timeout, for :continue support)
// Store session
if (result.sessionId) {
if (resumeSession || sessionId) {
// Update existing session
sessionMgr.updateSession(profile, result.sessionId, {
totalCost: result.totalCost,
});
sessionMgr.updateSession(profile, result.sessionId, { totalCost: result.totalCost });
} else {
// Store new session
sessionMgr.storeSession(profile, {
sessionId: result.sessionId,
totalCost: result.totalCost,
cwd: result.cwd,
cwd,
});
}
// Cleanup expired sessions periodically
if (Math.random() < 0.1) {
// 10% chance
sessionMgr.cleanupExpired();
}
if (Math.random() < 0.1) sessionMgr.cleanupExpired();
}
resolve(result);
@@ -529,57 +286,31 @@ export class HeadlessExecutor {
// Handle errors
proc.on('error', (error: Error) => {
if (progressInterval) {
clearInterval(progressInterval);
}
if (progressInterval) clearInterval(progressInterval);
reject(new Error(`Failed to execute Claude CLI: ${error.message}`));
});
// Handle timeout with graceful SIGTERM then forceful SIGKILL
let timedOut = false;
// Handle timeout
if (timeout > 0) {
const timeoutHandle = setTimeout(() => {
if (!proc.killed) {
timedOut = true;
if (progressInterval) {
clearInterval(progressInterval);
process.stderr.write('\r\x1b[K'); // Clear line
process.stderr.write('\r\x1b[K');
}
if (process.env.CCS_DEBUG) {
console.error(
warn(`Timeout reached after ${timeout}ms, sending SIGTERM for graceful shutdown...`)
);
}
// Send SIGTERM for graceful shutdown
proc.kill('SIGTERM');
// If process doesn't terminate within 10s, force kill
setTimeout(() => {
if (!proc.killed) {
if (process.env.CCS_DEBUG) {
console.error(warn('Process did not terminate gracefully, sending SIGKILL...'));
}
proc.kill('SIGKILL');
}
}, 10000); // Give 10s for graceful shutdown instead of 5s
if (!proc.killed) proc.kill('SIGKILL');
}, 10000);
}
}, timeout);
// Clear timeout on successful completion
proc.on('close', () => clearTimeout(timeoutHandle));
}
});
}
/**
* Validate permission mode
* @param mode - Permission mode
* @throws {Error} If mode is invalid
* @private
*/
/** Validate permission mode */
private static _validatePermissionMode(mode: string): void {
const VALID_MODES = ['default', 'plan', 'acceptEdits', 'bypassPermissions'];
if (!VALID_MODES.includes(mode)) {
@@ -587,34 +318,18 @@ export class HeadlessExecutor {
}
}
/**
* Detect Claude CLI executable
* @returns Path to claude CLI or null if not found
* @private
*/
/** Detect Claude CLI executable */
private static _detectClaudeCli(): string | null {
// Check environment variable override
if (process.env.CCS_CLAUDE_PATH) {
return process.env.CCS_CLAUDE_PATH;
}
// Try to find in PATH
if (process.env.CCS_CLAUDE_PATH) return process.env.CCS_CLAUDE_PATH;
const { execSync } = require('child_process');
try {
const result = execSync('command -v claude', { encoding: 'utf8' });
return result.trim();
} catch (_error) {
return execSync('command -v claude', { encoding: 'utf8' }).trim();
} catch {
return null;
}
}
/**
* Execute with retry logic
* @param profile - Profile name
* @param enhancedPrompt - Enhanced prompt
* @param options - Execution options
* @returns execution result
*/
/** Execute with retry logic */
static async executeWithRetry(
profile: string,
enhancedPrompt: string,
@@ -626,103 +341,59 @@ export class HeadlessExecutor {
for (let attempt = 0; attempt <= maxRetries; attempt++) {
try {
const result = await this.execute(profile, enhancedPrompt, execOptions);
// If successful, return immediately
if (result.success) {
return result;
}
// If not last attempt, retry
if (result.success) return result;
if (attempt < maxRetries) {
console.error(warn(`Attempt ${attempt + 1} failed, retrying...`));
await this._sleep(1000 * (attempt + 1)); // Exponential backoff
await this._sleep(1000 * (attempt + 1));
continue;
}
// Last attempt failed, return result anyway
return result;
} catch (error) {
lastError = error as Error;
if (attempt < maxRetries) {
console.error(
warn(`Attempt ${attempt + 1} errored: ${(error as Error).message}, retrying...`)
);
console.error(warn(`Attempt ${attempt + 1} errored, retrying...`));
await this._sleep(1000 * (attempt + 1));
}
}
}
// All retries exhausted
throw lastError || new Error('Execution failed after all retry attempts');
}
/**
* Sleep utility for retry backoff
* @param ms - Milliseconds to sleep
* @returns Promise<void>
* @private
*/
/** Sleep utility for retry backoff */
private static _sleep(ms: number): Promise<void> {
return new Promise((resolve) => setTimeout(resolve, ms));
}
/**
* Process prompt to detect and preserve slash commands
* Implements smart enhancement: preserves slash command at start, allows context in rest
* @param prompt - Original prompt (may contain slash command)
* @returns Processed prompt with slash command preserved
* @private
*/
/** Process prompt to detect and preserve slash commands */
private static _processSlashCommand(prompt: string): string {
const trimmed = prompt.trim();
// Case 1: Already starts with slash command - keep as-is
if (trimmed.match(/^\/[\w:-]+(\s|$)/)) {
return prompt;
}
// Case 1: Already starts with slash command
if (trimmed.match(/^\/[\w:-]+(\s|$)/)) return prompt;
// Case 2: Find slash command embedded in text
// Look for /command that's NOT part of a file path
// File paths: /home/user, /path/to/file (have / before or after)
// Commands: /cook, /plan (standalone, preceded by space/colon/start)
// Strategy: Find LAST occurrence that looks like a command, not a path
const embeddedSlash = trimmed.match(/(?:^|[^\w/])(\/[\w:-]+)(\s+[\s\S]*)?$/);
if (embeddedSlash) {
const command = embeddedSlash[1]; // e.g., "/cook"
const args = (embeddedSlash[2] || '').trim(); // Everything after command
// Calculate where the command starts (excluding preceding char if any)
const command = embeddedSlash[1];
const args = (embeddedSlash[2] || '').trim();
const matchIndex = embeddedSlash.index || 0;
const matchStart = matchIndex + (embeddedSlash[0][0] === '/' ? 0 : 1);
const beforeCommand = trimmed.substring(0, matchStart).trim();
// Restructure: command first, context after
if (beforeCommand && args) {
return `${command} ${args}\n\nContext: ${beforeCommand}`;
} else if (beforeCommand) {
return `${command}\n\nContext: ${beforeCommand}`;
}
if (beforeCommand && args) return `${command} ${args}\n\nContext: ${beforeCommand}`;
if (beforeCommand) return `${command}\n\nContext: ${beforeCommand}`;
return args ? `${command} ${args}` : command;
}
// No slash command detected, return as-is
return prompt;
}
/**
* Test if profile is executable (quick health check)
* @param profile - Profile name
* @returns True if profile can execute
*/
/** Test if profile is executable */
static async testProfile(profile: string): Promise<boolean> {
try {
const result = await this.execute(profile, 'Say "test successful"', {
timeout: 10000,
});
const result = await this.execute(profile, 'Say "test successful"', { timeout: 10000 });
return result.success;
} catch (_error) {
} catch {
return false;
}
}
+13
View File
@@ -0,0 +1,13 @@
/**
* Delegation module barrel export
*/
export { HeadlessExecutor } from './headless-executor';
export type { ExecutionOptions, ExecutionResult, StreamMessage } from './headless-executor';
export { SessionManager } from './session-manager';
export { SettingsParser } from './settings-parser';
export { ResultFormatter } from './result-formatter';
export { DelegationHandler } from './delegation-handler';
// Re-export executor sub-module
export * from './executor';
+3 -33
View File
@@ -9,40 +9,10 @@ import * as path from 'path';
import { execSync } from 'child_process';
import * as fs from 'fs';
import { ui } from '../utils/ui';
import type { ExecutionResult, ExecutionError, PermissionDenial } from './executor/types';
interface ExecutionResult {
profile: string;
cwd: string;
exitCode: number;
stdout: string;
stderr: string;
duration: number;
success: boolean;
content?: string;
sessionId?: string;
totalCost?: number;
numTurns?: number;
subtype?: string;
permissionDenials?: PermissionDenial[];
errors?: ErrorInfo[];
// json?: any; // Removed: unused parameter
timedOut?: boolean;
}
interface PermissionDenial {
tool_name?: string;
tool_input?: {
command?: string;
description?: string;
[key: string]: unknown;
};
}
interface ErrorInfo {
message?: string;
error?: string;
[key: string]: unknown;
}
// Alias for backward compatibility
type ErrorInfo = ExecutionError;
interface FileChanges {
created: string[];