custom_nodes/rgthree-comfy/web/common/utils_workflow.js

import { getResolver } from "./shared_utils.js";
import { getPngMetadata, getWebpMetadata } from "../../scripts/pnginfo.js";
function parseWorkflowJson(stringJson) {
    stringJson = stringJson || "null";
    stringJson = stringJson.replace(/:\s*NaN/g, ": null");
    return JSON.parse(stringJson);
}
export async function tryToGetWorkflowDataFromEvent(e) {
    var _a, _b, _c, _d;
    let work;
    for (const file of ((_a = e.dataTransfer) === null || _a === void 0 ? void 0 : _a.files) || []) {
        const data = await tryToGetWorkflowDataFromFile(file);
        if (data.workflow || data.prompt) {
            return data;
        }
    }
    const validTypes = ["text/uri-list", "text/x-moz-url"];
    const match = (((_b = e.dataTransfer) === null || _b === void 0 ? void 0 : _b.types) || []).find((t) => validTypes.find((v) => t === v));
    if (match) {
        const uri = (_d = (_c = e.dataTransfer.getData(match)) === null || _c === void 0 ? void 0 : _c.split("\n")) === null || _d === void 0 ? void 0 : _d[0];
        if (uri) {
            return tryToGetWorkflowDataFromFile(await (await fetch(uri)).blob());
        }
    }
    return { workflow: null, prompt: null };
}
export async function tryToGetWorkflowDataFromFile(file) {
    var _a;
    if (file.type === "image/png") {
        const pngInfo = await getPngMetadata(file);
        return {
            workflow: parseWorkflowJson(pngInfo === null || pngInfo === void 0 ? void 0 : pngInfo.workflow),
            prompt: parseWorkflowJson(pngInfo === null || pngInfo === void 0 ? void 0 : pngInfo.prompt),
        };
    }
    if (file.type === "image/webp") {
        const pngInfo = await getWebpMetadata(file);
        const workflow = parseWorkflowJson((pngInfo === null || pngInfo === void 0 ? void 0 : pngInfo.workflow) || (pngInfo === null || pngInfo === void 0 ? void 0 : pngInfo.Workflow) || "null");
        const prompt = parseWorkflowJson((pngInfo === null || pngInfo === void 0 ? void 0 : pngInfo.prompt) || (pngInfo === null || pngInfo === void 0 ? void 0 : pngInfo.Prompt) || "null");
        return { workflow, prompt };
    }
    if (file.type === "application/json" || ((_a = file.name) === null || _a === void 0 ? void 0 : _a.endsWith(".json"))) {
        const resolver = getResolver();
        const reader = new FileReader();
        reader.onload = async () => {
            const json = parseWorkflowJson(reader.result);
            const isApiJson = Object.values(json).every((v) => v.class_type);
            const prompt = isApiJson ? json : null;
            const workflow = !isApiJson && !(json === null || json === void 0 ? void 0 : json.templates) ? json : null;
            return { workflow, prompt };
        };
        return resolver.promise;
    }
    return { workflow: null, prompt: null };
}