feat: 新增expand-app装饰器

JessYan0913 · Jul 20, 2023 · c38cc8a · c38cc8a
1 parent c617742
commit c38cc8a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/packages/core/src/decorators/expand-app.ts b/packages/core/src/decorators/expand-app.ts
@@ -0,0 +1,9 @@
+import { App } from '../app';
+import { Plugin } from '../types';
+
+export function expandApp<T extends Plugin = Plugin>(methodName?: string) {
+  return function (target: T, propertyKey: string, descriptor: PropertyDescriptor) {
+    methodName = methodName ?? propertyKey;
+    Reflect.set(App.prototype, methodName, descriptor.value);
+  };
+}
diff --git a/packages/core/src/index.ts b/packages/core/src/index.ts
@@ -1,2 +1,3 @@
 export * from './app';
 export * from './types';
+export * from './decorators/expand-app';