import { execSync } from "node:child_process";
import path from "node:path";
import { test } from "./fixtures";
import {
  clickNewTabButton,
  gotoWorkspace,
  waitForLauncherPanel,
} from "./helpers/launcher";
import { createTempGitRepo } from "./helpers/workspace";
import {
  createAgentChatFromLauncher,
  createStandaloneTerminalFromLauncher,
  createTerminalAgentFromLauncher,
  expectTerminalCwd,
} from "./helpers/workspace-lifecycle";
import { connectWorkspaceSetupClient, seedProjectForWorkspaceSetup } from "./helpers/workspace-setup";

test.describe("Workspace lifecycle", () => {
  // The first test after a spec-file switch can intermittently fail because
  // the shared daemon still holds stale sessions from the previous spec.
  // One retry is enough for the daemon to stabilize.
  test.describe.configure({ retries: 1 });

  test.describe("Main checkout", () => {
    test("creates a terminal agent via provider tile", async ({ page }) => {
      test.setTimeout(60_000);

      const client = await connectWorkspaceSetupClient();
      const repo = await createTempGitRepo("lifecycle-main-agent-");

      try {
        await seedProjectForWorkspaceSetup(client, repo.path);
        const workspaceResult = await client.openProject(repo.path);
        if (!workspaceResult.workspace) {
          throw new Error(workspaceResult.error ?? `Failed to open project ${repo.path}`);
        }
        const workspaceId = String(workspaceResult.workspace.id);

        await gotoWorkspace(page, workspaceId);
        await clickNewTabButton(page);
        await waitForLauncherPanel(page);
        await createTerminalAgentFromLauncher(page, "Claude");
      } finally {
        await client.close();
        await repo.cleanup();
      }
    });

    test("creates an agent chat via New Chat", async ({ page }) => {
      test.setTimeout(60_000);

      const client = await connectWorkspaceSetupClient();
      const repo = await createTempGitRepo("lifecycle-main-chat-");

      try {
        await seedProjectForWorkspaceSetup(client, repo.path);
        const workspaceResult = await client.openProject(repo.path);
        if (!workspaceResult.workspace) {
          throw new Error(workspaceResult.error ?? `Failed to open project ${repo.path}`);
        }
        const workspaceId = String(workspaceResult.workspace.id);

        await gotoWorkspace(page, workspaceId);
        await clickNewTabButton(page);
        await waitForLauncherPanel(page);
        await createAgentChatFromLauncher(page);
      } finally {
        await client.close();
        await repo.cleanup();
      }
    });

    test("creates a terminal with correct CWD", async ({ page }) => {
      test.setTimeout(60_000);

      const client = await connectWorkspaceSetupClient();
      const repo = await createTempGitRepo("lifecycle-main-shell-");

      try {
        await seedProjectForWorkspaceSetup(client, repo.path);
        const workspaceResult = await client.openProject(repo.path);
        if (!workspaceResult.workspace) {
          throw new Error(workspaceResult.error ?? `Failed to open project ${repo.path}`);
        }
        const workspaceId = String(workspaceResult.workspace.id);

        await gotoWorkspace(page, workspaceId);
        await clickNewTabButton(page);
        await waitForLauncherPanel(page);
        await createStandaloneTerminalFromLauncher(page);
        await expectTerminalCwd(page, repo.path);
      } finally {
        await client.close();
        await repo.cleanup();
      }
    });
  });

  test.describe("Worktree workspace", () => {
    test("creates a terminal agent via provider tile", async ({ page }) => {
      test.setTimeout(90_000);

      const client = await connectWorkspaceSetupClient();
      const repo = await createTempGitRepo("lifecycle-wt-agent-");
      const worktreePath = path.join(
        "/tmp",
        `paseo-wt-${Date.now()}-${Math.random().toString(36).slice(2)}`,
      );
      const branchName = `lifecycle-wt-agent-${Date.now()}`;
      let worktreeCreated = false;

      try {
        await seedProjectForWorkspaceSetup(client, repo.path);

        execSync(`git worktree add ${JSON.stringify(worktreePath)} -b ${JSON.stringify(branchName)} main`, {
          cwd: repo.path,
          stdio: "ignore",
        });
        worktreeCreated = true;

        const workspaceResult = await client.openProject(worktreePath);
        if (!workspaceResult.workspace) {
          throw new Error(workspaceResult.error ?? `Failed to open project ${worktreePath}`);
        }
        const workspaceId = String(workspaceResult.workspace.id);

        await gotoWorkspace(page, workspaceId);
        await clickNewTabButton(page);
        await waitForLauncherPanel(page);
        await createTerminalAgentFromLauncher(page, "Claude");
      } finally {
        if (worktreeCreated) {
          try {
            execSync(`git worktree remove ${JSON.stringify(worktreePath)} --force`, {
              cwd: repo.path,
              stdio: "ignore",
            });
          } catch {
            // Best-effort cleanup so test failures preserve the original error.
          }
        }
        await client.close();
        await repo.cleanup();
      }
    });

    test("creates an agent chat via New Chat", async ({ page }) => {
      test.setTimeout(90_000);

      const client = await connectWorkspaceSetupClient();
      const repo = await createTempGitRepo("lifecycle-wt-chat-");
      const worktreePath = path.join(
        "/tmp",
        `paseo-wt-${Date.now()}-${Math.random().toString(36).slice(2)}`,
      );
      const branchName = `lifecycle-wt-chat-${Date.now()}`;
      let worktreeCreated = false;

      try {
        await seedProjectForWorkspaceSetup(client, repo.path);

        execSync(`git worktree add ${JSON.stringify(worktreePath)} -b ${JSON.stringify(branchName)} main`, {
          cwd: repo.path,
          stdio: "ignore",
        });
        worktreeCreated = true;

        const workspaceResult = await client.openProject(worktreePath);
        if (!workspaceResult.workspace) {
          throw new Error(workspaceResult.error ?? `Failed to open project ${worktreePath}`);
        }
        const workspaceId = String(workspaceResult.workspace.id);

        await gotoWorkspace(page, workspaceId);
        await clickNewTabButton(page);
        await waitForLauncherPanel(page);
        await createAgentChatFromLauncher(page);
      } finally {
        if (worktreeCreated) {
          try {
            execSync(`git worktree remove ${JSON.stringify(worktreePath)} --force`, {
              cwd: repo.path,
              stdio: "ignore",
            });
          } catch {
            // Best-effort cleanup so test failures preserve the original error.
          }
        }
        await client.close();
        await repo.cleanup();
      }
    });

    test("creates a terminal with correct CWD", async ({ page }) => {
      test.setTimeout(90_000);

      const client = await connectWorkspaceSetupClient();
      const repo = await createTempGitRepo("lifecycle-wt-shell-");
      const worktreePath = path.join(
        "/tmp",
        `paseo-wt-${Date.now()}-${Math.random().toString(36).slice(2)}`,
      );
      const branchName = `lifecycle-wt-shell-${Date.now()}`;
      let worktreeCreated = false;

      try {
        await seedProjectForWorkspaceSetup(client, repo.path);

        execSync(`git worktree add ${JSON.stringify(worktreePath)} -b ${JSON.stringify(branchName)} main`, {
          cwd: repo.path,
          stdio: "ignore",
        });
        worktreeCreated = true;

        const workspaceResult = await client.openProject(worktreePath);
        if (!workspaceResult.workspace) {
          throw new Error(workspaceResult.error ?? `Failed to open project ${worktreePath}`);
        }
        const workspaceId = String(workspaceResult.workspace.id);

        await gotoWorkspace(page, workspaceId);
        await clickNewTabButton(page);
        await waitForLauncherPanel(page);
        await createStandaloneTerminalFromLauncher(page);
        await expectTerminalCwd(page, worktreePath);
      } finally {
        if (worktreeCreated) {
          try {
            execSync(`git worktree remove ${JSON.stringify(worktreePath)} --force`, {
              cwd: repo.path,
              stdio: "ignore",
            });
          } catch {
            // Best-effort cleanup so test failures preserve the original error.
          }
        }
        await client.close();
        await repo.cleanup();
      }
    });
  });
});