From 73b907953860d1a5bff2a4a220919e06d2d2f2e1 Mon Sep 17 00:00:00 2001
From: Johnpaul <polymenjohn1@gmail.com>
Date: Tue, 13 Jan 2026 20:46:47 +0100
Subject: [PATCH] fix: remove redundant guard and export VueWidgets type

- Remove nested if check in getCustomVueWidgets registration by
  capturing method reference before async IIFE
- Export VueWidgets type for use by external extension code
---
 src/services/extensionService.ts | 7 +++----
 src/types/comfy.ts               | 2 +-
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/src/services/extensionService.ts b/src/services/extensionService.ts
index a8c1a1901..6bca3c031 100644
--- a/src/services/extensionService.ts
+++ b/src/services/extensionService.ts
@@ -82,11 +82,10 @@ export const useExtensionService = () => {
     }
 
     if (extension.getCustomVueWidgets) {
+      const getVueWidgets = extension.getCustomVueWidgets
       void (async () => {
-        if (extension.getCustomVueWidgets) {
-          const vueWidgets = await extension.getCustomVueWidgets(app)
-          registerVueWidgets(vueWidgets)
-        }
+        const vueWidgets = await getVueWidgets(app)
+        registerVueWidgets(vueWidgets)
       })()
     }
 
diff --git a/src/types/comfy.ts b/src/types/comfy.ts
index d1c5f2139..b89956780 100644
--- a/src/types/comfy.ts
+++ b/src/types/comfy.ts
@@ -31,7 +31,7 @@ export interface VueWidgetDefinition {
   aliases?: string[]
 }
 
-type VueWidgets = Record<string, VueWidgetDefinition>
+export type VueWidgets = Record<string, VueWidgetDefinition>
 
 export interface AboutPageBadge {
   label: string