import { type } from "arktype";
import { log } from "../utils/cli.ts";
import type { ToolContext } from "./server.ts";
import { execute, tool } from "./shared.ts";

export const SetOutputParams = type({
  value: type.string.describe("the output value to expose as a GitHub Action output"),
});

export function SetOutputTool(ctx: ToolContext) {
  return tool({
    name: "set_output",
    description:
      "Set the action output. When called by a subagent, returns a summary result to the orchestrator. When called in standalone mode, exposes the value as the 'result' GitHub Action output.",
    parameters: SetOutputParams,
    execute: execute(async (params) => {
      const activeId = ctx.toolState.activeSubagentId;
      if (activeId) {
        const subagent = ctx.toolState.subagents.get(activeId);
        if (subagent) {
          subagent.output = params.value;
          return { success: true, routed: "subagent" };
        }
        log.warning(
          `set_output: activeSubagentId=${activeId} but subagent not found in map — routing to action output`
        );
      }
      ctx.toolState.output = params.value;
      return { success: true, routed: "action_output" };
    }),
  });
}