From 21679435e531e729a4aea494e6cb9b7152ecdf75 Mon Sep 17 00:00:00 2001
From: DepFA <35278260+dfaker@users.noreply.github.com>
Date: Sat, 8 Oct 2022 05:46:42 +0100
Subject: [PATCH] implement removal

---
 javascript/contextMenus.js | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/javascript/contextMenus.js b/javascript/contextMenus.js
index 99d1d3f7..2d82269f 100644
--- a/javascript/contextMenus.js
+++ b/javascript/contextMenus.js
@@ -79,7 +79,13 @@ contextMenuInit = function(){
   }
 
   function removeContextMenuOption(uid){
-
+    menuSpecs.forEach(function(v,k) {
+      let index = -1
+      v.forEach(function(e,ei){if(e['id']==uid){index=ei}})
+      if(index>=0){
+        v.splice(index, 1);
+      }
+    })
   }
 
   function addContextMenuEventListener(){
@@ -148,7 +154,8 @@ cancelGenerateForever = function(){
 }
 
 appendContextMenuOption('#txt2img_interrupt','Cancel generate forever',cancelGenerateForever)
-appendContextMenuOption('#txt2img_generate','Cancel generate forever',cancelGenerateForever)
+appendContextMenuOption('#txt2img_generate', 'Cancel generate forever',cancelGenerateForever)
+
 
 appendContextMenuOption('#roll','Roll three',
   function(){ 
@@ -162,4 +169,4 @@ appendContextMenuOption('#roll','Roll three',
 
 onUiUpdate(function(){
   addContextMenuEventListener()
-});
\ No newline at end of file
+});