import test from "node:test";
import assert from "node:assert/strict";
import os from "node:os";
import path from "node:path";
import { mkdtemp, rm } from "node:fs/promises";

let runtimeRoot = "";
let saveAiAccount: (typeof import("../src/lib/boss-data"))["saveAiAccount"];
let updateProjectAgentControls: (typeof import("../src/lib/boss-data"))["updateProjectAgentControls"];
let updateMasterAgentPromptPolicy: (typeof import("../src/lib/boss-data"))["updateMasterAgentPromptPolicy"];
let updateUserMasterPrompt: (typeof import("../src/lib/boss-data"))["updateUserMasterPrompt"];
let resolveMasterAgentExecutionConfig: (typeof import("../src/lib/boss-master-agent"))["resolveMasterAgentExecutionConfig"];

async function setup() {
  if (runtimeRoot) return;

  runtimeRoot = await mkdtemp(path.join(os.tmpdir(), "boss-master-agent-config-"));
  process.env.BOSS_RUNTIME_ROOT = runtimeRoot;
  process.env.BOSS_STATE_FILE = path.join(runtimeRoot, "boss-state.json");

  const [data, masterAgent] = await Promise.all([
    import("../src/lib/boss-data.ts"),
    import("../src/lib/boss-master-agent.ts"),
  ]);

  saveAiAccount = data.saveAiAccount;
  updateProjectAgentControls = data.updateProjectAgentControls;
  updateMasterAgentPromptPolicy = data.updateMasterAgentPromptPolicy;
  updateUserMasterPrompt = data.updateUserMasterPrompt;
  resolveMasterAgentExecutionConfig = masterAgent.resolveMasterAgentExecutionConfig;
}

test.after(async () => {
  if (runtimeRoot) {
    await rm(runtimeRoot, { recursive: true, force: true });
  }
});

test("当前对话 override 优先于主控账号默认值", async () => {
  await setup();

  await saveAiAccount({
    accountId: "master-codex-primary",
    label: "主 GPT",
    role: "primary",
    provider: "master_codex_node",
    displayName: "Mac 上的 Master Codex Node",
    nodeId: "local-codex-node",
    nodeLabel: "本机 Codex",
    model: "gpt-4.1-mini",
    enabled: true,
    setActive: true,
    loginStatusNote: "通过绑定的 Master Codex Node 对话。",
  });

  await updateProjectAgentControls("master-agent", {
    modelOverride: "gpt-5.4",
    reasoningEffortOverride: "high",
  });

  assert.equal(typeof resolveMasterAgentExecutionConfig, "function");

  const resolved = await resolveMasterAgentExecutionConfig("master-agent");

  assert.equal(resolved.model, "gpt-5.4");
  assert.equal(resolved.reasoningEffort, "high");
  assert.equal(resolved.account.accountId, "master-codex-primary");
  assert.equal(resolved.account.model, "gpt-4.1-mini");
});

test("主 Agent 执行配置会合成管理员提示词、用户提示词和当前对话提示词", async () => {
  await setup();

  await saveAiAccount({
    accountId: "master-codex-primary",
    label: "主 GPT",
    role: "primary",
    provider: "master_codex_node",
    displayName: "Mac 上的 Master Codex Node",
    nodeId: "local-codex-node",
    nodeLabel: "本机 Codex",
    model: "gpt-4.1-mini",
    enabled: true,
    setActive: true,
    loginStatusNote: "通过绑定的 Master Codex Node 对话。",
  });

  await updateMasterAgentPromptPolicy({
    globalPrompt: "全局主提示词",
    updatedBy: "17600003315",
  });
  await updateUserMasterPrompt("17600003315", "用户私有主提示词");
  await updateProjectAgentControls("master-agent", {
    modelOverride: "gpt-5.4",
    reasoningEffortOverride: "high",
    promptOverride: "当前对话提示词",
  });

  const resolved = await resolveMasterAgentExecutionConfig("master-agent");

  assert.equal(resolved.promptPolicy?.globalPrompt, "全局主提示词");
  assert.equal(resolved.userPrompt?.content, "用户私有主提示词");
  assert.equal(resolved.projectPromptOverride, "当前对话提示词");
  assert.equal(resolved.promptPolicy?.updatedBy, "17600003315");
});