import { describe, expect, it } from "vitest";

import { resolveJavaScriptSourceMapReference } from "./javascriptSourceMapPaths.js";
import {
  createJavaScriptApplicationGraph,
  createJavaScriptApplicationNode,
} from "./javascriptApplicationGraph.js";
import { createHistoricalSourceGraph } from "../referenceSourceGraph.js";
import { compareUnicodeCodePoints } from "../unicodeCodePointOrder.js";
import { compareSourceToBundle } from "./sourceToBundleComparison.js";
import {
  SOURCE_TO_BUNDLE_SIGNAL_WEIGHTS,
  sourceToBundleComparisonResultSchema,
} from "./sourceToBundleComparisonSchemas.js";
import { artifactEvidence } from "./javascriptApplicationGraph.fixture.js";

const HASH = {
  artifact: "1".repeat(64),
  unchanged: "2".repeat(64),
  modified: "3".repeat(64),
  removed: "4".repeat(64),
  split: "5".repeat(64),
  merged: "6".repeat(64),
  duplicated: "7".repeat(64),
} as const;
const EVIDENCE_ID = `ev_${"a".repeat(64)}`;

describe("historical source to bundle comparison", () => {
  it("classifies exact, changed, absent, split, and merged mappings stably", () => {
    const reference = historicalGraph("complete");
    const graph = applicationGraph("complete");
    const input = {
      reference,
      application: {
        evidenceId: EVIDENCE_ID,
        rootArtifactSha256: HASH.artifact,
        graph,
      },
    };

    const first = compareSourceToBundle(input);
    const second = compareSourceToBundle(input);

    expect(first).toEqual(second);
    expect(() =>
      sourceToBundleComparisonResultSchema.parse(first),
    ).not.toThrow();
    expect(first.summary).toEqual({
      unchanged: 1,
      modified: 0,
      removed: 1,
      split: 1,
      merged: 2,
      duplicated: 1,
      unknown: 1,
    });
    expect(item(first, "src/unchanged.ts")).toMatchObject({
      status: "unchanged",
      confidence: "exact",
      candidates: [
        expect.objectContaining({
          signals: expect.arrayContaining([
            expect.objectContaining({
              kind: "exact-source-digest",
              weight: 100,
            }),
          ]),
        }),
      ],
    });
    expect(item(first, "src/modified.ts")).toMatchObject({
      status: "unknown",
      confidence: "unknown",
      candidates: [
        expect.objectContaining({
          signals: expect.arrayContaining([
            expect.objectContaining({ kind: "current-path-exact", weight: 60 }),
          ]),
        }),
      ],
    });
    expect(item(first, "src/removed.ts")).toMatchObject({
      status: "removed",
      current_node_ids: [],
    });
    expect(item(first, "src/split.ts")).toMatchObject({
      status: "split",
      confidence: "medium",
      current_node_ids: [expect.any(String), expect.any(String)],
    });
    expect(item(first, "src/merged-a.ts").status).toBe("merged");
    expect(item(first, "src/merged-b.ts").status).toBe("merged");
    expect(item(first, "src/duplicated.ts")).toMatchObject({
      status: "duplicated",
      candidates: [{}, {}],
      confidence: "exact",
      current_node_ids: [expect.any(String), expect.any(String)],
    });
    expect(first.coverage.status).toBe("complete-within-inputs");
  });

  it("keeps absence unknown when either inventory is incomplete", () => {
    const reference = historicalGraph("partial", ["src/removed.ts"]);
    const graph = applicationGraph("partial");
    const result = compareSourceToBundle({
      reference,
      application: {
        evidenceId: EVIDENCE_ID,
        rootArtifactSha256: HASH.artifact,
        graph,
      },
    });

    expect(result.summary).toMatchObject({ removed: 0, unknown: 1 });
    expect(result.items[0]).toMatchObject({
      source_path: "src/removed.ts",
      status: "unknown",
      confidence: "unknown",
    });
    expect(result.coverage.status).toBe("partial");
  });

  it("evaluates all indexed candidates for each source file", () => {
    const result = compareSourceToBundle({
      reference: historicalGraph("complete", ["src/split.ts"]),
      application: {
        evidenceId: EVIDENCE_ID,
        rootArtifactSha256: HASH.artifact,
        graph: applicationGraph("complete"),
      },
    });

    expect(result.items[0]).toMatchObject({ status: "split" });
    expect(result.coverage).toMatchObject({
      status: "complete-within-inputs",
      candidate_evaluations: 2,
    });
  });
});

describe("source to bundle digest changes", () => {
  it("reports a located source with a different digest as modified", () => {
    const node = createJavaScriptApplicationNode({
      kind: "source-module",
      identity: {
        strategy: "source-map-original",
        source_root: null,
        stability: "source-map-exact",
        source_map_sha256: HASH.artifact,
        original_source: "webpack://fixture/./src/modified.ts",
        source_sha256: HASH.unchanged,
      },
      observations: [
        {
          label: "display label",
          source_map_reference: resolveJavaScriptSourceMapReference(
            "webpack://fixture/./src/modified.ts",
            null,
            "dist/main.js.map",
          ),
          properties: {
            content_available: true,
            content_sha256: HASH.unchanged,
            source: "src/removed.ts",
          },
          evidence: artifactEvidence(HASH.artifact, "dist/main.js.map"),
        },
      ],
    });
    const graph = createJavaScriptApplicationGraph({
      schema: "JavaScriptApplicationGraph",
      root_node_ids: [node.node_id],
      nodes: [node],
      edges: [],
      coverage: {
        status: "complete",
        truncated: false,
        omitted_count: 0,
        limits: [],
      },
      limitations: [],
    });
    const result = compareSourceToBundle({
      reference: historicalGraph("complete", [
        "src/modified.ts",
        "src/removed.ts",
      ]),
      application: {
        evidenceId: EVIDENCE_ID,
        rootArtifactSha256: HASH.artifact,
        graph,
      },
    });

    expect(result.summary).toMatchObject({ modified: 1, unknown: 0 });
    expect(item(result, "src/removed.ts").candidates).toEqual([]);
    expect(item(result, "src/modified.ts")).toMatchObject({
      status: "modified",
      confidence: "high",
      current_node_ids: [node.node_id],
    });
  });
});

describe("source to bundle path syntax", () => {
  it.each([
    "src/modified#part.ts",
    "src/modified?part.ts",
    "pkg:app/src/modified#part.ts",
    "pkg:app/src/modified?part.ts",
    "data:/src/modified.ts",
  ])("preserves literal filesystem punctuation in %s", (path) => {
    const node = moduleNode("punctuated", { path });
    const graph = createJavaScriptApplicationGraph({
      schema: "JavaScriptApplicationGraph",
      root_node_ids: [node.node_id],
      nodes: [node],
      edges: [],
      coverage: {
        status: "complete",
        truncated: false,
        omitted_count: 0,
        limits: [],
      },
      limitations: [],
    });
    const result = compareSourceToBundle({
      reference: historicalGraph("complete", [path]),
      application: {
        evidenceId: EVIDENCE_ID,
        rootArtifactSha256: HASH.artifact,
        graph,
      },
    });
    expect(item(result, path)).toMatchObject({
      status: "unknown",
      current_node_ids: [node.node_id],
      candidates: [
        expect.objectContaining({
          signals: expect.arrayContaining([
            expect.objectContaining({ kind: "current-path-exact" }),
          ]),
        }),
      ],
    });
  });

  it.each([
    "https://example.test/src/modified.ts?revision=2#section",
    "webpack://fixture/./src/modified.ts?revision=2#section",
  ])("still strips URL query and fragment from %s", (path) => {
    const node = moduleNode("url", { path });
    const graph = createJavaScriptApplicationGraph({
      schema: "JavaScriptApplicationGraph",
      root_node_ids: [node.node_id],
      nodes: [node],
      edges: [],
      coverage: {
        status: "complete",
        truncated: false,
        omitted_count: 0,
        limits: [],
      },
      limitations: [],
    });
    const result = compareSourceToBundle({
      reference: historicalGraph("complete", ["src/modified.ts"]),
      application: {
        evidenceId: EVIDENCE_ID,
        rootArtifactSha256: HASH.artifact,
        graph,
      },
    });
    expect(item(result, "src/modified.ts")).toMatchObject({
      status: "unknown",
      current_node_ids: [node.node_id],
      candidates: [
        expect.objectContaining({
          signals: expect.arrayContaining([
            expect.objectContaining({ kind: "current-path-suffix" }),
          ]),
        }),
      ],
    });
  });
});

describe("source to bundle complete metadata", () => {
  it("accepts long source metadata and all reported limitations", () => {
    const result = compareSourceToBundle({
      reference: historicalGraph("complete"),
      application: {
        evidenceId: EVIDENCE_ID,
        rootArtifactSha256: HASH.artifact,
        graph: applicationGraph("complete"),
      },
    });
    const item = result.items[0];
    if (item === undefined) throw new Error("Expected a comparison item");

    const parsed = sourceToBundleComparisonResultSchema.parse({
      ...result,
      items: [
        {
          ...item,
          source_path: "p".repeat(4_097),
          source_language: "l".repeat(101),
          limitations: Array.from(
            { length: 101 },
            (_, index) => `gap-${index}`,
          ),
          candidates: item.candidates.map((candidate) => ({
            ...candidate,
            signals: Array.from({ length: 7 }, (_, index) => ({
              kind: SOURCE_TO_BUNDLE_SIGNAL_WEIGHTS[index % 6]?.[0],
              weight: SOURCE_TO_BUNDLE_SIGNAL_WEIGHTS[index % 6]?.[1],
              source_value: "v".repeat(4_097),
              current_values: ["v".repeat(4_097)],
            })),
          })),
        },
      ],
      limitations: Array.from({ length: 1_001 }, (_, index) => `gap-${index}`),
    });

    expect(parsed.items[0]?.limitations).toHaveLength(101);
    expect(parsed.limitations).toHaveLength(1_001);
  });
});

const historicalGraph = (
  inventoryState: "complete" | "partial",
  paths = [
    "src/duplicated.ts",
    "src/merged-a.ts",
    "src/merged-b.ts",
    "src/modified.ts",
    "src/removed.ts",
    "src/split.ts",
    "src/unchanged.ts",
  ],
) =>
  createHistoricalSourceGraph({
    schema: "HistoricalSourceGraph",
    authority: "historical-reference",
    root_alias: "$REFERENCE_ROOT",
    inventory_state: inventoryState,
    entries: [
      {
        path: "src",
        kind: "directory",
        classifications: ["source"],
        tree_state: inventoryState === "complete" ? "enumerated" : "partial",
        limitations:
          inventoryState === "complete"
            ? []
            : ["Fixture inventory is partial."],
      },
      ...paths.map((path) => ({
        path,
        kind: "file" as const,
        sha256: sourceDigest(path),
        size: 10,
        language: "TypeScript",
        classifications: ["source"] as const,
        content_state: "hashed" as const,
        limitations: [],
      })),
    ].sort((left, right) => compareUnicodeCodePoints(left.path, right.path)),
    relationships: [],
    parse_failures: [],
    exclusions: [],
    languages: ["TypeScript"],
    manifests: [],
    vcs: { kind: "none", head: null, dirty: null },
    provenance: {
      importer: "rea-test",
      importer_version: "1",
      caller: "source-to-bundle-test",
    },
    limitations:
      inventoryState === "complete" ? [] : ["Fixture inventory is partial."],
  });

const applicationGraph = (coverage: "complete" | "partial") => {
  const nodes = [
    moduleNode("duplicated-a", {
      path: "bundle-a.js",
      source_sha256: HASH.duplicated,
    }),
    moduleNode("duplicated-b", {
      path: "bundle-b.js",
      source_sha256: HASH.duplicated,
    }),
    moduleNode("unchanged", {
      path: "bundle.js",
      source_sha256: HASH.unchanged,
    }),
    createJavaScriptApplicationNode({
      kind: "javascript-asset",
      identity: {
        strategy: "canonical-path",
        stability: "artifact-version",
        artifact_sha256: HASH.artifact,
        path: "src/modified.ts",
      },
      observations: [
        {
          label: "src/modified.ts",
          properties: { path: "src/modified.ts" },
          evidence: artifactEvidence(HASH.artifact, "src/modified.ts"),
        },
      ],
    }),
    moduleNode("split-a", { path: "bundle-a/src/split.ts" }),
    moduleNode("split-b", { path: "bundle-b/src/split.ts" }),
    moduleNode("merged", {
      path: "bundle.js",
      source_sha256: HASH.merged,
    }),
  ];
  return createJavaScriptApplicationGraph({
    schema: "JavaScriptApplicationGraph",
    root_node_ids: nodes.map(({ node_id: nodeId }) => nodeId),
    nodes,
    edges: [],
    coverage:
      coverage === "complete"
        ? {
            status: "complete",
            truncated: false,
            omitted_count: 0,
            limits: [],
          }
        : {
            status: "partial",
            truncated: true,
            omitted_count: 1,
            limits: [{ name: "fixture", value: 5, unit: "items" }],
          },
    limitations:
      coverage === "complete" ? [] : ["Fixture graph coverage is partial."],
  });
};

const moduleNode = (key: string, properties: Record<string, string>) =>
  createJavaScriptApplicationNode({
    kind: "javascript-module",
    identity: {
      strategy: "artifact-local-key",
      stability: "artifact-version",
      artifact_sha256: HASH.artifact,
      namespace: "test",
      key,
    },
    observations: [
      {
        label: key,
        properties,
        evidence: artifactEvidence(HASH.artifact, "bundle.js"),
      },
    ],
  });

const sourceDigest = (path: string): string => {
  if (path.includes("duplicated")) return HASH.duplicated;
  if (path.includes("merged")) return HASH.merged;
  if (path.includes("modified")) return HASH.modified;
  if (path.includes("removed")) return HASH.removed;
  if (path.includes("split")) return HASH.split;
  return HASH.unchanged;
};

const item = (
  result: ReturnType<typeof compareSourceToBundle>,
  sourcePath: string,
) => {
  const value = result.items.find(
    ({ source_path: path }) => path === sourcePath,
  );
  if (value === undefined) throw new TypeError(`Missing item: ${sourcePath}`);
  return value;
};