From 8d4b5e03a8a5bfdd0b45d6f46a227d7414a0aba2 Mon Sep 17 00:00:00 2001
From: Taremin <taremin.vr@gmail.com>
Date: Mon, 22 Jan 2024 18:28:50 +0900
Subject: [PATCH] =?UTF-8?q?fix:=20=E5=8F=AF=E5=A4=89=E3=82=BD=E3=82=B1?=
 =?UTF-8?q?=E3=83=83=E3=83=88=E4=BB=A5=E5=A4=96=E3=81=AFPrimitiveNode?=
 =?UTF-8?q?=E3=81=A8=E6=8E=A5=E7=B6=9A=E3=81=A7=E3=81=8D=E3=82=8B=E3=82=88?=
 =?UTF-8?q?=E3=81=86=E3=81=AB=E3=81=99=E3=82=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/index.ts | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/src/index.ts b/src/index.ts
index 0bc1100..993f444 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -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