import { type RoleHost } from "./host-contracts.ts";
import { REVIEW_QUEUE_STATUSES, reviewSubmissionSchema, type ReviewQueueWord } from "./review-submission.ts";
import { registerFiledSubmissionTool } from "./filed-submission.ts";
import { roleSubmissionDeclaration } from "./role-submission-declarations.ts";
import { type GatekeeperSubject } from "./gatekeeper-role.ts";
import {
  JUDGE_OUTPUT_TOOL_NAME,
  type JudgeVerdict,
} from "./package-contracts/judge-output.ts";
import type { TerminalResult } from "./public-cli/terminal.ts";

export { JUDGE_OUTPUT_TOOL_NAME };
export type { JudgeVerdict };

/** The only judge audit order: 符宝郎, then 审刑院. */
const JUDGE_GATES: readonly GatekeeperSubject[] = [
  { kind: "judge_draft" },
  { kind: "judge_compliance" },
];

export async function runJudgeGates(input: {
  readonly gateAlreadyConverged: (subject: GatekeeperSubject) => Promise<boolean>;
  readonly runGate: (
    subject: GatekeeperSubject,
  ) => Promise<{
    readonly status?: unknown;
    readonly receipt?: unknown;
    readonly runId?: string;
    readonly runDirectory?: string;
    readonly terminal?: TerminalResult;
  } | void>;
}): Promise<{
  readonly status: ReviewQueueWord | "needs_reask";
  readonly passes: readonly {
    readonly subject: GatekeeperSubject;
    readonly status: ReviewQueueWord | "needs_reask";
    readonly receipt: unknown;
    readonly runId?: string;
    readonly runDirectory?: string;
    readonly terminal?: TerminalResult;
  }[];
}> {
  const passes: {
    subject: GatekeeperSubject;
    status: ReviewQueueWord | "needs_reask";
    receipt: unknown;
    runId?: string;
    runDirectory?: string;
    terminal?: TerminalResult;
  }[] = [];
  for (const subject of JUDGE_GATES) {
    if (await input.gateAlreadyConverged(subject)) continue;
    const pass = await input.runGate(subject);
    const received = pass?.status;
    if (pass === undefined || (
      typeof received !== "string" || (!REVIEW_QUEUE_STATUSES.has(received) && received !== "needs_reask")
    )) {
      throw new Error("judge gate returned no conclusion");
    }
    const status = received as ReviewQueueWord | "needs_reask";
    if (status === "needs_reask") {
      return {
        status,
        passes: [{
          subject,
          status,
          receipt: pass.receipt,
          ...(pass.runId === undefined ? {} : { runId: pass.runId }),
          ...(pass.runDirectory === undefined ? {} : { runDirectory: pass.runDirectory }),
          ...(pass.terminal === undefined ? {} : { terminal: pass.terminal }),
        }],
      };
    }
    passes.push({
      subject,
      status,
      receipt: pass.receipt,
      ...(pass.runId === undefined ? {} : { runId: pass.runId }),
      ...(pass.runDirectory === undefined ? {} : { runDirectory: pass.runDirectory }),
    });
    if (status === "continue" || status === "escalate") return { status, passes };
  }
  return { status: "converged", passes };
}

// #836 r16 class 1: fix/classes/note/decisionGate are LLM/human-read narrative
// content — 符宝郎/审刑院 read the raw receipt, no code branches on their length
// or nested presence. `status` alone is the machine discriminator the
// queue reads to pick converged/continue/escalate (below).
// The shared open schema preserves role-specific prose without adding another contract.
export const judgeVerdictSchema = reviewSubmissionSchema;

export type JudgeRoleDependencies = {
  loadSoul(): Promise<string>;
};

export function createJudgeRoleRuntime(
  pi: RoleHost,
  dependencies: JudgeRoleDependencies,
): { activate(): Promise<void> } {
  let soul: string | undefined;
  let lifecycleRegistered = false;

  return {
    async activate() {
      soul = (await dependencies.loadSoul()).trim();
      if (soul.length === 0) throw new Error("Judge soul is empty");
      if (!lifecycleRegistered) {
        lifecycleRegistered = true;
        registerFiledSubmissionTool(pi, roleSubmissionDeclaration("judge"), {
          readyError: () => (soul === undefined ? "大理寺职分未装载" : undefined),
        });
        pi.on("before_agent_start", (event) => {
          if (soul === undefined) throw new Error("大理寺职分未装载");
          return {
            systemPrompt:
              `${event.systemPrompt}\n\n<judge_soul>\n${soul}\n</judge_soul>`,
          };
        });
      }
    },
  };
}