automaker/apps/ui/tests/utils/project/setup.ts

import { Page } from '@playwright/test';
import { assertSafeProjectPath } from '../core/safe-paths';

/**
 * Store version constants - centralized to avoid hardcoding across tests
 * These MUST match the versions used in the actual stores
 */
const STORE_VERSIONS = {
  APP_STORE: 2, // Must match app-store.ts persist version
  SETUP_STORE: 1, // Must match setup-store.ts persist version
} as const;

/**
 * Project interface for test setup
 */
export interface TestProject {
  id: string;
  name: string;
  path: string;
  lastOpened?: string;
}

/**
 * Options for setting up the welcome view
 */
export interface WelcomeViewSetupOptions {
  /** Directory path to pre-configure as the workspace directory */
  workspaceDir?: string;
  /** Recent projects to show (but not as current project) */
  recentProjects?: TestProject[];
}

/**
 * Set up localStorage to show the welcome view with no current project
 * This is the cleanest way to test project creation flows
 *
 * @param page - Playwright page
 * @param options - Configuration options
 */
export async function setupWelcomeView(
  page: Page,
  options?: WelcomeViewSetupOptions
): Promise<void> {
  await page.addInitScript(
    ({
      opts,
      versions,
    }: {
      opts: WelcomeViewSetupOptions | undefined;
      versions: typeof STORE_VERSIONS;
    }) => {
      // Set up empty app state (no current project) - shows welcome view
      const appState = {
        state: {
          projects: opts?.recentProjects || [],
          currentProject: null,
          currentView: 'welcome',
          theme: 'dark',
          sidebarOpen: true,
          skipSandboxWarning: true,
          apiKeys: { anthropic: '', google: '' },
          chatSessions: [],
          chatHistoryOpen: false,
          maxConcurrency: 3,
        },
        version: versions.APP_STORE,
      };
      localStorage.setItem('automaker-storage', JSON.stringify(appState));

      // Mark setup as complete to skip the setup wizard
      const setupState = {
        state: {
          isFirstRun: false,
          setupComplete: true,
          skipClaudeSetup: false,
        },
        version: versions.SETUP_STORE,
      };
      localStorage.setItem('automaker-setup', JSON.stringify(setupState));

      // Set settings cache to ensure setupComplete is recognized on cold start.
      // This prevents the server's setupComplete value (which may be false on fresh CI)
      // from overriding the setup store and causing a redirect to /setup.
      const settingsCache: Record<string, unknown> = {
        setupComplete: true,
        isFirstRun: false,
        projects: opts?.recentProjects || [],
        // Explicitly set currentProjectId to null so the fast-hydrate path
        // does not restore a stale project from a previous test.
        currentProjectId: null,
        theme: 'dark',
        sidebarOpen: true,
        maxConcurrency: 3,
      };

      // Include lastProjectDir in settings cache so it's available during fast-hydrate.
      // The standalone localStorage key is a legacy fallback; the cache is the primary source.
      if (opts?.workspaceDir) {
        settingsCache.lastProjectDir = opts.workspaceDir;
      }

      localStorage.setItem('automaker-settings-cache', JSON.stringify(settingsCache));

      // Set workspace directory if provided (legacy fallback key)
      if (opts?.workspaceDir) {
        localStorage.setItem('automaker:lastProjectDir', opts.workspaceDir);
      }

      // Disable splash screen in tests
      localStorage.setItem('automaker-disable-splash', 'true');

      // Set up a mechanism to keep currentProject null even after settings hydration.
      // Settings API might restore a project, so we watch for changes and override.
      sessionStorage.setItem('automaker-test-welcome-view', 'true');

      // Use a MutationObserver + storage event to detect when hydration sets a project,
      // then immediately override it back to null. This is more reliable than a fixed timeout.
      const enforceWelcomeView = () => {
        const storage = localStorage.getItem('automaker-storage');
        if (storage) {
          try {
            const state = JSON.parse(storage);
            if (
              state.state &&
              sessionStorage.getItem('automaker-test-welcome-view') === 'true' &&
              state.state.currentProject !== null
            ) {
              state.state.currentProject = null;
              state.state.currentView = 'welcome';
              localStorage.setItem('automaker-storage', JSON.stringify(state));
            }
          } catch {
            // Ignore parse errors
          }
        }
      };

      // Listen for storage changes (catches hydration from settings API)
      window.addEventListener('storage', enforceWelcomeView);

      // Also poll briefly to catch synchronous hydration that doesn't fire storage events
      const pollInterval = setInterval(enforceWelcomeView, 200);
      setTimeout(() => {
        clearInterval(pollInterval);
        window.removeEventListener('storage', enforceWelcomeView);
      }, 5000); // Stop after 5s - hydration should be done by then
    },
    { opts: options, versions: STORE_VERSIONS }
  );
}

/**
 * Set up localStorage with a project at a real filesystem path
 * Use this when testing with actual files on disk.
 * Project path must be under test/ or temp to avoid affecting the main project's git.
 *
 * @param page - Playwright page
 * @param projectPath - Absolute path to the project directory
 * @param projectName - Display name for the project
 * @param options - Additional options
 */
export async function setupRealProject(
  page: Page,
  projectPath: string,
  projectName: string,
  options?: {
    /** Set as current project (opens board view) or just add to recent projects */
    setAsCurrent?: boolean;
    /** Additional recent projects to include */
    additionalProjects?: TestProject[];
    /** Optional project ID to use (if not provided, generates timestamp-based ID) */
    projectId?: string;
  }
): Promise<void> {
  assertSafeProjectPath(projectPath);
  await page.addInitScript(
    ({
      path,
      name,
      opts,
      versions,
    }: {
      path: string;
      name: string;
      opts: typeof options;
      versions: typeof STORE_VERSIONS;
    }) => {
      const projectId = opts?.projectId || `project-${Date.now()}`;
      const project: TestProject = {
        id: projectId,
        name: name,
        path: path,
        lastOpened: new Date().toISOString(),
      };

      const allProjects = [project, ...(opts?.additionalProjects || [])];
      const currentProject = opts?.setAsCurrent !== false ? project : null;

      const appState = {
        state: {
          projects: allProjects,
          currentProject: currentProject,
          currentView: currentProject ? 'board' : 'welcome',
          theme: 'dark',
          sidebarOpen: true,
          skipSandboxWarning: true,
          apiKeys: { anthropic: '', google: '' },
          chatSessions: [],
          chatHistoryOpen: false,
          maxConcurrency: 3,
        },
        version: versions.APP_STORE,
      };
      localStorage.setItem('automaker-storage', JSON.stringify(appState));

      // Mark setup as complete
      const setupState = {
        state: {
          isFirstRun: false,
          setupComplete: true,
          skipClaudeSetup: false,
        },
        version: versions.SETUP_STORE,
      };
      localStorage.setItem('automaker-setup', JSON.stringify(setupState));

      // Set settings cache to ensure setupComplete is recognized on cold start.
      // This prevents the server's setupComplete value (which may be false on fresh CI)
      // from overriding the setup store and causing a redirect to /setup.
      const settingsCache = {
        setupComplete: true,
        isFirstRun: false,
        projects: allProjects.map((p) => ({
          id: p.id,
          name: p.name,
          path: p.path,
          lastOpened: p.lastOpened,
        })),
        // Include currentProjectId so hydrateStoreFromSettings can restore
        // the current project directly (without relying on auto-open logic)
        currentProjectId: currentProject ? currentProject.id : null,
        theme: 'dark',
        sidebarOpen: true,
        maxConcurrency: 3,
      };
      localStorage.setItem('automaker-settings-cache', JSON.stringify(settingsCache));

      // Disable splash screen in tests
      localStorage.setItem('automaker-disable-splash', 'true');
    },
    { path: projectPath, name: projectName, opts: options, versions: STORE_VERSIONS }
  );
}

/**
 * Set up a mock project in localStorage to bypass the welcome screen
 * This simulates having opened a project before
 */
export async function setupMockProject(page: Page): Promise<void> {
  await page.addInitScript((versions: typeof STORE_VERSIONS) => {
    const mockProject = {
      id: 'test-project-1',
      name: 'Test Project',
      path: '/mock/test-project',
      lastOpened: new Date().toISOString(),
    };

    const mockState = {
      state: {
        projects: [mockProject],
        currentProject: mockProject,
        theme: 'dark',
        sidebarOpen: true,
        apiKeys: { anthropic: '', google: '' },
        chatSessions: [],
        chatHistoryOpen: false,
        maxConcurrency: 3,
      },
      version: versions.APP_STORE,
    };

    localStorage.setItem('automaker-storage', JSON.stringify(mockState));

    // Mark setup as complete to prevent redirect to /setup
    const setupState = {
      state: {
        isFirstRun: false,
        setupComplete: true,
        skipClaudeSetup: false,
      },
      version: versions.SETUP_STORE,
    };
    localStorage.setItem('automaker-setup', JSON.stringify(setupState));

    // Set settings cache so the fast hydrate path is taken on page load.
    const settingsCache = {
      setupComplete: true,
      isFirstRun: false,
      projects: [
        {
          id: mockProject.id,
          name: mockProject.name,
          path: mockProject.path,
          lastOpened: mockProject.lastOpened,
        },
      ],
      theme: 'dark',
      sidebarOpen: true,
      maxConcurrency: 3,
    };
    localStorage.setItem('automaker-settings-cache', JSON.stringify(settingsCache));

    // Disable splash screen in tests
    localStorage.setItem('automaker-disable-splash', 'true');
  }, STORE_VERSIONS);
}

/**
 * Set up a mock project with custom concurrency value
 */
export async function setupMockProjectWithConcurrency(
  page: Page,
  concurrency: number
): Promise<void> {
  await page.addInitScript(
    ({ maxConcurrency, versions }: { maxConcurrency: number; versions: typeof STORE_VERSIONS }) => {
      const mockProject = {
        id: 'test-project-1',
        name: 'Test Project',
        path: '/mock/test-project',
        lastOpened: new Date().toISOString(),
      };

      const mockState = {
        state: {
          projects: [mockProject],
          currentProject: mockProject,
          theme: 'dark',
          sidebarOpen: true,
          apiKeys: { anthropic: '', google: '' },
          chatSessions: [],
          chatHistoryOpen: false,
          maxConcurrency: maxConcurrency,
        },
        version: versions.APP_STORE,
      };

      localStorage.setItem('automaker-storage', JSON.stringify(mockState));

      // Mark setup as complete to prevent redirect to /setup
      const setupState = {
        state: { isFirstRun: false, setupComplete: true, skipClaudeSetup: false },
        version: versions.SETUP_STORE,
      };
      localStorage.setItem('automaker-setup', JSON.stringify(setupState));

      const settingsCache = {
        setupComplete: true,
        isFirstRun: false,
        projects: [
          {
            id: mockProject.id,
            name: mockProject.name,
            path: mockProject.path,
            lastOpened: mockProject.lastOpened,
          },
        ],
        theme: 'dark',
        sidebarOpen: true,
        maxConcurrency: maxConcurrency,
      };
      localStorage.setItem('automaker-settings-cache', JSON.stringify(settingsCache));
    },
    { maxConcurrency: concurrency, versions: STORE_VERSIONS }
  );
}

/**
 * Set up a mock project with specific running tasks to simulate concurrency limit
 */
export async function setupMockProjectAtConcurrencyLimit(
  page: Page,
  maxConcurrency: number = 1,
  runningTasks: string[] = ['running-task-1']
): Promise<void> {
  await page.addInitScript(
    ({
      maxConcurrency,
      runningTasks,
      versions,
    }: {
      maxConcurrency: number;
      runningTasks: string[];
      versions: typeof STORE_VERSIONS;
    }) => {
      const mockProject = {
        id: 'test-project-1',
        name: 'Test Project',
        path: '/mock/test-project',
        lastOpened: new Date().toISOString(),
      };

      const mockState = {
        state: {
          projects: [mockProject],
          currentProject: mockProject,
          theme: 'dark',
          sidebarOpen: true,
          apiKeys: { anthropic: '', google: '' },
          chatSessions: [],
          chatHistoryOpen: false,
          maxConcurrency: maxConcurrency,
          isAutoModeRunning: false,
          runningAutoTasks: runningTasks,
          autoModeActivityLog: [],
        },
        version: versions.APP_STORE,
      };

      localStorage.setItem('automaker-storage', JSON.stringify(mockState));

      const setupState = {
        state: { isFirstRun: false, setupComplete: true, skipClaudeSetup: false },
        version: versions.SETUP_STORE,
      };
      localStorage.setItem('automaker-setup', JSON.stringify(setupState));

      const settingsCache = {
        setupComplete: true,
        isFirstRun: false,
        projects: [
          {
            id: mockProject.id,
            name: mockProject.name,
            path: mockProject.path,
            lastOpened: mockProject.lastOpened,
          },
        ],
        theme: 'dark',
        sidebarOpen: true,
        maxConcurrency: maxConcurrency,
      };
      localStorage.setItem('automaker-settings-cache', JSON.stringify(settingsCache));

      // Disable splash screen in tests
      localStorage.setItem('automaker-disable-splash', 'true');
    },
    { maxConcurrency, runningTasks, versions: STORE_VERSIONS }
  );
}

/**
 * Set up a mock project with features in different states
 */
export async function setupMockProjectWithFeatures(
  page: Page,
  options?: {
    maxConcurrency?: number;
    runningTasks?: string[];
    features?: Array<{
      id: string;
      category: string;
      description: string;
      status: 'backlog' | 'in_progress' | 'verified';
      steps?: string[];
    }>;
  }
): Promise<void> {
  await page.addInitScript(
    ({ opts, versions }: { opts: typeof options; versions: typeof STORE_VERSIONS }) => {
      const mockProject = {
        id: 'test-project-1',
        name: 'Test Project',
        path: '/mock/test-project',
        lastOpened: new Date().toISOString(),
      };

      const mockFeatures = opts?.features || [];

      const mockState = {
        state: {
          projects: [mockProject],
          currentProject: mockProject,
          theme: 'dark',
          sidebarOpen: true,
          apiKeys: { anthropic: '', google: '' },
          chatSessions: [],
          chatHistoryOpen: false,
          maxConcurrency: opts?.maxConcurrency ?? 3,
          isAutoModeRunning: false,
          runningAutoTasks: opts?.runningTasks ?? [],
          autoModeActivityLog: [],
          features: mockFeatures,
        },
        version: versions.APP_STORE,
      };

      localStorage.setItem('automaker-storage', JSON.stringify(mockState));

      const setupState = {
        state: { isFirstRun: false, setupComplete: true, skipClaudeSetup: false },
        version: versions.SETUP_STORE,
      };
      localStorage.setItem('automaker-setup', JSON.stringify(setupState));

      const settingsCache = {
        setupComplete: true,
        isFirstRun: false,
        projects: [
          {
            id: mockProject.id,
            name: mockProject.name,
            path: mockProject.path,
            lastOpened: mockProject.lastOpened,
          },
        ],
        theme: 'dark',
        sidebarOpen: true,
        maxConcurrency: opts?.maxConcurrency ?? 3,
      };
      localStorage.setItem('automaker-settings-cache', JSON.stringify(settingsCache));

      // Also store features in a global variable that the mock electron API can use
      // This is needed because the board-view loads features from the file system
      (window as { __mockFeatures?: unknown[] }).__mockFeatures = mockFeatures;

      // Disable splash screen in tests
      localStorage.setItem('automaker-disable-splash', 'true');
    },
    { opts: options, versions: STORE_VERSIONS }
  );
}

/**
 * Set up a mock project with a feature context file
 * This simulates an agent having created context for a feature
 */
export async function setupMockProjectWithContextFile(
  page: Page,
  featureId: string,
  contextContent: string = '# Agent Context\n\nPrevious implementation work...'
): Promise<void> {
  await page.addInitScript(
    ({
      featureId,
      contextContent,
      versions,
    }: {
      featureId: string;
      contextContent: string;
      versions: typeof STORE_VERSIONS;
    }) => {
      const mockProject = {
        id: 'test-project-1',
        name: 'Test Project',
        path: '/mock/test-project',
        lastOpened: new Date().toISOString(),
      };

      const mockState = {
        state: {
          projects: [mockProject],
          currentProject: mockProject,
          theme: 'dark',
          sidebarOpen: true,
          apiKeys: { anthropic: '', google: '' },
          chatSessions: [],
          chatHistoryOpen: false,
          maxConcurrency: 3,
        },
        version: versions.APP_STORE,
      };

      localStorage.setItem('automaker-storage', JSON.stringify(mockState));

      const setupState = {
        state: { isFirstRun: false, setupComplete: true, skipClaudeSetup: false },
        version: versions.SETUP_STORE,
      };
      localStorage.setItem('automaker-setup', JSON.stringify(setupState));

      const settingsCache = {
        setupComplete: true,
        isFirstRun: false,
        projects: [
          {
            id: mockProject.id,
            name: mockProject.name,
            path: mockProject.path,
            lastOpened: mockProject.lastOpened,
          },
        ],
        theme: 'dark',
        sidebarOpen: true,
        maxConcurrency: 3,
      };
      localStorage.setItem('automaker-settings-cache', JSON.stringify(settingsCache));

      // Disable splash screen in tests
      localStorage.setItem('automaker-disable-splash', 'true');

      // Set up mock file system with a context file for the feature
      // This will be used by the mock electron API
      // Now uses features/{id}/agent-output.md path
      (
        window as { __mockContextFile?: { featureId: string; path: string; content: string } }
      ).__mockContextFile = {
        featureId,
        path: `/mock/test-project/.automaker/features/${featureId}/agent-output.md`,
        content: contextContent,
      };
    },
    { featureId, contextContent, versions: STORE_VERSIONS }
  );
}

/**
 * Set up a mock project with features that have startedAt timestamps
 */
export async function setupMockProjectWithInProgressFeatures(
  page: Page,
  options?: {
    maxConcurrency?: number;
    runningTasks?: string[];
    features?: Array<{
      id: string;
      category: string;
      description: string;
      status: 'backlog' | 'in_progress' | 'verified';
      steps?: string[];
      startedAt?: string;
    }>;
  }
): Promise<void> {
  await page.addInitScript(
    ({ opts, versions }: { opts: typeof options; versions: typeof STORE_VERSIONS }) => {
      const mockProject = {
        id: 'test-project-1',
        name: 'Test Project',
        path: '/mock/test-project',
        lastOpened: new Date().toISOString(),
      };

      const mockFeatures = opts?.features || [];

      const mockState = {
        state: {
          projects: [mockProject],
          currentProject: mockProject,
          theme: 'dark',
          sidebarOpen: true,
          apiKeys: { anthropic: '', google: '' },
          chatSessions: [],
          chatHistoryOpen: false,
          maxConcurrency: opts?.maxConcurrency ?? 3,
          isAutoModeRunning: false,
          runningAutoTasks: opts?.runningTasks ?? [],
          autoModeActivityLog: [],
          features: mockFeatures,
        },
        version: versions.APP_STORE,
      };

      localStorage.setItem('automaker-storage', JSON.stringify(mockState));

      const setupState = {
        state: { isFirstRun: false, setupComplete: true, skipClaudeSetup: false },
        version: versions.SETUP_STORE,
      };
      localStorage.setItem('automaker-setup', JSON.stringify(setupState));

      const settingsCache = {
        setupComplete: true,
        isFirstRun: false,
        projects: [
          {
            id: mockProject.id,
            name: mockProject.name,
            path: mockProject.path,
            lastOpened: mockProject.lastOpened,
          },
        ],
        theme: 'dark',
        sidebarOpen: true,
        maxConcurrency: opts?.maxConcurrency ?? 3,
      };
      localStorage.setItem('automaker-settings-cache', JSON.stringify(settingsCache));

      // Also store features in a global variable that the mock electron API can use
      // This is needed because the board-view loads features from the file system
      (window as { __mockFeatures?: unknown[] }).__mockFeatures = mockFeatures;
    },
    { opts: options, versions: STORE_VERSIONS }
  );
}

/**
 * Set up a mock project with a specific current view for route persistence testing
 */
export async function setupMockProjectWithView(page: Page, view: string): Promise<void> {
  await page.addInitScript(
    ({ currentView, versions }: { currentView: string; versions: typeof STORE_VERSIONS }) => {
      const mockProject = {
        id: 'test-project-1',
        name: 'Test Project',
        path: '/mock/test-project',
        lastOpened: new Date().toISOString(),
      };

      const mockState = {
        state: {
          projects: [mockProject],
          currentProject: mockProject,
          currentView: currentView,
          theme: 'dark',
          sidebarOpen: true,
          apiKeys: { anthropic: '', google: '' },
          chatSessions: [],
          chatHistoryOpen: false,
          maxConcurrency: 3,
        },
        version: versions.APP_STORE,
      };

      localStorage.setItem('automaker-storage', JSON.stringify(mockState));

      const setupState = {
        state: { isFirstRun: false, setupComplete: true, skipClaudeSetup: false },
        version: versions.SETUP_STORE,
      };
      localStorage.setItem('automaker-setup', JSON.stringify(setupState));

      const settingsCache = {
        setupComplete: true,
        isFirstRun: false,
        projects: [
          {
            id: mockProject.id,
            name: mockProject.name,
            path: mockProject.path,
            lastOpened: mockProject.lastOpened,
          },
        ],
        theme: 'dark',
        sidebarOpen: true,
        maxConcurrency: 3,
      };
      localStorage.setItem('automaker-settings-cache', JSON.stringify(settingsCache));
    },
    { currentView: view, versions: STORE_VERSIONS }
  );
}

/**
 * Set up an empty localStorage (no projects) to show welcome screen
 */
export async function setupEmptyLocalStorage(page: Page): Promise<void> {
  await page.addInitScript((versions: typeof STORE_VERSIONS) => {
    const mockState = {
      state: {
        projects: [],
        currentProject: null,
        currentView: 'welcome',
        theme: 'dark',
        sidebarOpen: true,
        apiKeys: { anthropic: '', google: '' },
        chatSessions: [],
        chatHistoryOpen: false,
        maxConcurrency: 3,
      },
      version: versions.APP_STORE,
    };
    localStorage.setItem('automaker-storage', JSON.stringify(mockState));

    const setupState = {
      state: { isFirstRun: false, setupComplete: true, skipClaudeSetup: false },
      version: versions.SETUP_STORE,
    };
    localStorage.setItem('automaker-setup', JSON.stringify(setupState));

    const settingsCache = {
      setupComplete: true,
      isFirstRun: false,
      projects: [],
      theme: 'dark',
      sidebarOpen: true,
      maxConcurrency: 3,
    };
    localStorage.setItem('automaker-settings-cache', JSON.stringify(settingsCache));

    // Disable splash screen in tests
    localStorage.setItem('automaker-disable-splash', 'true');
  }, STORE_VERSIONS);
}

/**
 * Set up mock projects in localStorage but with no current project (for recent projects list)
 */
export async function setupMockProjectsWithoutCurrent(page: Page): Promise<void> {
  await page.addInitScript((versions: typeof STORE_VERSIONS) => {
    const mockProjects = [
      {
        id: 'test-project-1',
        name: 'Test Project 1',
        path: '/mock/test-project-1',
        lastOpened: new Date().toISOString(),
      },
      {
        id: 'test-project-2',
        name: 'Test Project 2',
        path: '/mock/test-project-2',
        lastOpened: new Date(Date.now() - 86400000).toISOString(), // 1 day ago
      },
    ];

    const mockState = {
      state: {
        projects: mockProjects,
        currentProject: null,
        currentView: 'welcome',
        theme: 'dark',
        sidebarOpen: true,
        apiKeys: { anthropic: '', google: '' },
        chatSessions: [],
        chatHistoryOpen: false,
        maxConcurrency: 3,
      },
      version: versions.APP_STORE,
    };

    localStorage.setItem('automaker-storage', JSON.stringify(mockState));

    const setupState = {
      state: { isFirstRun: false, setupComplete: true, skipClaudeSetup: false },
      version: versions.SETUP_STORE,
    };
    localStorage.setItem('automaker-setup', JSON.stringify(setupState));

    const settingsCache = {
      setupComplete: true,
      isFirstRun: false,
      projects: mockProjects.map((p) => ({
        id: p.id,
        name: p.name,
        path: p.path,
        lastOpened: p.lastOpened,
      })),
      theme: 'dark',
      sidebarOpen: true,
      maxConcurrency: 3,
    };
    localStorage.setItem('automaker-settings-cache', JSON.stringify(settingsCache));

    // Disable splash screen in tests
    localStorage.setItem('automaker-disable-splash', 'true');
  }, STORE_VERSIONS);
}

/**
 * Set up a mock project with features that have skipTests enabled
 */
export async function setupMockProjectWithSkipTestsFeatures(
  page: Page,
  options?: {
    maxConcurrency?: number;
    runningTasks?: string[];
    features?: Array<{
      id: string;
      category: string;
      description: string;
      status: 'backlog' | 'in_progress' | 'verified';
      steps?: string[];
      startedAt?: string;
      skipTests?: boolean;
    }>;
  }
): Promise<void> {
  await page.addInitScript(
    ({ opts, versions }: { opts: typeof options; versions: typeof STORE_VERSIONS }) => {
      const mockProject = {
        id: 'test-project-1',
        name: 'Test Project',
        path: '/mock/test-project',
        lastOpened: new Date().toISOString(),
      };

      const mockFeatures = opts?.features || [];

      const mockState = {
        state: {
          projects: [mockProject],
          currentProject: mockProject,
          theme: 'dark',
          sidebarOpen: true,
          apiKeys: { anthropic: '', google: '' },
          chatSessions: [],
          chatHistoryOpen: false,
          maxConcurrency: opts?.maxConcurrency ?? 3,
          isAutoModeRunning: false,
          runningAutoTasks: opts?.runningTasks ?? [],
          autoModeActivityLog: [],
          features: mockFeatures,
        },
        version: versions.APP_STORE,
      };

      localStorage.setItem('automaker-storage', JSON.stringify(mockState));

      const setupState = {
        state: { isFirstRun: false, setupComplete: true, skipClaudeSetup: false },
        version: versions.SETUP_STORE,
      };
      localStorage.setItem('automaker-setup', JSON.stringify(setupState));

      const settingsCache = {
        setupComplete: true,
        isFirstRun: false,
        projects: [
          {
            id: mockProject.id,
            name: mockProject.name,
            path: mockProject.path,
            lastOpened: mockProject.lastOpened,
          },
        ],
        theme: 'dark',
        sidebarOpen: true,
        maxConcurrency: opts?.maxConcurrency ?? 3,
      };
      localStorage.setItem('automaker-settings-cache', JSON.stringify(settingsCache));

      // Disable splash screen in tests
      localStorage.setItem('automaker-disable-splash', 'true');
    },
    { opts: options, versions: STORE_VERSIONS }
  );
}

/**
 * Set up a mock state with multiple projects
 */
export async function setupMockMultipleProjects(
  page: Page,
  projectCount: number = 3
): Promise<void> {
  await page.addInitScript(
    ({ count, versions }: { count: number; versions: typeof STORE_VERSIONS }) => {
      const mockProjects: TestProject[] = [];
      for (let i = 0; i < count; i++) {
        mockProjects.push({
          id: `test-project-${i + 1}`,
          name: `Test Project ${i + 1}`,
          path: `/mock/test-project-${i + 1}`,
          lastOpened: new Date(Date.now() - i * 86400000).toISOString(),
        });
      }

      const mockState = {
        state: {
          projects: mockProjects,
          currentProject: mockProjects[0],
          currentView: 'board',
          theme: 'dark',
          sidebarOpen: true,
          apiKeys: { anthropic: '', google: '' },
          chatSessions: [],
          chatHistoryOpen: false,
          maxConcurrency: 3,
        },
        version: versions.APP_STORE,
      };

      localStorage.setItem('automaker-storage', JSON.stringify(mockState));

      // Mark setup as complete to prevent redirect to /setup
      const setupState = {
        state: {
          isFirstRun: false,
          setupComplete: true,
          skipClaudeSetup: false,
        },
        version: versions.SETUP_STORE,
      };
      localStorage.setItem('automaker-setup', JSON.stringify(setupState));

      // Set settings cache so the fast hydrate path is taken on page load.
      // This prevents the server's setupComplete value (which may be false on fresh CI)
      // from overwriting the setup store and causing a redirect to /setup.
      const settingsCache = {
        setupComplete: true,
        isFirstRun: false,
        projects: mockProjects.map((p) => ({
          id: p.id,
          name: p.name,
          path: p.path,
          lastOpened: p.lastOpened,
        })),
        // Include currentProjectId so hydrateStoreFromSettings can restore
        // the current project directly (without relying on auto-open logic)
        currentProjectId: mockProjects[0]?.id ?? null,
        theme: 'dark',
        sidebarOpen: true,
        maxConcurrency: 3,
      };
      localStorage.setItem('automaker-settings-cache', JSON.stringify(settingsCache));

      // Disable splash screen in tests
      localStorage.setItem('automaker-disable-splash', 'true');
    },
    { count: projectCount, versions: STORE_VERSIONS }
  );
}

/**
 * Set up a mock project with agent output content in the context file
 */
export async function setupMockProjectWithAgentOutput(
  page: Page,
  featureId: string,
  outputContent: string
): Promise<void> {
  await page.addInitScript(
    ({
      featureId,
      outputContent,
      versions,
    }: {
      featureId: string;
      outputContent: string;
      versions: typeof STORE_VERSIONS;
    }) => {
      const mockProject = {
        id: 'test-project-1',
        name: 'Test Project',
        path: '/mock/test-project',
        lastOpened: new Date().toISOString(),
      };

      const mockState = {
        state: {
          projects: [mockProject],
          currentProject: mockProject,
          theme: 'dark',
          sidebarOpen: true,
          apiKeys: { anthropic: '', google: '' },
          chatSessions: [],
          chatHistoryOpen: false,
          maxConcurrency: 3,
        },
        version: versions.APP_STORE,
      };

      localStorage.setItem('automaker-storage', JSON.stringify(mockState));

      const setupState = {
        state: { isFirstRun: false, setupComplete: true, skipClaudeSetup: false },
        version: versions.SETUP_STORE,
      };
      localStorage.setItem('automaker-setup', JSON.stringify(setupState));

      const settingsCache = {
        setupComplete: true,
        isFirstRun: false,
        projects: [
          {
            id: mockProject.id,
            name: mockProject.name,
            path: mockProject.path,
            lastOpened: mockProject.lastOpened,
          },
        ],
        theme: 'dark',
        sidebarOpen: true,
        maxConcurrency: 3,
      };
      localStorage.setItem('automaker-settings-cache', JSON.stringify(settingsCache));

      // Disable splash screen in tests
      localStorage.setItem('automaker-disable-splash', 'true');

      // Set up mock file system with output content for the feature
      // Now uses features/{id}/agent-output.md path
      (
        window as { __mockContextFile?: { featureId: string; path: string; content: string } }
      ).__mockContextFile = {
        featureId,
        path: `/mock/test-project/.automaker/features/${featureId}/agent-output.md`,
        content: outputContent,
      };
    },
    { featureId, outputContent, versions: STORE_VERSIONS }
  );
}

/**
 * Set up a mock project with features that include waiting_approval status
 */
export async function setupMockProjectWithWaitingApprovalFeatures(
  page: Page,
  options?: {
    maxConcurrency?: number;
    runningTasks?: string[];
    features?: Array<{
      id: string;
      category: string;
      description: string;
      status: 'backlog' | 'in_progress' | 'waiting_approval' | 'verified';
      steps?: string[];
      startedAt?: string;
      skipTests?: boolean;
    }>;
  }
): Promise<void> {
  await page.addInitScript(
    ({ opts, versions }: { opts: typeof options; versions: typeof STORE_VERSIONS }) => {
      const mockProject = {
        id: 'test-project-1',
        name: 'Test Project',
        path: '/mock/test-project',
        lastOpened: new Date().toISOString(),
      };

      const mockFeatures = opts?.features || [];

      const mockState = {
        state: {
          projects: [mockProject],
          currentProject: mockProject,
          theme: 'dark',
          sidebarOpen: true,
          apiKeys: { anthropic: '', google: '' },
          chatSessions: [],
          chatHistoryOpen: false,
          maxConcurrency: opts?.maxConcurrency ?? 3,
          isAutoModeRunning: false,
          runningAutoTasks: opts?.runningTasks ?? [],
          autoModeActivityLog: [],
          features: mockFeatures,
        },
        version: versions.APP_STORE,
      };

      localStorage.setItem('automaker-storage', JSON.stringify(mockState));

      const setupState = {
        state: { isFirstRun: false, setupComplete: true, skipClaudeSetup: false },
        version: versions.SETUP_STORE,
      };
      localStorage.setItem('automaker-setup', JSON.stringify(setupState));

      const settingsCache = {
        setupComplete: true,
        isFirstRun: false,
        projects: [
          {
            id: mockProject.id,
            name: mockProject.name,
            path: mockProject.path,
            lastOpened: mockProject.lastOpened,
          },
        ],
        theme: 'dark',
        sidebarOpen: true,
        maxConcurrency: opts?.maxConcurrency ?? 3,
      };
      localStorage.setItem('automaker-settings-cache', JSON.stringify(settingsCache));

      // Also store features in a global variable that the mock electron API can use
      (window as { __mockFeatures?: unknown[] }).__mockFeatures = mockFeatures;
    },
    { opts: options, versions: STORE_VERSIONS }
  );
}

/**
 * Set up the app store to show setup view (simulate first run)
 */
export async function setupFirstRun(page: Page): Promise<void> {
  await page.addInitScript((versions: typeof STORE_VERSIONS) => {
    // Clear any existing setup state to simulate first run
    localStorage.removeItem('automaker-setup');
    localStorage.removeItem('automaker-storage');

    // Set up the setup store state for first run
    const setupState = {
      state: {
        isFirstRun: true,
        setupComplete: false,
        currentStep: 'welcome',
        claudeCliStatus: null,
        claudeAuthStatus: null,
        claudeInstallProgress: {
          isInstalling: false,
          currentStep: '',
          progress: 0,
          output: [],
        },
        skipClaudeSetup: false,
      },
      version: versions.SETUP_STORE, // Must match setup-store.ts persist version
    };

    localStorage.setItem('automaker-setup', JSON.stringify(setupState));

    // Also set up app store to show setup view
    const appState = {
      state: {
        projects: [],
        currentProject: null,
        theme: 'dark',
        sidebarOpen: true,
        apiKeys: { anthropic: '', google: '' },
        chatSessions: [],
        chatHistoryOpen: false,
        maxConcurrency: 3,
        isAutoModeRunning: false,
        runningAutoTasks: [],
        autoModeActivityLog: [],
        currentView: 'setup',
      },
      version: versions.APP_STORE, // Must match app-store.ts persist version
    };

    localStorage.setItem('automaker-storage', JSON.stringify(appState));

    // Anchor the settings cache so CI cannot hydrate a conflicting setupComplete value.
    const settingsCache = {
      setupComplete: false,
      isFirstRun: true,
      projects: [],
      theme: 'dark',
      sidebarOpen: true,
      maxConcurrency: 3,
    };
    localStorage.setItem('automaker-settings-cache', JSON.stringify(settingsCache));

    // Disable splash screen in tests
    localStorage.setItem('automaker-disable-splash', 'true');
  }, STORE_VERSIONS);
}

/**
 * Set up the app to skip the setup wizard (setup already complete)
 */
export async function setupComplete(page: Page): Promise<void> {
  await page.addInitScript((versions: typeof STORE_VERSIONS) => {
    // Mark setup as complete
    const setupState = {
      state: {
        isFirstRun: false,
        setupComplete: true,
        currentStep: 'complete',
        skipClaudeSetup: false,
      },
      version: versions.SETUP_STORE,
    };

    localStorage.setItem('automaker-setup', JSON.stringify(setupState));

    // Disable splash screen in tests
    localStorage.setItem('automaker-disable-splash', 'true');
  }, STORE_VERSIONS);
}