import type { AgentResult, TestRunnerOptions, ValidationCheck } from "../utils.ts";
import { defineFixture, getAgentOutput } from "../utils.ts";

/**
 * delegateMulti test - validates multi-phase delegation with context passing.
 *
 * the orchestrator delegates twice using the tasks array API:
 * 1. first to Plan mode with a single-task array (subagent calls set_output with PHASE_1_MARKER)
 * 2. then to Plan mode again with context from phase 1 (subagent calls set_output with MULTI_DELEGATE_PASSED)
 *
 * validates that both delegations executed and the final set_output value is correct.
 */

const fixture = defineFixture(
  {
    prompt: `This is a multi-delegation test. You must delegate exactly twice using the tasks array format.

Phase 1: Select Plan mode via select_mode, then delegate with tasks: [{ label: "phase-1", instructions: "Your task is to call set_output with the value 'PHASE_1_MARKER'. Do not create plans or PRs.", effort: "mini" }]

Phase 2: After Phase 1 completes, select Plan mode again and delegate with tasks: [{ label: "phase-2", instructions: "Your task is to call set_output with the value 'MULTI_DELEGATE_PASSED'. Do not create plans or PRs.", effort: "mini" }]. Include the result from Phase 1 in the instructions if you want.

Both delegations must complete successfully.

When all delegations are complete, call set_output with the final result. This makes it available as the GitHub Action output.`,
    effort: "mini",
    timeout: "8m",
  },
  { localOnly: true }
);

function validator(result: AgentResult): ValidationCheck[] {
  const output = result.structuredOutput;
  const agentOutput = getAgentOutput(result);

  const setOutputCalled = output !== null;
  // the last set_output call wins — should be from Phase 2
  const finalValue = setOutputCalled && /MULTI_DELEGATE_PASSED/i.test(output);

  const delegationMatches = agentOutput.match(/» delegating \d+ task/g);
  const twoDelegations = delegationMatches !== null && delegationMatches.length >= 2;

  return [
    { name: "set_output", passed: setOutputCalled },
    { name: "final_value", passed: finalValue },
    { name: "two_delegations", passed: twoDelegations },
  ];
}

export const test: TestRunnerOptions = {
  name: "delegate-multi",
  fixture,
  validator,
  env: { GITHUB_REPOSITORY: "pullfrog/test-repo" },
  tags: ["agnostic"],
};