Limit AI Retries

src/OnboardingSPA/components/ErrorState/Step/SiteGen/index.js

@@ -1,6 +1,6 @@
 // WordPress
 import { useViewportMatch } from '@wordpress/compose';
-import { useEffect } from '@wordpress/element';
+import { useState, useEffect } from '@wordpress/element';
 import { useSelect, useDispatch } from '@wordpress/data';
 import { Button, Fill } from '@wordpress/components';
 
@@ -31,6 +31,7 @@
 const SiteGenStepErrorState = () => {
 	const navigate = useNavigate();
 	const isLargeViewport = useViewportMatch( 'small' );
+	const [ disableRetry, setDisableRetry ] = useState( false );
 
 	const {
 		setIsHeaderEnabled,
@@ -55,7 +56,11 @@
 		setIsHeaderNavigationEnabled( true );
 		setDrawerActiveView( false );
 		setSidebarActiveView( false );
+		setDisableRetry(
+			currentData.sitegen?.siteGenErrorMeta?.retryCount >
+				currentData.sitegen?.siteGenErrorMeta?.maxRetryCount
+		);
 	}, [] );
 
 	const { brandConfig, currentData, currentStep, previousStep, allSteps } =
 		useSelect( ( select ) => {
@@ -186,6 +191,7 @@
 								onClick={ () => {
 									handleRetry();
 								} }
+								disabled={ true === disableRetry }
 							>
 								<p className="nfd-onboarding-button--site-gen-next--text">
 									{ content.buttonText }
@@ -200,6 +206,7 @@
 									onClick={ () => {
 										handleRetry();
 									} }
+									disabled={ true === disableRetry }
 								>
 									<p className="nfd-onboarding-button--site-gen-next--text">
 										{ content.buttonText }

src/OnboardingSPA/store/reducer.js

@@ -181,13 +181,26 @@ export function data( state = {}, action ) {
 			};
 
 		case 'SET_SITEGEN_AI_ERROR_STATUS':
+			// Only update if the prev value was false and now there is an error else don't
+			const shouldUpdateRetryCount =
+				! state.flowData.sitegen.siteGenErrorMeta.status &&
+				action.siteGenErrorStatus;
 			return {
 				...state,
 				flowData: {
 					...state.flowData,
 					sitegen: {
 						...state.flowData.sitegen,
-						siteGenErrorStatus: action.siteGenErrorStatus,
+						siteGenErrorMeta: {
+							...state.flowData.sitegen.siteGenErrorMeta,
+							status: action.siteGenErrorStatus,
+							retryCount:
+								shouldUpdateRetryCount === true
+									? state.flowData.sitegen.siteGenErrorMeta
+										.retryCount + 1
+									: state.flowData.sitegen.siteGenErrorMeta
+										.retryCount,
+						},
 					},
 				},
 			};

src/OnboardingSPA/store/selectors.js

@@ -451,7 +451,7 @@ export function getCustomizeSidebarData( state ) {
 }
 
 export function getSiteGenErrorStatus( state ) {
-	return state.data.flowData.sitegen.siteGenErrorStatus;
+	return state.data.flowData.sitegen.siteGenErrorMeta.status;
 }
 
 export function getInteractionDisabled( state ) {