import { expect, test } from "bun:test";

type Scenario = "image-cleanup" | "scope-cleanup-switch" | "scope-cleanup-late-canvas-reference" | "video-commit-race" | "audio-commit-race" | "canvas-batch-commit-race" | "canvas-multi-output" | "canvas-copy-generation";

test("一次任务四图经真实消费链路堆叠并保存，刷新和重放保留四个独立资源", async () => {
    type Node = import("../src/types/canvas").CanvasNodeData;
    const result = await runScenario<{ edited: boolean; live: Node[]; restored: Node[]; replayed: Node[] }>("canvas-multi-output");
    expect(result.edited).toBe(true);
    for (const nodes of [result.live, result.restored, result.replayed]) {
        expect(nodes).toHaveLength(9);
        const root = nodes.find((node) => node.id === "node-0")!;
        expect(root).toMatchObject({ title: "用户改名", position: { x: 700, y: 500 }, metadata: { isBatchRoot: true, imageBatchExpanded: false, generationOutputCount: 4, assetId: "asset-0" } });
        expect(root.metadata?.batchChildIds).toHaveLength(4);
        const children = nodes.filter((node) => node.metadata?.batchRootId === root.id);
        expect(children).toHaveLength(4);
        expect(children.map((node) => node.metadata?.storageKey)).toEqual([0, 1, 2, 3].map((index) => `resource:image-${index}`));
        expect(children.map((node) => node.metadata?.assetId)).toEqual([0, 1, 2, 3].map((index) => `asset-${index}`));
        expect(children.map((node) => node.metadata?.naturalWidth)).toEqual([640, 641, 642, 643]);
        expect(nodes.filter((node) => node.metadata?.status === "error")).toHaveLength(3);
    }
});
type ScenarioResponse<T> = { ok: true; result: T } | { ok: false; error: string };

test("已生成图片的副本在普通保存和新生成结果持久化后仍保留，原节点不受影响", async () => {
    type Node = import("../src/types/canvas").CanvasNodeData;
    const result = await runScenario<{ beforeGeneration: Node[]; restored: Node[] }>("canvas-copy-generation");
    expect(result.beforeGeneration.map((node) => node.id)).toEqual(["source", "copy"]);
    expect(result.beforeGeneration.find((node) => node.id === "copy")?.metadata?.generationEffectKeys).toBeUndefined();
    expect(result.restored.map((node) => node.id)).toEqual(["source", "copy"]);
    expect(result.restored.find((node) => node.id === "source")?.metadata).toMatchObject({ content: "original-image", generationEffectKeys: ["attach-node:old-task:source:0"] });
    expect(result.restored.find((node) => node.id === "copy")?.metadata).toMatchObject({ content: "new-image", status: "success", generationEffectKeys: ["attach-node:new-task:copy:0"] });
});

test("并发生成结果经真实持久化消费链路及重新读取后保留成功、失败和用户编辑", async () => {
    type Node = import("../src/types/canvas").CanvasNodeData;
    const result = await runScenario<{ edited: boolean; live: Node[]; restored: Node[] }>("canvas-batch-commit-race");
    expect(result.edited).toBe(true);
    for (const nodes of [result.live, result.restored]) {
        expect(nodes).toHaveLength(6);
        expect(nodes.filter((node) => node.metadata?.status === "success")).toHaveLength(2);
        expect(nodes.filter((node) => node.metadata?.status === "error")).toHaveLength(3);
        expect(nodes.find((node) => node.id === "node-0")).toMatchObject({ title: "用户改名", position: { x: 700, y: 500 }, metadata: { assetId: "asset-0" } });
        expect(nodes.find((node) => node.id === "node-1")?.metadata?.assetId).toBe("asset-1");
    }
});

function runScenario<T>(scenario: Scenario): Promise<T> {
    return new Promise<T>((resolve, reject) => {
        const worker = new Worker(new URL("./helpers/generation-storage-consistency.worker.ts", import.meta.url).href, { type: "module" });
        const finish = () => {
            worker.terminate();
        };
        worker.onmessage = (event: MessageEvent<ScenarioResponse<T>>) => {
            finish();
            if (event.data.ok) resolve(event.data.result);
            else reject(new Error(event.data.error));
        };
        worker.onerror = (event) => {
            finish();
            reject(event.error ?? new Error(event.message));
        };
        worker.postMessage(scenario);
    });
}

test("generation image cleanup removes unused generation-image blobs and preserves referenced ones", async () => {
    const result = await runScenario<{ usedPresent: boolean; unusedPresent: boolean }>("image-cleanup");
    expect(result.usedPresent).toBe(true);
    expect(result.unusedPresent).toBe(false);
});

test("delayed cleanup keeps account-A Canvas-only image references after switching the active account", async () => {
    const result = await runScenario<{ referencedPresent: boolean; unusedPresent: boolean; otherScopePresent: boolean }>("scope-cleanup-switch");
    expect(result.referencedPresent).toBe(true);
    expect(result.unusedPresent).toBe(false);
    expect(result.otherScopePresent).toBe(true);
});

test("delayed cleanup preserves a same-scope Canvas-only reference added after cleanup was queued", async () => {
    const result = await runScenario<{ referencedPresent: boolean; unusedPresent: boolean }>("scope-cleanup-late-canvas-reference");
    expect(result.referencedPresent).toBe(true);
    expect(result.unusedPresent).toBe(false);
});

test("generation video and audio materialization cannot race cleanup into a catalog row whose blob is missing", async () => {
    for (const mediaType of ["video", "audio"] as const) {
        const result = await runScenario<{ kind?: string; storageKey?: string; blobPresent: boolean; generationAssetCount: number }>(`${mediaType}-commit-race`);
        expect(result.kind).toBe(mediaType);
        expect(result.storageKey).toMatch(new RegExp(`^generation-${mediaType}:generation-media-commit-race-${mediaType}:`));
        expect(result.blobPresent).toBe(true);
        expect(result.generationAssetCount).toBe(1);
    }
});