fix: resolve error on SSR with document (#117)

zakodium-oss · Mar 30, 2022 · 08869d0 · 08869d0
1 parent 1fa2e72
commit 08869d0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/layout/RootLayout.tsx b/src/layout/RootLayout.tsx
@@ -42,7 +42,9 @@ const style: CSSProperties = {
 };
 
 export function RootLayout(props: RootLayoutProps) {
-  const [state, setState] = useState<HTMLElement>(document.body);
+  const [state, setState] = useState<HTMLElement | null>(
+    typeof document !== 'undefined' ? document.body : null,
+  );
 
   const ref = useCallback((node: HTMLDivElement) => {
     if (node !== null) {

diff --git a/src/layout/context/RootLayoutContext.tsx b/src/layout/context/RootLayoutContext.tsx
@@ -13,7 +13,7 @@ export function useRootLayoutContext() {
 
 export function RootLayoutProvider(props: {
   children: ReactNode;
-  innerRef: HTMLElement;
+  innerRef: HTMLElement | null;
 }) {
   return (
     <rootLayoutContext.Provider value={props.innerRef}>