import type { Page } from "@playwright/test";
import { mockWorkbenchContext } from "./mock-workbench-context";

type WorkOrderPhase = "execution" | "reply";

export async function mockFlatWorkOrder(page: Page, phase: WorkOrderPhase) {
  const parentTitle = phase === "execution"
    ? "演示工单：北辰分区阀门开度调整"
    : "演示工单：泵站边界复核复令";
  const parentStatus = phase === "execution" ? "running" : "awaiting_reply";
  const subtaskTitles = phase === "execution"
    ? ["上传阀门操作前后照片", "记录最终开度", "复核北辰末端压力不低于 0.22MPa"]
    : ["上传泵站运行截图", "填写出水压力复核结果", "确认模型边界已同步更新"];
  const scheduledAt = shanghaiTimestamp(10, 30);
  const workOrderId = `mock-${phase}-work-order`;

  await mockWorkbenchContext(page, {
    availability: "not_configured",
    source: "none",
    scenario_availability: "not_configured",
    artifact_availability: "not_configured",
    schedule_availability: "not_configured",
    work_order_availability: "ready",
    alert_availability: "not_configured",
    generated_at: scheduledAt,
    conditions: [],
    alerts: [],
    scenarios: [],
    latest_analysis: null,
    work_orders: [{
      id: workOrderId,
      code: phase === "execution" ? "DEMO-EXEC-001" : "DEMO-REPLY-001",
      scheduled_at: scheduledAt,
      title: parentTitle,
      summary: phase === "execution"
        ? "现场调整边界阀门并逐项复核。"
        : "完成泵站边界复核并等待调度复令。",
      status: parentStatus,
      risk_level: phase === "execution" ? "attention" : "normal",
      updated_at: scheduledAt,
      detail: "浏览器测试使用的后端工单投影。",
      recommendation: "按回执要求完成并回传。",
      duration_minutes: 45,
      source: "生产工单系统",
      dispatcher: "调度中心",
      priority: phase === "execution" ? "urgent" : "normal",
      location: phase === "execution" ? "北辰分区边界阀门 BV-07" : "北辰二级泵站 P-02",
      assignee: phase === "execution" ? "北辰抢修二组" : "泵站运行班组",
      feedback_requirements: subtaskTitles,
      activity: [{ sequence: 1, occurred_at: scheduledAt, actor: "调度中心", kind: "created",
        title: "演示工单创建", detail: "预约任务已创建。" }]
    }]
  });
}

function shanghaiTimestamp(hour: number, minute: number) {
  const target = new Date();
  const parts = new Intl.DateTimeFormat("en-CA", {
    timeZone: "Asia/Shanghai",
    year: "numeric",
    month: "2-digit",
    day: "2-digit"
  }).formatToParts(target);
  const get = (type: Intl.DateTimeFormatPartTypes) =>
    parts.find((part) => part.type === type)?.value ?? "";
  const normalizedHour = hour + Math.floor(minute / 60);
  return `${get("year")}-${get("month")}-${get("day")}T${String(normalizedHour).padStart(2, "0")}:${String(minute % 60).padStart(2, "0")}:00+08:00`;
}