From e32072ab3465827c4d0d3124e6f5809ea0ca9c3c Mon Sep 17 00:00:00 2001
From: johnqiao <qiaoshun8888@gmail.com>
Date: Fri, 24 Nov 2023 19:06:40 -0700
Subject: [PATCH] Fix the original missing menu options

---
 js/comfyui-manager.js | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/js/comfyui-manager.js b/js/comfyui-manager.js
index d2e64916..091185c5 100644
--- a/js/comfyui-manager.js
+++ b/js/comfyui-manager.js
@@ -766,10 +766,19 @@ app.registerExtension({
 	},
 
 	_addExtraNodeContextMenu(node, app) {
-	    node.prototype.getExtraMenuOptions = function (_, options) {
-	    	if (isOutputNode(node)) {
+        const origGetExtraMenuOptions = node.prototype.getExtraMenuOptions;
+        node.prototype.getExtraMenuOptions = function (_, options) {
+            origGetExtraMenuOptions?.apply?.(this, arguments);
+            if (isOutputNode(node)) {
 	    	    const { potential_outputs } = getPotentialOutputsAndOutputNodes([this]);
 	    	    const hasOutput = potential_outputs.length > 0;
+
+	    	    // Check if the previous menu option is `null`. If it's not,
+	    	    // then we need to add a `null` as a separator.
+	    	    if (options[options.length - 1] !== null) {
+	    	        options.push(null);
+	    	    }
+
                 options.push({
 				    content: "🏞️ Share Output",
 				    disabled: !hasOutput,
@@ -793,6 +802,6 @@ app.registerExtension({
 					}
 				}, null);
 			}
-	    }
+        }
 	},
 });