From cacbe3ee1a4c6af8907dc523df3e54909e494492 Mon Sep 17 00:00:00 2001
From: "Alexander G. Morano" <amorano@gmail.com>
Date: Tue, 18 Mar 2025 04:02:05 -0400
Subject: [PATCH] * vectors now convert in-line

---
 README.md                   |  2 +-
 web/widget/widget_vector.js | 55 +++++++++++++++++++++----------------
 2 files changed, 33 insertions(+), 24 deletions(-)

diff --git a/README.md b/README.md
index 92d13af..f1c049a 100644
--- a/README.md
+++ b/README.md
@@ -96,7 +96,7 @@ You can colorize nodes via their `title background`, `node body` or `title text`
 
 **IF YOU ARE USING AN OLD VERSION THAT HAS COLOR CONFIGURATIONS THAT NO LONGER WORK, I CAN HELP YOU MIGRATE THEM. JUST START A TICKET OR SEND A MESSAGE**
 
-**2025/03/17** @1.7.40:
+**2025/03/18** @1.7.40:
 * allow vectors to be inline inputs + widget cause waiting for ComfyUI team is painful
 * changed precision default to 3
 * merged new action trigger from ComfyUI team
diff --git a/web/widget/widget_vector.js b/web/widget/widget_vector.js
index b13bd23..7e59b15 100644
--- a/web/widget/widget_vector.js
+++ b/web/widget/widget_vector.js
@@ -11,38 +11,39 @@ const TYPES = ['RGB', 'VEC2', 'VEC3', 'VEC4', 'VEC2INT', 'VEC3INT', 'VEC4INT']
 const TYPES_ACCEPT = 'RGB, VEC2, VEC3, VEC4, VEC2INT, VEC3INT, VEC4INT, FLOAT, INT, BOOLEAN'
 
 function convertToInput(node, widget) {
-    // hideWidget(node, widget);
-    //const { type } = getWidgetType(config);
-    // const [oldWidth, oldHeight] = node.size;
-    /*
-    const config = [
-        widget.type,
-        widget.options || {}
-    ];
-    */
-    //const inputIsOptional = !!widget.options?.inputIsOptional;
-    //widget.options.type = TYPES_ACCEPT;
     const input = node.addInput(widget.name, widget.type, {
         // @ts-expect-error [GET_CONFIG] is not a valid property of IWidget
         widget: { name: widget.name, [GET_CONFIG]: () => [
             widget.type,
             widget.options || {}
         ] },
-        //...inputIsOptional ? { shape: LiteGraph.SlotShape.HollowCircle } : {}
         ...{ shape: LiteGraph.SlotShape.HollowCircle }
     });
-    /*
-    for (const widget2 of node.widgets) {
-        widget2.last_y += LiteGraph.NODE_SLOT_HEIGHT;
-    }
-    node.setSize([
-        Math.max(oldWidth, node.size[0]),
-        Math.max(oldHeight, node.size[1])
-    ]);
-    */
     return input;
 }
 
+function removeConvertToWidgetEntries(menuArray, widgetNames) {
+    function recursiveFilter(menu) {
+        return menu.filter(item => {
+            // Keep non-matching items
+            if (!item || !item.content) return true;
+
+            // Remove this item
+            if (widgetNames.some(name => item.content === `Convert ${name} to Widget`)) {
+                return false;
+            }
+
+            // Recursively filter submenus if they exist
+            if (item.submenu && item.submenu.options) {
+                item.submenu.options = recursiveFilter(item.submenu.options);
+            }
+            return true;
+        });
+    }
+
+    return recursiveFilter(menuArray);
+}
+
 function isVersionLess(v1, v2) {
     const parts1 = v1.split('.').map(Number);
     const parts2 = v2.split('.').map(Number);
@@ -350,9 +351,10 @@ app.registerExtension({
             return;
         }
 
+        const getExtraMenuOptions = nodeType.prototype.getExtraMenuOptions;
+
         const version = window.__COMFYUI_FRONTEND_VERSION__;
         if (!isVersionLess(version, "1.10.3")) {
-
             const onNodeCreated = nodeType.prototype.onNodeCreated;
             nodeType.prototype.onNodeCreated = async function () {
                 const me = onNodeCreated?.apply(this);
@@ -364,11 +366,18 @@ app.registerExtension({
                 });
                 return me;
             }
+
+            nodeType.prototype.getExtraMenuOptions = function (_, options) {
+                const me = getExtraMenuOptions?.apply(this, arguments);
+                const widgets = matchingTypes.map(item => item[0]);
+                options = removeConvertToWidgetEntries(options, widgets);
+                console.info(options)
+                return me;
+            }
             return;
         }
 
         // MENU CONVERSIONS
-        const getExtraMenuOptions = nodeType.prototype.getExtraMenuOptions;
         nodeType.prototype.getExtraMenuOptions = function (_, options) {
             const me = getExtraMenuOptions?.apply(this, arguments);
             const widgetToInputArray = [];