Fix first time visit modal issue

PREreview · Aug 14, 2020 · c39de50 · c39de50
1 parent c0fdb9b
commit c39de50
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/src/components/home.js b/src/components/home.js
@@ -42,7 +42,7 @@ export default function Home() {
   const [showLeftPanel, setShowLeftPanel] = useState(!isMobile);
   const [loginModalOpenNext, setLoginModalOpenNext] = useState(null);
   const isNewVisitor = useIsNewVisitor();
-  const [displayExtensionBanner, closeExtensionBanner] = useDisplayExtensionBanner();
+  const [displayExtensionBanner, closeExtensionBanner, checkForExtension] = useDisplayExtensionBanner();
   const [isWelcomeModalOpen, setIsWelcomeModalOpen] = useState(true);
   const [newPreprints, setNewPreprints] = useNewPreprints();
 
@@ -120,6 +120,7 @@ export default function Home() {
         <WelcomeModal
           onClose={() => {
             setIsWelcomeModalOpen(false);
+            checkForExtension();
           }}
         />
       )}

diff --git a/src/hooks/ui-hooks.js b/src/hooks/ui-hooks.js
@@ -163,12 +163,16 @@ export function usePrevious(value) {
 export function useDisplayExtensionBanner() {
   const [displayExtensionBanner, setDisplayExtensionBanner] = useState(false);
 
+  const checkForExtension = () => {
+    setTimeout(() => {
+      const shouldDisplayExtensionBanner = localStorage.getItem('displayExtensionBanner') !== 'false' && !document.getElementById(CSS_SCOPE_ID);
+      setDisplayExtensionBanner(shouldDisplayExtensionBanner);
+    },500);
+  }
+
   useEffect(() => {
     if(document.readyState === 'interactive' || document.readyState === 'complete'){
-      setTimeout(() => {
-        const shouldDisplayExtensionBanner = localStorage.getItem('displayExtensionBanner') !== 'false' && !document.getElementById(CSS_SCOPE_ID);
-        setDisplayExtensionBanner(shouldDisplayExtensionBanner);
-      },500)
+      checkForExtension();
     }
   }, []);
 
@@ -177,5 +181,5 @@ export function useDisplayExtensionBanner() {
     setDisplayExtensionBanner(false);
   }, []);
 
-  return [displayExtensionBanner, localSet];
+  return [displayExtensionBanner, localSet, checkForExtension];
 }