import { parseComfyWorkflow } from "../../src/types/comfyWorkflow";
import { defaultGraph } from "../../src/scripts/defaultGraph";
import fs from "fs";

const WORKFLOW_DIR = "tests-ui/workflows";

describe("parseComfyWorkflow", () => {
    it("parses valid workflow", async () => {
        fs.readdirSync(WORKFLOW_DIR).forEach(async (file) => {
            if (file.endsWith(".json")) {
                const data = fs.readFileSync(`${WORKFLOW_DIR}/${file}`, "utf-8");
                await expect(parseComfyWorkflow(data)).resolves.not.toThrow();
            }
        });
    });

    it("workflow.nodes", async () => {
        const workflow = JSON.parse(JSON.stringify(defaultGraph));
        workflow.nodes = undefined;
        await expect(parseComfyWorkflow(JSON.stringify(workflow))).rejects.toThrow();

        workflow.nodes = null;
        await expect(parseComfyWorkflow(JSON.stringify(workflow))).rejects.toThrow();

        workflow.nodes = [];
        await expect(parseComfyWorkflow(JSON.stringify(workflow))).resolves.not.toThrow();
    });

    it("workflow.version", async () => {
        const workflow = JSON.parse(JSON.stringify(defaultGraph));
        workflow.version = undefined;
        await expect(parseComfyWorkflow(JSON.stringify(workflow))).rejects.toThrow();

        workflow.version = "1.0.1"; // Invalid format.
        await expect(parseComfyWorkflow(JSON.stringify(workflow))).rejects.toThrow();

        workflow.version = 1;
        await expect(parseComfyWorkflow(JSON.stringify(workflow))).resolves.not.toThrow();
    });

    it("workflow.extra", async () => {
        const workflow = JSON.parse(JSON.stringify(defaultGraph));
        workflow.extra = undefined;
        await expect(parseComfyWorkflow(JSON.stringify(workflow))).resolves.not.toThrow();

        workflow.extra = null;
        await expect(parseComfyWorkflow(JSON.stringify(workflow))).resolves.not.toThrow();

        workflow.extra = {};
        await expect(parseComfyWorkflow(JSON.stringify(workflow))).resolves.not.toThrow();

        workflow.extra = { foo: "bar" }; // Should accept extra fields.
        await expect(parseComfyWorkflow(JSON.stringify(workflow))).resolves.not.toThrow();
    });

    it("workflow.nodes.pos", async () => {
        const workflow = JSON.parse(JSON.stringify(defaultGraph));
        workflow.nodes[0].pos = [1, 2, 3];
        await expect(parseComfyWorkflow(JSON.stringify(workflow))).rejects.toThrow();

        workflow.nodes[0].pos = [1, 2];
        await expect(parseComfyWorkflow(JSON.stringify(workflow))).resolves.not.toThrow();

        // Should automatically transform the legacy format object to array.
        workflow.nodes[0].pos = {"0": 3, "1": 4};
        let parsedWorkflow = await parseComfyWorkflow(JSON.stringify(workflow));
        expect(parsedWorkflow.nodes[0].pos).toEqual([3, 4]);

        workflow.nodes[0].pos = {0: 3, 1: 4};
        parsedWorkflow = await parseComfyWorkflow(JSON.stringify(workflow));
        expect(parsedWorkflow.nodes[0].pos).toEqual([3, 4]);
    });

    it("workflow.nodes.widget_values", async () => {
        const workflow = JSON.parse(JSON.stringify(defaultGraph));
        workflow.nodes[0].widgets_values = ["foo", "bar"];
        await expect(parseComfyWorkflow(JSON.stringify(workflow))).resolves.not.toThrow();

        workflow.nodes[0].widgets_values = "foo";
        await expect(parseComfyWorkflow(JSON.stringify(workflow))).rejects.toThrow();

        workflow.nodes[0].widgets_values = undefined;
        await expect(parseComfyWorkflow(JSON.stringify(workflow))).resolves.not.toThrow();

        // Should automatically transform the legacy format object to array.
        workflow.nodes[0].widgets_values = {0: "foo", 1: "bar"};
        const parsedWorkflow = await parseComfyWorkflow(JSON.stringify(workflow));
        expect(parsedWorkflow.nodes[0].widgets_values).toEqual(["foo", "bar"]);
    });
});