🐛 fix: fix compatible with ssr

lobehub · Nov 20, 2023 · e276fc8 · e276fc8
1 parent 15dda5e
commit e276fc8
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 10 deletions.
diff --git a/src/client/fetch/message.ts b/src/client/fetch/message.ts
@@ -9,11 +9,11 @@ export const fetchPluginMessage = <T = any>() =>
         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
@@ -16,11 +16,11 @@ export const fetchPluginPayload = <T = any>() =>
         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
@@ -6,11 +6,11 @@ export const fetchPluginSettings = <T = any>() =>
       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
@@ -6,11 +6,11 @@ export const fetchPluginState = <T = any>(key: string) =>
       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 }, '*');
   });
diff --git a/src/client/hooks/useWatchPluginMessage.ts b/src/client/hooks/useWatchPluginMessage.ts
@@ -17,12 +17,12 @@ export const useWatchPluginMessage = <T = any>() => {
   };
 
   useEffect(() => {
-    window.addEventListener('message', receiverData);
+    window?.addEventListener('message', receiverData);
 
     top?.postMessage({ type: PluginChannel.pluginReadyForRender }, '*');
 
     return () => {
-      window.removeEventListener('message', receiverData);
+      window?.removeEventListener('message', receiverData);
     };
   }, []);