automaker/apps/server/src/services/auto-mode-service.ts

/**
 * Auto Mode Service - Autonomous feature implementation using Claude Agent SDK
 *
 * Manages:
 * - Worktree creation for isolated development
 * - Feature execution with Claude
 * - Concurrent execution with max concurrency limits
 * - Progress streaming via events
 * - Verification and merge workflows
 */

import { ProviderFactory } from "../providers/provider-factory.js";
import type { ExecuteOptions } from "../providers/types.js";
import { exec } from "child_process";
import { promisify } from "util";
import path from "path";
import fs from "fs/promises";
import type { EventEmitter } from "../lib/events.js";
import { buildPromptWithImages } from "../lib/prompt-builder.js";
import { resolveModelString, DEFAULT_MODELS } from "../lib/model-resolver.js";
import { createAutoModeOptions } from "../lib/sdk-options.js";
import { classifyError } from "../lib/error-handler.js";
import { resolveDependencies, areDependenciesSatisfied } from "../lib/dependency-resolver.js";
import type { Feature } from "./feature-loader.js";
import {
  getFeatureDir,
  getFeaturesDir,
  getAutomakerDir,
  getWorktreesDir,
} from "../lib/automaker-paths.js";

const execAsync = promisify(exec);

// Planning mode types for spec-driven development
type PlanningMode = 'skip' | 'lite' | 'spec' | 'full';

interface ParsedTask {
  id: string;          // e.g., "T001"
  description: string; // e.g., "Create user model"
  filePath?: string;   // e.g., "src/models/user.ts"
  phase?: string;      // e.g., "Phase 1: Foundation" (for full mode)
  status: 'pending' | 'in_progress' | 'completed' | 'failed';
}

interface PlanSpec {
  status: 'pending' | 'generating' | 'generated' | 'approved' | 'rejected';
  content?: string;
  version: number;
  generatedAt?: string;
  approvedAt?: string;
  reviewedByUser: boolean;
  tasksCompleted?: number;
  tasksTotal?: number;
  currentTaskId?: string;
  tasks?: ParsedTask[];
}

const PLANNING_PROMPTS = {
  lite: `## Planning Phase (Lite Mode)

IMPORTANT: Do NOT output exploration text, tool usage, or thinking before the plan. Start DIRECTLY with the planning outline format below. Silently analyze the codebase first, then output ONLY the structured plan.

Create a brief planning outline:

1. **Goal**: What are we accomplishing? (1 sentence)
2. **Approach**: How will we do it? (2-3 sentences)
3. **Files to Touch**: List files and what changes
4. **Tasks**: Numbered task list (3-7 items)
5. **Risks**: Any gotchas to watch for

After generating the outline, output:
"[PLAN_GENERATED] Planning outline complete."

Then proceed with implementation.`,

  lite_with_approval: `## Planning Phase (Lite Mode)

IMPORTANT: Do NOT output exploration text, tool usage, or thinking before the plan. Start DIRECTLY with the planning outline format below. Silently analyze the codebase first, then output ONLY the structured plan.

Create a brief planning outline:

1. **Goal**: What are we accomplishing? (1 sentence)
2. **Approach**: How will we do it? (2-3 sentences)
3. **Files to Touch**: List files and what changes
4. **Tasks**: Numbered task list (3-7 items)
5. **Risks**: Any gotchas to watch for

After generating the outline, output:
"[SPEC_GENERATED] Please review the planning outline above. Reply with 'approved' to proceed or provide feedback for revisions."

DO NOT proceed with implementation until you receive explicit approval.`,

  spec: `## Specification Phase (Spec Mode)

IMPORTANT: Do NOT output exploration text, tool usage, or thinking before the spec. Start DIRECTLY with the specification format below. Silently analyze the codebase first, then output ONLY the structured specification.

Generate a specification with an actionable task breakdown. WAIT for approval before implementing.

### Specification Format

1. **Problem**: What problem are we solving? (user perspective)

2. **Solution**: Brief approach (1-2 sentences)

3. **Acceptance Criteria**: 3-5 items in GIVEN-WHEN-THEN format
   - GIVEN [context], WHEN [action], THEN [outcome]

4. **Files to Modify**:
   | File | Purpose | Action |
   |------|---------|--------|
   | path/to/file | description | create/modify/delete |

5. **Implementation Tasks**:
   Use this EXACT format for each task (the system will parse these):
   \`\`\`tasks
   - [ ] T001: [Description] | File: [path/to/file]
   - [ ] T002: [Description] | File: [path/to/file]
   - [ ] T003: [Description] | File: [path/to/file]
   \`\`\`

   Task ID rules:
   - Sequential: T001, T002, T003, etc.
   - Description: Clear action (e.g., "Create user model", "Add API endpoint")
   - File: Primary file affected (helps with context)
   - Order by dependencies (foundational tasks first)

6. **Verification**: How to confirm feature works

After generating the spec, output on its own line:
"[SPEC_GENERATED] Please review the specification above. Reply with 'approved' to proceed or provide feedback for revisions."

DO NOT proceed with implementation until you receive explicit approval.

When approved, execute tasks SEQUENTIALLY in order. For each task:
1. BEFORE starting, output: "[TASK_START] T###: Description"
2. Implement the task
3. AFTER completing, output: "[TASK_COMPLETE] T###: Brief summary"

This allows real-time progress tracking during implementation.`,

  full: `## Full Specification Phase (Full SDD Mode)

IMPORTANT: Do NOT output exploration text, tool usage, or thinking before the spec. Start DIRECTLY with the specification format below. Silently analyze the codebase first, then output ONLY the structured specification.

Generate a comprehensive specification with phased task breakdown. WAIT for approval before implementing.

### Specification Format

1. **Problem Statement**: 2-3 sentences from user perspective

2. **User Story**: As a [user], I want [goal], so that [benefit]

3. **Acceptance Criteria**: Multiple scenarios with GIVEN-WHEN-THEN
   - **Happy Path**: GIVEN [context], WHEN [action], THEN [expected outcome]
   - **Edge Cases**: GIVEN [edge condition], WHEN [action], THEN [handling]
   - **Error Handling**: GIVEN [error condition], WHEN [action], THEN [error response]

4. **Technical Context**:
   | Aspect | Value |
   |--------|-------|
   | Affected Files | list of files |
   | Dependencies | external libs if any |
   | Constraints | technical limitations |
   | Patterns to Follow | existing patterns in codebase |

5. **Non-Goals**: What this feature explicitly does NOT include

6. **Implementation Tasks**:
   Use this EXACT format for each task (the system will parse these):
   \`\`\`tasks
   ## Phase 1: Foundation
   - [ ] T001: [Description] | File: [path/to/file]
   - [ ] T002: [Description] | File: [path/to/file]

   ## Phase 2: Core Implementation
   - [ ] T003: [Description] | File: [path/to/file]
   - [ ] T004: [Description] | File: [path/to/file]

   ## Phase 3: Integration & Testing
   - [ ] T005: [Description] | File: [path/to/file]
   - [ ] T006: [Description] | File: [path/to/file]
   \`\`\`

   Task ID rules:
   - Sequential across all phases: T001, T002, T003, etc.
   - Description: Clear action verb + target
   - File: Primary file affected
   - Order by dependencies within each phase
   - Phase structure helps organize complex work

7. **Success Metrics**: How we know it's done (measurable criteria)

8. **Risks & Mitigations**:
   | Risk | Mitigation |
   |------|------------|
   | description | approach |

After generating the spec, output on its own line:
"[SPEC_GENERATED] Please review the comprehensive specification above. Reply with 'approved' to proceed or provide feedback for revisions."

DO NOT proceed with implementation until you receive explicit approval.

When approved, execute tasks SEQUENTIALLY by phase. For each task:
1. BEFORE starting, output: "[TASK_START] T###: Description"
2. Implement the task
3. AFTER completing, output: "[TASK_COMPLETE] T###: Brief summary"

After completing all tasks in a phase, output:
"[PHASE_COMPLETE] Phase N complete"

This allows real-time progress tracking during implementation.`
};

/**
 * Parse tasks from generated spec content
 * Looks for the ```tasks code block and extracts task lines
 * Format: - [ ] T###: Description | File: path/to/file
 */
function parseTasksFromSpec(specContent: string): ParsedTask[] {
  const tasks: ParsedTask[] = [];

  // Extract content within ```tasks ... ``` block
  const tasksBlockMatch = specContent.match(/```tasks\s*([\s\S]*?)```/);
  if (!tasksBlockMatch) {
    // Try fallback: look for task lines anywhere in content
    const taskLines = specContent.match(/- \[ \] T\d{3}:.*$/gm);
    if (!taskLines) {
      return tasks;
    }
    // Parse fallback task lines
    let currentPhase: string | undefined;
    for (const line of taskLines) {
      const parsed = parseTaskLine(line, currentPhase);
      if (parsed) {
        tasks.push(parsed);
      }
    }
    return tasks;
  }

  const tasksContent = tasksBlockMatch[1];
  const lines = tasksContent.split('\n');

  let currentPhase: string | undefined;

  for (const line of lines) {
    const trimmedLine = line.trim();

    // Check for phase header (e.g., "## Phase 1: Foundation")
    const phaseMatch = trimmedLine.match(/^##\s*(.+)$/);
    if (phaseMatch) {
      currentPhase = phaseMatch[1].trim();
      continue;
    }

    // Check for task line
    if (trimmedLine.startsWith('- [ ]')) {
      const parsed = parseTaskLine(trimmedLine, currentPhase);
      if (parsed) {
        tasks.push(parsed);
      }
    }
  }

  return tasks;
}

/**
 * Parse a single task line
 * Format: - [ ] T###: Description | File: path/to/file
 */
function parseTaskLine(line: string, currentPhase?: string): ParsedTask | null {
  // Match pattern: - [ ] T###: Description | File: path
  const taskMatch = line.match(/- \[ \] (T\d{3}):\s*([^|]+)(?:\|\s*File:\s*(.+))?$/);
  if (!taskMatch) {
    // Try simpler pattern without file
    const simpleMatch = line.match(/- \[ \] (T\d{3}):\s*(.+)$/);
    if (simpleMatch) {
      return {
        id: simpleMatch[1],
        description: simpleMatch[2].trim(),
        phase: currentPhase,
        status: 'pending',
      };
    }
    return null;
  }

  return {
    id: taskMatch[1],
    description: taskMatch[2].trim(),
    filePath: taskMatch[3]?.trim(),
    phase: currentPhase,
    status: 'pending',
  };
}

// Feature type is imported from feature-loader.js
// Extended type with planning fields for local use
interface FeatureWithPlanning extends Feature {
  planningMode?: PlanningMode;
  planSpec?: PlanSpec;
  requirePlanApproval?: boolean;
}

interface RunningFeature {
  featureId: string;
  projectPath: string;
  worktreePath: string | null;
  branchName: string | null;
  abortController: AbortController;
  isAutoMode: boolean;
  startTime: number;
}

interface AutoModeConfig {
  maxConcurrency: number;
  useWorktrees: boolean;
  projectPath: string;
}

interface PendingApproval {
  resolve: (result: { approved: boolean; editedPlan?: string; feedback?: string }) => void;
  reject: (error: Error) => void;
  featureId: string;
  projectPath: string;
}

export class AutoModeService {
  private events: EventEmitter;
  private runningFeatures = new Map<string, RunningFeature>();
  private autoLoopRunning = false;
  private autoLoopAbortController: AbortController | null = null;
  private config: AutoModeConfig | null = null;
  private pendingApprovals = new Map<string, PendingApproval>();

  constructor(events: EventEmitter) {
    this.events = events;
  }

  /**
   * Start the auto mode loop - continuously picks and executes pending features
   */
  async startAutoLoop(projectPath: string, maxConcurrency = 3): Promise<void> {
    if (this.autoLoopRunning) {
      throw new Error("Auto mode is already running");
    }

    this.autoLoopRunning = true;
    this.autoLoopAbortController = new AbortController();
    this.config = {
      maxConcurrency,
      useWorktrees: true,
      projectPath,
    };

    this.emitAutoModeEvent("auto_mode_started", {
      message: `Auto mode started with max ${maxConcurrency} concurrent features`,
      projectPath,
    });

    // Run the loop in the background
    this.runAutoLoop().catch((error) => {
      console.error("[AutoMode] Loop error:", error);
      const errorInfo = classifyError(error);
      this.emitAutoModeEvent("auto_mode_error", {
        error: errorInfo.message,
        errorType: errorInfo.type,
      });
    });
  }

  private async runAutoLoop(): Promise<void> {
    while (
      this.autoLoopRunning &&
      this.autoLoopAbortController &&
      !this.autoLoopAbortController.signal.aborted
    ) {
      try {
        // Check if we have capacity
        if (this.runningFeatures.size >= (this.config?.maxConcurrency || 3)) {
          await this.sleep(5000);
          continue;
        }

        // Load pending features
        const pendingFeatures = await this.loadPendingFeatures(
          this.config!.projectPath
        );

        if (pendingFeatures.length === 0) {
          this.emitAutoModeEvent("auto_mode_idle", {
            message: "No pending features - auto mode idle",
            projectPath: this.config!.projectPath,
          });
          await this.sleep(10000);
          continue;
        }

        // Find a feature not currently running
        const nextFeature = pendingFeatures.find(
          (f) => !this.runningFeatures.has(f.id)
        );

        if (nextFeature) {
          // Start feature execution in background
          this.executeFeature(
            this.config!.projectPath,
            nextFeature.id,
            this.config!.useWorktrees,
            true
          ).catch((error) => {
            console.error(`[AutoMode] Feature ${nextFeature.id} error:`, error);
          });
        }

        await this.sleep(2000);
      } catch (error) {
        console.error("[AutoMode] Loop iteration error:", error);
        await this.sleep(5000);
      }
    }

    this.autoLoopRunning = false;
  }

  /**
   * Stop the auto mode loop
   */
  async stopAutoLoop(): Promise<number> {
    const wasRunning = this.autoLoopRunning;
    this.autoLoopRunning = false;
    if (this.autoLoopAbortController) {
      this.autoLoopAbortController.abort();
      this.autoLoopAbortController = null;
    }

    // Emit stop event immediately when user explicitly stops
    if (wasRunning) {
      this.emitAutoModeEvent("auto_mode_stopped", {
        message: "Auto mode stopped",
        projectPath: this.config?.projectPath,
      });
    }

    return this.runningFeatures.size;
  }

  /**
   * Execute a single feature
   * @param projectPath - The main project path
   * @param featureId - The feature ID to execute
   * @param useWorktrees - Whether to use worktrees for isolation
   * @param isAutoMode - Whether this is running in auto mode
   * @param providedWorktreePath - Optional: use this worktree path instead of creating a new one
   */
  async executeFeature(
    projectPath: string,
    featureId: string,
    useWorktrees = false,
    isAutoMode = false,
    providedWorktreePath?: string,
    options?: {
      continuationPrompt?: string;
    }
  ): Promise<void> {
    if (this.runningFeatures.has(featureId)) {
      throw new Error(`Feature ${featureId} is already running`);
    }

    const abortController = new AbortController();
    const branchName = `feature/${featureId}`;
    let worktreePath: string | null = null;

    // Use provided worktree path if given, otherwise setup new worktree if enabled
    if (providedWorktreePath) {
      // Resolve to absolute path - critical for cross-platform compatibility
      // On Windows, relative paths or paths with forward slashes may not work correctly with cwd
      // On all platforms, absolute paths ensure commands execute in the correct directory
      try {
        // Resolve relative paths relative to projectPath, absolute paths as-is
        const resolvedPath = path.isAbsolute(providedWorktreePath)
          ? path.resolve(providedWorktreePath)
          : path.resolve(projectPath, providedWorktreePath);
        
        // Verify the path exists before using it
        await fs.access(resolvedPath);
        worktreePath = resolvedPath;
        console.log(`[AutoMode] Using provided worktree path (resolved): ${worktreePath}`);
      } catch (error) {
        console.error(`[AutoMode] Provided worktree path invalid or doesn't exist: ${providedWorktreePath}`, error);
        // Fall through to create new worktree or use project path
      }
    }
    
    if (!worktreePath && useWorktrees) {
      // No specific worktree provided, create a new one for this feature
      worktreePath = await this.setupWorktree(
        projectPath,
        featureId,
        branchName
      );
    }

    // Ensure workDir is always an absolute path for cross-platform compatibility
    const workDir = worktreePath ? path.resolve(worktreePath) : path.resolve(projectPath);

    this.runningFeatures.set(featureId, {
      featureId,
      projectPath,
      worktreePath,
      branchName,
      abortController,
      isAutoMode,
      startTime: Date.now(),
    });

    // Emit feature start event
    this.emitAutoModeEvent("auto_mode_feature_start", {
      featureId,
      projectPath,
      feature: {
        id: featureId,
        title: "Loading...",
        description: "Feature is starting",
      },
    });

    try {
      // Load feature details
      const feature = await this.loadFeature(projectPath, featureId);
      if (!feature) {
        throw new Error(`Feature ${featureId} not found`);
      }

      // Update feature status to in_progress
      await this.updateFeatureStatus(projectPath, featureId, "in_progress");

      // Build the prompt - use continuation prompt if provided (for recovery after plan approval)
      let prompt: string;
      if (options?.continuationPrompt) {
        // Continuation prompt is used when recovering from a plan approval
        // The plan was already approved, so skip the planning phase
        prompt = options.continuationPrompt;
        console.log(`[AutoMode] Using continuation prompt for feature ${featureId}`);
      } else {
        // Normal flow: build prompt with planning phase
        const featurePrompt = this.buildFeaturePrompt(feature);
        const planningPrefix = this.getPlanningPromptPrefix(feature);
        prompt = planningPrefix + featurePrompt;

        // Emit planning mode info
        if (feature.planningMode && feature.planningMode !== 'skip') {
          this.emitAutoModeEvent('planning_started', {
            featureId: feature.id,
            mode: feature.planningMode,
            message: `Starting ${feature.planningMode} planning phase`
          });
        }
      }

      // Extract image paths from feature
      const imagePaths = feature.imagePaths?.map((img) =>
        typeof img === "string" ? img : img.path
      );

      // Get model from feature
      const model = resolveModelString(feature.model, DEFAULT_MODELS.claude);
      console.log(
        `[AutoMode] Executing feature ${featureId} with model: ${model}`
      );

      // Run the agent with the feature's model and images
      await this.runAgent(
        workDir,
        featureId,
        prompt,
        abortController,
        imagePaths,
        model,
        {
          projectPath,
          planningMode: feature.planningMode,
          requirePlanApproval: feature.requirePlanApproval,
        }
      );

      // Mark as waiting_approval for user review
      await this.updateFeatureStatus(
        projectPath,
        featureId,
        "waiting_approval"
      );

      this.emitAutoModeEvent("auto_mode_feature_complete", {
        featureId,
        passes: true,
        message: `Feature completed in ${Math.round(
          (Date.now() - this.runningFeatures.get(featureId)!.startTime) / 1000
        )}s`,
        projectPath,
      });
    } catch (error) {
      const errorInfo = classifyError(error);

      if (errorInfo.isAbort) {
        this.emitAutoModeEvent("auto_mode_feature_complete", {
          featureId,
          passes: false,
          message: "Feature stopped by user",
          projectPath,
        });
      } else {
        console.error(`[AutoMode] Feature ${featureId} failed:`, error);
        await this.updateFeatureStatus(projectPath, featureId, "backlog");
        this.emitAutoModeEvent("auto_mode_error", {
          featureId,
          error: errorInfo.message,
          errorType: errorInfo.type,
          projectPath,
        });
      }
    } finally {
      console.log(`[AutoMode] Feature ${featureId} execution ended, cleaning up runningFeatures`);
      console.log(`[AutoMode] Pending approvals at cleanup: ${Array.from(this.pendingApprovals.keys()).join(', ') || 'none'}`);
      this.runningFeatures.delete(featureId);
    }
  }

  /**
   * Stop a specific feature
   */
  async stopFeature(featureId: string): Promise<boolean> {
    const running = this.runningFeatures.get(featureId);
    if (!running) {
      return false;
    }

    // Cancel any pending plan approval for this feature
    this.cancelPlanApproval(featureId);

    running.abortController.abort();
    return true;
  }

  /**
   * Resume a feature (continues from saved context)
   */
  async resumeFeature(
    projectPath: string,
    featureId: string,
    useWorktrees = false
  ): Promise<void> {
    // Check if context exists in .automaker directory
    const featureDir = getFeatureDir(projectPath, featureId);
    const contextPath = path.join(featureDir, "agent-output.md");

    let hasContext = false;
    try {
      await fs.access(contextPath);
      hasContext = true;
    } catch {
      // No context
    }

    if (hasContext) {
      // Load previous context and continue
      const context = await fs.readFile(contextPath, "utf-8");
      return this.executeFeatureWithContext(
        projectPath,
        featureId,
        context,
        useWorktrees
      );
    }

    // No context, start fresh
    return this.executeFeature(projectPath, featureId, useWorktrees, false);
  }

  /**
   * Follow up on a feature with additional instructions
   */
  async followUpFeature(
    projectPath: string,
    featureId: string,
    prompt: string,
    imagePaths?: string[],
    providedWorktreePath?: string
  ): Promise<void> {
    if (this.runningFeatures.has(featureId)) {
      throw new Error(`Feature ${featureId} is already running`);
    }

    const abortController = new AbortController();

    // Use the provided worktreePath (from the feature's assigned branch)
    // Fall back to project path if not provided
    let workDir = path.resolve(projectPath);
    let worktreePath: string | null = null;

    if (providedWorktreePath) {
      try {
        // Resolve to absolute path - critical for cross-platform compatibility
        // On Windows, relative paths or paths with forward slashes may not work correctly with cwd
        // On all platforms, absolute paths ensure commands execute in the correct directory
        const resolvedPath = path.isAbsolute(providedWorktreePath)
          ? path.resolve(providedWorktreePath)
          : path.resolve(projectPath, providedWorktreePath);
        
        await fs.access(resolvedPath);
        workDir = resolvedPath;
        worktreePath = resolvedPath;
      } catch {
        // Worktree path provided but doesn't exist, use project path
        console.log(`[AutoMode] Provided worktreePath doesn't exist: ${providedWorktreePath}, using project path`);
      }
    }

    // Load feature info for context
    const feature = await this.loadFeature(projectPath, featureId);

    // Load previous agent output if it exists
    const featureDir = getFeatureDir(projectPath, featureId);
    const contextPath = path.join(featureDir, "agent-output.md");
    let previousContext = "";
    try {
      previousContext = await fs.readFile(contextPath, "utf-8");
    } catch {
      // No previous context
    }

    // Build complete prompt with feature info, previous context, and follow-up instructions
    let fullPrompt = `## Follow-up on Feature Implementation

${feature ? this.buildFeaturePrompt(feature) : `**Feature ID:** ${featureId}`}
`;

    if (previousContext) {
      fullPrompt += `
## Previous Agent Work
The following is the output from the previous implementation attempt:

${previousContext}
`;
    }

    fullPrompt += `
## Follow-up Instructions
${prompt}

## Task
Address the follow-up instructions above. Review the previous work and make the requested changes or fixes.`;

    this.runningFeatures.set(featureId, {
      featureId,
      projectPath,
      worktreePath,
      branchName: worktreePath ? path.basename(worktreePath) : null,
      abortController,
      isAutoMode: false,
      startTime: Date.now(),
    });

    this.emitAutoModeEvent("auto_mode_feature_start", {
      featureId,
      projectPath,
      feature: feature || {
        id: featureId,
        title: "Follow-up",
        description: prompt.substring(0, 100),
      },
    });

    try {
      // Get model from feature (already loaded above)
      const model = resolveModelString(feature?.model, DEFAULT_MODELS.claude);
      console.log(
        `[AutoMode] Follow-up for feature ${featureId} using model: ${model}`
      );

      // Update feature status to in_progress
      await this.updateFeatureStatus(projectPath, featureId, "in_progress");

      // Copy follow-up images to feature folder
      const copiedImagePaths: string[] = [];
      if (imagePaths && imagePaths.length > 0) {
        const featureDirForImages = getFeatureDir(projectPath, featureId);
        const featureImagesDir = path.join(featureDirForImages, "images");

        await fs.mkdir(featureImagesDir, { recursive: true });

        for (const imagePath of imagePaths) {
          try {
            // Get the filename from the path
            const filename = path.basename(imagePath);
            const destPath = path.join(featureImagesDir, filename);

            // Copy the image
            await fs.copyFile(imagePath, destPath);

            // Store the absolute path (external storage uses absolute paths)
            copiedImagePaths.push(destPath);
          } catch (error) {
            console.error(
              `[AutoMode] Failed to copy follow-up image ${imagePath}:`,
              error
            );
          }
        }
      }

      // Update feature object with new follow-up images BEFORE building prompt
      if (copiedImagePaths.length > 0 && feature) {
        const currentImagePaths = feature.imagePaths || [];
        const newImagePaths = copiedImagePaths.map((p) => ({
          path: p,
          filename: path.basename(p),
          mimeType: "image/png", // Default, could be improved
        }));

        feature.imagePaths = [...currentImagePaths, ...newImagePaths];
      }

      // Combine original feature images with new follow-up images
      const allImagePaths: string[] = [];

      // Add all images from feature (now includes both original and new)
      if (feature?.imagePaths) {
        const allPaths = feature.imagePaths.map((img) =>
          typeof img === "string" ? img : img.path
        );
        allImagePaths.push(...allPaths);
      }

      // Save updated feature.json with new images
      if (copiedImagePaths.length > 0 && feature) {
        const featureDirForSave = getFeatureDir(projectPath, featureId);
        const featurePath = path.join(featureDirForSave, "feature.json");

        try {
          await fs.writeFile(featurePath, JSON.stringify(feature, null, 2));
        } catch (error) {
          console.error(`[AutoMode] Failed to save feature.json:`, error);
        }
      }

      // Use fullPrompt (already built above) with model and all images
      // Note: Follow-ups skip planning mode - they continue from previous work
      // Pass previousContext so the history is preserved in the output file
      await this.runAgent(
        workDir,
        featureId,
        fullPrompt,
        abortController,
        allImagePaths.length > 0 ? allImagePaths : imagePaths,
        model,
        {
          projectPath,
          planningMode: 'skip', // Follow-ups don't require approval
          previousContent: previousContext || undefined,
        }
      );

      // Mark as waiting_approval for user review
      await this.updateFeatureStatus(
        projectPath,
        featureId,
        "waiting_approval"
      );

      this.emitAutoModeEvent("auto_mode_feature_complete", {
        featureId,
        passes: true,
        message: "Follow-up completed successfully",
        projectPath,
      });
    } catch (error) {
      const errorInfo = classifyError(error);
      if (!errorInfo.isCancellation) {
        this.emitAutoModeEvent("auto_mode_error", {
          featureId,
          error: errorInfo.message,
          errorType: errorInfo.type,
          projectPath,
        });
      }
    } finally {
      this.runningFeatures.delete(featureId);
    }
  }

  /**
   * Verify a feature's implementation
   */
  async verifyFeature(
    projectPath: string,
    featureId: string
  ): Promise<boolean> {
    // Worktrees are in project dir
    const worktreePath = path.join(projectPath, ".worktrees", featureId);
    let workDir = projectPath;

    try {
      await fs.access(worktreePath);
      workDir = worktreePath;
    } catch {
      // No worktree
    }

    // Run verification - check if tests pass, build works, etc.
    const verificationChecks = [
      { cmd: "npm run lint", name: "Lint" },
      { cmd: "npm run typecheck", name: "Type check" },
      { cmd: "npm test", name: "Tests" },
      { cmd: "npm run build", name: "Build" },
    ];

    let allPassed = true;
    const results: Array<{ check: string; passed: boolean; output?: string }> =
      [];

    for (const check of verificationChecks) {
      try {
        const { stdout, stderr } = await execAsync(check.cmd, {
          cwd: workDir,
          timeout: 120000,
        });
        results.push({
          check: check.name,
          passed: true,
          output: stdout || stderr,
        });
      } catch (error) {
        allPassed = false;
        results.push({
          check: check.name,
          passed: false,
          output: (error as Error).message,
        });
        break; // Stop on first failure
      }
    }

    this.emitAutoModeEvent("auto_mode_feature_complete", {
      featureId,
      passes: allPassed,
      message: allPassed
        ? "All verification checks passed"
        : `Verification failed: ${
            results.find((r) => !r.passed)?.check || "Unknown"
          }`,
    });

    return allPassed;
  }

  /**
   * Commit feature changes
   * @param projectPath - The main project path
   * @param featureId - The feature ID to commit
   * @param providedWorktreePath - Optional: the worktree path where the feature's changes are located
   */
  async commitFeature(
    projectPath: string,
    featureId: string,
    providedWorktreePath?: string
  ): Promise<string | null> {
    let workDir = projectPath;

    // Use the provided worktree path if given
    if (providedWorktreePath) {
      try {
        await fs.access(providedWorktreePath);
        workDir = providedWorktreePath;
        console.log(`[AutoMode] Committing in provided worktree: ${workDir}`);
      } catch {
        console.log(`[AutoMode] Provided worktree path doesn't exist: ${providedWorktreePath}, using project path`);
      }
    } else {
      // Fallback: try to find worktree at legacy location
      const legacyWorktreePath = path.join(projectPath, ".worktrees", featureId);
      try {
        await fs.access(legacyWorktreePath);
        workDir = legacyWorktreePath;
        console.log(`[AutoMode] Committing in legacy worktree: ${workDir}`);
      } catch {
        console.log(`[AutoMode] No worktree found, committing in project path: ${workDir}`);
      }
    }

    try {
      // Check for changes
      const { stdout: status } = await execAsync("git status --porcelain", {
        cwd: workDir,
      });
      if (!status.trim()) {
        return null; // No changes
      }

      // Load feature for commit message
      const feature = await this.loadFeature(projectPath, featureId);
      const commitMessage = feature
        ? `feat: ${this.extractTitleFromDescription(
            feature.description
          )}\n\nImplemented by Automaker auto-mode`
        : `feat: Feature ${featureId}`;

      // Stage and commit
      await execAsync("git add -A", { cwd: workDir });
      await execAsync(`git commit -m "${commitMessage.replace(/"/g, '\\"')}"`, {
        cwd: workDir,
      });

      // Get commit hash
      const { stdout: hash } = await execAsync("git rev-parse HEAD", {
        cwd: workDir,
      });

      this.emitAutoModeEvent("auto_mode_feature_complete", {
        featureId,
        passes: true,
        message: `Changes committed: ${hash.trim().substring(0, 8)}`,
      });

      return hash.trim();
    } catch (error) {
      console.error(`[AutoMode] Commit failed for ${featureId}:`, error);
      return null;
    }
  }

  /**
   * Check if context exists for a feature
   */
  async contextExists(
    projectPath: string,
    featureId: string
  ): Promise<boolean> {
    // Context is stored in .automaker directory
    const featureDir = getFeatureDir(projectPath, featureId);
    const contextPath = path.join(featureDir, "agent-output.md");

    try {
      await fs.access(contextPath);
      return true;
    } catch {
      return false;
    }
  }

  /**
   * Analyze project to gather context
   */
  async analyzeProject(projectPath: string): Promise<void> {
    const abortController = new AbortController();

    const analysisFeatureId = `analysis-${Date.now()}`;
    this.emitAutoModeEvent("auto_mode_feature_start", {
      featureId: analysisFeatureId,
      projectPath,
      feature: {
        id: analysisFeatureId,
        title: "Project Analysis",
        description: "Analyzing project structure",
      },
    });

    const prompt = `Analyze this project and provide a summary of:
1. Project structure and architecture
2. Main technologies and frameworks used
3. Key components and their responsibilities
4. Build and test commands
5. Any existing conventions or patterns

Format your response as a structured markdown document.`;

    try {
      // Use default Claude model for analysis (can be overridden in the future)
      const analysisModel = resolveModelString(
        undefined,
        DEFAULT_MODELS.claude
      );
      const provider = ProviderFactory.getProviderForModel(analysisModel);

      const options: ExecuteOptions = {
        prompt,
        model: analysisModel,
        maxTurns: 5,
        cwd: projectPath,
        allowedTools: ["Read", "Glob", "Grep"],
        abortController,
      };

      const stream = provider.executeQuery(options);
      let analysisResult = "";

      for await (const msg of stream) {
        if (msg.type === "assistant" && msg.message?.content) {
          for (const block of msg.message.content) {
            if (block.type === "text") {
              analysisResult = block.text || "";
              this.emitAutoModeEvent("auto_mode_progress", {
                featureId: analysisFeatureId,
                content: block.text,
                projectPath,
              });
            }
          }
        } else if (msg.type === "result" && msg.subtype === "success") {
          analysisResult = msg.result || analysisResult;
        }
      }

      // Save analysis to .automaker directory
      const automakerDir = getAutomakerDir(projectPath);
      const analysisPath = path.join(automakerDir, "project-analysis.md");
      await fs.mkdir(automakerDir, { recursive: true });
      await fs.writeFile(analysisPath, analysisResult);

      this.emitAutoModeEvent("auto_mode_feature_complete", {
        featureId: analysisFeatureId,
        passes: true,
        message: "Project analysis completed",
        projectPath,
      });
    } catch (error) {
      const errorInfo = classifyError(error);
      this.emitAutoModeEvent("auto_mode_error", {
        featureId: analysisFeatureId,
        error: errorInfo.message,
        errorType: errorInfo.type,
        projectPath,
      });
    }
  }

  /**
   * Get current status
   */
  getStatus(): {
    isRunning: boolean;
    autoLoopRunning: boolean;
    runningFeatures: string[];
    runningCount: number;
  } {
    return {
      isRunning: this.autoLoopRunning || this.runningFeatures.size > 0,
      autoLoopRunning: this.autoLoopRunning,
      runningFeatures: Array.from(this.runningFeatures.keys()),
      runningCount: this.runningFeatures.size,
    };
  }

  /**
   * Get detailed info about all running agents
   */
  getRunningAgents(): Array<{
    featureId: string;
    projectPath: string;
    projectName: string;
    isAutoMode: boolean;
  }> {
    return Array.from(this.runningFeatures.values()).map((rf) => ({
      featureId: rf.featureId,
      projectPath: rf.projectPath,
      projectName: path.basename(rf.projectPath),
      isAutoMode: rf.isAutoMode,
    }));
  }

  /**
   * Wait for plan approval from the user.
   * Returns a promise that resolves when the user approves/rejects the plan.
   */
  waitForPlanApproval(
    featureId: string,
    projectPath: string
  ): Promise<{ approved: boolean; editedPlan?: string; feedback?: string }> {
    console.log(`[AutoMode] Registering pending approval for feature ${featureId}`);
    console.log(`[AutoMode] Current pending approvals: ${Array.from(this.pendingApprovals.keys()).join(', ') || 'none'}`);
    return new Promise((resolve, reject) => {
      this.pendingApprovals.set(featureId, {
        resolve,
        reject,
        featureId,
        projectPath,
      });
      console.log(`[AutoMode] Pending approval registered for feature ${featureId}`);
    });
  }

  /**
   * Resolve a pending plan approval.
   * Called when the user approves or rejects the plan via API.
   */
  async resolvePlanApproval(
    featureId: string,
    approved: boolean,
    editedPlan?: string,
    feedback?: string,
    projectPathFromClient?: string
  ): Promise<{ success: boolean; error?: string }> {
    console.log(`[AutoMode] resolvePlanApproval called for feature ${featureId}, approved=${approved}`);
    console.log(`[AutoMode] Current pending approvals: ${Array.from(this.pendingApprovals.keys()).join(', ') || 'none'}`);
    const pending = this.pendingApprovals.get(featureId);

    if (!pending) {
      console.log(`[AutoMode] No pending approval in Map for feature ${featureId}`);

      // RECOVERY: If no pending approval but we have projectPath from client,
      // check if feature's planSpec.status is 'generated' and handle recovery
      if (projectPathFromClient) {
        console.log(`[AutoMode] Attempting recovery with projectPath: ${projectPathFromClient}`);
        const feature = await this.loadFeature(projectPathFromClient, featureId);

        if (feature?.planSpec?.status === 'generated') {
          console.log(`[AutoMode] Feature ${featureId} has planSpec.status='generated', performing recovery`);

          if (approved) {
            // Update planSpec to approved
            await this.updateFeaturePlanSpec(projectPathFromClient, featureId, {
              status: 'approved',
              approvedAt: new Date().toISOString(),
              reviewedByUser: true,
              content: editedPlan || feature.planSpec.content,
            });

            // Build continuation prompt and re-run the feature
            const planContent = editedPlan || feature.planSpec.content || '';
            let continuationPrompt = `The plan/specification has been approved. `;
            if (feedback) {
              continuationPrompt += `\n\nUser feedback: ${feedback}\n\n`;
            }
            continuationPrompt += `Now proceed with the implementation as specified in the plan:\n\n${planContent}\n\nImplement the feature now.`;

            console.log(`[AutoMode] Starting recovery execution for feature ${featureId}`);

            // Start feature execution with the continuation prompt (async, don't await)
            // Pass undefined for providedWorktreePath, use options for continuation prompt
            this.executeFeature(projectPathFromClient, featureId, true, false, undefined, {
              continuationPrompt,
            })
              .catch((error) => {
                console.error(`[AutoMode] Recovery execution failed for feature ${featureId}:`, error);
              });

            return { success: true };
          } else {
            // Rejected - update status and emit event
            await this.updateFeaturePlanSpec(projectPathFromClient, featureId, {
              status: 'rejected',
              reviewedByUser: true,
            });

            await this.updateFeatureStatus(projectPathFromClient, featureId, 'backlog');

            this.emitAutoModeEvent('plan_rejected', {
              featureId,
              projectPath: projectPathFromClient,
              feedback,
            });

            return { success: true };
          }
        }
      }

      console.log(`[AutoMode] ERROR: No pending approval found for feature ${featureId} and recovery not possible`);
      return { success: false, error: `No pending approval for feature ${featureId}` };
    }
    console.log(`[AutoMode] Found pending approval for feature ${featureId}, proceeding...`);

    const { projectPath } = pending;

    // Update feature's planSpec status
    await this.updateFeaturePlanSpec(projectPath, featureId, {
      status: approved ? 'approved' : 'rejected',
      approvedAt: approved ? new Date().toISOString() : undefined,
      reviewedByUser: true,
      content: editedPlan, // Update content if user provided an edited version
    });

    // If rejected with feedback, we can store it for the user to see
    if (!approved && feedback) {
      // Emit event so client knows the rejection reason
      this.emitAutoModeEvent('plan_rejected', {
        featureId,
        projectPath,
        feedback,
      });
    }

    // Resolve the promise with all data including feedback
    pending.resolve({ approved, editedPlan, feedback });
    this.pendingApprovals.delete(featureId);

    return { success: true };
  }

  /**
   * Cancel a pending plan approval (e.g., when feature is stopped).
   */
  cancelPlanApproval(featureId: string): void {
    console.log(`[AutoMode] cancelPlanApproval called for feature ${featureId}`);
    console.log(`[AutoMode] Current pending approvals: ${Array.from(this.pendingApprovals.keys()).join(', ') || 'none'}`);
    const pending = this.pendingApprovals.get(featureId);
    if (pending) {
      console.log(`[AutoMode] Found and cancelling pending approval for feature ${featureId}`);
      pending.reject(new Error('Plan approval cancelled - feature was stopped'));
      this.pendingApprovals.delete(featureId);
    } else {
      console.log(`[AutoMode] No pending approval to cancel for feature ${featureId}`);
    }
  }

  /**
   * Check if a feature has a pending plan approval.
   */
  hasPendingApproval(featureId: string): boolean {
    return this.pendingApprovals.has(featureId);
  }

  // Private helpers

  /**
   * Find an existing worktree for a given branch by checking git worktree list
   */
  private async findExistingWorktreeForBranch(
    projectPath: string,
    branchName: string
  ): Promise<string | null> {
    try {
      const { stdout } = await execAsync("git worktree list --porcelain", {
        cwd: projectPath,
      });

      const lines = stdout.split("\n");
      let currentPath: string | null = null;
      let currentBranch: string | null = null;

      for (const line of lines) {
        if (line.startsWith("worktree ")) {
          currentPath = line.slice(9);
        } else if (line.startsWith("branch ")) {
          currentBranch = line.slice(7).replace("refs/heads/", "");
        } else if (line === "" && currentPath && currentBranch) {
          // End of a worktree entry
          if (currentBranch === branchName) {
            // Resolve to absolute path - git may return relative paths
            // On Windows, this is critical for cwd to work correctly
            // On all platforms, absolute paths ensure consistent behavior
            const resolvedPath = path.isAbsolute(currentPath)
              ? path.resolve(currentPath)
              : path.resolve(projectPath, currentPath);
            return resolvedPath;
          }
          currentPath = null;
          currentBranch = null;
        }
      }

      // Check the last entry (if file doesn't end with newline)
      if (currentPath && currentBranch && currentBranch === branchName) {
        // Resolve to absolute path for cross-platform compatibility
        const resolvedPath = path.isAbsolute(currentPath)
          ? path.resolve(currentPath)
          : path.resolve(projectPath, currentPath);
        return resolvedPath;
      }

      return null;
    } catch {
      return null;
    }
  }

  private async setupWorktree(
    projectPath: string,
    featureId: string,
    branchName: string
  ): Promise<string> {
    // First, check if git already has a worktree for this branch (anywhere)
    const existingWorktree = await this.findExistingWorktreeForBranch(projectPath, branchName);
    if (existingWorktree) {
      // Path is already resolved to absolute in findExistingWorktreeForBranch
      console.log(`[AutoMode] Found existing worktree for branch "${branchName}" at: ${existingWorktree}`);
      return existingWorktree;
    }

    // Git worktrees stay in project directory
    const worktreesDir = path.join(projectPath, ".worktrees");
    const worktreePath = path.join(worktreesDir, featureId);

    await fs.mkdir(worktreesDir, { recursive: true });

    // Check if worktree directory already exists (might not be linked to branch)
    try {
      await fs.access(worktreePath);
      // Return absolute path for cross-platform compatibility
      return path.resolve(worktreePath);
    } catch {
      // Create new worktree
    }

    // Create branch if it doesn't exist
    try {
      await execAsync(`git branch ${branchName}`, { cwd: projectPath });
    } catch {
      // Branch may already exist
    }

    // Create worktree
    try {
      await execAsync(`git worktree add "${worktreePath}" ${branchName}`, {
        cwd: projectPath,
      });
      // Return absolute path for cross-platform compatibility
      return path.resolve(worktreePath);
    } catch (error) {
      // Worktree creation failed, fall back to direct execution
      console.error(`[AutoMode] Worktree creation failed:`, error);
      return path.resolve(projectPath);
    }
  }

  private async loadFeature(
    projectPath: string,
    featureId: string
  ): Promise<Feature | null> {
    // Features are stored in .automaker directory
    const featureDir = getFeatureDir(projectPath, featureId);
    const featurePath = path.join(featureDir, "feature.json");

    try {
      const data = await fs.readFile(featurePath, "utf-8");
      return JSON.parse(data);
    } catch {
      return null;
    }
  }

  private async updateFeatureStatus(
    projectPath: string,
    featureId: string,
    status: string
  ): Promise<void> {
    // Features are stored in .automaker directory
    const featureDir = getFeatureDir(projectPath, featureId);
    const featurePath = path.join(featureDir, "feature.json");

    try {
      const data = await fs.readFile(featurePath, "utf-8");
      const feature = JSON.parse(data);
      feature.status = status;
      feature.updatedAt = new Date().toISOString();
      // Set justFinishedAt timestamp when moving to waiting_approval (agent just completed)
      // Badge will show for 2 minutes after this timestamp
      if (status === "waiting_approval") {
        feature.justFinishedAt = new Date().toISOString();
      } else {
        // Clear the timestamp when moving to other statuses
        feature.justFinishedAt = undefined;
      }
      await fs.writeFile(featurePath, JSON.stringify(feature, null, 2));
    } catch {
      // Feature file may not exist
    }
  }

  /**
   * Update the planSpec of a feature
   */
  private async updateFeaturePlanSpec(
    projectPath: string,
    featureId: string,
    updates: Partial<PlanSpec>
  ): Promise<void> {
    const featurePath = path.join(
      projectPath,
      ".automaker",
      "features",
      featureId,
      "feature.json"
    );

    try {
      const data = await fs.readFile(featurePath, "utf-8");
      const feature = JSON.parse(data);

      // Initialize planSpec if it doesn't exist
      if (!feature.planSpec) {
        feature.planSpec = {
          status: 'pending',
          version: 1,
          reviewedByUser: false,
        };
      }

      // Apply updates
      Object.assign(feature.planSpec, updates);

      // If content is being updated and it's a new version, increment version
      if (updates.content && updates.content !== feature.planSpec.content) {
        feature.planSpec.version = (feature.planSpec.version || 0) + 1;
      }

      feature.updatedAt = new Date().toISOString();
      await fs.writeFile(featurePath, JSON.stringify(feature, null, 2));
    } catch (error) {
      console.error(`[AutoMode] Failed to update planSpec for ${featureId}:`, error);
    }
  }

  private async loadPendingFeatures(projectPath: string): Promise<Feature[]> {
    // Features are stored in .automaker directory
    const featuresDir = getFeaturesDir(projectPath);

    try {
      const entries = await fs.readdir(featuresDir, { withFileTypes: true });
      const allFeatures: Feature[] = [];
      const pendingFeatures: Feature[] = [];

      // Load all features (for dependency checking)
      for (const entry of entries) {
        if (entry.isDirectory()) {
          const featurePath = path.join(
            featuresDir,
            entry.name,
            "feature.json"
          );
          try {
            const data = await fs.readFile(featurePath, "utf-8");
            const feature = JSON.parse(data);
            allFeatures.push(feature);

            // Track pending features separately
            if (
              feature.status === "pending" ||
              feature.status === "ready" ||
              feature.status === "backlog"
            ) {
              pendingFeatures.push(feature);
            }
          } catch {
            // Skip invalid features
          }
        }
      }

      // Apply dependency-aware ordering
      const { orderedFeatures } = resolveDependencies(pendingFeatures);

      // Filter to only features with satisfied dependencies
      const readyFeatures = orderedFeatures.filter(feature =>
        areDependenciesSatisfied(feature, allFeatures)
      );

      return readyFeatures;
    } catch {
      return [];
    }
  }

  /**
   * Extract a title from feature description (first line or truncated)
   */
  private extractTitleFromDescription(description: string): string {
    if (!description || !description.trim()) {
      return "Untitled Feature";
    }

    // Get first line, or first 60 characters if no newline
    const firstLine = description.split("\n")[0].trim();
    if (firstLine.length <= 60) {
      return firstLine;
    }

    // Truncate to 60 characters and add ellipsis
    return firstLine.substring(0, 57) + "...";
  }

  /**
   * Get the planning prompt prefix based on feature's planning mode
   */
  private getPlanningPromptPrefix(feature: Feature): string {
    const mode = feature.planningMode || 'skip';

    if (mode === 'skip') {
      return ''; // No planning phase
    }

    // For lite mode, use the approval variant if requirePlanApproval is true
    let promptKey: string = mode;
    if (mode === 'lite' && feature.requirePlanApproval === true) {
      promptKey = 'lite_with_approval';
    }

    const planningPrompt = PLANNING_PROMPTS[promptKey as keyof typeof PLANNING_PROMPTS];
    if (!planningPrompt) {
      return '';
    }

    return planningPrompt + '\n\n---\n\n## Feature Request\n\n';
  }

  private buildFeaturePrompt(feature: Feature): string {
    const title = this.extractTitleFromDescription(feature.description);

    let prompt = `## Feature Implementation Task

**Feature ID:** ${feature.id}
**Title:** ${title}
**Description:** ${feature.description}
`;

    if (feature.spec) {
      prompt += `
**Specification:**
${feature.spec}
`;
    }

    // Add images note (like old implementation)
    if (feature.imagePaths && feature.imagePaths.length > 0) {
      const imagesList = feature.imagePaths
        .map((img, idx) => {
          const path = typeof img === "string" ? img : img.path;
          const filename =
            typeof img === "string"
              ? path.split("/").pop()
              : img.filename || path.split("/").pop();
          const mimeType =
            typeof img === "string" ? "image/*" : img.mimeType || "image/*";
          return `   ${
            idx + 1
          }. ${filename} (${mimeType})\n      Path: ${path}`;
        })
        .join("\n");

      prompt += `
**📎 Context Images Attached:**
The user has attached ${feature.imagePaths.length} image(s) for context. These images are provided both visually (in the initial message) and as files you can read:

${imagesList}

You can use the Read tool to view these images at any time during implementation. Review them carefully before implementing.
`;
    }

    prompt += `
## Instructions

Implement this feature by:
1. First, explore the codebase to understand the existing structure
2. Plan your implementation approach
3. Write the necessary code changes
4. Add or update tests as needed
5. Ensure the code follows existing patterns and conventions

When done, wrap your final summary in <summary> tags like this:

<summary>
## Summary: [Feature Title]

### Changes Implemented
- [List of changes made]

### Files Modified
- [List of files]

### Notes for Developer
- [Any important notes]
</summary>

This helps parse your summary correctly in the output logs.`;

    return prompt;
  }

  private async runAgent(
    workDir: string,
    featureId: string,
    prompt: string,
    abortController: AbortController,
    imagePaths?: string[],
    model?: string,
    options?: {
      projectPath?: string;
      planningMode?: PlanningMode;
      requirePlanApproval?: boolean;
      previousContent?: string;
    }
  ): Promise<void> {
    const projectPath = options?.projectPath || workDir;
    const planningMode = options?.planningMode || 'skip';
    const previousContent = options?.previousContent;

    // Check if this planning mode can generate a spec/plan that needs approval
    // - spec and full always generate specs
    // - lite only generates approval-ready content when requirePlanApproval is true
    const planningModeRequiresApproval =
      planningMode === 'spec' ||
      planningMode === 'full' ||
      (planningMode === 'lite' && options?.requirePlanApproval === true);
    const requiresApproval = planningModeRequiresApproval && options?.requirePlanApproval === true;

    // CI/CD Mock Mode: Return early with mock response when AUTOMAKER_MOCK_AGENT is set
    // This prevents actual API calls during automated testing
    if (process.env.AUTOMAKER_MOCK_AGENT === "true") {
      console.log(`[AutoMode] MOCK MODE: Skipping real agent execution for feature ${featureId}`);

      // Simulate some work being done
      await this.sleep(500);

      // Emit mock progress events to simulate agent activity
      this.emitAutoModeEvent("auto_mode_progress", {
        featureId,
        content: "Mock agent: Analyzing the codebase...",
      });

      await this.sleep(300);

      this.emitAutoModeEvent("auto_mode_progress", {
        featureId,
        content: "Mock agent: Implementing the feature...",
      });

      await this.sleep(300);

      // Create a mock file with "yellow" content as requested in the test
      const mockFilePath = path.join(workDir, "yellow.txt");
      await fs.writeFile(mockFilePath, "yellow");

      this.emitAutoModeEvent("auto_mode_progress", {
        featureId,
        content: "Mock agent: Created yellow.txt file with content 'yellow'",
      });

      await this.sleep(200);

      // Save mock agent output
      const configProjectPath = this.config?.projectPath || workDir;
      const featureDirForOutput = getFeatureDir(configProjectPath, featureId);
      const outputPath = path.join(featureDirForOutput, "agent-output.md");

      const mockOutput = `# Mock Agent Output

## Summary
This is a mock agent response for CI/CD testing.

## Changes Made
- Created \`yellow.txt\` with content "yellow"

## Notes
This mock response was generated because AUTOMAKER_MOCK_AGENT=true was set.
`;

      await fs.mkdir(path.dirname(outputPath), { recursive: true });
      await fs.writeFile(outputPath, mockOutput);

      console.log(`[AutoMode] MOCK MODE: Completed mock execution for feature ${featureId}`);
      return;
    }

    // Build SDK options using centralized configuration for feature implementation
    const sdkOptions = createAutoModeOptions({
      cwd: workDir,
      model: model,
      abortController,
    });

    // Extract model, maxTurns, and allowedTools from SDK options
    const finalModel = sdkOptions.model!;
    const maxTurns = sdkOptions.maxTurns;
    const allowedTools = sdkOptions.allowedTools as string[] | undefined;

    console.log(
      `[AutoMode] runAgent called for feature ${featureId} with model: ${finalModel}, planningMode: ${planningMode}, requiresApproval: ${requiresApproval}`
    );

    // Get provider for this model
    const provider = ProviderFactory.getProviderForModel(finalModel);

    console.log(
      `[AutoMode] Using provider "${provider.getName()}" for model "${finalModel}"`
    );

    // Build prompt content with images using utility
    const { content: promptContent } = await buildPromptWithImages(
      prompt,
      imagePaths,
      workDir,
      false // don't duplicate paths in text
    );

    const executeOptions: ExecuteOptions = {
      prompt: promptContent,
      model: finalModel,
      maxTurns: maxTurns,
      cwd: workDir,
      allowedTools: allowedTools,
      abortController,
    };

    // Execute via provider
    const stream = provider.executeQuery(executeOptions);
    // Initialize with previous content if this is a follow-up, with a separator
    let responseText = previousContent
      ? `${previousContent}\n\n---\n\n## Follow-up Session\n\n`
      : "";
    let specDetected = false;

    // Agent output goes to .automaker directory
    // Note: We use the original projectPath here (from config), not workDir
    // because workDir might be a worktree path
    const configProjectPath = this.config?.projectPath || workDir;
    const featureDirForOutput = getFeatureDir(configProjectPath, featureId);
    const outputPath = path.join(featureDirForOutput, "agent-output.md");

    // Incremental file writing state
    let writeTimeout: ReturnType<typeof setTimeout> | null = null;
    const WRITE_DEBOUNCE_MS = 500; // Batch writes every 500ms

    // Helper to write current responseText to file
    const writeToFile = async (): Promise<void> => {
      try {
        await fs.mkdir(path.dirname(outputPath), { recursive: true });
        await fs.writeFile(outputPath, responseText);
      } catch (error) {
        // Log but don't crash - file write errors shouldn't stop execution
        console.error(`[AutoMode] Failed to write agent output for ${featureId}:`, error);
      }
    };

    // Debounced write - schedules a write after WRITE_DEBOUNCE_MS
    const scheduleWrite = (): void => {
      if (writeTimeout) {
        clearTimeout(writeTimeout);
      }
      writeTimeout = setTimeout(() => {
        writeToFile();
      }, WRITE_DEBOUNCE_MS);
    };

    streamLoop: for await (const msg of stream) {
      if (msg.type === "assistant" && msg.message?.content) {
        for (const block of msg.message.content) {
          if (block.type === "text") {
            // Add separator before new text if we already have content and it doesn't end with newlines
            if (responseText.length > 0 && !responseText.endsWith('\n\n')) {
              if (responseText.endsWith('\n')) {
                responseText += '\n';
              } else {
                responseText += '\n\n';
              }
            }
            responseText += block.text || "";

            // Check for authentication errors in the response
            if (
              block.text &&
              (block.text.includes("Invalid API key") ||
                block.text.includes("authentication_failed") ||
                block.text.includes("Fix external API key"))
            ) {
              throw new Error(
                "Authentication failed: Invalid or expired API key. " +
                  "Please check your ANTHROPIC_API_KEY or GOOGLE_API_KEY, or run 'claude login' to re-authenticate."
              );
            }

            // Schedule incremental file write (debounced)
            scheduleWrite();

            // Check for [SPEC_GENERATED] marker in planning modes (spec or full)
            if (planningModeRequiresApproval && !specDetected && responseText.includes('[SPEC_GENERATED]')) {
              specDetected = true;

              // Extract plan content (everything before the marker)
              const markerIndex = responseText.indexOf('[SPEC_GENERATED]');
              const planContent = responseText.substring(0, markerIndex).trim();

              // Parse tasks from the generated spec (for spec and full modes)
              // Use let since we may need to update this after plan revision
              let parsedTasks = parseTasksFromSpec(planContent);
              const tasksTotal = parsedTasks.length;

              console.log(`[AutoMode] Parsed ${tasksTotal} tasks from spec for feature ${featureId}`);
              if (parsedTasks.length > 0) {
                console.log(`[AutoMode] Tasks: ${parsedTasks.map(t => t.id).join(', ')}`);
              }

              // Update planSpec status to 'generated' and save content with parsed tasks
              await this.updateFeaturePlanSpec(projectPath, featureId, {
                status: 'generated',
                content: planContent,
                version: 1,
                generatedAt: new Date().toISOString(),
                reviewedByUser: false,
                tasks: parsedTasks,
                tasksTotal,
                tasksCompleted: 0,
              });

              let approvedPlanContent = planContent;
              let userFeedback: string | undefined;
              let currentPlanContent = planContent;
              let planVersion = 1;

              // Only pause for approval if requirePlanApproval is true
              if (requiresApproval) {
                // ========================================
                // PLAN REVISION LOOP
                // Keep regenerating plan until user approves
                // ========================================
                let planApproved = false;

                while (!planApproved) {
                  console.log(`[AutoMode] Spec v${planVersion} generated for feature ${featureId}, waiting for approval`);

                  // CRITICAL: Register pending approval BEFORE emitting event
                  const approvalPromise = this.waitForPlanApproval(featureId, projectPath);

                  // Emit plan_approval_required event
                  this.emitAutoModeEvent('plan_approval_required', {
                    featureId,
                    projectPath,
                    planContent: currentPlanContent,
                    planningMode,
                    planVersion,
                  });

                  // Wait for user response
                  try {
                    const approvalResult = await approvalPromise;

                    if (approvalResult.approved) {
                      // User approved the plan
                      console.log(`[AutoMode] Plan v${planVersion} approved for feature ${featureId}`);
                      planApproved = true;

                      // If user provided edits, use the edited version
                      if (approvalResult.editedPlan) {
                        approvedPlanContent = approvalResult.editedPlan;
                        await this.updateFeaturePlanSpec(projectPath, featureId, {
                          content: approvalResult.editedPlan,
                        });
                      } else {
                        approvedPlanContent = currentPlanContent;
                      }

                      // Capture any additional feedback for implementation
                      userFeedback = approvalResult.feedback;

                      // Emit approval event
                      this.emitAutoModeEvent('plan_approved', {
                        featureId,
                        projectPath,
                        hasEdits: !!approvalResult.editedPlan,
                        planVersion,
                      });

                    } else {
                      // User rejected - check if they provided feedback for revision
                      const hasFeedback = approvalResult.feedback && approvalResult.feedback.trim().length > 0;
                      const hasEdits = approvalResult.editedPlan && approvalResult.editedPlan.trim().length > 0;

                      if (!hasFeedback && !hasEdits) {
                        // No feedback or edits = explicit cancel
                        console.log(`[AutoMode] Plan rejected without feedback for feature ${featureId}, cancelling`);
                        throw new Error('Plan cancelled by user');
                      }

                      // User wants revisions - regenerate the plan
                      console.log(`[AutoMode] Plan v${planVersion} rejected with feedback for feature ${featureId}, regenerating...`);
                      planVersion++;

                      // Emit revision event
                      this.emitAutoModeEvent('plan_revision_requested', {
                        featureId,
                        projectPath,
                        feedback: approvalResult.feedback,
                        hasEdits: !!hasEdits,
                        planVersion,
                      });

                      // Build revision prompt
                      let revisionPrompt = `The user has requested revisions to the plan/specification.

## Previous Plan (v${planVersion - 1})
${hasEdits ? approvalResult.editedPlan : currentPlanContent}

## User Feedback
${approvalResult.feedback || 'Please revise the plan based on the edits above.'}

## Instructions
Please regenerate the specification incorporating the user's feedback.
Keep the same format with the \`\`\`tasks block for task definitions.
After generating the revised spec, output:
"[SPEC_GENERATED] Please review the revised specification above."
`;

                      // Update status to regenerating
                      await this.updateFeaturePlanSpec(projectPath, featureId, {
                        status: 'generating',
                        version: planVersion,
                      });

                      // Make revision call
                      const revisionStream = provider.executeQuery({
                        prompt: revisionPrompt,
                        model: finalModel,
                        maxTurns: maxTurns || 100,
                        cwd: workDir,
                        allowedTools: allowedTools,
                        abortController,
                      });

                      let revisionText = "";
                      for await (const msg of revisionStream) {
                        if (msg.type === "assistant" && msg.message?.content) {
                          for (const block of msg.message.content) {
                            if (block.type === "text") {
                              revisionText += block.text || "";
                              this.emitAutoModeEvent("auto_mode_progress", {
                                featureId,
                                content: block.text,
                              });
                            }
                          }
                        } else if (msg.type === "error") {
                          throw new Error(msg.error || "Error during plan revision");
                        } else if (msg.type === "result" && msg.subtype === "success") {
                          revisionText += msg.result || "";
                        }
                      }

                      // Extract new plan content
                      const markerIndex = revisionText.indexOf('[SPEC_GENERATED]');
                      if (markerIndex > 0) {
                        currentPlanContent = revisionText.substring(0, markerIndex).trim();
                      } else {
                        currentPlanContent = revisionText.trim();
                      }

                      // Re-parse tasks from revised plan
                      const revisedTasks = parseTasksFromSpec(currentPlanContent);
                      console.log(`[AutoMode] Revised plan has ${revisedTasks.length} tasks`);

                      // Update planSpec with revised content
                      await this.updateFeaturePlanSpec(projectPath, featureId, {
                        status: 'generated',
                        content: currentPlanContent,
                        version: planVersion,
                        tasks: revisedTasks,
                        tasksTotal: revisedTasks.length,
                        tasksCompleted: 0,
                      });

                      // Update parsedTasks for implementation
                      parsedTasks = revisedTasks;

                      responseText += revisionText;
                    }

                  } catch (error) {
                    if ((error as Error).message.includes('cancelled')) {
                      throw error;
                    }
                    throw new Error(`Plan approval failed: ${(error as Error).message}`);
                  }
                }

              } else {
                // Auto-approve: requirePlanApproval is false, just continue without pausing
                console.log(`[AutoMode] Spec generated for feature ${featureId}, auto-approving (requirePlanApproval=false)`);

                // Emit info event for frontend
                this.emitAutoModeEvent('plan_auto_approved', {
                  featureId,
                  projectPath,
                  planContent,
                  planningMode,
                });

                approvedPlanContent = planContent;
              }

              // CRITICAL: After approval, we need to make a second call to continue implementation
              // The agent is waiting for "approved" - we need to send it and continue
              console.log(`[AutoMode] Making continuation call after plan approval for feature ${featureId}`);

              // Update planSpec status to approved (handles both manual and auto-approval paths)
              await this.updateFeaturePlanSpec(projectPath, featureId, {
                status: 'approved',
                approvedAt: new Date().toISOString(),
                reviewedByUser: requiresApproval,
              });

              // ========================================
              // MULTI-AGENT TASK EXECUTION
              // Each task gets its own focused agent call
              // ========================================

              if (parsedTasks.length > 0) {
                console.log(`[AutoMode] Starting multi-agent execution: ${parsedTasks.length} tasks for feature ${featureId}`);

                // Execute each task with a separate agent
                for (let taskIndex = 0; taskIndex < parsedTasks.length; taskIndex++) {
                  const task = parsedTasks[taskIndex];

                  // Check for abort
                  if (abortController.signal.aborted) {
                    throw new Error('Feature execution aborted');
                  }

                  // Emit task started
                  console.log(`[AutoMode] Starting task ${task.id}: ${task.description}`);
                  this.emitAutoModeEvent("auto_mode_task_started", {
                    featureId,
                    projectPath,
                    taskId: task.id,
                    taskDescription: task.description,
                    taskIndex,
                    tasksTotal: parsedTasks.length,
                  });

                  // Update planSpec with current task
                  await this.updateFeaturePlanSpec(projectPath, featureId, {
                    currentTaskId: task.id,
                  });

                  // Build focused prompt for this specific task
                  const taskPrompt = this.buildTaskPrompt(task, parsedTasks, taskIndex, approvedPlanContent, userFeedback);

                  // Execute task with dedicated agent
                  const taskStream = provider.executeQuery({
                    prompt: taskPrompt,
                    model: finalModel,
                    maxTurns: Math.min(maxTurns || 100, 50), // Limit turns per task
                    cwd: workDir,
                    allowedTools: allowedTools,
                    abortController,
                  });

                  let taskOutput = "";

                  // Process task stream
                  for await (const msg of taskStream) {
                    if (msg.type === "assistant" && msg.message?.content) {
                      for (const block of msg.message.content) {
                        if (block.type === "text") {
                          taskOutput += block.text || "";
                          responseText += block.text || "";
                          this.emitAutoModeEvent("auto_mode_progress", {
                            featureId,
                            content: block.text,
                          });
                        } else if (block.type === "tool_use") {
                          this.emitAutoModeEvent("auto_mode_tool", {
                            featureId,
                            tool: block.name,
                            input: block.input,
                          });
                        }
                      }
                    } else if (msg.type === "error") {
                      throw new Error(msg.error || `Error during task ${task.id}`);
                    } else if (msg.type === "result" && msg.subtype === "success") {
                      taskOutput += msg.result || "";
                      responseText += msg.result || "";
                    }
                  }

                  // Emit task completed
                  console.log(`[AutoMode] Task ${task.id} completed for feature ${featureId}`);
                  this.emitAutoModeEvent("auto_mode_task_complete", {
                    featureId,
                    projectPath,
                    taskId: task.id,
                    tasksCompleted: taskIndex + 1,
                    tasksTotal: parsedTasks.length,
                  });

                  // Update planSpec with progress
                  await this.updateFeaturePlanSpec(projectPath, featureId, {
                    tasksCompleted: taskIndex + 1,
                  });

                  // Check for phase completion (group tasks by phase)
                  if (task.phase) {
                    const nextTask = parsedTasks[taskIndex + 1];
                    if (!nextTask || nextTask.phase !== task.phase) {
                      // Phase changed, emit phase complete
                      const phaseMatch = task.phase.match(/Phase\s*(\d+)/i);
                      if (phaseMatch) {
                        this.emitAutoModeEvent("auto_mode_phase_complete", {
                          featureId,
                          projectPath,
                          phaseNumber: parseInt(phaseMatch[1], 10),
                        });
                      }
                    }
                  }
                }

                console.log(`[AutoMode] All ${parsedTasks.length} tasks completed for feature ${featureId}`);
              } else {
                // No parsed tasks - fall back to single-agent execution
                console.log(`[AutoMode] No parsed tasks, using single-agent execution for feature ${featureId}`);

                const continuationPrompt = `The plan/specification has been approved. Now implement it.
${userFeedback ? `\n## User Feedback\n${userFeedback}\n` : ''}
## Approved Plan

${approvedPlanContent}

## Instructions

Implement all the changes described in the plan above.`;

                const continuationStream = provider.executeQuery({
                  prompt: continuationPrompt,
                  model: finalModel,
                  maxTurns: maxTurns,
                  cwd: workDir,
                  allowedTools: allowedTools,
                  abortController,
                });

                for await (const msg of continuationStream) {
                  if (msg.type === "assistant" && msg.message?.content) {
                    for (const block of msg.message.content) {
                      if (block.type === "text") {
                        responseText += block.text || "";
                        this.emitAutoModeEvent("auto_mode_progress", {
                          featureId,
                          content: block.text,
                        });
                      } else if (block.type === "tool_use") {
                        this.emitAutoModeEvent("auto_mode_tool", {
                          featureId,
                          tool: block.name,
                          input: block.input,
                        });
                      }
                    }
                  } else if (msg.type === "error") {
                    throw new Error(msg.error || "Unknown error during implementation");
                  } else if (msg.type === "result" && msg.subtype === "success") {
                    responseText += msg.result || "";
                  }
                }
              }

              console.log(`[AutoMode] Implementation completed for feature ${featureId}`);
              // Exit the original stream loop since continuation is done
              break streamLoop;
            }

            // Only emit progress for non-marker text (marker was already handled above)
            if (!specDetected) {
              this.emitAutoModeEvent("auto_mode_progress", {
                featureId,
                content: block.text,
              });
            }
          } else if (block.type === "tool_use") {
            // Emit event for real-time UI
            this.emitAutoModeEvent("auto_mode_tool", {
              featureId,
              tool: block.name,
              input: block.input,
            });

            // Also add to file output for persistence
            if (responseText.length > 0 && !responseText.endsWith('\n')) {
              responseText += '\n';
            }
            responseText += `\n🔧 Tool: ${block.name}\n`;
            if (block.input) {
              responseText += `Input: ${JSON.stringify(block.input, null, 2)}\n`;
            }
            scheduleWrite();
          }
        }
      } else if (msg.type === "error") {
        // Handle error messages
        throw new Error(msg.error || "Unknown error");
      } else if (msg.type === "result" && msg.subtype === "success") {
        // Don't replace responseText - the accumulated content is the full history
        // The msg.result is just a summary which would lose all tool use details
        // Just ensure final write happens
        scheduleWrite();
      }
    }

    // Clear any pending timeout and do a final write to ensure all content is saved
    if (writeTimeout) {
      clearTimeout(writeTimeout);
    }
    // Final write - ensure all accumulated content is saved
    await writeToFile();
  }

  private async executeFeatureWithContext(
    projectPath: string,
    featureId: string,
    context: string,
    useWorktrees: boolean
  ): Promise<void> {
    const feature = await this.loadFeature(projectPath, featureId);
    if (!feature) {
      throw new Error(`Feature ${featureId} not found`);
    }

    const prompt = `## Continuing Feature Implementation

${this.buildFeaturePrompt(feature)}

## Previous Context
The following is the output from a previous implementation attempt. Continue from where you left off:

${context}

## Instructions
Review the previous work and continue the implementation. If the feature appears complete, verify it works correctly.`;

    return this.executeFeature(projectPath, featureId, useWorktrees, false, undefined, {
      continuationPrompt: prompt,
    });
  }

  /**
   * Build a focused prompt for executing a single task.
   * Each task gets minimal context to keep the agent focused.
   */
  private buildTaskPrompt(
    task: ParsedTask,
    allTasks: ParsedTask[],
    taskIndex: number,
    planContent: string,
    userFeedback?: string
  ): string {
    const completedTasks = allTasks.slice(0, taskIndex);
    const remainingTasks = allTasks.slice(taskIndex + 1);

    let prompt = `# Task Execution: ${task.id}

You are executing a specific task as part of a larger feature implementation.

## Your Current Task

**Task ID:** ${task.id}
**Description:** ${task.description}
${task.filePath ? `**Primary File:** ${task.filePath}` : ''}
${task.phase ? `**Phase:** ${task.phase}` : ''}

## Context

`;

    // Show what's already done
    if (completedTasks.length > 0) {
      prompt += `### Already Completed (${completedTasks.length} tasks)
${completedTasks.map(t => `- [x] ${t.id}: ${t.description}`).join('\n')}

`;
    }

    // Show remaining tasks
    if (remainingTasks.length > 0) {
      prompt += `### Coming Up Next (${remainingTasks.length} tasks remaining)
${remainingTasks.slice(0, 3).map(t => `- [ ] ${t.id}: ${t.description}`).join('\n')}
${remainingTasks.length > 3 ? `... and ${remainingTasks.length - 3} more tasks` : ''}

`;
    }

    // Add user feedback if any
    if (userFeedback) {
      prompt += `### User Feedback
${userFeedback}

`;
    }

    // Add relevant excerpt from plan (just the task-related part to save context)
    prompt += `### Reference: Full Plan
<details>
${planContent}
</details>

## Instructions

1. Focus ONLY on completing task ${task.id}: "${task.description}"
2. Do not work on other tasks
3. Use the existing codebase patterns
4. When done, summarize what you implemented

Begin implementing task ${task.id} now.`;

    return prompt;
  }

  /**
   * Emit an auto-mode event wrapped in the correct format for the client.
   * All auto-mode events are sent as type "auto-mode:event" with the actual
   * event type and data in the payload.
   */
  private emitAutoModeEvent(
    eventType: string,
    data: Record<string, unknown>
  ): void {
    // Wrap the event in auto-mode:event format expected by the client
    this.events.emit("auto-mode:event", {
      type: eventType,
      ...data,
    });
  }

  private sleep(ms: number): Promise<void> {
    return new Promise((resolve) => setTimeout(resolve, ms));
  }
}