From 63d874bd93d52d52370e762d4e9ec4997da94efa Mon Sep 17 00:00:00 2001
From: Chenlei Hu <chenlei.hu@mail.utoronto.ca>
Date: Mon, 22 Jul 2024 10:09:38 -0400
Subject: [PATCH] Close side bar on menu location set as disabled (#194)

---
 src/components/sidebar/SideToolBar.vue | 20 +++++++++++++++++++-
 src/scripts/ui/menu/index.ts           |  1 +
 2 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/src/components/sidebar/SideToolBar.vue b/src/components/sidebar/SideToolBar.vue
index f2e58746d..d210307d0 100644
--- a/src/components/sidebar/SideToolBar.vue
+++ b/src/components/sidebar/SideToolBar.vue
@@ -22,7 +22,7 @@ import SideBarThemeToggleIcon from "./SideBarThemeToggleIcon.vue";
 import SideBarSettingsToggleIcon from "./SideBarSettingsToggleIcon.vue";
 import NodeDetailSideBarItem from "./items/NodeDetailSideBarItem.vue";
 import QueueSideBarItem from "./items/QueueSideBarItem.vue";
-import { markRaw, ref, watch } from "vue";
+import { markRaw, onMounted, onUnmounted, ref, watch } from "vue";
 
 const items = ref([
   { icon: "pi pi-map", component: markRaw(NodeDetailSideBarItem) },
@@ -42,6 +42,24 @@ const emit = defineEmits(["change"]);
 watch(selectedItem, (newVal) => {
   emit("change", newVal !== null);
 });
+
+const onBetaMenuDisabled = () => {
+  selectedItem.value = null;
+};
+
+onMounted(() => {
+  document.addEventListener(
+    "comfy:setting:beta-menu-disabled",
+    onBetaMenuDisabled
+  );
+});
+
+onUnmounted(() => {
+  document.removeEventListener(
+    "comfy:setting:beta-menu-disabled",
+    onBetaMenuDisabled
+  );
+});
 </script>
 
 <style>
diff --git a/src/scripts/ui/menu/index.ts b/src/scripts/ui/menu/index.ts
index e8e26411a..f593de0ad 100644
--- a/src/scripts/ui/menu/index.ts
+++ b/src/scripts/ui/menu/index.ts
@@ -182,6 +182,7 @@ export class ComfyAppMenu {
           app.ui.menuContainer.style.removeProperty("display");
           this.element.style.display = "none";
           app.ui.restoreMenuPosition();
+          document.dispatchEvent(new Event("comfy:setting:beta-menu-disabled"));
         }
         window.dispatchEvent(new Event("resize"));
       },