From 5e600e6a6ad3feb944a0fd37a59b3e5bc31676e8 Mon Sep 17 00:00:00 2001
From: ssit <ssit5568@gmail.com>
Date: Mon, 11 Sep 2023 13:49:06 -0400
Subject: [PATCH] Avoid primitives being able to connect to wildcard reroutes

---
 web/extensions/core/rerouteNode.js | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/web/extensions/core/rerouteNode.js b/web/extensions/core/rerouteNode.js
index 0b65a51e8..9a11ff310 100644
--- a/web/extensions/core/rerouteNode.js
+++ b/web/extensions/core/rerouteNode.js
@@ -111,12 +111,13 @@ app.registerExtension({
 								} else {
 									// We've found an output
 									const nodeOutType = node.inputs && node.inputs[link?.target_slot] && node.inputs[link.target_slot].type ? node.inputs[link.target_slot].type : null;
-									outputWidget = node.inputs[link.target_slot]?.widget ?? null;
 									if (inputType && nodeOutType !== inputType && inputType !== "*") {
 										// The output doesnt match our input and is not a wildcard so disconnect it
 										node.disconnectInput(link.target_slot);
 									} else {
 										outputType = nodeOutType;
+										// We are not disconnecting the output node, so store the widget of the output node
+										outputWidget = node.inputs[link.target_slot]?.widget ?? null;
 									}
 								}
 							}
@@ -141,9 +142,10 @@ app.registerExtension({
 						node.__outputType = displayType;
 						node.outputs[0].name = node.properties.showOutputText ? displayType : "";
 						if (outputWidget) {
-							// Inherit the widget from the connected input 
+							// Inherit the widget from the connected input
 							node.inputs[0].widget = outputWidget;
 						} else {
+							// Delete the widget if no output widget is found
 							delete node.inputs[0].widget;
 						}
 						node.size = node.computeSize();