nodes/melMass-comfy_mtb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: ✨ define dynamic input on def load

Browse Source
This commit is contained in:
melMass
2023-07-10 20:00:47 +02:00
parent 22b9b94679
commit f77ddbd6a3
1 changed files with 26 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
web/mtb_widget_register.js
+26 -23
View File
@@ -72,6 +72,26 @@ const custom = (node, inputName, inputData, app) => {
}
const dynamic_connection = (node, index, connected, connectionPrefix = "input_", connectionType = "PSDLAYER") => {
// remove all non connected inputs
if (!connected && node.inputs.length > 1) {
node.removeInput(index)
// make inputs sequential again
for (let i = 0; i < node.inputs.length; i++) {
node.inputs[i].name = `${connectionPrefix}${i + 1}`
}
}
// add an extra input
if (node.inputs[node.inputs.length - 1].link != undefined) {
node.addInput(`${connectionPrefix}${node.inputs.length + 1}`, connectionType)
}
}
/**
@@ -179,36 +199,19 @@ const mtb_widgets = {
// // Register the custom widget in LiteGraph.js
// LiteGraph.registered_widgets["VIDEO"] = videoWidget;
// },
/**
*
* @param {import("./types/litegraph").LGraphNode} node
*/
async nodeCreated(node, app) {
if (node.comfyClass === "Psd Save (mtb)") {
node.onConnectionsChange = function (type, index, connected, link_info) {
// remove all non connected inputs
if (!connected && node.inputs.length > 1) {
node.removeInput(index)
// make inputs sequential again
for (let i = 0; i < node.inputs.length; i++) {
node.inputs[i].name = `input_${i + 1}`
}
}
async beforeRegisterNodeDef(nodeType, nodeData, app) {
if (nodeData.name === "Psd Save (mtb)") {
// add an extra input
if (node.inputs[node.inputs.length - 1].link != undefined) {
node.addInput(`input_${node.inputs.length + 1}`, "PSDLAYER")
// const onConnectionsChange = nodeType.prototype.onConnectionsChange;
nodeType.prototype.onConnectionsChange = function (type, index, connected, link_info) {
// const r = onConnectionsChange ? onConnectionsChange.apply(this, arguments) : undefined;
dynamic_connection(this, index, connected)
}
}
}
},
async beforeRegisterNodeDef(nodeType, nodeData, app) {
// console.log(nodeData.output)
// if (nodeData.output.includes("VIDEO") && nodeData.output_node) {
// console.log(`Found video output for ${nodeType}`)