import { z } from "zod";
import { supplyLayerSchema } from "./task-presentation.js";
import { answerUiBlockSchema } from "../chat/answer-ui.js";

const text = z.string().trim().min(1);
export function mermaidDeclaresNodeId(source: string, nodeId: string): boolean {
  const escaped = nodeId.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
  return new RegExp(
    `(?:^|[^A-Za-z0-9_-])${escaped}(?=\\s*(?:[\\[({@;]|--|==|-\\.|$))`,
    "m",
  ).test(source);
}

export const investigationDiagramSchema = z.object({
  title: text,
  mermaid: text.max(20000).refine((source) =>
    /^\s*(?:flowchart|graph)\s+(?:TD|TB|BT|LR|RL)\b/.test(source)
    && !source.includes("```") && !source.includes("%%{") && !/(?:^|[;\n])\s*click\b/.test(source),
  "调查图只接受不含指令或回调的 Mermaid 流程描述"),
  node_refs: z.array(z.object({
    node_id: z.string().regex(/^[A-Za-z_][A-Za-z0-9_-]{0,79}$/),
    label: text,
    evidence_refs: z.array(text).default([]),
    action_refs: z.array(z.string().uuid()).default([]),
    status: z.enum(["pending", "running", "completed", "waiting", "waiting_human", "failed", "skipped", "cancelled"]).nullish().describe("Evidence-confirmed node state; waiting means prerequisites/external/system result, waiting_human requires manual authorization/approval/feedback. Unknown stays omitted/null; host controls colors."),
    map_targets: z.array(z.object({
      source_id: supplyLayerSchema,
      feature_id: text.max(128),
    }).strict()).max(100).default([]),
  }).strict()).max(100).default([]),
}).strict().superRefine((diagram, ctx) => {
  if (new Set(diagram.node_refs.map((node) => node.node_id)).size !== diagram.node_refs.length) {
    ctx.addIssue({ code: z.ZodIssueCode.custom, message: "调查图节点引用重复" });
  }
  diagram.node_refs.forEach((node, index) => {
    if (!mermaidDeclaresNodeId(diagram.mermaid, node.node_id)) {
      ctx.addIssue({
        code: z.ZodIssueCode.custom,
        path: ["node_refs", index, "node_id"],
        message: `调查图节点引用 ${node.node_id} 未在 Mermaid 源码中声明`,
      });
    }
  });
});
export const taskProgressSchema = z.object({
  summary: text.max(20000),
  diagram: investigationDiagramSchema.nullable().default(null),
}).strict();

export type TaskProgress = z.infer<typeof taskProgressSchema>;
const dataGapSchema = z.object({
  gap_id: text,
  node_id: text.nullish(),
  missing_data: text,
  scope: text,
  impact: text,
  required_data: text,
  evidence_refs: z.array(text).min(1),
  blocking: z.boolean()
}).strict();

/** New report writes require explicit business outcomes; storage reads remain compatible. */
export const taskReportSchema = z.object({
  title: text,
  current_conclusion: text,
  conclusion_boundary: text,
  evidence_refs: z.array(text).default([]),
  recommendations: z.array(text).default([]),
  pending_items: z.array(text).default([]),
  stage: z.enum(["interim", "final"]),
  outcome: z.enum(["concluded", "insufficient_data"]),
  stop_reason: text.nullable().default(null),
  data_gaps: z.array(dataGapSchema),
  diagram: investigationDiagramSchema.nullish(),
  ui_blocks: z.array(z.object({
    block: answerUiBlockSchema,
    evidence_refs: z.array(text.max(128)).min(1).max(100),
  }).strict()).max(8).optional(),
}).strict().superRefine((report, ctx) => {
  const invalid = (message: string) => ctx.addIssue({ code: z.ZodIssueCode.custom, message });
  const blocks = report.ui_blocks ?? [];
  if (Buffer.byteLength(JSON.stringify(blocks), "utf8") > 400000) invalid("报告交互内容超过 400 KB");
  if (new Set(blocks.map((item) => item.block.block_id)).size !== blocks.length) invalid("报告界面块编号重复");
  for (const item of blocks) {
    if (item.block.phase === "process") invalid("正式报告只能包含 final 界面块；旧版未标注阶段的块保持兼容");
    if (item.evidence_refs.some((ref) => !report.evidence_refs.includes(ref))) invalid("界面依据必须属于报告依据");
    for (const node of Object.values(item.block.spec.elements)) {
      if (node.type === "Evidence" && !item.evidence_refs.includes(String(node.props.evidence_id ?? "")))
        invalid("证据组件必须引用界面块的已有依据");
    }
  }
  if (new Set(report.data_gaps.map((gap) => gap.gap_id)).size !== report.data_gaps.length) invalid("数据缺口编号重复");
  const blocking = report.data_gaps.some((gap) => gap.blocking);
  if (report.outcome === "insufficient_data") {
    if (report.stage !== "final" || !report.stop_reason || !blocking) invalid("证据不足必须提供最终报告、结束原因和关键缺口");
    if (report.recommendations.length || report.pending_items.length) invalid("证据不足退出后不再生成后续处置建议或待办");
  } else if (report.stop_reason != null || blocking) invalid("已形成结论不能包含结束原因或阻塞缺口");
});

export type TaskReport = z.infer<typeof taskReportSchema>;