🐛 fix: fix fetchPluginState bug

lobehub · Oct 24, 2023 · d2ff3b6 · d2ff3b6
1 parent 3c9ad14
commit d2ff3b6
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 8 deletions.
diff --git a/src/client/const.ts b/src/client/const.ts
@@ -5,5 +5,6 @@ export enum PluginChannel {
   initStandalonePlugin = 'lobe-chat:init-standalone-plugin',
   pluginReadyForRender = 'lobe-chat:plugin-ready-for-render',
   renderPlugin = 'lobe-chat:render-plugin',
+  renderPluginState = 'lobe-chat:render-plugin-state',
   updatePluginState = 'lobe-chat:update-plugin-state',
 }
diff --git a/src/client/fetch/message.ts b/src/client/fetch/message.ts
@@ -1,13 +1,16 @@
+import { PluginRenderProps } from '@/client';
+
 import { PluginChannel } from '../const';
-import { onReceiveData } from '../utils';
 
 export const fetchPluginMessage = <T = any>() =>
   new Promise<T>((resolve) => {
     const receiverData = (e: MessageEvent) => {
-      onReceiveData(e, (data) => {
-        resolve(data.content as T);
+      if (e.data.type === PluginChannel.renderPlugin) {
+        const props = e.data.props as PluginRenderProps<T>;
+        resolve(props.content as T);
+
         window.removeEventListener('message', receiverData);
-      });
+      }
     };
 
     window.addEventListener('message', receiverData);

diff --git a/src/client/fetch/pluginState.ts b/src/client/fetch/pluginState.ts
@@ -1,13 +1,13 @@
 import { PluginChannel } from '../const';
-import { onReceiveData } from '../utils';
 
 export const fetchPluginState = <T = any>() =>
   new Promise<T>((resolve) => {
     const receiverData = (e: MessageEvent) => {
-      onReceiveData(e, (data) => {
-        resolve(data.content as T);
+      if (e.data.type === PluginChannel.renderPluginState) {
+        resolve(e.data.props);
+
         window.removeEventListener('message', receiverData);
-      });
+      }
     };
 
     window.addEventListener('message', receiverData);