fix(Portal): state init on SSR (#1954)

kiwicom · Jun 16, 2020 · 1f43f7a · 1f43f7a
1 parent bd7f59f
commit 1f43f7a
Showing 1 changed file with 16 additions and 8 deletions.
diff --git a/src/Portal/index.js b/src/Portal/index.js
@@ -5,18 +5,26 @@ import ReactDOM from "react-dom";
 import type { Props } from "./index";
 
 const Portal = ({ renderInto, children }: Props) => {
-  const [el] = React.useState(() => document.createElement("div"));
-  const [node] = React.useState(() =>
-    renderInto && document.getElementById(renderInto)
-      ? document.getElementById(renderInto)
-      : document.body,
-  );
+  const [el] = React.useState(() => {
+    if (typeof window !== "undefined") {
+      return document.createElement("div");
+    }
+    return null;
+  });
+  const [node] = React.useState(() => {
+    if (typeof window !== "undefined") {
+      return renderInto && document.getElementById(renderInto)
+        ? document.getElementById(renderInto)
+        : document.body;
+    }
+    return null;
+  });
   React.useLayoutEffect(() => {
-    if (node) {
+    if (node && el) {
       node.appendChild(el);
     }
     return () => {
-      if (node) {
+      if (node && el) {
         node.removeChild(el);
       }
     };