nodes/Taremin-comfyui-string-tools
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: 可変ソケット以外はPrimitiveNodeと接続できるようにする

Browse Source
This commit is contained in:
Taremin
2024-01-22 18:28:50 +09:00
parent cd1d8177e0
commit 8d4b5e03a8
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/index.ts
+8 -5
View File
@@ -69,7 +69,7 @@ function createCallback(nodename: string, basename: string, inputType: string) {
const onConnectInputOriginal = this.onConnectInput
this.onConnectInput = function(targetSlot: number, type: string, output:any, originNode:any, originSlot:number) {
let retVal = onConnectInputOriginal ? onConnectInputOriginal.apply(this, arguments) : void 0
if (originNode.type === "PrimitiveNode") {
if (originNode.type === "PrimitiveNode" && getInputBasename(this.inputs[targetSlot]) === basename) {
return false
}
this.removeCancel = targetSlot
@@ -81,11 +81,14 @@ function createCallback(nodename: string, basename: string, inputType: string) {
this.onInputDblClick = function(slot: number) {
if (onInputDblClickOriginal) {
const originalCreateNode = LiteGraph.createNode
LiteGraph.createNode = function(nodeType: string) {
if (nodeData !== "PrimitiveNode") {
return originalCreateNode.apply(this, arguments)
if (getInputBasename(this.inputs[slot]) === basename) {
LiteGraph.createNode = function(nodeType: string) {
if (nodeType !== "PrimitiveNode") {
return originalCreateNode.apply(this, arguments)
}
return originalCreateNode.call(this, "StringToolsText")
}
return originalCreateNode.call(this, "StringToolsText")
}
onInputDblClickOriginal.call(this, slot)
LiteGraph.createNode = originalCreateNode