jaidaken
f09734b0ee
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Python Linting / Run Pylint (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Has been cancelled
Execution Tests / test (macos-latest) (push) Has been cancelled
Execution Tests / test (ubuntu-latest) (push) Has been cancelled
Execution Tests / test (windows-latest) (push) Has been cancelled
Test server launches without errors / test (push) Has been cancelled
Unit Tests / test (macos-latest) (push) Has been cancelled
Unit Tests / test (ubuntu-latest) (push) Has been cancelled
Unit Tests / test (windows-2022) (push) Has been cancelled
Includes 30 custom nodes committed directly, 7 Civitai-exclusive loras stored via Git LFS, and a setup script that installs all dependencies and downloads HuggingFace-hosted models on vast.ai. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
554 lines
23 KiB
JavaScript
554 lines
23 KiB
JavaScript
import {app} from "../../../../scripts/app.js";
|
|
import {api} from "../../../../scripts/api.js";
|
|
import {$el} from "../../../../scripts/ui.js";
|
|
|
|
const propmts = ["easy wildcards", "easy positive", "easy negative", "easy stylesSelector", "easy promptConcat", "easy promptReplace"]
|
|
const loaders = ["easy a1111Loader", "easy comfyLoader", "easy fullLoader", "easy svdLoader", "easy cascadeLoader", "easy sv3dLoader"]
|
|
const preSamplingNodes = ["easy preSampling", "easy preSamplingAdvanced", "easy preSamplingNoiseIn", "easy preSamplingCustom", "easy preSamplingDynamicCFG","easy preSamplingSdTurbo", "easy preSamplingLayerDiffusion"]
|
|
const kSampler = ["easy kSampler", "easy kSamplerTiled","easy kSamplerInpainting", "easy kSamplerDownscaleUnet", "easy kSamplerSDTurbo"]
|
|
const controlNetNodes = ["easy controlnetLoader", "easy controlnetLoaderADV"]
|
|
const instantIDNodes = ["easy instantIDApply", "easy instantIDApplyADV"]
|
|
const ipadapterNodes = ["easy ipadapterApply", "easy ipadapterApplyADV" ,"easy ipadapterApplyFaceIDKolors", "easy ipadapterStyleComposition"]
|
|
const pipeNodes = ['easy pipeIn','easy pipeOut', 'easy pipeEdit']
|
|
const xyNodes = ['easy XYPlot', 'easy XYPlotAdvanced']
|
|
const extraNodes = ['easy setNode']
|
|
const modelNormalNodes = [...['RescaleCFG','LoraLoaderModelOnly','LoraLoader','FreeU','FreeU_v2'],...ipadapterNodes,...extraNodes]
|
|
const suggestions = {
|
|
// prompt
|
|
"easy seed":{
|
|
"from":{
|
|
"INT": [...preSamplingNodes,...['easy fullkSampler']]
|
|
}
|
|
},
|
|
"easy positive":{
|
|
"from":{
|
|
"STRING": [...propmts]
|
|
}
|
|
},
|
|
"easy negative":{
|
|
"from":{
|
|
"STRING": [...propmts]
|
|
}
|
|
},
|
|
"easy wildcards":{
|
|
"from":{
|
|
"STRING": [...["Reroute","easy showAnything"],...propmts,]
|
|
}
|
|
},
|
|
"easy stylesSelector":{
|
|
"from":{
|
|
"STRING": [...["Reroute","easy showAnything"],...propmts,]
|
|
}
|
|
},
|
|
"easy promptConcat":{
|
|
"from":{
|
|
"STRING": [...["Reroute","easy showAnything"],...propmts,]
|
|
}
|
|
},
|
|
"easy promptReplace":{
|
|
"from":{
|
|
"STRING": [...["Reroute","easy showAnything"],...propmts,]
|
|
}
|
|
},
|
|
// sd相关
|
|
"easy fullLoader": {
|
|
"from":{
|
|
"PIPE_LINE": [...preSamplingNodes,...['easy fullkSampler'],...pipeNodes,...extraNodes],
|
|
"MODEL":modelNormalNodes
|
|
},
|
|
"to":{
|
|
"STRING": [...propmts]
|
|
}
|
|
},
|
|
"easy a1111Loader": {
|
|
"from": {
|
|
"PIPE_LINE": [ ...preSamplingNodes, ...controlNetNodes, ...instantIDNodes, ...pipeNodes, ...extraNodes],
|
|
"MODEL": modelNormalNodes
|
|
},
|
|
"to":{
|
|
"STRING": [...propmts]
|
|
}
|
|
},
|
|
"easy comfyLoader": {
|
|
"from": {
|
|
"PIPE_LINE": [ ...preSamplingNodes, ...controlNetNodes, ...instantIDNodes, ...pipeNodes, ...extraNodes],
|
|
"MODEL": modelNormalNodes
|
|
},
|
|
"to":{
|
|
"STRING": [...propmts]
|
|
}
|
|
},
|
|
"easy svdLoader":{
|
|
"from": {
|
|
"PIPE_LINE": [ ...["easy preSampling", "easy preSamplingAdvanced", "easy preSamplingDynamicCFG"], ...pipeNodes, ...extraNodes],
|
|
"MODEL": modelNormalNodes
|
|
},
|
|
"to":{
|
|
"STRING": [...propmts]
|
|
}
|
|
},
|
|
"easy zero123Loader":{
|
|
"from": {
|
|
"PIPE_LINE": [ ...["easy preSampling", "easy preSamplingAdvanced", "easy preSamplingDynamicCFG"], ...pipeNodes, ...extraNodes],
|
|
"MODEL": modelNormalNodes
|
|
},
|
|
"to":{
|
|
"STRING": [...propmts]
|
|
}
|
|
},
|
|
"easy sv3dLoader":{
|
|
"from": {
|
|
"PIPE_LINE": [ ...["easy preSampling", "easy preSamplingAdvanced", "easy preSamplingDynamicCFG"], ...pipeNodes, ...extraNodes],
|
|
"MODEL": modelNormalNodes
|
|
},
|
|
"to":{
|
|
"STRING": [...propmts]
|
|
}
|
|
},
|
|
"easy preSampling": {
|
|
"from": {
|
|
"PIPE_LINE": [ ...kSampler, ...pipeNodes, ...controlNetNodes, ...xyNodes, ...extraNodes]
|
|
},
|
|
},
|
|
"easy preSamplingAdvanced": {
|
|
"from": {
|
|
"PIPE_LINE": [ ...kSampler, ...pipeNodes, ...controlNetNodes, ...xyNodes, ...extraNodes]
|
|
}
|
|
},
|
|
"easy preSamplingDynamicCFG": {
|
|
"from": {
|
|
"PIPE_LINE": [ ...kSampler, ...pipeNodes, ...controlNetNodes, ...xyNodes, ...extraNodes]
|
|
}
|
|
},
|
|
"easy preSamplingCustom": {
|
|
"from": {
|
|
"PIPE_LINE": [ ...kSampler, ...pipeNodes, ...controlNetNodes, ...xyNodes, ...extraNodes]
|
|
}
|
|
},
|
|
"easy preSamplingLayerDiffusion": {
|
|
"from": {
|
|
"PIPE_LINE": [...["easy kSamplerLayerDiffusion"], ...kSampler, ...pipeNodes, ...controlNetNodes, ...xyNodes, ...extraNodes]
|
|
}
|
|
},
|
|
"easy preSamplingNoiseIn": {
|
|
"from": {
|
|
"PIPE_LINE": [ ...kSampler, ...pipeNodes, ...controlNetNodes, ...xyNodes, ...extraNodes]
|
|
}
|
|
},
|
|
// ksampler
|
|
"easy fullkSampler": {
|
|
"from": {
|
|
"PIPE_LINE": [ ...pipeNodes.reverse(), ...['easy preDetailerFix', 'easy preMaskDetailerFix'], ...preSamplingNodes, ...extraNodes]
|
|
}
|
|
},
|
|
"easy kSampler": {
|
|
"from": {
|
|
"PIPE_LINE": [ ...pipeNodes.reverse(), ...['easy preDetailerFix', 'easy preMaskDetailerFix', 'easy hiresFix'], ...preSamplingNodes, ...extraNodes],
|
|
}
|
|
},
|
|
// cn
|
|
"easy controlnetLoader": {
|
|
"from": {
|
|
"PIPE_LINE": [ ...preSamplingNodes, ...controlNetNodes, ...instantIDNodes, ...pipeNodes, ...extraNodes]
|
|
}
|
|
},
|
|
"easy controlnetLoaderADV":{
|
|
"from": {
|
|
"PIPE_LINE": [ ...preSamplingNodes, ...controlNetNodes, ...instantIDNodes, ...pipeNodes, ...extraNodes]
|
|
}
|
|
},
|
|
// instant
|
|
"easy instantIDApply": {
|
|
"from": {
|
|
"PIPE_LINE": [ ...preSamplingNodes, ...controlNetNodes, ...instantIDNodes, ...pipeNodes, ...extraNodes],
|
|
"MODEL": modelNormalNodes
|
|
},
|
|
"to":{
|
|
"COMBO": [...["easy promptLine"]]
|
|
}
|
|
},
|
|
"easy instantIDApplyADV":{
|
|
"from": {
|
|
"PIPE_LINE": [ ...preSamplingNodes, ...controlNetNodes, ...instantIDNodes, ...pipeNodes, ...extraNodes],
|
|
"MODEL": modelNormalNodes
|
|
},
|
|
"to":{
|
|
"COMBO": [...["easy promptLine"]]
|
|
}
|
|
},
|
|
"easy ipadapterApply":{
|
|
"to":{
|
|
"COMBO": [...["easy promptLine"]]
|
|
}
|
|
},
|
|
"easy ipadapterApplyADV":{
|
|
"to":{
|
|
"STRING": [...["easy sliderControl"], ...propmts],
|
|
"COMBO": [...["easy promptLine"]]
|
|
}
|
|
},
|
|
"easy ipadapterStyleComposition":{
|
|
"to":{
|
|
"COMBO": [...["easy promptLine"]]
|
|
}
|
|
},
|
|
// fix
|
|
"easy preDetailerFix":{
|
|
"from": {
|
|
"PIPE_LINE": [...["easy detailerFix"], ...pipeNodes, ...extraNodes]
|
|
},
|
|
"to":{
|
|
"PIPE_LINE": [...["easy ultralyticsDetectorPipe", "easy samLoaderPipe", "easy kSampler", "easy fullkSampler"]]
|
|
}
|
|
},
|
|
"easy preMaskDetailerFix":{
|
|
"from": {
|
|
"PIPE_LINE": [...["easy detailerFix"], ...pipeNodes, ...extraNodes]
|
|
}
|
|
},
|
|
"easy samLoaderPipe": {
|
|
"from":{
|
|
"PIPE_LINE": [...["easy preDetailerFix"], ...pipeNodes, ...extraNodes]
|
|
}
|
|
},
|
|
"easy ultralyticsDetectorPipe": {
|
|
"from":{
|
|
"PIPE_LINE": [...["easy preDetailerFix"], ...pipeNodes, ...extraNodes]
|
|
}
|
|
},
|
|
// cascade相关
|
|
"easy cascadeLoader":{
|
|
"from": {
|
|
"PIPE_LINE": [ ...["easy fullCascadeKSampler", 'easy preSamplingCascade'], ...controlNetNodes, ...pipeNodes, ...extraNodes],
|
|
"MODEL": modelNormalNodes.filter(cate => !ipadapterNodes.includes(cate))
|
|
}
|
|
},
|
|
"easy fullCascadeKSampler":{
|
|
"from": {
|
|
"PIPE_LINE": [ ...["easy preSampling", "easy preSamplingAdvanced"], ...pipeNodes, ...extraNodes]
|
|
}
|
|
},
|
|
"easy preSamplingCascade":{
|
|
"from": {
|
|
"PIPE_LINE": [ ...["easy cascadeKSampler",], ...pipeNodes, ...extraNodes]
|
|
}
|
|
},
|
|
"easy cascadeKSampler": {
|
|
"from": {
|
|
"PIPE_LINE": [ ...["easy preSampling", "easy preSamplingAdvanced"], ...pipeNodes, ...extraNodes]
|
|
}
|
|
},
|
|
}
|
|
|
|
class NullGraphError extends Error {
|
|
constructor(message="Attempted to access LGraph reference that was null or undefined.", cause) {
|
|
super(message, {cause})
|
|
this.name = "NullGraphError"
|
|
}
|
|
}
|
|
app.registerExtension({
|
|
name: "comfy.easyuse.suggestions",
|
|
async setup() {
|
|
const createDefaultNodeForSlot = LGraphCanvas.prototype.createDefaultNodeForSlot;
|
|
LGraphCanvas.prototype.createDefaultNodeForSlot = function(optPass) { // addNodeMenu for connection
|
|
const opts = Object.assign({ nodeFrom: null // input
|
|
,slotFrom: null // input
|
|
,nodeTo: null // output
|
|
,slotTo: null // output
|
|
,position: [] // pass the event coords
|
|
,nodeType: null // choose a nodetype to add, AUTO to set at first good
|
|
,posAdd:[0,0] // adjust x,y
|
|
,posSizeFix:[0,0] // alpha, adjust the position x,y based on the new node size w,h
|
|
}
|
|
, optPass || {}
|
|
);
|
|
const { afterRerouteId } = opts
|
|
const that = this;
|
|
|
|
const isFrom = opts.nodeFrom && opts.slotFrom!==null;
|
|
const isTo = !isFrom && opts.nodeTo && opts.slotTo!==null;
|
|
const node = isFrom ? opts.nodeFrom : opts.nodeTo
|
|
// Not an Easy Use node, skip showConnectionMenu hijack
|
|
if(!node || !Object.keys(suggestions).includes(node.type)){
|
|
return createDefaultNodeForSlot.call(this, optPass)
|
|
}
|
|
|
|
if (!isFrom && !isTo){
|
|
console.warn("No data passed to createDefaultNodeForSlot "+opts.nodeFrom+" "+opts.slotFrom+" "+opts.nodeTo+" "+opts.slotTo);
|
|
return false;
|
|
}
|
|
if (!opts.nodeType){
|
|
console.warn("No type to createDefaultNodeForSlot");
|
|
return false;
|
|
}
|
|
|
|
const nodeX = isFrom ? opts.nodeFrom : opts.nodeTo;
|
|
const nodeType = nodeX.type
|
|
let slotX = isFrom ? opts.slotFrom : opts.slotTo;
|
|
|
|
let iSlotConn = false;
|
|
switch (typeof slotX){
|
|
case "string":
|
|
iSlotConn = isFrom ? nodeX.findOutputSlot(slotX,false) : nodeX.findInputSlot(slotX,false);
|
|
slotX = isFrom ? nodeX.outputs[slotX] : nodeX.inputs[slotX];
|
|
break;
|
|
case "object":
|
|
// ok slotX
|
|
iSlotConn = isFrom ? nodeX.findOutputSlot(slotX.name) : nodeX.findInputSlot(slotX.name);
|
|
break;
|
|
case "number":
|
|
iSlotConn = slotX;
|
|
slotX = isFrom ? nodeX.outputs[slotX] : nodeX.inputs[slotX];
|
|
break;
|
|
case "undefined":
|
|
default:
|
|
// bad ?
|
|
//iSlotConn = 0;
|
|
console.warn("Cant get slot information "+slotX);
|
|
return false;
|
|
}
|
|
|
|
if (slotX===false || iSlotConn===false){
|
|
console.warn("createDefaultNodeForSlot bad slotX "+slotX+" "+iSlotConn);
|
|
}
|
|
|
|
// check for defaults nodes for this slottype
|
|
var fromSlotType = slotX.type==LiteGraph.EVENT?"_event_":slotX.type;
|
|
var slotTypesDefault = isFrom ? LiteGraph.slot_types_default_out : LiteGraph.slot_types_default_in;
|
|
if(slotTypesDefault && slotTypesDefault[fromSlotType]){
|
|
if (slotX.link !== null) {
|
|
// is connected
|
|
}else{
|
|
// is not not connected
|
|
}
|
|
let nodeNewType = false;
|
|
const fromOrTo = isFrom ? 'from' : 'to'
|
|
if(suggestions[nodeType] && suggestions[nodeType][fromOrTo] && suggestions[nodeType][fromOrTo][fromSlotType]?.length>0){
|
|
for(var typeX in suggestions[nodeType][fromOrTo][fromSlotType]){
|
|
if (opts.nodeType == suggestions[nodeType][fromOrTo][fromSlotType][typeX] || opts.nodeType == "AUTO") {
|
|
nodeNewType = suggestions[nodeType][fromOrTo][fromSlotType][typeX];
|
|
break
|
|
}
|
|
}
|
|
}
|
|
else if(typeof slotTypesDefault[fromSlotType] == "object" || typeof slotTypesDefault[fromSlotType] == "array"){
|
|
for(var typeX in slotTypesDefault[fromSlotType]){
|
|
if (opts.nodeType == slotTypesDefault[fromSlotType][typeX] || opts.nodeType == "AUTO"){
|
|
nodeNewType = slotTypesDefault[fromSlotType][typeX];
|
|
break;
|
|
}
|
|
}
|
|
}else{
|
|
if (opts.nodeType == slotTypesDefault[fromSlotType] || opts.nodeType == "AUTO") nodeNewType = slotTypesDefault[fromSlotType];
|
|
}
|
|
if (nodeNewType) {
|
|
var nodeNewOpts = false;
|
|
if (typeof nodeNewType == "object" && nodeNewType.node){
|
|
nodeNewOpts = nodeNewType;
|
|
nodeNewType = nodeNewType.node;
|
|
}
|
|
|
|
//that.graph.beforeChange();
|
|
|
|
var newNode = LiteGraph.createNode(nodeNewType);
|
|
if(newNode){
|
|
// if is object pass options
|
|
if (nodeNewOpts){
|
|
if (nodeNewOpts.properties) {
|
|
for (var i in nodeNewOpts.properties) {
|
|
newNode.addProperty( i, nodeNewOpts.properties[i] );
|
|
}
|
|
}
|
|
if (nodeNewOpts.inputs) {
|
|
newNode.inputs = [];
|
|
for (var i in nodeNewOpts.inputs) {
|
|
newNode.addOutput(
|
|
nodeNewOpts.inputs[i][0],
|
|
nodeNewOpts.inputs[i][1]
|
|
);
|
|
}
|
|
}
|
|
if (nodeNewOpts.outputs) {
|
|
newNode.outputs = [];
|
|
for (var i in nodeNewOpts.outputs) {
|
|
newNode.addOutput(
|
|
nodeNewOpts.outputs[i][0],
|
|
nodeNewOpts.outputs[i][1]
|
|
);
|
|
}
|
|
}
|
|
if (nodeNewOpts.title) {
|
|
newNode.title = nodeNewOpts.title;
|
|
}
|
|
if (nodeNewOpts.json) {
|
|
newNode.configure(nodeNewOpts.json);
|
|
}
|
|
|
|
}
|
|
|
|
// add the node
|
|
that.graph.add(newNode);
|
|
newNode.pos = [ opts.position[0]+opts.posAdd[0]+(opts.posSizeFix[0]?opts.posSizeFix[0]*newNode.size[0]:0)
|
|
,opts.position[1]+opts.posAdd[1]+(opts.posSizeFix[1]?opts.posSizeFix[1]*newNode.size[1]:0)]; //that.last_click_position; //[e.canvasX+30, e.canvasX+5];*/
|
|
|
|
// connect the two!
|
|
if (isFrom){
|
|
opts.nodeFrom.connectByType( iSlotConn, newNode, fromSlotType );
|
|
}else{
|
|
opts.nodeTo.connectByTypeOutput( iSlotConn, newNode, fromSlotType );
|
|
}
|
|
|
|
return true;
|
|
|
|
}else{
|
|
console.log("failed creating "+nodeNewType);
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
let showConnectionMenu = LGraphCanvas.prototype.showConnectionMenu
|
|
LGraphCanvas.prototype.showConnectionMenu = function(optPass) { // addNodeMenu for connection
|
|
const opts = Object.assign({
|
|
nodeFrom: null, // input
|
|
slotFrom: null, // input
|
|
nodeTo: null, // output
|
|
slotTo: null, // output
|
|
e: undefined,
|
|
allow_searchbox: this.allow_searchbox,
|
|
showSearchBox: this.showSearchBox,
|
|
}
|
|
,optPass || {}
|
|
);
|
|
const that = this;
|
|
const { graph } = this
|
|
const { afterRerouteId } = opts
|
|
const isFrom = opts.nodeFrom && opts.slotFrom;
|
|
const isTo = !isFrom && opts.nodeTo && opts.slotTo;
|
|
const node = isFrom ? opts.nodeFrom : opts.nodeTo
|
|
|
|
// Not an Easy Use node, skip showConnectionMenu hijack
|
|
if(!node || !Object.keys(suggestions).includes(node.type)){
|
|
return showConnectionMenu.call(this, optPass)
|
|
}
|
|
|
|
if (!isFrom && !isTo){
|
|
console.warn("No data passed to showConnectionMenu");
|
|
return false;
|
|
}
|
|
|
|
const nodeX = isFrom ? opts.nodeFrom : opts.nodeTo;
|
|
if (!nodeX) throw new TypeError("nodeX was null when creating default node for slot.")
|
|
let slotX = isFrom ? opts.slotFrom : opts.slotTo;
|
|
|
|
let iSlotConn = false;
|
|
switch (typeof slotX){
|
|
case "string":
|
|
iSlotConn = isFrom ? nodeX.findOutputSlot(slotX,false) : nodeX.findInputSlot(slotX,false);
|
|
slotX = isFrom ? nodeX.outputs[slotX] : nodeX.inputs[slotX];
|
|
break;
|
|
case "object":
|
|
// ok slotX
|
|
iSlotConn = isFrom ? nodeX.findOutputSlot(slotX.name) : nodeX.findInputSlot(slotX.name);
|
|
break;
|
|
case "number":
|
|
iSlotConn = slotX;
|
|
slotX = isFrom ? nodeX.outputs[slotX] : nodeX.inputs[slotX];
|
|
break;
|
|
default:
|
|
// bad ?
|
|
//iSlotConn = 0;
|
|
console.warn("Cant get slot information "+slotX);
|
|
return false;
|
|
}
|
|
|
|
const options = ["Add Node", "Add Reroute", null]
|
|
if (opts.allow_searchbox){
|
|
options.push("Search");
|
|
options.push(null);
|
|
}
|
|
|
|
// get defaults nodes for this slottype
|
|
var fromSlotType = slotX.type==LiteGraph.EVENT?"_event_":slotX.type;
|
|
var slotTypesDefault = isFrom ? LiteGraph.slot_types_default_out : LiteGraph.slot_types_default_in;
|
|
var nodeType = nodeX.type
|
|
if(slotTypesDefault && slotTypesDefault[fromSlotType]){
|
|
const fromOrTo = isFrom ? 'from' : 'to'
|
|
if(suggestions[nodeType] && suggestions[nodeType][fromOrTo] && suggestions[nodeType][fromOrTo][fromSlotType]?.length>0){
|
|
for(var typeX in suggestions[nodeType][fromOrTo][fromSlotType]){
|
|
options.push(suggestions[nodeType][fromOrTo][fromSlotType][typeX]);
|
|
}
|
|
}
|
|
else if(typeof slotTypesDefault[fromSlotType] == "object" || typeof slotTypesDefault[fromSlotType] == "array"){
|
|
for(var typeX in slotTypesDefault[fromSlotType]){
|
|
options.push(slotTypesDefault[fromSlotType][typeX]);
|
|
}
|
|
}else{
|
|
options.push(slotTypesDefault[fromSlotType]);
|
|
}
|
|
}
|
|
|
|
// build menu
|
|
var menu = new LiteGraph.ContextMenu(options, {
|
|
event: opts.e,
|
|
extra: slotX,
|
|
title:
|
|
(slotX && slotX.name != ""
|
|
? slotX.name + (fromSlotType ? " | " : "")
|
|
: "") + (slotX && fromSlotType ? fromSlotType : ""),
|
|
callback: inner_clicked,
|
|
});
|
|
|
|
// callback
|
|
function inner_clicked(v,options,e) {
|
|
//console.log("Process showConnectionMenu selection");
|
|
switch (v) {
|
|
case "Add Node":
|
|
LGraphCanvas.onMenuAdd(null, null, e, menu, function (node) {
|
|
if (!node) return
|
|
|
|
if (isFrom) {
|
|
opts.nodeFrom?.connectByType(iSlotConn, node, fromSlotType, { afterRerouteId })
|
|
} else {
|
|
opts.nodeTo?.connectByTypeOutput(iSlotConn, node, fromSlotType, { afterRerouteId })
|
|
}
|
|
})
|
|
break;
|
|
case "Add Reroute":
|
|
const node = isFrom ? opts.nodeFrom : opts.nodeTo
|
|
const slot = options.extra
|
|
if (!graph) throw new NullGraphError()
|
|
if (!node) throw new TypeError("Cannot add reroute: node was null")
|
|
if (!slot) throw new TypeError("Cannot add reroute: slot was null")
|
|
if (!opts.e) throw new TypeError("Cannot add reroute: CanvasPointerEvent was null")
|
|
|
|
const reroute = node.connectFloatingReroute([opts.e.canvasX, opts.e.canvasY], slot, afterRerouteId)
|
|
if (!reroute) throw new Error("Failed to create reroute")
|
|
|
|
that.dirty_canvas = true
|
|
that.dirty_bgcanvas = true
|
|
break
|
|
case "Search":
|
|
if(isFrom){
|
|
opts.showSearchBox(e,{node_from: opts.nodeFrom, slot_from: slotX, type_filter_in: fromSlotType});
|
|
}else{
|
|
opts.showSearchBox(e,{node_to: opts.nodeTo, slot_from: slotX, type_filter_out: fromSlotType});
|
|
}
|
|
break;
|
|
default:
|
|
const customProps = {
|
|
position: [opts.e?.canvasX ?? 0, opts.e?.canvasY ?? 0],
|
|
nodeType: v,
|
|
afterRerouteId,
|
|
}
|
|
// check for defaults nodes for this slottype
|
|
that.createDefaultNodeForSlot(Object.assign(opts, customProps))
|
|
break;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
};
|
|
}
|
|
}) |