|
import { ComfyApp, app } from "../../scripts/app.js";
|
|
import { ComfyDialog, $el } from "../../scripts/ui.js";
|
|
import { api } from "../../scripts/api.js";
|
|
|
|
app.registerExtension({
|
|
name: "Comfy.Inspire.Regional",
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (nodeData.name === 'ApplyRegionalIPAdapters //Inspire') {
|
|
var input_name = "input";
|
|
var base_slot = 0;
|
|
|
|
switch(nodeData.name) {
|
|
case 'ApplyRegionalIPAdapters //Inspire':
|
|
input_name = "regional_ipadapter";
|
|
base_slot = 1;
|
|
break;
|
|
}
|
|
|
|
const onConnectionsChange = nodeType.prototype.onConnectionsChange;
|
|
nodeType.prototype.onConnectionsChange = function (type, index, connected, link_info) {
|
|
if(!link_info || type == 2)
|
|
return;
|
|
|
|
if(this.inputs[0].type == '*'){
|
|
const node = app.graph.getNodeById(link_info.origin_id);
|
|
let origin_type = node.outputs[link_info.origin_slot].type;
|
|
|
|
if(origin_type == '*') {
|
|
this.disconnectInput(link_info.target_slot);
|
|
return;
|
|
}
|
|
|
|
for(let i in this.inputs) {
|
|
let input_i = this.inputs[i];
|
|
if(input_i.name != 'select' && input_i.name != 'sel_mode')
|
|
input_i.type = origin_type;
|
|
}
|
|
}
|
|
|
|
if (!connected && (this.inputs.length > base_slot+1)) {
|
|
const stackTrace = new Error().stack;
|
|
|
|
if(
|
|
!stackTrace.includes('LGraphNode.prototype.connect') &&
|
|
!stackTrace.includes('LGraphNode.connect') &&
|
|
!stackTrace.includes('loadGraphData')) {
|
|
this.removeInput(index);
|
|
}
|
|
}
|
|
|
|
let slot_i = 1;
|
|
for (let i = base_slot; i < this.inputs.length; i++) {
|
|
let input_i = this.inputs[i];
|
|
input_i.name = `${input_name}${slot_i}`
|
|
slot_i++;
|
|
}
|
|
|
|
let last_slot = this.inputs[this.inputs.length - 1];
|
|
if (last_slot.link != undefined) {
|
|
this.addInput(`${input_name}${slot_i}`, this.inputs[base_slot].type);
|
|
}
|
|
}
|
|
}
|
|
}}); |
