import assert from "node:assert/strict";
import test from "node:test";

import {
  applyThinkingSectionProgress,
  consultationComposePrompt,
  consultationContinuePrompt,
  consultationReportHeadings,
  consultationSectionPrompt,
  consultationSpokenHeadingRule,
  dailyConsultationThinkingPlan,
  DAILY_HEADING,
  natalConsultationThinkingPlan,
  REPORT_HEADING,
  splitAnswerByHeadings,
  thinkPlanSteps,
  visibleThinkingSteps,
  windowConsultationThinkingPlan,
} from "../src/lib/consultation-thinking-plan.ts";

test("multi-domain thinking plan uses Chinese product labels and hides tool ids", () => {
  const sections = natalConsultationThinkingPlan({
    domains: ["career", "wealth"],
    requiredBlocks: [
      "raw_structure",
      "raman_six_step",
      "yoga_table",
      "timing",
      "synthesis",
      "technique_audit_table",
      "modern_wrap",
    ],
    mustUseLayers: ["D1", "run-jyotish-consultation", "D10", "skill_read"],
  });
  const encoded = JSON.stringify(sections);
  assert.deepEqual(sections.map((section) => section.heading), [
    REPORT_HEADING.question,
    REPORT_HEADING.support,
    REPORT_HEADING.timing,
    REPORT_HEADING.action,
  ]);
  assert.ok(sections.some((section) => section.id === "support"));
  assert.ok(JSON.stringify(sections).includes("事业"));
  assert.equal(encoded.includes("run-jyotish"), false);
  assert.equal(encoded.includes("skill_read"), false);
  assert.ok(sections.some((section) => section.steps.some((step) => step.label.includes("D1"))));
});

test("window thinking plan stays on stable-layer copy", () => {
  const [section] = windowConsultationThinkingPlan();
  assert.equal(section?.heading, REPORT_HEADING.question);
  assert.equal(JSON.stringify(section).includes("run-jyotish"), false);
});

test("visible steps collapse extras into a remaining count", () => {
  const steps = Array.from({ length: 6 }, (_, index) => ({
    id: `s${index}`,
    label: `步骤 ${index + 1}`,
    status: "pending" as const,
  }));
  const visible = visibleThinkingSteps(steps);
  assert.equal(visible.visible.length, 4);
  assert.equal(visible.hiddenCount, 2);
});

test("answer split follows the required heading order", () => {
  const headings = consultationReportHeadings(["career", "wealth"]);
  const text = [
    `## ${REPORT_HEADING.question}`,
    "外松内紧。",
    `## ${REPORT_HEADING.support}`,
    "事业宫被土星压着。",
    `## ${REPORT_HEADING.timing}`,
    "这轮大运还没过完。",
    `## ${REPORT_HEADING.action}`,
    "这周只把合同条款写清楚。",
  ].join("\n");
  const { slices } = splitAnswerByHeadings(text, headings);
  assert.match(slices[REPORT_HEADING.question] ?? "", /外松内紧/);
  assert.match(slices[REPORT_HEADING.support] ?? "", /土星/);
  assert.match(slices[REPORT_HEADING.timing] ?? "", /大运/);
  assert.match(slices[REPORT_HEADING.action] ?? "", /合同条款/);
});

test("settled answers mark leftover thinking sections done", () => {
  const sections = natalConsultationThinkingPlan({ domains: ["career"] });
  const progressed = applyThinkingSectionProgress(
    sections,
    "今日行运已经写完，但没有领域二级标题。",
    { settled: true },
  );
  assert.equal(progressed.every((section) => section.steps.every((step) => step.status === "done")), true);
});

test("progress marks the current heading active and completed ones done", () => {
  const sections = natalConsultationThinkingPlan({ domains: ["career"] });
  const progressed = applyThinkingSectionProgress(sections, `## ${REPORT_HEADING.question}\n外松内紧。\n`);
  assert.equal(progressed[0]?.steps[0]?.status, "done");
  assert.equal(progressed[1]?.steps[0]?.status, "active");
});

test("continue prompt asks to resume after the last complete heading", () => {
  const prompt = consultationContinuePrompt("## 事业\n方向稳定。\n");
  assert.match(prompt, /最后一个完整标题是：## 事业/);
  assert.match(prompt, /不要重复已写出的段落/);
  assert.match(prompt, /不要写思考过程清单/);
});

test("natal spoken answers must use markdown lists for parallel points", () => {
  assert.match(consultationSpokenHeadingRule("natal"), /Markdown bullet lists/);
});

test("think plan flattens natal sections into v2 steps", () => {
  const steps = thinkPlanSteps(natalConsultationThinkingPlan({
    domains: ["career", "wealth"],
  }));
  assert.deepEqual(steps.map((step) => step.id), ["question", "support", "timing", "action"]);
});

test("section prompt asks for one heading and forbids another calculation", () => {
  const prompt = consultationSectionPrompt("事业", `## ${REPORT_HEADING.question}\n外松内紧。\n`);
  assert.match(prompt, /只写这一个二级标题及其正文：## 事业/);
  assert.match(prompt, /不要再调用排盘工具/);
  assert.match(consultationComposePrompt(), /不要写「统一参数与原始结构」/);
});

test("daily thinking plan is three sections without a close heading", () => {
  const plan = dailyConsultationThinkingPlan();
  assert.deepEqual(plan.map((section) => section.heading), [
    DAILY_HEADING.trend,
    DAILY_HEADING.actAvoid,
    DAILY_HEADING.action,
  ]);
  assert.equal(plan.at(-1)?.title, "一个行动 + 边界句");
  assert.equal(plan.some((section) => section.id === "close"), false);
  assert.deepEqual(
    thinkPlanSteps(plan).map((step) => step.id),
    ["daily-trend", "daily-act-avoid", "daily-action"],
  );
  const written = [
    `## ${DAILY_HEADING.trend}\n今日宜把已谈妥的细节落成文字。`,
    `## ${DAILY_HEADING.actAvoid}\n适合推进合同核对，避开新开战场。`,
    `## ${DAILY_HEADING.action}\n今晚只做一件：把待确认条款列成清单。探索性日提示，不是确定预测。`,
  ].join("\n\n");
  const next = applyThinkingSectionProgress(plan, written);
  assert.equal(next.every((section) => section.steps.every((step) => step.status === "done")), true);
  assert.equal(written.startsWith(`## ${DAILY_HEADING.trend}`), true);
  assert.match(consultationSpokenHeadingRule("daily"), /今日趋势/);
});

test("empty-retry section prompt asks to write the heading directly", () => {
  const prompt = consultationSectionPrompt("今日趋势", "", { reason: "empty-retry" });
  assert.match(prompt, /上一段没有输出正文，请直接写这一节/);
  assert.match(prompt, /只写这一个二级标题及其正文：## 今日趋势/);
});