Fix window is not defined

mustofa-id · Jul 2, 2019 · 089bbae · 089bbae
1 parent d4912fc
commit 089bbae
Showing 1 changed file with 14 additions and 11 deletions.
diff --git a/src/components/widget/masonry.js b/src/components/widget/masonry.js
@@ -6,26 +6,29 @@ const Masonry = ({ column, children, breakpoint }) => {
   const [columnCount, setColumnCount] = useState(getColumnCount())
 
   useEffect(() => {
-    if (window) {
+    if (typeof window !== 'undefined') {
       updateColumn()
       window.addEventListener('resize', updateColumn)
       return () => window.removeEventListener('resize', updateColumn)
     }
   })
 
   function getColumnCount () {
-    const width = (window && window.innerWidth) || Infinity
-    let matchedBreakpoint = Infinity
-    let columns = breakpoint.default
-    for (let bp in breakpoint) {
-      const optBreakpoint = parseInt(bp)
-      const isCurrentBreakpoint = optBreakpoint > 0 && width <= optBreakpoint
-      if (isCurrentBreakpoint && optBreakpoint < matchedBreakpoint) {
-        matchedBreakpoint = optBreakpoint
-        columns = breakpoint[bp]
+    if (typeof window !== 'undefined') {
+      const width = (window.innerWidth) || Infinity
+      let matchedBreakpoint = Infinity
+      let columns = breakpoint.default
+      for (let bp in breakpoint) {
+        const optBreakpoint = parseInt(bp)
+        const isCurrentBreakpoint = optBreakpoint > 0 && width <= optBreakpoint
+        if (isCurrentBreakpoint && optBreakpoint < matchedBreakpoint) {
+          matchedBreakpoint = optBreakpoint
+          columns = breakpoint[bp]
+        }
       }
+      return Math.max(1, parseInt(columns) || 1)
     }
-    return Math.max(1, parseInt(columns) || 1)
+    return 0
   }
 
   function updateColumn () {