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 readState: (typeof import("../src/lib/boss-data"))["readState"];
let writeState: (typeof import("../src/lib/boss-data"))["writeState"];
let saveAiAccount: (typeof import("../src/lib/boss-data"))["saveAiAccount"];
let updateMasterAgentPromptPolicy: (typeof import("../src/lib/boss-data"))["updateMasterAgentPromptPolicy"];
let updateUserMasterPrompt: (typeof import("../src/lib/boss-data"))["updateUserMasterPrompt"];
let createUserMasterMemory: (typeof import("../src/lib/boss-data"))["createUserMasterMemory"];
let updateProjectAgentControls: (typeof import("../src/lib/boss-data"))["updateProjectAgentControls"];
let resolveMasterAgentExecutionConfig: (typeof import("../src/lib/boss-master-agent"))["resolveMasterAgentExecutionConfig"];
let replyToMasterAgentUserMessage: (typeof import("../src/lib/boss-master-agent"))["replyToMasterAgentUserMessage"];

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

  runtimeRoot = await mkdtemp(path.join(os.tmpdir(), "boss-master-agent-thread-status-"));
  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"),
  ]);

  readState = data.readState;
  writeState = data.writeState;
  saveAiAccount = data.saveAiAccount;
  updateMasterAgentPromptPolicy = data.updateMasterAgentPromptPolicy;
  updateUserMasterPrompt = data.updateUserMasterPrompt;
  createUserMasterMemory = data.createUserMasterMemory;
  updateProjectAgentControls = data.updateProjectAgentControls;
  resolveMasterAgentExecutionConfig = masterAgent.resolveMasterAgentExecutionConfig;
  replyToMasterAgentUserMessage = masterAgent.replyToMasterAgentUserMessage;
}

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

test("主 Agent 执行 prompt 命中线程时只读取相关状态文档和最近进展事件，不再常态注入深拉兜底", async () => {
  await setup();

  await saveAiAccount({
    accountId: "master-codex-primary",
    label: "主 GPT",
    role: "primary",
    provider: "master_codex_node",
    displayName: "17600003315 · Master Codex Node",
    nodeId: "mac-studio",
    nodeLabel: "Mac Studio",
    enabled: true,
    setActive: true,
    status: "ready",
    loginStatusNote: "主节点可用。",
  });
  await updateMasterAgentPromptPolicy({
    globalPrompt: "管理员全局主提示词",
    updatedBy: "17600003315",
  });
  await updateUserMasterPrompt("17600003315", "用户私有主提示词");
  await updateProjectAgentControls("master-agent", {
    promptOverride: "当前对话提示词",
  });
  await createUserMasterMemory({
    account: "17600003315",
    scope: "project",
    projectId: "master-agent",
    title: "项目记忆",
    content: "项目记忆正文",
    memoryType: "project_progress",
    tags: ["线程状态"],
  });
  await createUserMasterMemory({
    account: "master-codex-primary",
    scope: "project",
    projectId: "master-agent",
    title: "项目记忆",
    content: "项目记忆正文",
    memoryType: "project_progress",
    tags: ["线程状态"],
  });

  const state = await readState();
  const auditProject = state.projects.find((project) => project.id === "audit-collab");
  assert.ok(auditProject, "expected seeded audit-collab project");
  const masterProject = state.projects.find((project) => project.id === "master-agent");
  assert.ok(masterProject, "expected seeded master-agent project");
  auditProject!.projectUnderstanding = {
    projectGoal: "深拉兜底目标",
    currentProgress: "深拉兜底进度",
    technicalArchitecture: "深拉兜底架构",
    currentBlockers: "深拉兜底阻塞",
    recommendedNextStep: "深拉兜底下一步",
    sourceTaskId: "task-deep-pull",
    updatedAt: "2026-04-04T18:00:00+08:00",
    sourceKind: "thread_sync",
  };
  masterProject!.projectUnderstanding = {
    projectGoal: "主 Agent 旧目标",
    currentProgress: "主 Agent 旧进度",
    technicalArchitecture: "主 Agent 旧架构",
    currentBlockers: "主 Agent 旧阻塞",
    recommendedNextStep: "主 Agent 旧下一步",
    sourceTaskId: "task-master-legacy",
    updatedAt: "2026-04-04T17:50:00+08:00",
    sourceKind: "thread_sync",
  };
  state.threadStatusDocuments = [
    {
      documentId: "thread-status-doc-master",
      projectId: "master-agent",
      threadId: "thread-master-main",
      threadDisplayName: "主 Agent 汇总",
      folderName: "主控线程",
      deviceId: "mac-studio",
      projectGoal: "主 Agent 额外状态目标",
      currentPhase: "主 Agent 额外阶段",
      currentProgress: "主 Agent 额外进度",
      technicalArchitecture: "主 Agent 额外架构",
      currentBlockers: "主 Agent 额外阻塞",
      recommendedNextStep: "主 Agent 额外下一步",
      keyFiles: ["src/lib/boss-master-agent.ts"],
      keyCommands: ["npm run lint"],
      updatedAt: "2026-04-04T18:03:00+08:00",
      sourceTaskId: "task-master-status",
      sourceKind: "incremental_sync",
    },
    {
      documentId: "thread-status-doc-1",
      projectId: "audit-collab",
      threadId: "thread-audit-chief",
      threadDisplayName: "审计对话",
      folderName: "审计群聊",
      deviceId: "mac-studio",
      projectGoal: "线程状态目标",
      currentPhase: "线程状态阶段",
      currentProgress: "线程状态进度",
      technicalArchitecture: "线程状态架构",
      currentBlockers: "线程状态阻塞",
      recommendedNextStep: "线程状态下一步",
      keyFiles: ["src/lib/boss-master-agent.ts"],
      keyCommands: ["npm run build"],
      updatedAt: "2026-04-04T18:01:00+08:00",
      sourceTaskId: "task-thread-status",
      sourceKind: "incremental_sync",
    },
  ];
  state.threadProgressEvents = [
    {
      eventId: "thread-progress-event-master",
      projectId: "master-agent",
      threadId: "thread-master-main",
      threadDisplayName: "主 Agent 汇总",
      deviceId: "mac-studio",
      eventType: "progress_updated",
      summary: "主 Agent 额外进展摘要",
      phase: "主 Agent 额外阶段",
      blockerDelta: "主 Agent 额外阻塞",
      nextStepDelta: "主 Agent 额外下一步",
      createdAt: "2026-04-04T18:03:30+08:00",
      sourceTaskId: "task-master-progress",
    },
    {
      eventId: "thread-progress-event-1",
      projectId: "audit-collab",
      threadId: "thread-audit-chief",
      threadDisplayName: "审计对话",
      deviceId: "mac-studio",
      eventType: "progress_updated",
      summary: "最近进展事件摘要",
      phase: "线程状态阶段",
      blockerDelta: "线程状态阻塞",
      nextStepDelta: "线程状态下一步",
      createdAt: "2026-04-04T18:02:00+08:00",
      sourceTaskId: "task-thread-progress",
    },
  ];
  await writeState(state);

  const resolved = await resolveMasterAgentExecutionConfig(
    "master-agent",
    "17600003315",
    "审计对话，请继续推进线程状态同步",
  );
  assert.ok(resolved.projectMemories.length > 0);
  assert.equal(resolved.projectMemories[0]?.content, "项目记忆正文");
  assert.ok(resolved.executionPrompt.includes("当前对话提示词"));
  assert.ok(
    resolved.executionPrompt.indexOf("管理员全局主提示词：") <
      resolved.executionPrompt.indexOf("用户私有主提示词：") &&
      resolved.executionPrompt.indexOf("用户私有主提示词：") <
        resolved.executionPrompt.indexOf("当前对话附加提示词：") &&
      resolved.executionPrompt.indexOf("当前对话附加提示词：") <
        resolved.executionPrompt.indexOf("当前消息："),
  );

  const reply = await replyToMasterAgentUserMessage({
    requestText: "审计对话，请继续推进线程状态同步",
    requestedBy: "Boss 超级管理员",
    requestedByAccount: "17600003315",
    mode: "enqueue",
  });
  assert.equal(reply.ok, true);
  assert.equal(reply.masterReplyState, "queued");

  const queuedTask = (await readState()).masterAgentTasks.find(
    (task) => task.projectId === "master-agent" && task.requestText === "审计对话，请继续推进线程状态同步",
  );
  assert.ok(queuedTask, "expected master-agent task to be queued");
  assert.ok(queuedTask?.executionPrompt.includes("线程状态文档："));
  assert.ok(queuedTask?.executionPrompt.includes("线程状态目标"));
  assert.ok(!queuedTask?.executionPrompt.includes("主 Agent 额外状态目标"));
  assert.ok(queuedTask?.executionPrompt.includes("最近进展事件："));
  assert.ok(queuedTask?.executionPrompt.includes("最近进展事件摘要"));
  assert.ok(!queuedTask?.executionPrompt.includes("主 Agent 额外进展摘要"));
  assert.ok(!queuedTask?.executionPrompt.includes("深拉兜底目标"));
  assert.ok(!queuedTask?.executionPrompt.includes("关键时刻深拉线程兜底："));
});

test("主 Agent 执行 prompt 在未命中时退回最近活跃项目，且不会常态注入深拉兜底", async () => {
  await setup();

  const state = await readState();
  const auditProject = state.projects.find((project) => project.id === "audit-collab");
  const masterProject = state.projects.find((project) => project.id === "master-agent");
  assert.ok(auditProject, "expected seeded audit-collab project");
  assert.ok(masterProject, "expected seeded master-agent project");

  auditProject!.projectUnderstanding = {
    projectGoal: "审计兜底目标",
    currentProgress: "审计兜底进度",
    technicalArchitecture: "审计兜底架构",
    currentBlockers: "审计兜底阻塞",
    recommendedNextStep: "审计兜底下一步",
    sourceTaskId: "task-audit-fallback",
    updatedAt: "2026-04-04T17:55:00+08:00",
    sourceKind: "thread_sync",
  };
  masterProject!.projectUnderstanding = {
    projectGoal: "最近活跃目标",
    currentProgress: "最近活跃进度",
    technicalArchitecture: "最近活跃架构",
    currentBlockers: "最近活跃阻塞",
    recommendedNextStep: "最近活跃下一步",
    sourceTaskId: "task-master-active",
    updatedAt: "2026-04-04T18:05:00+08:00",
    sourceKind: "thread_sync",
  };
  state.threadStatusDocuments = [
    {
      documentId: "thread-status-doc-audit-fallback",
      projectId: "audit-collab",
      threadId: "thread-audit-chief",
      threadDisplayName: "审计对话",
      folderName: "审计群聊",
      deviceId: "mac-studio",
      projectGoal: "审计兜底状态",
      currentPhase: "审计兜底阶段",
      currentProgress: "审计兜底进度",
      technicalArchitecture: "审计兜底架构",
      currentBlockers: "审计兜底阻塞",
      recommendedNextStep: "审计兜底下一步",
      keyFiles: ["src/lib/boss-master-agent.ts"],
      keyCommands: ["npm run build"],
      updatedAt: "2026-04-04T17:56:00+08:00",
      sourceTaskId: "task-audit-status",
      sourceKind: "incremental_sync",
    },
    {
      documentId: "thread-status-doc-master-fallback",
      projectId: "master-agent",
      threadId: "thread-master-main",
      threadDisplayName: "主 Agent 汇总",
      folderName: "主控线程",
      deviceId: "mac-studio",
      projectGoal: "最近活跃状态",
      currentPhase: "最近活跃阶段",
      currentProgress: "最近活跃进度",
      technicalArchitecture: "最近活跃架构",
      currentBlockers: "最近活跃阻塞",
      recommendedNextStep: "最近活跃下一步",
      keyFiles: ["src/lib/boss-data.ts"],
      keyCommands: ["npm run lint"],
      updatedAt: "2026-04-04T18:06:00+08:00",
      sourceTaskId: "task-master-status-fallback",
      sourceKind: "incremental_sync",
    },
  ];
  state.threadProgressEvents = [
    {
      eventId: "thread-progress-event-audit-fallback",
      projectId: "audit-collab",
      threadId: "thread-audit-chief",
      threadDisplayName: "审计对话",
      deviceId: "mac-studio",
      eventType: "progress_updated",
      summary: "审计兜底进展摘要",
      phase: "审计兜底阶段",
      blockerDelta: "审计兜底阻塞",
      nextStepDelta: "审计兜底下一步",
      createdAt: "2026-04-04T17:56:30+08:00",
      sourceTaskId: "task-audit-progress",
    },
    {
      eventId: "thread-progress-event-master-fallback",
      projectId: "master-agent",
      threadId: "thread-master-main",
      threadDisplayName: "主 Agent 汇总",
      deviceId: "mac-studio",
      eventType: "progress_updated",
      summary: "最近活跃进展摘要",
      phase: "最近活跃阶段",
      blockerDelta: "最近活跃阻塞",
      nextStepDelta: "最近活跃下一步",
      createdAt: "2026-04-04T18:06:30+08:00",
      sourceTaskId: "task-master-progress-fallback",
    },
  ];
  await writeState(state);

  const reply = await replyToMasterAgentUserMessage({
    requestText: "请继续推进线程状态同步（仅深拉兜底）",
    requestedBy: "Boss 超级管理员",
    requestedByAccount: "17600003315",
    mode: "enqueue",
  });
  assert.equal(reply.ok, true);

  const queuedTask = (await readState()).masterAgentTasks.find(
    (task) => task.projectId === "master-agent" && task.requestText === "请继续推进线程状态同步（仅深拉兜底）",
  );
  assert.ok(queuedTask, "expected master-agent task to be queued");
  assert.ok(queuedTask?.executionPrompt.includes("最近活跃状态"));
  assert.ok(queuedTask?.executionPrompt.includes("最近活跃进展摘要"));
  assert.ok(!queuedTask?.executionPrompt.includes("关键时刻深拉线程兜底："));
});

test("主 Agent 执行 prompt 在没有线程状态文档和进展事件时才会注入深拉兜底", async () => {
  await setup();

  const state = await readState();
  const auditProject = state.projects.find((project) => project.id === "audit-collab");
  assert.ok(auditProject, "expected seeded audit-collab project");
  auditProject!.projectUnderstanding = {
    projectGoal: "深拉兜底目标",
    currentProgress: "深拉兜底进度",
    technicalArchitecture: "深拉兜底架构",
    currentBlockers: "深拉兜底阻塞",
    recommendedNextStep: "深拉兜底下一步",
    sourceTaskId: "task-deep-pull",
    updatedAt: "2026-04-04T18:00:00+08:00",
    sourceKind: "thread_sync",
  };
  state.threadStatusDocuments = [];
  state.threadProgressEvents = [];
  await writeState(state);

  const reply = await replyToMasterAgentUserMessage({
    requestText: "请继续推进线程状态同步",
    requestedBy: "Boss 超级管理员",
    requestedByAccount: "17600003315",
    mode: "enqueue",
  });
  assert.equal(reply.ok, true);

  const queuedTask = (await readState()).masterAgentTasks.find(
    (task) => task.projectId === "master-agent" && task.requestText === "请继续推进线程状态同步",
  );
  assert.ok(queuedTask, "expected master-agent task to be queued");
  assert.ok(queuedTask?.executionPrompt.includes("关键时刻深拉线程兜底："));
  assert.ok(queuedTask?.executionPrompt.includes("深拉兜底目标"));
});