92 lines
3.0 KiB
JavaScript
92 lines
3.0 KiB
JavaScript
import { app } from "../../scripts/app.js";
|
|
import { ComfyWidgets } from "../../scripts/widgets.js";
|
|
// Adds defaults for quickly adding nodes with middle click on the input/output
|
|
|
|
app.registerExtension({
|
|
name: "Comfy.SlotDefaults",
|
|
suggestionsNumber: null,
|
|
init() {
|
|
LiteGraph.search_filter_enabled = true;
|
|
LiteGraph.middle_click_slot_add_default_node = true;
|
|
this.suggestionsNumber = app.ui.settings.addSetting({
|
|
id: "Comfy.NodeSuggestions.number",
|
|
name: "Number of nodes suggestions",
|
|
type: "slider",
|
|
attrs: {
|
|
min: 1,
|
|
max: 100,
|
|
step: 1,
|
|
},
|
|
defaultValue: 5,
|
|
onChange: (newVal, oldVal) => {
|
|
this.setDefaults(newVal);
|
|
}
|
|
});
|
|
},
|
|
slot_types_default_out: {},
|
|
slot_types_default_in: {},
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
var nodeId = nodeData.name;
|
|
var inputs = [];
|
|
inputs = nodeData["input"]["required"]; //only show required inputs to reduce the mess also not logical to create node with optional inputs
|
|
for (const inputKey in inputs) {
|
|
var input = (inputs[inputKey]);
|
|
if (typeof input[0] !== "string") continue;
|
|
|
|
var type = input[0]
|
|
if (type in ComfyWidgets) {
|
|
var customProperties = input[1]
|
|
if (!(customProperties?.forceInput)) continue; //ignore widgets that don't force input
|
|
}
|
|
|
|
if (!(type in this.slot_types_default_out)) {
|
|
this.slot_types_default_out[type] = ["Reroute"];
|
|
}
|
|
if (this.slot_types_default_out[type].includes(nodeId)) continue;
|
|
this.slot_types_default_out[type].push(nodeId);
|
|
|
|
// Input types have to be stored as lower case
|
|
// Store each node that can handle this input type
|
|
const lowerType = type.toLocaleLowerCase();
|
|
if (!(lowerType in LiteGraph.registered_slot_in_types)) {
|
|
LiteGraph.registered_slot_in_types[lowerType] = { nodes: [] };
|
|
}
|
|
LiteGraph.registered_slot_in_types[lowerType].nodes.push(nodeType.comfyClass);
|
|
}
|
|
|
|
var outputs = nodeData["output"];
|
|
for (const key in outputs) {
|
|
var type = outputs[key];
|
|
if (!(type in this.slot_types_default_in)) {
|
|
this.slot_types_default_in[type] = ["Reroute"];// ["Reroute", "Primitive"]; primitive doesn't always work :'()
|
|
}
|
|
|
|
this.slot_types_default_in[type].push(nodeId);
|
|
|
|
// Store each node that can handle this output type
|
|
if (!(type in LiteGraph.registered_slot_out_types)) {
|
|
LiteGraph.registered_slot_out_types[type] = { nodes: [] };
|
|
}
|
|
LiteGraph.registered_slot_out_types[type].nodes.push(nodeType.comfyClass);
|
|
|
|
if(!LiteGraph.slot_types_out.includes(type)) {
|
|
LiteGraph.slot_types_out.push(type);
|
|
}
|
|
}
|
|
var maxNum = this.suggestionsNumber.value;
|
|
this.setDefaults(maxNum);
|
|
},
|
|
setDefaults(maxNum) {
|
|
|
|
LiteGraph.slot_types_default_out = {};
|
|
LiteGraph.slot_types_default_in = {};
|
|
|
|
for (const type in this.slot_types_default_out) {
|
|
LiteGraph.slot_types_default_out[type] = this.slot_types_default_out[type].slice(0, maxNum);
|
|
}
|
|
for (const type in this.slot_types_default_in) {
|
|
LiteGraph.slot_types_default_in[type] = this.slot_types_default_in[type].slice(0, maxNum);
|
|
}
|
|
}
|
|
});
|