86 lines
3.5 KiB
JavaScript
86 lines
3.5 KiB
JavaScript
import { app } from "../../../scripts/app.js";
|
|
|
|
app.registerExtension({
|
|
name: "KlinterNodes.nodevalue2stringmulti",
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (nodeData.name !== "nodevalue2stringmulti") {
|
|
return;
|
|
}
|
|
|
|
const orig_onNodeCreated = nodeType.prototype.onNodeCreated;
|
|
nodeType.prototype.onNodeCreated = function() {
|
|
if (orig_onNodeCreated) {
|
|
orig_onNodeCreated.apply(this, arguments);
|
|
}
|
|
|
|
// Add update button
|
|
this.addWidget("button", "Update inputs", null, () => {
|
|
const target_inputs = this.widgets.find(w => w.name === "inputcount")?.value || 2;
|
|
const current_input_count = this.inputs.length;
|
|
|
|
// Calculate how many new inputs we need
|
|
const inputs_to_add = target_inputs - current_input_count;
|
|
|
|
if (inputs_to_add > 0) {
|
|
// Add new inputs
|
|
for (let i = 0; i < inputs_to_add; i++) {
|
|
const inputName = `value_${current_input_count + i + 1}`;
|
|
this.addInput(inputName, ["STRING", "INT", "FLOAT"], {
|
|
// Add a custom property to track the connected node name
|
|
connectedNodeName: "Not Connected"
|
|
});
|
|
}
|
|
} else if (inputs_to_add < 0) {
|
|
// Remove excess inputs from the end
|
|
for (let i = 0; i < -inputs_to_add; i++) {
|
|
this.removeInput(this.inputs.length - 1);
|
|
}
|
|
}
|
|
});
|
|
|
|
// Initial setup
|
|
setTimeout(() => {
|
|
const widget = this.widgets.find(w => w.name === "Update inputs");
|
|
if (widget) {
|
|
widget.callback();
|
|
}
|
|
}, 100);
|
|
};
|
|
|
|
// Override connections change to update input names
|
|
const orig_onConnectionsChange = nodeType.prototype.onConnectionsChange;
|
|
nodeType.prototype.onConnectionsChange = function(type, index, connected, link_info) {
|
|
if (orig_onConnectionsChange) {
|
|
orig_onConnectionsChange.apply(this, arguments);
|
|
}
|
|
|
|
// Focus on input connections
|
|
if (type === LiteGraph.INPUT) {
|
|
if (connected && link_info) {
|
|
const sourceNode = app.graph.getNodeById(link_info.origin_id);
|
|
|
|
if (sourceNode) {
|
|
// Prefer title, fallback to type
|
|
const nodeName = sourceNode.title || sourceNode.type || `Node_${sourceNode.id}`;
|
|
|
|
// Update the input's name to show the connected node
|
|
this.inputs[index].label = nodeName;
|
|
|
|
// Store the node name as a custom property
|
|
this.inputs[index].connectedNodeName = nodeName;
|
|
}
|
|
} else {
|
|
// Reset input label when disconnected
|
|
this.inputs[index].label = null;
|
|
this.inputs[index].connectedNodeName = "Not Connected";
|
|
}
|
|
|
|
// Force graph to redraw
|
|
if (this.graph) {
|
|
this.graph.setDirtyCanvas(true);
|
|
}
|
|
}
|
|
};
|
|
}
|
|
});
|