docs: add comprehensive JSDoc comments to core modules

Add detailed JSDoc documentation to:
- inner-loop-tracker.ts: Ralph Wiggum loop detection and todo tracking
- respawn-controller.ts: Autonomous session cycling state machine
- task-tracker.ts: Background task detection and hierarchy

Includes:
- @fileoverview module descriptions
- Detailed class and method documentation
- @param, @returns, @fires annotations
- Code examples in class docs
- Configuration constant descriptions
- Pre-compiled regex pattern documentation

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
arkon
2026-01-21 03:46:11 +01:00
co-authored by Claude Opus 4.5
parent cdb57a1104
commit 8cfc02785f
3 changed files with 1045 additions and 139 deletions
+451 -74
View File
@@ -1,3 +1,18 @@
/**
* @fileoverview Inner Loop Tracker for Ralph Wiggum detection
*
* This module parses terminal output from Claude Code sessions to detect:
* - Ralph Wiggum loop state (active, completion phrase, iteration count)
* - Todo list items from the TodoWrite tool
* - Completion phrases signaling loop completion
*
* The tracker is DISABLED by default and auto-enables when Ralph-related
* patterns are detected in the output stream, reducing overhead for
* sessions not using autonomous loops.
*
* @module inner-loop-tracker
*/
import { EventEmitter } from 'node:events';
import {
InnerLoopState,
@@ -6,111 +21,275 @@ import {
createInitialInnerLoopState,
} from './types.js';
// Maximum number of todo items to track per session
// ========== Configuration Constants ==========
/**
* Maximum number of todo items to track per session.
* Older items are removed when this limit is reached.
*/
const MAX_TODO_ITEMS = 50;
// Todo items older than this will be auto-expired (1 hour)
/**
* Todo items older than this duration (in milliseconds) will be auto-expired.
* Default: 1 hour (60 * 60 * 1000)
*/
const TODO_EXPIRY_MS = 60 * 60 * 1000;
// Throttle cleanup checks (every 30 seconds)
/**
* Minimum interval between cleanup checks (in milliseconds).
* Prevents running cleanup on every data chunk.
* Default: 30 seconds
*/
const CLEANUP_THROTTLE_MS = 30 * 1000;
// Pre-compiled regex patterns for performance (avoid re-compilation on each call)
// ========== Pre-compiled Regex Patterns ==========
// Pre-compiled for performance (avoid re-compilation on each call)
// Completion phrase detection: <promise>PHRASE</promise>
/**
* Matches completion phrase tags: `<promise>PHRASE</promise>`
* Used to detect when Claude signals task completion.
* Capture group 1: The completion phrase text
*/
const PROMISE_PATTERN = /<promise>([^<]+)<\/promise>/;
// Todo item patterns - multiple formats Claude Code uses
// Format 1: Checkbox format in markdown: "- [ ] Task" or "- [x] Task"
// ---------- Todo Item Patterns ----------
// Claude Code outputs todos in multiple formats; we detect all of them
/**
* Format 1: Markdown checkbox format
* Matches: "- [ ] Task" or "- [x] Task" (also with * bullet)
* Capture group 1: Checkbox state ('x', 'X', or ' ')
* Capture group 2: Task content
*/
const TODO_CHECKBOX_PATTERN = /^[-*]\s*\[([xX ])\]\s+(.+)$/gm;
// Format 2: Todo with indicator icons: "Todo: ☐ Task", "Todo: ◐ Task", "Todo: ✓ Task"
/**
* Format 2: Todo with indicator icons
* Matches: "Todo: ☐ Task", "Todo: ◐ Task", "Todo: ✓ Task"
* Capture group 1: Status icon
* Capture group 2: Task content
*/
const TODO_INDICATOR_PATTERN = /Todo:\s*(☐|◐|✓|⏳|✅|⌛|🔄)\s+(.+)/g;
// Format 3: Status in parentheses: "(pending)", "(in_progress)", "(completed)"
/**
* Format 3: Status in parentheses
* Matches: "- Task (pending)", "- Task (in_progress)", "- Task (completed)"
* Capture group 1: Task content
* Capture group 2: Status string
*/
const TODO_STATUS_PATTERN = /[-*]\s*(.+?)\s+\((pending|in_progress|completed)\)/g;
// Format 4: Claude Code native TodoWrite output: "☐ Task", "☒ Task", "◐ Task"
// These appear in terminal with optional leading whitespace/brackets like "⎿ ☐ Task"
// Matches: start of line with optional whitespace/bracket, then checkbox, then task text
/**
* Format 4: Claude Code native TodoWrite output
* Matches: "☐ Task", "☒ Task", "◐ Task"
* These appear with optional leading whitespace/brackets like "⎿ ☐ Task"
* Capture group 1: Checkbox icon (☐=pending, ☒=completed, ◐=in_progress)
* Capture group 2: Task content (min 3 chars, excludes checkbox icons)
*/
const TODO_NATIVE_PATTERN = /^[\s⎿]*(☐|☒|◐)\s+([^☐☒◐\n]{3,})/gm;
// Patterns to exclude from todo detection (tool invocations, etc.)
/**
* Patterns to exclude from todo detection
* Prevents false positives from tool invocations and Claude commentary
*/
const TODO_EXCLUDE_PATTERNS = [
/^(?:Bash|Search|Read|Write|Glob|Grep|Edit|Task)\s*\(/i, // Tool invocations
/^(?:I'll |Let me |Now I|First,|Task \d+:|Result:|Error:)/i, // Claude commentary (with context)
/^(?:I'll |Let me |Now I|First,|Task \d+:|Result:|Error:)/i, // Claude commentary
/^\S+\([^)]+\)$/, // Generic function call pattern
];
// Loop status patterns (does NOT include <promise> - that's handled by PROMISE_PATTERN)
// ---------- Loop Status Patterns ----------
// Note: <promise> tags are handled separately by PROMISE_PATTERN
/**
* Matches generic loop start messages
* Examples: "Loop started at", "Starting main loop", "Ralph loop started"
*/
const LOOP_START_PATTERN = /Loop started at|Starting.*loop|Ralph loop started/i;
/**
* Matches elapsed time output
* Example: "Elapsed: 2.5 hours"
* Capture group 1: Hours as decimal number
*/
const ELAPSED_TIME_PATTERN = /Elapsed:\s*(\d+(?:\.\d+)?)\s*hours?/i;
/**
* Matches cycle count indicators (legacy format)
* Examples: "cycle #5", "respawn cycle #3"
* Capture groups 1 or 2: Cycle number
*/
const CYCLE_PATTERN = /cycle\s*#?(\d+)|respawn cycle #(\d+)/i;
// New patterns for improved Ralph detection (based on official Ralph Wiggum plugin)
// Iteration patterns: "Iteration 5/50", "[5/50]", "iteration #5"
// ---------- Ralph Wiggum Plugin Patterns ----------
// Based on the official Ralph Wiggum plugin output format
/**
* Matches iteration progress indicators
* Examples: "Iteration 5/50", "[5/50]", "iteration #5", "iter. 3 of 10"
* Capture groups: (1,2) for "Iteration X/Y" format, (3,4) for "[X/Y]" format
*/
const ITERATION_PATTERN = /(?:iteration|iter\.?)\s*#?(\d+)(?:\s*[\/of]\s*(\d+))?|\[(\d+)\/(\d+)\]/i;
// Ralph loop start: "/ralph-loop:ralph-loop" command or "Starting Ralph loop"
// Pattern matches /ralph-loop anywhere to catch both skill invocations and output
/**
* Matches Ralph loop start command or announcement
* Examples: "/ralph-loop:ralph-loop", "Starting Ralph Wiggum loop", "ralph loop beginning"
*/
const RALPH_START_PATTERN = /\/ralph-loop|starting ralph(?:\s+wiggum)?\s+loop|ralph loop (?:started|beginning)/i;
// Max iterations: "max-iterations 50" or "maxIterations: 50" or "max_iterations=50"
/**
* Matches max iterations configuration
* Examples: "max-iterations 50", "maxIterations: 50", "max_iterations=50"
* Capture group 1: Maximum iteration count
*/
const MAX_ITERATIONS_PATTERN = /max[_-]?iterations?\s*[=:]\s*(\d+)/i;
// TodoWrite tool output - detect the tool being used
/**
* Matches TodoWrite tool usage indicators
* Examples: "TodoWrite", "todos updated", "Todos have been modified"
*/
const TODOWRITE_PATTERN = /TodoWrite|todo(?:s)?\s*(?:updated|written|saved)|Todos have been modified/i;
// All tasks complete patterns - detect when Claude says all work is done
// Includes patterns like "All 8 files have been created", "All tasks completed"
// ---------- Task Completion Detection Patterns ----------
/**
* Matches "all tasks complete" announcements
* Examples: "All 8 files have been created", "All tasks completed", "Everything is done"
* Used to mark all tracked todos as complete at once
*/
const ALL_COMPLETE_PATTERN = /all\s+(?:\d+\s+)?(?:tasks?|files?|items?)\s+(?:have\s+been\s+|are\s+)?(?:completed?|done|finished|created)|completed?\s+all\s+(?:\d+\s+)?tasks?|all\s+done|everything\s+(?:is\s+)?(?:completed?|done)|finished\s+all\s+tasks?/i;
// Pattern to extract count from "All 8 files created" type messages
/**
* Extracts count from "all N items" messages
* Example: "All 8 files created" → captures "8"
* Capture group 1: The count
*/
const ALL_COUNT_PATTERN = /all\s+(\d+)\s+(?:tasks?|files?|items?)/i;
// Individual task completion patterns - detect when Claude marks a specific task done
/**
* Matches individual task completion messages
* Examples: "Task #5 is done", "marked as completed", "todo 3 finished"
* Used to update specific todo items by number
*/
const TASK_DONE_PATTERN = /(?:task|item|todo)\s*(?:#?\d+|"\s*[^"]+\s*")?\s*(?:is\s+)?(?:done|completed?|finished)|(?:completed?|done|finished)\s+(?:task|item)\s*(?:#?\d+)?|marking\s+(?:.*?\s+)?(?:as\s+)?completed?|marked\s+(?:.*?\s+)?(?:as\s+)?completed?/i;
// Generic completion signals (be careful with these - can be false positives)
/**
* Matches generic standalone completion signals
* Examples: "Done!", "Completed", "Finished", "All set"
* Note: Use with caution - high false positive potential
* @deprecated Currently unused due to false positive risk
*/
const COMPLETION_SIGNAL_PATTERN = /^(?:done|completed?|finished|all\s+set)!?\s*$/i;
// ANSI escape code removal for cleaner parsing
// Matches color codes (\x1b[...m), cursor movement (\x1b[...H, \x1b[...C, etc.), and other sequences
// ---------- Utility Patterns ----------
/**
* Removes ANSI escape codes from terminal output for cleaner parsing.
* Matches: color codes (\x1b[...m), cursor movement (\x1b[...H, \x1b[...C), etc.
*/
const ANSI_ESCAPE_PATTERN = /\x1b\[[0-9;]*[A-Za-z]/g;
// ========== Event Types ==========
/**
* Events emitted by InnerLoopTracker
* @event loopUpdate - Fired when loop state changes (active, iteration, completion phrase)
* @event todoUpdate - Fired when todo list changes (items added, status changed)
* @event completionDetected - Fired when completion phrase is detected (task complete)
* @event enabled - Fired when tracker auto-enables due to Ralph pattern detection
*/
export interface InnerLoopTrackerEvents {
/** Emitted when loop state changes */
loopUpdate: (state: InnerLoopState) => void;
/** Emitted when todo list is modified */
todoUpdate: (todos: InnerTodoItem[]) => void;
/** Emitted when completion phrase detected (loop finished) */
completionDetected: (phrase: string) => void;
enabled: () => void; // Emitted when tracker auto-enables
/** Emitted when tracker auto-enables from disabled state */
enabled: () => void;
}
/**
* InnerLoopTracker parses terminal output from Claude Code sessions to detect:
* 1. Ralph Wiggum loop state (active, completion phrase, cycle count)
* 2. Todo list items from the TodoWrite tool
* InnerLoopTracker - Parses terminal output to detect Ralph Wiggum loops and todos
*
* The tracker is DISABLED by default and auto-enables when Ralph-related
* This class monitors Claude Code session output to detect:
* 1. **Ralph Wiggum loop state** - Active loops, completion phrases, iteration counts
* 2. **Todo list items** - From TodoWrite tool in various formats
* 3. **Completion signals** - `<promise>PHRASE</promise>` tags
*
* ## Lifecycle
*
* The tracker is **DISABLED by default** and auto-enables when Ralph-related
* patterns are detected (e.g., /ralph-loop:ralph-loop, <promise>, todos).
* This reduces overhead for sessions not using autonomous loops.
*
* ## Completion Detection
*
* Uses occurrence-based detection to distinguish prompt from actual completion:
* - 1st occurrence of `<promise>X</promise>`: Stored as expected phrase (likely in prompt)
* - 2nd occurrence: Emits `completionDetected` event (actual completion)
* - If loop already active: Emits immediately on first occurrence
*
* ## Events
*
* - `loopUpdate` - Loop state changed (status, iteration, phrase)
* - `todoUpdate` - Todo list modified (add, status change)
* - `completionDetected` - Loop completion phrase detected
* - `enabled` - Tracker auto-enabled from disabled state
*
* @extends EventEmitter
* @example
* ```typescript
* const tracker = new InnerLoopTracker();
* tracker.on('completionDetected', (phrase) => {
* console.log('Loop completed with phrase:', phrase);
* });
* tracker.processTerminalData(ptyOutput);
* ```
*/
export class InnerLoopTracker extends EventEmitter {
/** Current state of the detected loop */
private _loopState: InnerLoopState;
/** Map of todo items by ID for O(1) lookup */
private _todos: Map<string, InnerTodoItem> = new Map();
/** Buffer for incomplete lines from terminal data */
private _lineBuffer: string = '';
// Track occurrences of completion phrases to distinguish prompt from actual completion
/**
* Tracks occurrences of completion phrases.
* Used to distinguish prompt echo (1st) from actual completion (2nd+).
*/
private _completionPhraseCount: Map<string, number> = new Map();
// Throttle cleanup to avoid running on every data chunk
/** Timestamp of last cleanup check for throttling */
private _lastCleanupTime: number = 0;
/**
* Creates a new InnerLoopTracker instance.
* Starts in disabled state until Ralph patterns are detected.
*/
constructor() {
super();
this._loopState = createInitialInnerLoopState();
}
/**
* Whether the tracker is enabled and actively monitoring
* Whether the tracker is enabled and actively monitoring output.
* Disabled by default; auto-enables when Ralph patterns detected.
* @returns True if tracker is processing terminal data
*/
get enabled(): boolean {
return this._loopState.enabled;
}
/**
* Enable the tracker (called automatically when Ralph patterns detected)
* Enable the tracker to start monitoring terminal output.
* Called automatically when Ralph patterns are detected.
* Emits 'enabled' event when transitioning from disabled state.
* @fires enabled
* @fires loopUpdate
*/
enable(): void {
if (!this._loopState.enabled) {
@@ -122,7 +301,9 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Disable the tracker
* Disable the tracker to stop monitoring terminal output.
* Terminal data will be ignored until re-enabled.
* @fires loopUpdate
*/
disable(): void {
if (this._loopState.enabled) {
@@ -133,8 +314,20 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Reset the tracker to initial state (for when a new task/loop starts)
* Clears all todos, completion phrase, and loop state while keeping enabled status
* Soft reset - clears state but keeps enabled status.
* Use when a new task/loop starts within the same session.
*
* Clears:
* - All todo items
* - Completion phrase tracking
* - Loop state (active, iterations)
* - Line buffer
*
* Preserves:
* - Enabled status
*
* @fires loopUpdate
* @fires todoUpdate
*/
reset(): void {
const wasEnabled = this._loopState.enabled;
@@ -148,7 +341,11 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Full reset including enabled state (for complete cleanup)
* Full reset - clears all state including enabled status.
* Use when session is closed or completely cleared.
* Returns tracker to initial disabled state.
* @fires loopUpdate
* @fires todoUpdate
*/
fullReset(): void {
this._loopState = createInitialInnerLoopState();
@@ -159,16 +356,39 @@ export class InnerLoopTracker extends EventEmitter {
this.emit('todoUpdate', this.todos);
}
/**
* Get a copy of the current loop state.
* @returns Shallow copy of loop state (safe to modify)
*/
get loopState(): InnerLoopState {
return { ...this._loopState };
}
/**
* Get all tracked todo items as an array.
* @returns Array of todo items (copy, safe to modify)
*/
get todos(): InnerTodoItem[] {
return Array.from(this._todos.values());
}
/**
* Process raw terminal data to detect inner loop patterns
* Process raw terminal data to detect inner loop patterns.
*
* This is the main entry point for parsing output. Call this with each
* chunk of data from the PTY. The tracker will:
*
* 1. Strip ANSI escape codes
* 2. Auto-enable if disabled and Ralph patterns detected
* 3. Buffer data and process complete lines
* 4. Detect loop status, todos, and completion phrases
* 5. Periodically clean up expired todos
*
* @param data - Raw terminal data (may include ANSI codes)
* @fires loopUpdate - When loop state changes
* @fires todoUpdate - When todos are detected or updated
* @fires completionDetected - When completion phrase found
* @fires enabled - When tracker auto-enables
*/
processTerminalData(data: string): void {
// Remove ANSI escape codes for cleaner parsing
@@ -203,7 +423,21 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Check if the data contains patterns that should auto-enable the tracker
* Check if data contains patterns that should auto-enable the tracker.
*
* The tracker auto-enables when any of these patterns are detected:
* - `/ralph-loop:ralph-loop` command
* - `<promise>PHRASE</promise>` completion tags
* - TodoWrite tool usage indicators
* - Iteration patterns (`Iteration 5/50`, `[5/50]`)
* - Todo checkboxes (`- [ ]`, `- [x]`)
* - Todo indicator icons (`☐`, `◐`, `☒`)
* - Loop start messages (`Loop started at`)
* - All tasks complete announcements
* - Task completion signals
*
* @param data - ANSI-cleaned terminal data
* @returns True if any Ralph-related pattern is detected
*/
private shouldAutoEnable(data: string): boolean {
// Ralph loop command: /ralph-loop:ralph-loop
@@ -264,7 +498,9 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Process a single line of terminal output
* Process a single line of terminal output.
* Runs all detection methods in sequence.
* @param line - Single line of ANSI-cleaned terminal output
*/
private processLine(line: string): void {
const trimmed = line.trim();
@@ -287,8 +523,23 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Detect "all tasks complete" messages
* When detected: marks all todos as complete AND emits completion event
* Detect "all tasks complete" messages.
*
* When a valid "all complete" message is detected:
* 1. Marks all tracked todos as completed
* 2. Emits completion event if a completion phrase is set
*
* Validation criteria:
* - Line must match ALL_COMPLETE_PATTERN
* - Line must be reasonably short (<100 chars) to avoid matching commentary
* - Must not look like prompt text (no "output:" or `<promise>`)
* - Must have at least one tracked todo
* - If count is mentioned, should roughly match tracked todo count
*
* @param line - Single line to check
* @fires todoUpdate - If any todos marked complete
* @fires completionDetected - If completion phrase was set
* @fires loopUpdate - If loop state changes
*/
private detectAllTasksComplete(line: string): void {
// Only trigger if line is a clear standalone completion message
@@ -365,7 +616,9 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Check for multi-line patterns in the data chunk
* Check for multi-line patterns that might span line boundaries.
* Completion phrases can be split across PTY chunks.
* @param data - The full data chunk (may contain multiple lines)
*/
private checkMultiLinePatterns(data: string): void {
// Completion phrase can span lines, so check the whole chunk
@@ -376,8 +629,18 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Detect <promise>PHRASE</promise> completion phrases
* Also detects bare phrase (without tags) if we already know the expected phrase
* Detect completion phrases in a line.
*
* Handles two formats:
* 1. Tagged: `<promise>PHRASE</promise>` - Processed via handleCompletionPhrase
* 2. Bare: Just `PHRASE` - Only if we already know the expected phrase
*
* Bare phrase detection avoids false positives by requiring:
* - The phrase was previously seen in tagged form
* - Line is standalone or ends with the phrase
* - Line doesn't look like prompt context
*
* @param line - Single line to check
*/
private detectCompletionPhrase(line: string): void {
// First check for tagged phrase: <promise>PHRASE</promise>
@@ -404,9 +667,21 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Handle a bare completion phrase (without tags)
* Emits completion if we've already seen the tagged version in the prompt
* and this appears to be actual completion output (not prompt echo)
* Handle a bare completion phrase (without XML tags).
*
* Only fires completion if:
* 1. The phrase was previously seen in tagged form (from prompt)
* 2. This is the first bare occurrence (prevents double-firing)
*
* When triggered:
* - Marks all todos as complete
* - Emits completionDetected event
* - Sets loop to inactive
*
* @param phrase - The completion phrase text
* @fires todoUpdate - If any todos marked complete
* @fires completionDetected - When completion triggered
* @fires loopUpdate - When loop state changes
*/
private handleBareCompletionPhrase(phrase: string): void {
// Only count if this phrase was already seen in tagged form (from the prompt)
@@ -481,7 +756,13 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Helper: Activate the loop if not already active
* Activate the loop if not already active.
*
* Sets loop state to active and initializes counters.
* No-op if loop is already active.
*
* @returns True if loop was activated, false if already active
* @fires loopUpdate - When loop state changes
*/
private activateLoopIfNeeded(): boolean {
if (this._loopState.active) return false;
@@ -497,7 +778,19 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Detect loop start and status indicators
* Detect loop start and status indicators.
*
* Patterns detected:
* - Ralph loop start commands (`/ralph-loop:ralph-loop`)
* - Loop start messages (`Loop started at`, `Starting Ralph loop`)
* - Max iterations setting (`max-iterations 50`)
* - Iteration progress (`Iteration 5/50`, `[5/50]`)
* - Elapsed time (`Elapsed: 2.5 hours`)
* - Cycle count (`cycle #5`, `respawn cycle #3`)
* - TodoWrite tool usage
*
* @param line - Single line to check
* @fires loopUpdate - When any loop state changes
*/
private detectLoopStatus(line: string): void {
// Check for Ralph loop start command (/ralph-loop:ralph-loop)
@@ -562,7 +855,19 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Detect todo items in various formats
* Detect todo items in various formats from Claude Code output.
*
* Supported formats:
* - Format 1: Checkbox markdown (`- [ ] Task`, `- [x] Task`)
* - Format 2: Indicator icons (`Todo: ☐ Task`, `Todo: ✓ Task`)
* - Format 3: Status in parentheses (`- Task (pending)`)
* - Format 4: Native TodoWrite (`☐ Task`, `☒ Task`, `◐ Task`)
*
* Uses quick pre-check to skip lines that can't contain todos.
* Excludes tool invocations and Claude commentary patterns.
*
* @param line - Single line to check
* @fires todoUpdate - When any todos are detected or updated
*/
private detectTodoItems(line: string): void {
// Quick check: skip lines that can't possibly contain todos
@@ -629,7 +934,15 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Convert todo icon to status
* Convert a todo icon character to its corresponding status.
*
* Icon mappings:
* - Completed: `✓`, `✅`, `☒`, `◉`, `●`
* - In Progress: `◐`, `⏳`, `⌛`, `🔄`
* - Pending: `☐`, `○`, and anything else (default)
*
* @param icon - Single character icon
* @returns Corresponding InnerTodoStatus
*/
private iconToStatus(icon: string): InnerTodoStatus {
switch (icon) {
@@ -652,7 +965,17 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Add or update a todo item
* Add a new todo item or update an existing one.
*
* Behavior:
* - Content is cleaned (ANSI removed, whitespace collapsed)
* - Content under 5 chars is skipped
* - ID is generated from normalized content (stable hash)
* - Existing item: Updates status and timestamp
* - New item: Adds to map, evicts oldest if at MAX_TODO_ITEMS
*
* @param content - Raw todo content text
* @param status - Status to set
*/
private upsertTodo(content: string, status: InnerTodoStatus): void {
// Skip empty or whitespace-only content
@@ -693,10 +1016,18 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Normalize todo content for consistent matching
* - Collapse multiple whitespace to single space
* - Remove trailing garbage characters
* - Trim whitespace
* Normalize todo content for consistent matching.
*
* Normalization steps:
* 1. Collapse multiple whitespace to single space
* 2. Remove special characters (keep alphanumeric + basic punctuation)
* 3. Trim whitespace
* 4. Convert to lowercase
*
* This prevents duplicate todos from terminal rendering artifacts.
*
* @param content - Raw todo content
* @returns Normalized lowercase string
*/
private normalizeTodoContent(content: string): string {
if (!content) return '';
@@ -708,8 +1039,13 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Generate a stable ID from todo content using djb2 hash
* Content is normalized first to prevent duplicates from terminal artifacts
* Generate a stable ID from todo content using djb2 hash.
*
* Uses the djb2 hash algorithm for good distribution across strings.
* Content is normalized first to prevent duplicates from terminal artifacts.
*
* @param content - Todo content text
* @returns Stable ID in format `todo-{hash}` (base36 encoded)
*/
private generateTodoId(content: string): string {
if (!content) return 'todo-empty';
@@ -728,7 +1064,9 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Find the oldest todo item
* Find the todo item with the oldest detectedAt timestamp.
* Used for LRU eviction when at MAX_TODO_ITEMS limit.
* @returns Oldest todo item, or undefined if map is empty
*/
private findOldestTodo(): InnerTodoItem | undefined {
let oldest: InnerTodoItem | undefined;
@@ -741,7 +1079,8 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Throttled cleanup - only runs every CLEANUP_THROTTLE_MS
* Conditionally run cleanup, throttled to CLEANUP_THROTTLE_MS.
* Prevents cleanup from running on every data chunk (performance).
*/
private maybeCleanupExpiredTodos(): void {
const now = Date.now();
@@ -753,7 +1092,9 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Remove expired todo items
* Remove todo items older than TODO_EXPIRY_MS.
* Emits todoUpdate if any items were removed.
* @fires todoUpdate - When expired items are removed
*/
private cleanupExpiredTodos(): void {
const now = Date.now();
@@ -774,8 +1115,17 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Mark the loop as started (can be called externally)
* Also enables the tracker if not already enabled
* Programmatically start a loop (external API).
*
* Use when starting a loop from outside terminal detection,
* such as from a user action or API call.
*
* Automatically enables the tracker if not already enabled.
*
* @param completionPhrase - Optional phrase that signals completion
* @param maxIterations - Optional maximum iteration count
* @fires enabled - If tracker was disabled
* @fires loopUpdate - When loop state changes
*/
startLoop(completionPhrase?: string, maxIterations?: number): void {
this.enable(); // Ensure tracker is enabled
@@ -792,7 +1142,10 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Update max iterations (can be called externally)
* Update the maximum iteration count (external API).
*
* @param maxIterations - New max iterations, or null to remove limit
* @fires loopUpdate - When loop state changes
*/
setMaxIterations(maxIterations: number | null): void {
this._loopState.maxIterations = maxIterations;
@@ -801,7 +1154,12 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Mark the loop as stopped
* Programmatically stop the loop (external API).
*
* Sets loop to inactive. Does not disable the tracker
* or clear todos - use reset() or clear() for that.
*
* @fires loopUpdate - When loop state changes
*/
stopLoop(): void {
this._loopState.active = false;
@@ -810,8 +1168,13 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Clear all state (e.g., when session is cleared)
* Resets to disabled state
* Clear all state and disable the tracker.
*
* Use when the session is cleared or closed.
* Resets everything to initial disabled state.
*
* @fires loopUpdate - With initial state
* @fires todoUpdate - With empty array
*/
clear(): void {
this._loopState = createInitialInnerLoopState(); // This sets enabled: false
@@ -823,7 +1186,13 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Get todo completion stats
* Get aggregated statistics about tracked todos.
*
* @returns Object with counts by status:
* - total: Total number of tracked todos
* - pending: Todos not yet started
* - inProgress: Todos currently in progress
* - completed: Finished todos
*/
getTodoStats(): { total: number; pending: number; inProgress: number; completed: number } {
let pending = 0;
@@ -853,7 +1222,15 @@ export class InnerLoopTracker extends EventEmitter {
}
/**
* Restore state from persisted data
* Restore tracker state from persisted data.
*
* Use after loading state from StateStore. Handles backwards
* compatibility by defaulting missing `enabled` flag to false.
*
* Note: Does not emit events (caller should handle if needed).
*
* @param loopState - Persisted loop state object
* @param todos - Persisted todo items array
*/
restoreState(loopState: InnerLoopState, todos: InnerTodoItem[]): void {
// Ensure enabled flag exists (backwards compatibility)
+392 -47
View File
@@ -1,64 +1,192 @@
/**
* @fileoverview Respawn Controller for autonomous Claude Code session cycling
*
* The RespawnController manages automatic respawning of Claude Code sessions.
* When Claude finishes working (detected by idle prompt), it automatically
* cycles through update → clear → init steps to keep the session productive.
*
* ## State Machine
*
* ```
* WATCHING → SENDING_UPDATE → WAITING_UPDATE → SENDING_CLEAR → WAITING_CLEAR
* ↑ │
* │ ▼
* │ SENDING_INIT → WAITING_INIT → MONITORING_INIT ───────┘
* │ │
* │ ▼ (if no work triggered)
* └── SENDING_KICKSTART → WAITING_KICKSTART ──┘
* ```
*
* ## Configuration
*
* - `sendClear`: Whether to send /clear after update (default: true)
* - `sendInit`: Whether to send /init after clear (default: true)
* - `kickstartPrompt`: Optional prompt if /init doesn't trigger work
*
* @module respawn-controller
*/
import { EventEmitter } from 'node:events';
import { Session } from './session.js';
// Maximum terminal buffer size for respawn controller (1MB)
const MAX_RESPAWN_BUFFER_SIZE = 1024 * 1024;
// Keep this much when trimming (512KB)
const RESPAWN_BUFFER_TRIM_SIZE = 512 * 1024;
// Pre-compiled patterns for performance
const ANSI_ESCAPE_PATTERN = /\x1b\[[0-9;]*[HJKmsu?lh]/g;
const WHITESPACE_PATTERN = /\s+/g;
// The definitive "ready for input" indicator - when Claude shows a suggestion
const READY_INDICATOR = '↵ send';
// ========== Configuration Constants ==========
/**
* Respawn sequence states
* Maximum terminal buffer size for respawn controller.
* Buffer is trimmed when this limit is exceeded to prevent memory issues.
*/
const MAX_RESPAWN_BUFFER_SIZE = 1024 * 1024; // 1MB
/**
* Size to trim buffer to when MAX_RESPAWN_BUFFER_SIZE is exceeded.
* Keeps the most recent output for pattern detection.
*/
const RESPAWN_BUFFER_TRIM_SIZE = 512 * 1024; // 512KB
// ========== Pre-compiled Regex Patterns ==========
/**
* Matches ANSI escape codes for terminal control sequences.
* Used to strip formatting before pattern matching.
* @deprecated Currently unused but kept for potential future use
*/
const ANSI_ESCAPE_PATTERN = /\x1b\[[0-9;]*[HJKmsu?lh]/g;
/**
* Matches whitespace sequences for normalization.
* @deprecated Currently unused but kept for potential future use
*/
const WHITESPACE_PATTERN = /\s+/g;
/**
* The definitive "ready for input" indicator.
* When Claude shows a suggestion, this appears and indicates idle state.
*/
const READY_INDICATOR = '↵ send';
// ========== Type Definitions ==========
/**
* Respawn sequence states.
*
* The controller cycles through these states:
* WATCHING → SENDING_UPDATE → WAITING_UPDATE → SENDING_CLEAR → WAITING_CLEAR → SENDING_INIT → WAITING_INIT → MONITORING_INIT → (maybe SENDING_KICKSTART → WAITING_KICKSTART) → WATCHING
* ```
* WATCHING → SENDING_UPDATE → WAITING_UPDATE →
* SENDING_CLEAR → WAITING_CLEAR →
* SENDING_INIT → WAITING_INIT →
* MONITORING_INIT → (maybe SENDING_KICKSTART → WAITING_KICKSTART) →
* WATCHING (repeat)
* ```
*
* Steps can be skipped via config (`sendClear: false`, `sendInit: false`).
*/
export type RespawnState =
| 'watching' // Watching for idle, ready to start respawn sequence
| 'sending_update' // About to send the update docs prompt
| 'waiting_update' // Waiting for update to complete
| 'sending_clear' // About to send /clear
| 'waiting_clear' // Waiting for clear to complete
| 'sending_init' // About to send /init
| 'waiting_init' // Waiting for init to complete
| 'monitoring_init' // Monitoring if /init triggered work
| 'sending_kickstart' // About to send kickstart prompt
| 'waiting_kickstart' // Waiting for kickstart to complete
| 'stopped'; // Controller stopped
/** Watching for idle, ready to start respawn sequence */
| 'watching'
/** About to send the update docs prompt */
| 'sending_update'
/** Waiting for update to complete */
| 'waiting_update'
/** About to send /clear command */
| 'sending_clear'
/** Waiting for clear to complete */
| 'waiting_clear'
/** About to send /init command */
| 'sending_init'
/** Waiting for init to complete */
| 'waiting_init'
/** Monitoring if /init triggered work */
| 'monitoring_init'
/** About to send kickstart prompt */
| 'sending_kickstart'
/** Waiting for kickstart to complete */
| 'waiting_kickstart'
/** Controller stopped (not running) */
| 'stopped';
/**
* Configuration options for the RespawnController.
*/
export interface RespawnConfig {
/** How long to wait after seeing prompt before considering truly idle (ms) */
/**
* How long to wait after seeing prompt before considering truly idle.
* Prevents premature cycling when user is about to type.
* @default 10000 (10 seconds)
*/
idleTimeoutMs: number;
/** The prompt to send for updating docs */
/**
* The prompt to send when updating docs.
* Sent at the start of each respawn cycle.
* @default 'update all the docs and CLAUDE.md'
*/
updatePrompt: string;
/** Delay between sending steps (ms) */
/**
* Delay between sending steps (ms).
* Gives Claude time to process each command.
* @default 1000 (1 second)
*/
interStepDelayMs: number;
/** Whether to enable respawn loop */
/**
* Whether the respawn loop is enabled.
* When false, start() will be a no-op.
* @default true
*/
enabled: boolean;
/** Whether to send /clear after update prompt */
/**
* Whether to send /clear after update prompt completes.
* Resets Claude's context for fresh start.
* @default true
*/
sendClear: boolean;
/** Whether to send /init after /clear */
/**
* Whether to send /init after /clear completes.
* Re-initializes Claude with CLAUDE.md context.
* @default true
*/
sendInit: boolean;
/** Optional prompt to send if /init doesn't trigger work */
/**
* Optional prompt to send if /init doesn't trigger work.
* Used as a fallback when /init completes but Claude doesn't start working.
* @default undefined
*/
kickstartPrompt?: string;
}
/**
* Events emitted by RespawnController.
*
* @event stateChanged - Fired when state machine transitions
* @event respawnCycleStarted - Fired when a new cycle begins
* @event respawnCycleCompleted - Fired when a cycle finishes
* @event stepSent - Fired when a command is sent to the session
* @event stepCompleted - Fired when a step finishes (ready indicator detected)
* @event error - Fired on errors
* @event log - Fired for debug logging
*/
export interface RespawnEvents {
/** State machine transition */
stateChanged: (state: RespawnState, prevState: RespawnState) => void;
/** New respawn cycle started */
respawnCycleStarted: (cycleNumber: number) => void;
/** Respawn cycle finished */
respawnCycleCompleted: (cycleNumber: number) => void;
/** Command sent to session */
stepSent: (step: string, input: string) => void;
/** Step completed (ready indicator detected) */
stepCompleted: (step: string) => void;
/** Error occurred */
error: (error: Error) => void;
/** Debug log message */
log: (message: string) => void;
}
/** Default configuration values */
const DEFAULT_CONFIG: RespawnConfig = {
idleTimeoutMs: 10000, // 10 seconds of no activity after prompt
updatePrompt: 'update all the docs and CLAUDE.md',
@@ -69,29 +197,90 @@ const DEFAULT_CONFIG: RespawnConfig = {
};
/**
* RespawnController manages automatic respawning of Claude Code sessions
* RespawnController - Automatic session cycling for continuous Claude work.
*
* When Claude finishes working (detected by idle prompt), it:
* 1. Sends an update docs prompt
* 2. Waits for completion
* 3. Sends /clear
* 4. Sends /init
* 5. Repeats
* Monitors a Claude Code session for idle state and automatically cycles
* through update → clear → init steps to keep the session productive.
*
* ## How It Works
*
* 1. **Idle Detection**: Watches terminal output for `↵ send` indicator
* 2. **Update**: Sends configured prompt (e.g., "update all docs")
* 3. **Clear**: Sends `/clear` to reset context (optional)
* 4. **Init**: Sends `/init` to re-initialize with CLAUDE.md (optional)
* 5. **Kickstart**: If /init doesn't trigger work, sends fallback prompt (optional)
* 6. **Repeat**: Returns to watching state for next cycle
*
* ## Idle Detection
*
* Primary indicator: `↵ send` - Claude's suggestion prompt
* Fallback indicators: Various prompt characters (❯, ⏵, etc.)
*
* Working indicators: Thinking, Writing, spinner characters, etc.
*
* ## Events
*
* - `stateChanged`: State machine transition
* - `respawnCycleStarted`: New cycle began
* - `respawnCycleCompleted`: Cycle finished
* - `stepSent`: Command sent to session
* - `stepCompleted`: Step finished
* - `log`: Debug messages
*
* @extends EventEmitter
* @example
* ```typescript
* const respawn = new RespawnController(session, {
* updatePrompt: 'continue working on the task',
* idleTimeoutMs: 5000,
* });
*
* respawn.on('respawnCycleCompleted', (cycle) => {
* console.log(`Completed cycle ${cycle}`);
* });
*
* respawn.start();
* ```
*/
export class RespawnController extends EventEmitter {
/** The session being controlled */
private session: Session;
/** Current configuration */
private config: RespawnConfig;
/** Current state machine state */
private _state: RespawnState = 'stopped';
/** Timer for idle detection timeout */
private idleTimer: NodeJS.Timeout | null = null;
/** Timer for step delays */
private stepTimer: NodeJS.Timeout | null = null;
/** Number of completed respawn cycles */
private cycleCount: number = 0;
/** Timestamp of last terminal activity */
private lastActivityTime: number = 0;
/** Buffer for recent terminal output */
private terminalBuffer: string = '';
/** Whether a prompt indicator was detected */
private promptDetected: boolean = false;
/** Whether a working indicator was detected */
private workingDetected: boolean = false;
/** Reference to terminal event handler (for cleanup) */
private terminalHandler: ((data: string) => void) | null = null;
// Terminal patterns - detect when Claude is ready for input
/**
* Patterns indicating Claude is ready for input.
* Primary: `↵ send` (suggestion prompt)
* Fallback: Various prompt characters
*/
private readonly PROMPT_PATTERNS = [
'↵ send', // Suggestion ready to send (strongest indicator of idle)
'❯', // Standard prompt
@@ -100,6 +289,11 @@ export class RespawnController extends EventEmitter {
'> ', // Fallback
'tokens', // The status line shows "X tokens" when at prompt
];
/**
* Patterns indicating Claude is actively working.
* When detected, resets idle detection.
*/
private readonly WORKING_PATTERNS = [
'Thinking', 'Writing', 'Reading', 'Running', 'Searching',
'Editing', 'Creating', 'Deleting', 'Analyzing', 'Executing',
@@ -108,24 +302,51 @@ export class RespawnController extends EventEmitter {
'✻', '✽', // Activity indicators (spinning star)
];
/**
* Creates a new RespawnController.
*
* @param session - The Session instance to control
* @param config - Partial configuration (merged with defaults)
*/
constructor(session: Session, config: Partial<RespawnConfig> = {}) {
super();
this.session = session;
this.config = { ...DEFAULT_CONFIG, ...config };
}
/**
* Get the current state machine state.
* @returns Current RespawnState
*/
get state(): RespawnState {
return this._state;
}
/**
* Get the current respawn cycle count.
* Increments each time a new cycle starts.
* @returns Number of cycles started
*/
get currentCycle(): number {
return this.cycleCount;
}
/**
* Check if the controller is currently running.
* @returns True if state is not 'stopped'
*/
get isRunning(): boolean {
return this._state !== 'stopped';
}
/**
* Transition to a new state.
* Emits 'stateChanged' event with old and new states.
* No-op if already in the target state.
*
* @param newState - State to transition to
* @fires stateChanged
*/
private setState(newState: RespawnState): void {
if (newState === this._state) return;
@@ -135,13 +356,27 @@ export class RespawnController extends EventEmitter {
this.emit('stateChanged', newState, prevState);
}
/**
* Emit a timestamped log message.
* @param message - Log message content
* @fires log
*/
private log(message: string): void {
const timestamp = new Date().toISOString();
this.emit('log', `[${timestamp}] [Respawn] ${message}`);
}
/**
* Start watching the session for idle state
* Start watching the session for idle state.
*
* Begins monitoring terminal output for idle indicators.
* When idle is detected, starts the respawn cycle.
*
* No-op if:
* - `config.enabled` is false
* - Already running (state !== 'stopped')
*
* @fires stateChanged - Transitions to 'watching'
*/
start(): void {
if (!this.config.enabled) {
@@ -161,7 +396,12 @@ export class RespawnController extends EventEmitter {
}
/**
* Stop the respawn controller
* Stop the respawn controller.
*
* Clears all timers, removes terminal listener, and sets state to 'stopped'.
* Safe to call multiple times.
*
* @fires stateChanged - Transitions to 'stopped'
*/
stop(): void {
this.log('Stopping respawn controller');
@@ -174,7 +414,11 @@ export class RespawnController extends EventEmitter {
}
/**
* Pause respawn (keeps listening but won't trigger)
* Pause respawn without stopping.
*
* Clears timers but keeps listening to terminal.
* State is preserved; won't trigger idle detection while paused.
* Use resume() to continue.
*/
pause(): void {
this.log('Pausing respawn');
@@ -183,7 +427,10 @@ export class RespawnController extends EventEmitter {
}
/**
* Resume respawn
* Resume respawn after pause.
*
* If in 'watching' state, immediately checks for idle condition.
* Otherwise, continues from current state.
*/
resume(): void {
this.log('Resuming respawn');
@@ -192,6 +439,10 @@ export class RespawnController extends EventEmitter {
}
}
/**
* Set up terminal output listener on the session.
* Removes any previous listener first to avoid duplicates.
*/
private setupTerminalListener(): void {
// Remove our previous listener if any (don't remove other listeners!)
if (this.terminalHandler) {
@@ -204,6 +455,16 @@ export class RespawnController extends EventEmitter {
this.session.on('terminal', this.terminalHandler);
}
/**
* Process terminal data for idle/working detection.
*
* 1. Buffers data (with size limit)
* 2. Detects working indicators → resets idle
* 3. Detects ready indicator → triggers state-specific action
* 4. Detects prompt indicators → starts idle timer
*
* @param data - Raw terminal output data
*/
private handleTerminalData(data: string): void {
this.terminalBuffer += data;
@@ -282,7 +543,12 @@ export class RespawnController extends EventEmitter {
}
}
// Step completion handlers - called when ready indicator is detected
/**
* Handle update step completion.
* Called when ready indicator detected in waiting_update state.
* Proceeds to clear, init, or completes cycle based on config.
* @fires stepCompleted - With step 'update'
*/
private checkUpdateComplete(): void {
this.clearIdleTimer();
this.log('Update completed (ready indicator)');
@@ -297,6 +563,11 @@ export class RespawnController extends EventEmitter {
}
}
/**
* Handle /clear step completion.
* Proceeds to init or completes cycle based on config.
* @fires stepCompleted - With step 'clear'
*/
private checkClearComplete(): void {
this.clearIdleTimer();
this.log('/clear completed (ready indicator)');
@@ -309,6 +580,12 @@ export class RespawnController extends EventEmitter {
}
}
/**
* Handle /init step completion.
* If kickstart is configured, monitors for work.
* Otherwise completes cycle.
* @fires stepCompleted - With step 'init' (if no kickstart)
*/
private checkInitComplete(): void {
this.clearIdleTimer();
this.log('/init completed (ready indicator)');
@@ -322,6 +599,11 @@ export class RespawnController extends EventEmitter {
}
}
/**
* Start monitoring to see if /init triggered work.
* Enters 'monitoring_init' state and waits 3s grace period.
* If no work detected, sends kickstart prompt.
*/
private startMonitoringInit(): void {
this.setState('monitoring_init');
this.terminalBuffer = '';
@@ -337,6 +619,11 @@ export class RespawnController extends EventEmitter {
}, 3000); // 3 second grace period for /init to trigger work
}
/**
* Handle monitoring timeout when /init didn't trigger work.
* Sends kickstart prompt as fallback.
* @fires stepCompleted - With step 'init'
*/
private checkMonitoringInitIdle(): void {
this.clearIdleTimer();
if (this.stepTimer) {
@@ -348,6 +635,10 @@ export class RespawnController extends EventEmitter {
this.sendKickstart();
}
/**
* Send the kickstart prompt to get Claude working.
* @fires stepSent - With step 'kickstart'
*/
private sendKickstart(): void {
this.setState('sending_kickstart');
this.terminalBuffer = '';
@@ -363,6 +654,10 @@ export class RespawnController extends EventEmitter {
}, this.config.interStepDelayMs);
}
/**
* Handle kickstart step completion.
* @fires stepCompleted - With step 'kickstart'
*/
private checkKickstartComplete(): void {
this.clearIdleTimer();
this.log('Kickstart completed (ready indicator)');
@@ -370,6 +665,10 @@ export class RespawnController extends EventEmitter {
this.completeCycle();
}
/**
* Start the idle detection timer.
* After idleTimeoutMs, triggers onIdleDetected if still idle.
*/
private startIdleTimer(): void {
this.clearIdleTimer();
@@ -383,6 +682,7 @@ export class RespawnController extends EventEmitter {
}, this.config.idleTimeoutMs);
}
/** Clear the idle detection timer if running */
private clearIdleTimer(): void {
if (this.idleTimer) {
clearTimeout(this.idleTimer);
@@ -390,6 +690,7 @@ export class RespawnController extends EventEmitter {
}
}
/** Clear all timers (idle and step) */
private clearTimers(): void {
this.clearIdleTimer();
if (this.stepTimer) {
@@ -398,6 +699,11 @@ export class RespawnController extends EventEmitter {
}
}
/**
* Handle confirmed idle detection.
* Starts a new respawn cycle.
* @fires respawnCycleStarted
*/
private onIdleDetected(): void {
if (this._state !== 'watching') {
return;
@@ -411,6 +717,10 @@ export class RespawnController extends EventEmitter {
this.sendUpdateDocs();
}
/**
* Send the update docs prompt (first step of cycle).
* @fires stepSent - With step 'update'
*/
private sendUpdateDocs(): void {
this.setState('sending_update');
this.terminalBuffer = ''; // Clear buffer for fresh detection
@@ -426,6 +736,10 @@ export class RespawnController extends EventEmitter {
}, this.config.interStepDelayMs);
}
/**
* Send /clear command.
* @fires stepSent - With step 'clear'
*/
private sendClear(): void {
this.setState('sending_clear');
this.terminalBuffer = '';
@@ -439,6 +753,10 @@ export class RespawnController extends EventEmitter {
}, this.config.interStepDelayMs);
}
/**
* Send /init command.
* @fires stepSent - With step 'init'
*/
private sendInit(): void {
this.setState('sending_init');
this.terminalBuffer = '';
@@ -453,6 +771,11 @@ export class RespawnController extends EventEmitter {
}, this.config.interStepDelayMs);
}
/**
* Complete the current respawn cycle.
* Returns to watching state for next cycle.
* @fires respawnCycleCompleted
*/
private completeCycle(): void {
this.log(`Respawn cycle #${this.cycleCount} completed`);
this.emit('respawnCycleCompleted', this.cycleCount);
@@ -464,6 +787,10 @@ export class RespawnController extends EventEmitter {
this.workingDetected = false;
}
/**
* Check if already idle and start cycle if so.
* Used when resuming from pause.
*/
private checkIdleAndMaybeStart(): void {
// Check if already idle
const timeSinceActivity = Date.now() - this.lastActivityTime;
@@ -473,7 +800,13 @@ export class RespawnController extends EventEmitter {
}
/**
* Update configuration
* Update configuration at runtime.
*
* Merges provided config with existing config.
* Takes effect immediately for new operations.
*
* @param config - Partial configuration to merge
* @fires log - With updated config details
*/
updateConfig(config: Partial<RespawnConfig>): void {
this.config = { ...this.config, ...config };
@@ -481,14 +814,26 @@ export class RespawnController extends EventEmitter {
}
/**
* Get current configuration
* Get current configuration.
* @returns Copy of current config (safe to modify)
*/
getConfig(): RespawnConfig {
return { ...this.config };
}
/**
* Get status information
* Get comprehensive status information.
*
* Useful for debugging and monitoring.
*
* @returns Status object with:
* - state: Current state machine state
* - cycleCount: Number of cycles started
* - lastActivityTime: Timestamp of last activity
* - timeSinceActivity: Milliseconds since last activity
* - promptDetected: Whether prompt indicator seen
* - workingDetected: Whether working indicator seen
* - config: Current configuration
*/
getStatus() {
return {
+202 -18
View File
@@ -1,60 +1,180 @@
/**
* @fileoverview Background Task Tracker for Claude Code sessions
*
* This module tracks background tasks (subagents) spawned by Claude Code
* during session execution. It parses both JSON messages and terminal output
* to detect when tasks are created, updated, and completed.
*
* ## Task Hierarchy
*
* Tasks can be nested (parent-child relationships) when Claude spawns
* a subagent from within another subagent. The tracker maintains a stack
* to track nesting and a tree structure for visualization.
*
* ## Detection Methods
*
* 1. **JSON Messages**: Parses `tool_use` blocks for Task tool invocations
* and `tool_result` blocks for completion
* 2. **Terminal Output**: Fallback pattern matching for launch/complete messages
*
* @module task-tracker
*/
import { EventEmitter } from 'node:events';
// Maximum number of completed tasks to keep in memory
// ========== Configuration Constants ==========
/**
* Maximum number of completed tasks to keep in memory.
* Oldest completed tasks are removed when this limit is exceeded.
*/
const MAX_COMPLETED_TASKS = 100;
// Pre-compiled patterns for performance
// ========== Pre-compiled Regex Patterns ==========
/**
* Patterns that indicate a new task/agent is being launched.
* Used as fallback when JSON parsing doesn't capture the launch.
* Capture group 1: Agent/task type name
*/
const LAUNCH_PATTERNS = [
/Launching\s+(\w+)\s+agent/i,
/Starting\s+(\w+)\s+task/i,
/Spawning\s+(\w+)\s+agent/i,
];
/**
* Patterns that indicate a task has completed.
* Used as fallback when JSON parsing doesn't capture the result.
*/
const COMPLETE_PATTERNS = [
/Task\s+completed/i,
/Agent\s+finished/i,
/Background\s+task\s+done/i,
];
// ========== Type Definitions ==========
/**
* Represents a background task spawned by Claude Code
* Represents a background task spawned by Claude Code.
*
* Tasks form a tree structure where a parent task can spawn child tasks.
* This enables tracking of nested agent invocations.
*/
export interface BackgroundTask {
/** Unique task identifier (usually the tool_use ID from Claude) */
id: string;
/** Parent task ID if this is a nested task, null for root tasks */
parentId: string | null;
/** Human-readable description of what the task is doing */
description: string;
/** Type of subagent (e.g., 'explore', 'bash', 'general-purpose') */
subagentType: string;
/** Current execution status */
status: 'running' | 'completed' | 'failed';
/** Timestamp when task was created (milliseconds since epoch) */
startTime: number;
/** Timestamp when task finished (milliseconds since epoch) */
endTime?: number;
/** Output/result from the task execution */
output?: string;
/** IDs of child tasks spawned by this task */
children: string[];
}
/**
* Events emitted by TaskTracker.
*
* @event taskCreated - New task detected and added
* @event taskUpdated - Task state changed (rarely used)
* @event taskCompleted - Task finished successfully
* @event taskFailed - Task finished with error
*/
export interface TaskTrackerEvents {
/** New task created */
taskCreated: (task: BackgroundTask) => void;
/** Task state updated */
taskUpdated: (task: BackgroundTask) => void;
/** Task completed successfully */
taskCompleted: (task: BackgroundTask) => void;
/** Task failed with error */
taskFailed: (task: BackgroundTask, error: string) => void;
}
/**
* TaskTracker parses Claude Code's output to detect and track background tasks.
* TaskTracker - Detects and tracks background tasks in Claude Code sessions.
*
* Claude Code outputs JSON messages. When it spawns a task via the Task tool,
* we see tool_use blocks with the task parameters. We track these and match
* them with tool_result blocks to track completion.
* ## How It Works
*
* Claude Code outputs JSON messages when executing. When it spawns a subagent
* via the Task tool, we see:
*
* 1. `tool_use` block with `name: "Task"` and input parameters
* 2. ... task execution output ...
* 3. `tool_result` block with the result or error
*
* The tracker maintains:
* - A map of all tasks by ID
* - A stack of currently running tasks (for nesting)
* - Parent-child relationships between tasks
*
* ## Usage
*
* ```typescript
* const tracker = new TaskTracker();
*
* tracker.on('taskCreated', (task) => {
* console.log(`New task: ${task.description}`);
* });
*
* tracker.on('taskCompleted', (task) => {
* console.log(`Task done: ${task.id}`);
* });
*
* // Feed in Claude messages
* tracker.processMessage(claudeJsonMessage);
*
* // Or terminal output as fallback
* tracker.processTerminalOutput(ptyData);
* ```
*
* @extends EventEmitter
*/
export class TaskTracker extends EventEmitter {
/** Map of task ID to task object */
private tasks: Map<string, BackgroundTask> = new Map();
private taskStack: string[] = []; // Stack of active task IDs for nesting
/** Stack of active task IDs for tracking nesting depth */
private taskStack: string[] = [];
/** Pending tool_use blocks waiting for results */
private pendingToolUses: Map<string, { description: string; subagentType: string; parentId: string | null }> = new Map();
/**
* Creates a new TaskTracker instance.
*/
constructor() {
super();
}
/**
* Process a Claude message to detect task events
* Process a Claude JSON message to detect task events.
*
* Looks for `tool_use` blocks with `name: "Task"` and `tool_result` blocks
* to track task lifecycle.
*
* @param msg - Parsed Claude JSON message object
* @fires taskCreated - When a new task is detected
* @fires taskCompleted - When a task finishes successfully
* @fires taskFailed - When a task finishes with error
*/
processMessage(msg: any): void {
if (!msg || !msg.message?.content) return;
@@ -69,9 +189,17 @@ export class TaskTracker extends EventEmitter {
}
/**
* Process raw terminal output to detect task patterns
* This is a fallback for when JSON parsing doesn't capture everything
* Uses pre-compiled patterns for performance
* Process raw terminal output to detect task patterns.
*
* This is a fallback for when JSON parsing doesn't capture everything.
* Uses pre-compiled patterns to detect launch and completion messages.
*
* Note: May create duplicate tasks in some cases; deduplication is
* handled by checking for existing running tasks of the same type.
*
* @param data - Raw terminal output string
* @fires taskCreated - When a launch pattern is matched
* @fires taskCompleted - When a complete pattern is matched
*/
processTerminalOutput(data: string): void {
// Detect task launch patterns in terminal output
@@ -108,6 +236,13 @@ export class TaskTracker extends EventEmitter {
}
}
/**
* Handle a tool_use block for the Task tool.
* Creates a new task and pushes it onto the stack.
*
* @param block - The tool_use content block
* @fires taskCreated
*/
private handleTaskToolUse(block: any): void {
const toolUseId = block.id;
const params = block.input || {};
@@ -146,6 +281,14 @@ export class TaskTracker extends EventEmitter {
this.emit('taskCreated', task);
}
/**
* Handle a tool_result block.
* Marks the corresponding task as completed or failed.
*
* @param block - The tool_result content block
* @fires taskCompleted - If result is success
* @fires taskFailed - If result is error
*/
private handleToolResult(block: any): void {
const toolUseId = block.tool_use_id;
const task = this.tasks.get(toolUseId);
@@ -216,6 +359,14 @@ export class TaskTracker extends EventEmitter {
}
}
/**
* Create a task from terminal pattern detection.
* Used as fallback when JSON messages aren't available.
*
* @param agentType - Type of agent detected
* @param context - Terminal context for debugging
* @fires taskCreated
*/
private createTaskFromTerminal(agentType: string, context: string): void {
const taskId = `terminal-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
const parentId = this.taskStack.length > 0 ? this.taskStack[this.taskStack.length - 1] : null;
@@ -243,6 +394,12 @@ export class TaskTracker extends EventEmitter {
this.emit('taskCreated', task);
}
/**
* Mark a task as completed.
*
* @param taskId - ID of task to complete
* @fires taskCompleted - If task was running
*/
private completeTask(taskId: string): void {
const task = this.tasks.get(taskId);
if (task && task.status === 'running') {
@@ -258,6 +415,12 @@ export class TaskTracker extends EventEmitter {
}
}
/**
* Get the most recently started running task.
* Returns the task at the top of the stack (most nested).
*
* @returns Most recent running task, or undefined if none
*/
private getMostRecentRunningTask(): BackgroundTask | undefined {
// Return the task at the top of the stack
if (this.taskStack.length > 0) {
@@ -268,7 +431,12 @@ export class TaskTracker extends EventEmitter {
}
/**
* Get the task tree as a nested structure
* Get root-level tasks as a list.
*
* Child tasks can be accessed via the `children` array on each task.
* Use this for displaying a task tree in the UI.
*
* @returns Array of tasks without parents (root level)
*/
getTaskTree(): BackgroundTask[] {
const rootTasks: BackgroundTask[] = [];
@@ -283,21 +451,28 @@ export class TaskTracker extends EventEmitter {
}
/**
* Get all tasks as a flat map
* Get all tasks as a flat Map.
*
* @returns Copy of the internal tasks map (safe to modify)
*/
getAllTasks(): Map<string, BackgroundTask> {
return new Map(this.tasks);
}
/**
* Get a specific task by ID
* Get a specific task by its ID.
*
* @param taskId - The task ID to look up
* @returns The task if found, undefined otherwise
*/
getTask(taskId: string): BackgroundTask | undefined {
return this.tasks.get(taskId);
}
/**
* Get count of currently running tasks
* Get the count of currently running tasks.
*
* @returns Number of tasks with status 'running'
*/
getRunningCount(): number {
let count = 0;
@@ -308,7 +483,13 @@ export class TaskTracker extends EventEmitter {
}
/**
* Get summary statistics
* Get aggregated statistics about all tracked tasks.
*
* @returns Object with counts:
* - total: Total number of tracked tasks
* - running: Tasks currently executing
* - completed: Successfully finished tasks
* - failed: Tasks that ended with errors
*/
getStats(): { total: number; running: number; completed: number; failed: number } {
let running = 0, completed = 0, failed = 0;
@@ -325,7 +506,10 @@ export class TaskTracker extends EventEmitter {
}
/**
* Clear all tasks (e.g., when session is cleared)
* Clear all tracked tasks.
*
* Use when the session is cleared or closed.
* Resets the task map, stack, and pending tool uses.
*/
clear(): void {
this.tasks.clear();