import { describe, expect, it } from "vitest";
import {
  createPagedDocumentPayload,
  defaultExportConfig
} from "@md-to-pdf/core";
import {
  createElectronPdfPrintOptions,
  isAllowedPdfRuntimeUrl,
  parsePagedDocumentPayload
} from "../src/pdf-contract.js";
import {
  parseMarkdownRenderRequest,
  parseThemeId,
  parseThemeResourceUrl
} from "../src/application-contract.js";

const payload = createPagedDocumentPayload({
  document: {
    rendererVersion: 1,
    articleHtml: '<article id="write"><h1>测试</h1></article>',
    bodyHtml: "<h1>测试</h1>",
    metadata: {
      title: "测试",
      author: "",
      subject: "",
      keywords: [],
      language: "zh-CN"
    },
    semanticDocument: {
      schemaVersion: 1,
      titlePolicy: {
        metadataTitle: "suppress",
        firstBodyHeading: "keep"
      },
      regions: []
    },
    features: ["mermaid"],
    warnings: []
  },
  fileName: "测试.md",
  themeCss: "#write { color: black; }",
  exportConfig: defaultExportConfig
});

describe("桌面 PDF 载荷", () => {
  it("校验并保留合法的共享分页载荷", () => {
    expect(parsePagedDocumentPayload(payload)).toEqual(payload);
  });

  it("拒绝无效功能标识", () => {
    expect(() =>
      parsePagedDocumentPayload({
        ...payload,
        features: ["remote-script"]
      })
    ).toThrow();
  });

  it("映射为与 Playwright 对齐的 Electron 打印参数", () => {
    expect(
      createElectronPdfPrintOptions(payload, {
        pageCount: 3,
        echartsErrors: [],
        mermaidErrors: [],
        timings: {
          setupMs: 0,
          echartsMs: 0,
          echartsFitMs: 0,
          mermaidMs: 0,
          mermaidFitMs: 0,
          mermaidConversionMs: 0,
          resourceWaitMs: 0,
          paginationMs: 0,
          finalizeMs: 0,
          totalMs: 0
        }
      })
    ).toEqual({
      displayHeaderFooter: false,
      margins: { top: 0, right: 0, bottom: 0, left: 0 },
      pageRanges: "1-3",
      preferCSSPageSize: true,
      printBackground: true,
      scale: 1
    });
  });
});

describe("桌面 PDF 网络边界", () => {
  const renderUrl =
    "http://localhost:5173/preview-frame.html?target=pdf";

  it("允许同源和内嵌资源", () => {
    expect(
      isAllowedPdfRuntimeUrl(
        "http://localhost:5173/assets/runtime.js",
        renderUrl
      )
    ).toBe(true);
    expect(isAllowedPdfRuntimeUrl("data:image/png;base64,AA", renderUrl))
      .toBe(true);
    expect(
      isAllowedPdfRuntimeUrl(
        "mdpdf://theme/typora-like/font.woff2",
        renderUrl
      )
    ).toBe(true);
  });

  it("拒绝外部网络资源", () => {
    expect(
      isAllowedPdfRuntimeUrl("https://example.com/image.png", renderUrl)
    ).toBe(false);
  });
});

describe("桌面应用服务 IPC 载荷", () => {
  it("只提取 Markdown 渲染所需字段", () => {
    expect(
      parseMarkdownRenderRequest({
        markdown: "# 测试",
        language: "zh-CN",
        ignored: true
      })
    ).toEqual({
      markdown: "# 测试",
      language: "zh-CN"
    });
  });

  it("校验主题 ID 与主题资源地址", () => {
    expect(parseThemeId("typora-like")).toBe("typora-like");
    expect(
      parseThemeResourceUrl(
        "mdpdf://theme/typora-like/fonts/test.woff2"
      )
    ).toEqual({
      themeId: "typora-like",
      assetPath: "fonts/test.woff2"
    });
    expect(() => parseThemeId("../outside")).toThrow("主题 ID 无效");
    expect(
      parseThemeResourceUrl("mdpdf://bundle/index.html")
    ).toBeUndefined();
  });
});