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");
  auditProject!.projectUnderstanding = {
    projectGoal: "深拉兜底目标",
    currentProgress: "深拉兜底进度",
    technicalArchitecture: "深拉兜底架构",
    currentBlockers: "深拉兜底阻塞",
    recommendedNextStep: "深拉兜底下一步",
    sourceTaskId: "task-deep-pull",
    updatedAt: "2026-04-04T18:00:00+08:00",
    sourceKind: "thread_sync",
  };
  state.threadStatusDocuments = [
    {
      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-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("最近进展事件："));
  assert.ok(queuedTask?.executionPrompt.includes("最近进展事件摘要"));
  assert.ok(queuedTask?.executionPrompt.includes("关键时刻深拉线程兜底："));
  assert.ok(queuedTask?.executionPrompt.includes("深拉兜底目标"));

  assert.ok(
    queuedTask?.executionPrompt.indexOf("管理员全局主提示词：") <
      queuedTask.executionPrompt.indexOf("用户私有主提示词：") &&
      queuedTask.executionPrompt.indexOf("用户私有主提示词：") <
        queuedTask.executionPrompt.indexOf("当前对话附加提示词：") &&
      queuedTask.executionPrompt.indexOf("当前对话附加提示词：") <
        queuedTask.executionPrompt.indexOf("项目记忆：") &&
      queuedTask.executionPrompt.indexOf("项目记忆：") <
        queuedTask.executionPrompt.indexOf("当前消息：") &&
      queuedTask.executionPrompt.indexOf("当前消息：") <
        queuedTask.executionPrompt.indexOf("当前对话覆盖：") &&
      queuedTask.executionPrompt.indexOf("当前对话覆盖：") <
        queuedTask.executionPrompt.indexOf("线程状态文档：") &&
      queuedTask.executionPrompt.indexOf("线程状态文档：") <
        queuedTask.executionPrompt.indexOf("最近进展事件：") &&
      queuedTask.executionPrompt.indexOf("最近进展事件：") <
        queuedTask.executionPrompt.indexOf("关键时刻深拉线程兜底："),
  );
});