united-workforce/packages/cli/src/__tests__/thread-start-cwd-cli.test.ts

import { execFileSync } from "node:child_process";
import { mkdir, rm, writeFile } from "node:fs/promises";
import { tmpdir } from "node:os";
import { dirname, join } from "node:path";
import { fileURLToPath } from "node:url";
import type { CasRef, StartNodePayload, ThreadId } from "@united-workforce/protocol";
import { describe, expect, test } from "vitest";
import { cmdThreadStart } from "../commands/thread.js";
import { createUwfStore, getThread } from "../store.js";

describe("thread start --cwd CLI option", () => {
  let tmpDir: string;
  let storageRoot: string;
  let casDir: string;
  let originalEnv: string | undefined;

  async function setupTestEnv() {
    tmpDir = join(tmpdir(), `uwf-test-cwd-cli-${Date.now()}`);
    storageRoot = join(tmpDir, "storage");
    casDir = join(tmpDir, "cas");
    await mkdir(storageRoot, { recursive: true });
    await mkdir(casDir, { recursive: true });

    // Set OCAS_HOME for this test
    originalEnv = process.env.OCAS_HOME;
    process.env.OCAS_HOME = casDir;
  }

  async function teardown() {
    if (tmpDir) {
      await rm(tmpDir, { recursive: true, force: true });
    }
    // Restore original environment
    if (originalEnv === undefined) {
      delete process.env.OCAS_HOME;
    } else {
      process.env.OCAS_HOME = originalEnv;
    }
  }

  async function createTestWorkflow(): Promise<string> {
    const workflowYaml = `
name: test-cwd-cli
description: Test workflow for CLI cwd option
roles:
  planner:
    description: Plans the work
    goal: Plan implementation
    capabilities: ["planning"]
    procedure: Plan
    output: |
      $status: "ready"
    frontmatter:
      type: object
      required: ["$status"]
      properties:
        $status: { type: string, enum: ["ready"] }
graph:
  $START:
    _:
      role: planner
      prompt: "Plan the work"
      location: null
  planner:
    ready:
      role: $END
      prompt: "Done"
      location: null
`;

    const workflowPath = join(tmpDir, "test-cwd-cli.yaml");
    await writeFile(workflowPath, workflowYaml, "utf8");
    return workflowPath;
  }

  async function getStartNodeCwd(threadId: string): Promise<string> {
    const uwf = await createUwfStore(storageRoot);
    const entry = getThread(uwf.varStore, threadId as ThreadId);
    const headHash = entry!.head;
    expect(headHash).toBeDefined();

    const startNode = uwf.store.cas.get(headHash as CasRef);
    expect(startNode).not.toBe(null);
    expect(startNode?.type).toBe(uwf.schemas.startNode);

    const startPayload = startNode?.payload as StartNodePayload;
    return startPayload.cwd;
  }

  test("thread start with custom cwd via cmdThreadStart", async () => {
    await setupTestEnv();

    const workflowPath = await createTestWorkflow();
    const testCwd = "/test/custom/path";

    const result = await cmdThreadStart(storageRoot, workflowPath, "test prompt", tmpDir, testCwd);

    expect(result.thread).toBeDefined();
    const actualCwd = await getStartNodeCwd(result.thread);
    expect(actualCwd).toBe(testCwd);

    await teardown();
  });

  test("thread start without cwd defaults to process.cwd()", async () => {
    await setupTestEnv();

    const workflowPath = await createTestWorkflow();

    // Call without cwd parameter (it defaults to process.cwd())
    const result = await cmdThreadStart(storageRoot, workflowPath, "test prompt", tmpDir);

    expect(result.thread).toBeDefined();
    const actualCwd = await getStartNodeCwd(result.thread);
    expect(actualCwd).toBe(process.cwd());

    await teardown();
  });

  test("thread start with relative path fails", async () => {
    await setupTestEnv();

    const workflowPath = await createTestWorkflow();

    await expect(
      cmdThreadStart(storageRoot, workflowPath, "test", tmpDir, "relative/path"),
    ).rejects.toThrow();

    await teardown();
  });

  test("CLI accepts --cwd option without error", async () => {
    await setupTestEnv();

    const workflowPath = await createTestWorkflow();
    const testCwd = "/test/cli/path";
    const pkgRoot = dirname(dirname(dirname(fileURLToPath(import.meta.url))));
    const uwfBin = join(pkgRoot, "dist", "cli.js");

    // Register the workflow
    execFileSync(process.execPath, [uwfBin, "workflow", "add", workflowPath], {
      env: { ...process.env, UWF_HOME: storageRoot, OCAS_HOME: casDir },
      encoding: "utf8",
    });

    // Verify CLI accepts --cwd option (no error thrown)
    const output = execFileSync(
      process.execPath,
      [uwfBin, "thread", "start", "test-cwd-cli", "-p", "test prompt", "--cwd", testCwd],
      {
        env: { ...process.env, UWF_HOME: storageRoot, OCAS_HOME: casDir },
        encoding: "utf8",
      },
    );

    const result = JSON.parse(output);
    expect(result.thread).toBeDefined();
    expect(result.workflow).toBeDefined();

    // The fact that we got here without throwing means CLI accepted the --cwd option
    // The actual cwd functionality is tested by the other tests using cmdThreadStart directly
    await teardown();
  });
});