Merge pull request #455 from newfold-labs/siteErrorPreviewFix

Fixed the retry issue in preview page
newfold-labs · Feb 8, 2024 · 7d4e4eb · 7d4e4eb
2 parents 8f07296 + b9bc1b1
commit 7d4e4eb
Showing 1 changed file with 15 additions and 7 deletions.
diff --git a/src/OnboardingSPA/steps/SiteGen/Preview/index.js b/src/OnboardingSPA/steps/SiteGen/Preview/index.js
@@ -34,13 +34,17 @@ const SiteGenPreview = () => {
 		updateSiteGenErrorStatus,
 	} = useDispatch( nfdOnboardingStore );
 
-	const { currentData, nextStep } = useSelect( ( select ) => {
-		return {
-			currentData:
-				select( nfdOnboardingStore ).getCurrentOnboardingData(),
-			nextStep: select( nfdOnboardingStore ).getNextStep(),
-		};
-	} );
+	const { currentData, nextStep, siteGenErrorStatus } = useSelect(
+		( select ) => {
+			return {
+				currentData:
+					select( nfdOnboardingStore ).getCurrentOnboardingData(),
+				nextStep: select( nfdOnboardingStore ).getNextStep(),
+				siteGenErrorStatus:
+					select( nfdOnboardingStore ).getSiteGenErrorStatus(),
+			};
+		}
+	);
 
 	useEffect( () => {
 		setIsHeaderEnabled( true );
@@ -49,6 +53,10 @@ const SiteGenPreview = () => {
 		setHeaderActiveView( HEADER_SITEGEN );
 		setDrawerActiveView( false );
 		updateInitialize( true );
+		if ( ! siteGenErrorStatus ) {
+			loadHomepages();
+			loadGlobalStyles();
+		}
 	}, [ currentData ] );
 
 	const loadHomepages = async () => {