feat(plugin): add ctx.getPluginApi method

Kingcrypto38 · Dec 15, 2023 · 5976534 · 5976534
1 parent 2466402
commit 5976534
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 2 deletions.
diff --git a/src/renderer/__tests__/core/plugin.ts b/src/renderer/__tests__/core/plugin.ts
@@ -19,4 +19,11 @@ test('plugin usage', () => {
   }, 123)
 
   expect(fn).toHaveBeenCalledTimes(1)
+
+  plugin.register({
+    name: 'test2',
+    register: () => '12345',
+  }, 123)
+
+  expect(plugin.getApi('test2')).toBe('12345')
 })
diff --git a/src/renderer/context/index.ts b/src/renderer/context/index.ts
@@ -69,6 +69,7 @@ const ctx = Object.freeze({
   showExtensionManager: extension.showManager,
   getExtensionLoadStatus: extension.getLoadStatus,
   getExtensionInitialized: extension.getInitialized,
+  getPluginApi: plugin.getApi,
   version: __APP_VERSION__,
 })
 

diff --git a/src/renderer/core/plugin.ts b/src/renderer/core/plugin.ts
@@ -4,10 +4,11 @@ const logger = getLogger('plugin')
 
 export interface Plugin<Ctx = any> {
   name: string;
-  register?: (ctx: Ctx) => void;
+  register?: (ctx: Ctx) => any;
 }
 
 const plugins: {[name: string]: Plugin} = {}
+const apis: {[name: string]: any} = {}
 
 /**
  * Register a plugin.
@@ -23,7 +24,16 @@ export function register <Ctx> (plugin: Plugin<Ctx>, ctx: Ctx) {
   }
 
   plugins[plugin.name] = plugin
-  plugin.register && plugin.register(ctx)
+  apis[plugin.name] = plugin.register && plugin.register(ctx)
+}
+
+/**
+ * Get a plugin exported api.
+ * @param name
+ * @returns
+ */
+export function getApi <T = any> (name: string): T {
+  return apis[name]
 }
 
 /**