From 40880dbb59b25c864f18e3e185f902a6ed21eebc Mon Sep 17 00:00:00 2001
From: Chenlei Hu <huchenlei@proton.me>
Date: Thu, 7 Nov 2024 08:47:46 -0500
Subject: [PATCH] Move refresh button from actionbar to 'Edit' menu (#1451)

---
 src/components/actionbar/ComfyActionbar.vue | 15 ---------------
 src/stores/menuItemStore.ts                 |  1 +
 2 files changed, 1 insertion(+), 15 deletions(-)

diff --git a/src/components/actionbar/ComfyActionbar.vue b/src/components/actionbar/ComfyActionbar.vue
index 978e7a697e..5021de982c 100644
--- a/src/components/actionbar/ComfyActionbar.vue
+++ b/src/components/actionbar/ComfyActionbar.vue
@@ -8,16 +8,6 @@
       <span class="drag-handle cursor-move mr-2 p-0!" ref="dragHandleRef">
       </span>
       <ComfyQueueButton />
-      <Divider layout="vertical" class="mx-1" />
-      <ButtonGroup class="flex flex-nowrap">
-        <Button
-          v-tooltip.bottom="$t('menu.refresh')"
-          icon="pi pi-refresh"
-          severity="secondary"
-          text
-          @click="() => commandStore.execute('Comfy.RefreshNodeDefinitions')"
-        />
-      </ButtonGroup>
     </div>
   </Panel>
 </template>
@@ -25,12 +15,8 @@
 <script lang="ts" setup>
 import { computed, inject, nextTick, onMounted, Ref, ref, watch } from 'vue'
 import Panel from 'primevue/panel'
-import Divider from 'primevue/divider'
-import Button from 'primevue/button'
-import ButtonGroup from 'primevue/buttongroup'
 import ComfyQueueButton from './ComfyQueueButton.vue'
 import { useSettingStore } from '@/stores/settingStore'
-import { useCommandStore } from '@/stores/commandStore'
 import {
   useDraggable,
   useElementBounding,
@@ -42,7 +28,6 @@ import {
 import { clamp } from 'lodash'
 
 const settingsStore = useSettingStore()
-const commandStore = useCommandStore()
 
 const visible = computed(
   () => settingsStore.get('Comfy.UseNewMenu') !== 'Disabled'
diff --git a/src/stores/menuItemStore.ts b/src/stores/menuItemStore.ts
index 6d4762d1de..f294223388 100644
--- a/src/stores/menuItemStore.ts
+++ b/src/stores/menuItemStore.ts
@@ -95,6 +95,7 @@ export const useMenuItemStore = defineStore('menuItem', () => {
   )
 
   registerCommands(['Edit'], ['Comfy.Undo', 'Comfy.Redo'])
+  registerCommands(['Edit'], ['Comfy.RefreshNodeDefinitions'])
   registerCommands(['Edit'], ['Comfy.ClearWorkflow'])
   registerCommands(['Edit'], ['Comfy.OpenClipspace'])