🐛 fix: fetchPluginMessage 支持定义出参泛型

lobehub · Sep 5, 2023 · e3059da · e3059da
1 parent b47e60f
commit e3059da
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/render/message.ts b/src/render/message.ts
@@ -1,11 +1,11 @@
 import { PluginChannel } from './const';
 import { onReceiveData } from './utils';
 
-export const fetchPluginMessage = () =>
-  new Promise<any>((resolve) => {
+export const fetchPluginMessage = <T = any>() =>
+  new Promise<T>((resolve) => {
     const receiverData = (e: MessageEvent) => {
       onReceiveData(e, (data) => {
-        resolve(data.content);
+        resolve(data.content as T);
         window.removeEventListener('message', receiverData);
       });
     };