import { afterEach, describe, expect, it, vi } from "vitest";
import {
  getInstallationTarget,
  withInstallationTarget,
} from "../../infra/installation-target-context.js";
import { withEnvAsync } from "../../test-utils/env.js";
import { buildPreparedCliRunContext } from "../cli-runner.test-helpers.js";
import { executePreparedCliRun as executePreparedCliRunImpl } from "./execute.js";
import {
  createManagedRun,
  createSuccessfulProcessExit,
  supervisorSpawnMock,
  wrapPreparedCliRunWithTestAdmission,
} from "./execute.test-support.js";

const executePreparedCliRun = wrapPreparedCliRunWithTestAdmission(executePreparedCliRunImpl);

afterEach(() => supervisorSpawnMock.mockReset());

describe("CLI installation target", () => {
  it.each(["process", "plugin", "node"] as const)(
    "projects local child environment and fences %s placement",
    async (kind) => {
      const target = {
        stateDir: "/fixture/diagnosed",
        configPath: "/fixture/custom.json",
        defaultWorkspaceDir: "/fixture/default-workspace",
      };
      const context = buildPreparedCliRunContext({
        model: "fixture-model",
        backend: {
          command: "/bin/sh",
          args: [],
          output: "text",
          systemPromptFileArg: undefined,
          input: "stdin",
        },
      });
      let childEnv: NodeJS.ProcessEnv | undefined;
      const pluginExecute = vi.fn(async function* (execution: { env: NodeJS.ProcessEnv }) {
        childEnv = execution.env;
        yield { type: "result", subtype: "success", result: "done" };
      });
      if (kind === "plugin") {
        context.executionTarget = { kind, execute: pluginExecute };
        context.preparedBackend.backend.output = "jsonl";
        context.preparedBackend.backend.jsonlDialect = "claude-stream-json";
      } else if (kind === "node") {
        context.executionTarget = { kind, placement: { nodeId: "fixture-node" } };
      }
      supervisorSpawnMock.mockResolvedValue(
        createManagedRun({
          ...createSuccessfulProcessExit(),
          durationMs: 1,
          stdout: "done",
        }),
      );
      await withEnvAsync(
        {
          OPENCLAW_STATE_DIR: "/fixture/scratch",
          OPENCLAW_CONFIG_PATH: undefined,
          OPENCLAW_WORKSPACE_DIR: "/fixture/execution-cwd",
        },
        async () => {
          const run = withInstallationTarget(target, () => executePreparedCliRun(context));
          expect(getInstallationTarget()).toBeUndefined();
          if (kind === "node") {
            await expect(run).rejects.toThrow("saved prompt");
            expect(supervisorSpawnMock).not.toHaveBeenCalled();
            expect(pluginExecute).not.toHaveBeenCalled();
            return;
          }
          await expect(run).resolves.toMatchObject({ text: "done" });
          const expectedEnv = {
            OPENCLAW_STATE_DIR: target.stateDir,
            OPENCLAW_CONFIG_PATH: target.configPath,
            OPENCLAW_WORKSPACE_DIR: target.defaultWorkspaceDir,
          };
          if (kind === "process") {
            expect(supervisorSpawnMock).toHaveBeenLastCalledWith(
              expect.objectContaining({ env: expect.objectContaining(expectedEnv) }),
            );
          } else {
            expect(childEnv).toMatchObject(expectedEnv);
          }
          expect(process.env.OPENCLAW_STATE_DIR).toBe("/fixture/scratch");
          expect(process.env.OPENCLAW_CONFIG_PATH).toBeUndefined();
          expect(process.env.OPENCLAW_WORKSPACE_DIR).toBe("/fixture/execution-cwd");
          await executePreparedCliRun(context);
          if (kind === "process") {
            expect(supervisorSpawnMock).toHaveBeenLastCalledWith(
              expect.objectContaining({
                env: expect.objectContaining({
                  OPENCLAW_STATE_DIR: "/fixture/scratch",
                  OPENCLAW_WORKSPACE_DIR: "/fixture/execution-cwd",
                }),
              }),
            );
            expect(supervisorSpawnMock).not.toHaveBeenLastCalledWith(
              expect.objectContaining({
                env: expect.objectContaining({ OPENCLAW_CONFIG_PATH: expect.anything() }),
              }),
            );
          } else {
            expect(childEnv?.OPENCLAW_STATE_DIR).toBe("/fixture/scratch");
            expect(childEnv?.OPENCLAW_CONFIG_PATH).toBeUndefined();
            expect(childEnv?.OPENCLAW_WORKSPACE_DIR).toBe("/fixture/execution-cwd");
          }
        },
      );
    },
  );
});