| 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 }; | |
| } | |