fix: ensure that privacy shield is always rendered in SSR to avoid hy…

…dration issues
hashbite · Feb 26, 2024 · 49efbfe · 49efbfe
1 parent e2b0220
commit 49efbfe
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/packages/core/src/index.tsx b/packages/core/src/index.tsx
@@ -1,4 +1,10 @@
-import React, { Dispatch, SetStateAction, useContext } from 'react'
+import React, {
+  Dispatch,
+  SetStateAction,
+  useContext,
+  useEffect,
+  useState,
+} from 'react'
 import { Location } from 'history'
 import {
   FallbackComponentProps,
@@ -78,10 +84,13 @@ export const PrivacyShield: React.FC<PrivacyShieldProps> = ({
   children,
   ...props
 }) => {
+  const [isSSR, setIsSSR] = useState(true)
   const [decision] = useDecision(id)
   const DefaultFallbackComponent = useFallbackComponent()
 
-  if (decision) {
+  useEffect(() => setIsSSR(false))
+
+  if (!isSSR && decision) {
     return <React.Fragment>{children}</React.Fragment>
   }