handle undefined document or body

codecov · Sep 3, 2024 · b02f732 · b02f732
1 parent d95694a
commit b02f732
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/src/shared/ThemeContext/ThemeContext.tsx b/src/shared/ThemeContext/ThemeContext.tsx
@@ -34,18 +34,21 @@ export const ThemeContextProvider: FC<ThemeContextProviderProps> = ({
   const initialRender = useRef(true)
 
   if (initialRender.current) {
-    document.body.classList.remove(Theme.LIGHT, Theme.DARK)
-    document.body.classList.add(theme)
-
-    localStorage.setItem('theme', theme)
+    if (typeof document !== 'undefined' && document.body) {
+      document.body.classList.remove(Theme.LIGHT, Theme.DARK)
+      document.body.classList.add(theme)
+      localStorage.setItem('theme', theme)
+    }
     initialRender.current = false
   }
 
   const handleTheme = useCallback((theme: Theme) => {
-    document.body.classList.remove(Theme.LIGHT, Theme.DARK)
-    document.body.classList.add(theme)
-    localStorage.setItem('theme', theme)
-    setTheme(theme)
+    if (typeof document !== 'undefined' && document.body) {
+      document.body.classList.remove(Theme.LIGHT, Theme.DARK)
+      document.body.classList.add(theme)
+      localStorage.setItem('theme', theme)
+      setTheme(theme)
+    }
   }, [])
 
   return (