|
|
import { app } from "../../scripts/app.js";
|
|
|
|
|
|
function mimicNode(node, target, slot) {
|
|
|
const getWidget = (widget) => {
|
|
|
if (typeof widget.origType === "undefined") return widget;
|
|
|
const newWidget = new Proxy(widget, {
|
|
|
has(target, prop) {
|
|
|
return prop in target && prop !== "origType" && prop !== "computeSize";
|
|
|
},
|
|
|
get(target, prop) {
|
|
|
if (prop.startsWith("orig") || prop === "computeSize") {
|
|
|
return undefined;
|
|
|
} else if (prop === "type") {
|
|
|
return target.origType;
|
|
|
} else if (prop === "name") {
|
|
|
return inputName;
|
|
|
} else {
|
|
|
return target[prop];
|
|
|
}
|
|
|
},
|
|
|
set(target, prop, value) {
|
|
|
target[prop] = value;
|
|
|
return true;
|
|
|
}
|
|
|
});
|
|
|
return newWidget;
|
|
|
}
|
|
|
const input = target.inputs[slot];
|
|
|
const inputName = node.inputs.length > 0 ? node.inputs[0].name : node.widgets[0].name;
|
|
|
if (typeof input.widget === "undefined") {
|
|
|
node.inputs = [{ ...input, name: inputName }];
|
|
|
node.widgets = [];
|
|
|
} else {
|
|
|
node.widgets = [getWidget(target.widgets.find(w => w.name === input.name))];
|
|
|
node.inputs = [];
|
|
|
node.size = node.computeSize();
|
|
|
requestAnimationFrame(() => {
|
|
|
if (node.onResize) {
|
|
|
node.onResize(node.size);
|
|
|
}
|
|
|
});
|
|
|
}
|
|
|
}
|
|
|
|
|
|
app.registerExtension({
|
|
|
name: "Comfy.DynamicInput",
|
|
|
dynamicNodes: ["CPackInputAny", "CPackInputFile"],
|
|
|
|
|
|
async beforeRegisterNodeDef(nodeType, nodeData) {
|
|
|
if (!this.dynamicNodes.includes(nodeData.name)) return;
|
|
|
|
|
|
nodeType.prototype.onConnectOutput = function () {
|
|
|
if (this.outputs[0].links?.length > 0) return false;
|
|
|
const target = arguments[3];
|
|
|
if (this.type === "CPackInputFile" && !["COMBO", "STRING"].includes(target.inputs[arguments[4]].type))
|
|
|
return false;
|
|
|
mimicNode(this, target, arguments[4]);
|
|
|
this.title = target.inputs[arguments[4]].name;
|
|
|
}
|
|
|
},
|
|
|
|
|
|
async setup(app) {
|
|
|
app.graph.nodes.forEach((node) => {
|
|
|
if (!this.dynamicNodes.includes(node.type)) return;
|
|
|
if (node.outputs.length > 0 && node.outputs[0].links.length > 0) {
|
|
|
const link = node.graph.links[node.outputs[0].links[0]];
|
|
|
mimicNode(node, app.graph.getNodeById(link.target_id), link.target_slot);
|
|
|
}
|
|
|
})
|
|
|
},
|
|
|
|
|
|
async init(app) {
|
|
|
const originalToPrompt = app.graphToPrompt;
|
|
|
const self = this;
|
|
|
app.graphToPrompt = async function(graph = app.graph, clean = true) {
|
|
|
const { workflow, output } = await originalToPrompt(graph, clean);
|
|
|
Object.entries(output).forEach(([id, nodeData]) => {
|
|
|
if (!nodeData.class_type.startsWith("CPackInput")) return;
|
|
|
const node = graph.getNodeById(parseInt(id));
|
|
|
if (!nodeData["_meta"]) {
|
|
|
nodeData["_meta"] = { title: node.title };
|
|
|
}
|
|
|
if (node.widgets.length === 0) return;
|
|
|
const widget = node.widgets[0];
|
|
|
nodeData["_meta"] = { ...nodeData["_meta"], options: widget.options };
|
|
|
});
|
|
|
return { workflow, output };
|
|
|
};
|
|
|
}
|
|
|
});
|
|
|
|
