nodes/JeyzerMC-comfyui-flakes
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(nodes): robustly remove hidden widget input pins via splice and onConfigure

Browse Source
This commit is contained in:
Mehdi Chaid
2026-05-06 16:22:04 -04:00
parent a0092fae3c
commit e2c4e64cc7
1 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
web/flake_stack.js
+17 -4
View File
@@ -5,6 +5,15 @@ import { setupFlakeModelPresetWidget } from "./widgets/flake-model-preset.js";
import { setupFlakeModelComboWidget } from "./widgets/flake-model-combo.js";
import "./queue.js";
function removeHiddenInputs(node, names) {
for (let i = node.inputs.length - 1; i >= 0; i--) {
if (names.includes(node.inputs[i].name)) {
node.inputs.splice(i, 1);
}
}
node.setSize(node.computeSize());
}
app.registerExtension({
name: "comfyui-flakes.FlakeStack",
async beforeRegisterNodeDef(nodeType, nodeData) {
@@ -13,13 +22,14 @@ app.registerExtension({
nodeType.prototype.onNodeCreated = function () {
const r = origOnNodeCreated?.apply(this, arguments);
setupFlakeWidget(this);
this.inputs = this.inputs.filter(inp => inp.name !== "model_family" && inp.name !== "flakes_json");
removeHiddenInputs(this, ["model_family", "flakes_json"]);
return r;
};
const origOnConfigure = nodeType.prototype.onConfigure;
nodeType.prototype.onConfigure = function () {
const r = origOnConfigure?.apply(this, arguments);
removeHiddenInputs(this, ["model_family", "flakes_json"]);
this._flakes_render?.();
return r;
};
@@ -29,13 +39,14 @@ app.registerExtension({
nodeType.prototype.onNodeCreated = function () {
const r = origOnNodeCreated?.apply(this, arguments);
setupFlakeModelPresetWidget(this);
this.inputs = this.inputs.filter(inp => inp.name !== "model_family" && inp.name !== "preset");
removeHiddenInputs(this, ["model_family", "preset"]);
return r;
};
const origOnConfigure = nodeType.prototype.onConfigure;
nodeType.prototype.onConfigure = function () {
const r = origOnConfigure?.apply(this, arguments);
removeHiddenInputs(this, ["model_family", "preset"]);
this._preset_render?.();
return r;
};
@@ -45,13 +56,14 @@ app.registerExtension({
nodeType.prototype.onNodeCreated = function () {
const r = origOnNodeCreated?.apply(this, arguments);
setupFlakeComboWidget(this);
this.inputs = this.inputs.filter(inp => inp.name !== "model_family" && inp.name !== "flakes_json");
removeHiddenInputs(this, ["model_family", "flakes_json"]);
return r;
};
const origOnConfigure = nodeType.prototype.onConfigure;
nodeType.prototype.onConfigure = function () {
const r = origOnConfigure?.apply(this, arguments);
removeHiddenInputs(this, ["model_family", "flakes_json"]);
this._combo_render?.();
return r;
};
@@ -61,13 +73,14 @@ app.registerExtension({
nodeType.prototype.onNodeCreated = function () {
const r = origOnNodeCreated?.apply(this, arguments);
setupFlakeModelComboWidget(this);
this.inputs = this.inputs.filter(inp => inp.name !== "model_family" && inp.name !== "preset");
removeHiddenInputs(this, ["model_family", "preset"]);
return r;
};
const origOnConfigure = nodeType.prototype.onConfigure;
nodeType.prototype.onConfigure = function () {
const r = origOnConfigure?.apply(this, arguments);
removeHiddenInputs(this, ["model_family", "preset"]);
this._model_combo_render?.();
return r;
};