jaidaken
f09734b0ee
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Python Linting / Run Pylint (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Has been cancelled
Execution Tests / test (macos-latest) (push) Has been cancelled
Execution Tests / test (ubuntu-latest) (push) Has been cancelled
Execution Tests / test (windows-latest) (push) Has been cancelled
Test server launches without errors / test (push) Has been cancelled
Unit Tests / test (macos-latest) (push) Has been cancelled
Unit Tests / test (ubuntu-latest) (push) Has been cancelled
Unit Tests / test (windows-2022) (push) Has been cancelled
Includes 30 custom nodes committed directly, 7 Civitai-exclusive loras stored via Git LFS, and a setup script that installs all dependencies and downloads HuggingFace-hosted models on vast.ai. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
56 lines
2.7 KiB
JavaScript
56 lines
2.7 KiB
JavaScript
import { getResolver } from "./shared_utils.js";
|
|
import { getPngMetadata, getWebpMetadata } from "./comfyui_shim.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 };
|
|
}
|