feat: pass extensionContext to setup function

kermanx · Oct 26, 2024 · 87935ee · 87935ee
1 parent 73d772d
commit 87935ee
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/packages/core/src/utils/defineExtension.ts b/packages/core/src/utils/defineExtension.ts
@@ -15,13 +15,13 @@ export const extensionScope = effectScope()
  *
  * @category lifecycle
  */
-export function defineExtension<T>(setup: () => T) {
+export function defineExtension<T>(setup: (context: ExtensionContext) => T) {
   return {
-    activate: (extCtx: ExtensionContext) => {
-      extensionContext.value = extCtx
+    activate: (context: ExtensionContext) => {
+      extensionContext.value = context
       return extensionScope.run(() => {
-        activateCbs.map(fn => fn(extCtx))
-        return setup()
+        activateCbs.map(fn => fn(context))
+        return setup(context)
       })
     },
     deactivate: () => {