🐛 fix: fix compatible with ssr

lobehub · Nov 21, 2023 · 2337060 · 2337060
1 parent 89980b4
commit 2337060
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 8 deletions.
diff --git a/src/client/fetch/message.ts b/src/client/fetch/message.ts
@@ -4,16 +4,20 @@ import { PluginChannel } from '../const';
 
 export const fetchPluginMessage = <T = any>() =>
   new Promise<T>((resolve) => {
+    if (typeof window === 'undefined') {
+      resolve(undefined as any);
+      return;
+    }
     const receiverData = (e: MessageEvent) => {
       if (e.data.type === PluginChannel.renderPlugin) {
         const props = e.data.props as PluginRenderProps<T>;
         resolve(props.content as T);
 
-        window?.removeEventListener('message', receiverData);
+        window.removeEventListener('message', receiverData);
       }
     };
 
-    window?.addEventListener('message', receiverData);
+    window.addEventListener('message', receiverData);
 
     top?.postMessage({ type: PluginChannel.fetchPluginMessage }, '*');
   });
diff --git a/src/client/fetch/pluginPayload.ts b/src/client/fetch/pluginPayload.ts
@@ -9,18 +9,22 @@ export interface PluginPayload<T = any> {
 
 export const fetchPluginPayload = <T = any>() =>
   new Promise<PluginPayload<T>>((resolve) => {
+    if (typeof window === 'undefined') {
+      resolve(undefined as any);
+      return;
+    }
     const receiverData = (e: MessageEvent) => {
       if (e.data.type === PluginChannel.initStandalonePlugin) {
         const payload = e.data.props as PluginRequestPayload;
         const func = payload.apiName;
         const args = JSON.parse(payload.arguments || '{}');
         resolve({ arguments: args, name: func });
 
-        window?.removeEventListener('message', receiverData);
+        window.removeEventListener('message', receiverData);
       }
     };
 
-    window?.addEventListener('message', receiverData);
+    window.addEventListener('message', receiverData);
 
     top?.postMessage({ type: PluginChannel.pluginReadyForRender }, '*');
   });
diff --git a/src/client/fetch/pluginSettings.ts b/src/client/fetch/pluginSettings.ts
@@ -2,15 +2,20 @@ import { PluginChannel } from '../const';
 
 export const fetchPluginSettings = <T = any>() =>
   new Promise<T>((resolve) => {
+    if (typeof window === 'undefined') {
+      resolve(undefined as any);
+      return;
+    }
+
     const receiverData = (e: MessageEvent) => {
       if (e.data.type === PluginChannel.renderPluginSettings) {
         resolve(e.data.value);
 
-        window?.removeEventListener('message', receiverData);
+        window.removeEventListener('message', receiverData);
       }
     };
 
-    window?.addEventListener('message', receiverData);
+    window.addEventListener('message', receiverData);
 
     top?.postMessage({ type: PluginChannel.fetchPluginSettings }, '*');
   });
diff --git a/src/client/fetch/pluginState.ts b/src/client/fetch/pluginState.ts
@@ -2,15 +2,20 @@ import { PluginChannel } from '../const';
 
 export const fetchPluginState = <T = any>(key: string) =>
   new Promise<T>((resolve) => {
+    if (typeof window === 'undefined') {
+      resolve(undefined as any);
+      return;
+    }
+
     const receiverData = (e: MessageEvent) => {
       if (e.data.type === PluginChannel.renderPluginState && e.data.key === key) {
         resolve(e.data.value);
 
-        window?.removeEventListener('message', receiverData);
+        window.removeEventListener('message', receiverData);
       }
     };
 
-    window?.addEventListener('message', receiverData);
+    window.addEventListener('message', receiverData);
 
     top?.postMessage({ key, type: PluginChannel.fetchPluginState }, '*');
   });