From 187c80d213731414f8462265de8191eceb10b448 Mon Sep 17 00:00:00 2001 From: Johnpaul Date: Wed, 24 Dec 2025 02:14:58 +0100 Subject: [PATCH] feat: invoke getCustomVueWidgets hook when registering extensions --- src/services/extensionService.ts | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/src/services/extensionService.ts b/src/services/extensionService.ts index 75b159e11e..a8c1a19012 100644 --- a/src/services/extensionService.ts +++ b/src/services/extensionService.ts @@ -2,6 +2,7 @@ import { useCurrentUser } from '@/composables/auth/useCurrentUser' import { useErrorHandling } from '@/composables/useErrorHandling' import { legacyMenuCompat } from '@/lib/litegraph/src/contextMenuCompat' import { useSettingStore } from '@/platform/settings/settingStore' +import { registerVueWidgets } from '@/renderer/extensions/vueNodes/widgets/registry/widgetRegistry' import { api } from '@/scripts/api' import { app } from '@/scripts/app' import { useCommandStore } from '@/stores/commandStore' @@ -80,6 +81,15 @@ export const useExtensionService = () => { })() } + if (extension.getCustomVueWidgets) { + void (async () => { + if (extension.getCustomVueWidgets) { + const vueWidgets = await extension.getCustomVueWidgets(app) + registerVueWidgets(vueWidgets) + } + })() + } + if (extension.onAuthUserResolved) { const { onUserResolved } = useCurrentUser() const handleUserResolved = wrapWithErrorHandlingAsync(