From c5499d8ef842f99e9c0065a1518789038a0ffda1 Mon Sep 17 00:00:00 2001
From: dkmyta <dkmyta10@gmail.com>
Date: Sat, 21 Dec 2024 13:35:32 -0800
Subject: [PATCH] Add functionality to return to all fixers

---
 .../components/threats-modal/index.tsx        |  2 ++
 .../threats-modal/styles.module.scss          |  4 ++++
 .../threats-modal/threat-actions.tsx          | 24 +++++++++++++++++--
 .../threats-modal/threat-fix-confirmation.tsx | 16 +++++++++++--
 4 files changed, 42 insertions(+), 4 deletions(-)

diff --git a/projects/js-packages/components/components/threats-modal/index.tsx b/projects/js-packages/components/components/threats-modal/index.tsx
index 3b31f05204a31..9936021a99841 100644
--- a/projects/js-packages/components/components/threats-modal/index.tsx
+++ b/projects/js-packages/components/components/threats-modal/index.tsx
@@ -9,6 +9,7 @@ import ThreatFixConfirmation from './threat-fix-confirmation';
 
 interface ThreatModalContextType {
 	closeModal: () => void;
+	currentThreats: Threat[];
 	threatsList: Threat[];
 	setThreatsList: Dispatch< SetStateAction< Threat[] > >;
 	isBulk: boolean;
@@ -110,6 +111,7 @@ export default function ThreatsModal( {
 				<ThreatsModalContext.Provider
 					value={ {
 						closeModal: modalProps.onRequestClose,
+						currentThreats,
 						threatsList,
 						setThreatsList,
 						isBulk,
diff --git a/projects/js-packages/components/components/threats-modal/styles.module.scss b/projects/js-packages/components/components/threats-modal/styles.module.scss
index 31e9f442132b5..0db446a781e9c 100644
--- a/projects/js-packages/components/components/threats-modal/styles.module.scss
+++ b/projects/js-packages/components/components/threats-modal/styles.module.scss
@@ -66,6 +66,10 @@
 	}
 }
 
+.view-bulk {
+	margin-right: auto;
+}
+
 .section {
 	display: flex;
 	flex-direction: column;
diff --git a/projects/js-packages/components/components/threats-modal/threat-actions.tsx b/projects/js-packages/components/components/threats-modal/threat-actions.tsx
index 77aebd9290180..f6b4bdba699d7 100644
--- a/projects/js-packages/components/components/threats-modal/threat-actions.tsx
+++ b/projects/js-packages/components/components/threats-modal/threat-actions.tsx
@@ -1,5 +1,5 @@
 import { getFixerState, getDetailedFixerAction, type Threat } from '@automattic/jetpack-scan';
-import { __ } from '@wordpress/i18n';
+import { __, sprintf } from '@wordpress/i18n';
 import { useCallback, useContext, useMemo } from 'react';
 import { Button } from '@automattic/jetpack-components';
 import FixerStateNotice from './fixer-state-notice';
@@ -11,12 +11,23 @@ import { ThreatsModalContext } from '.';
  *
  * @param {object}   props                 - The props.
  * @param {Threat[]} props.selectedThreats - The selected threats.
+ * @param {boolean}  props.canReturnToBulk - Whether the user can return to the bulk view.
+ * @param {Function} props.viewBulkThreats - The function to view all threats.
  *
  * @return {JSX.Element | null} The rendered action buttons or null if no actions are available.
  */
-const ThreatActions = ( { selectedThreats }: { selectedThreats: Threat[] } ): JSX.Element => {
+const ThreatActions = ( {
+	selectedThreats,
+	canReturnToBulk,
+	viewBulkThreats,
+}: {
+	selectedThreats: Threat[];
+	canReturnToBulk: boolean;
+	viewBulkThreats: () => void;
+} ): JSX.Element => {
 	const {
 		closeModal,
+		currentThreats,
 		isBulk,
 		actionToConfirm,
 		handleFixThreatClick,
@@ -66,6 +77,15 @@ const ThreatActions = ( { selectedThreats }: { selectedThreats: Threat[] } ): JS
 
 	const renderIndividualActions = () => (
 		<>
+			{ canReturnToBulk && (
+				<Button className={ styles[ 'view-bulk' ] } onClick={ viewBulkThreats }>
+					{ sprintf(
+						// translators: %d is the number of threats.
+						__( 'Show auto fixers (%d)', 'jetpack-components' ),
+						currentThreats.length
+					) }
+				</Button>
+			) }
 			{ firstThreat.status === 'ignored' && (
 				<Button disabled={ disabled } isDestructive variant="secondary" onClick={ onUnignoreClick }>
 					{ __( 'Un-ignore threat', 'jetpack-components' ) }
diff --git a/projects/js-packages/components/components/threats-modal/threat-fix-confirmation.tsx b/projects/js-packages/components/components/threats-modal/threat-fix-confirmation.tsx
index 16a5f8691b8b0..ca53c5b6ce900 100644
--- a/projects/js-packages/components/components/threats-modal/threat-fix-confirmation.tsx
+++ b/projects/js-packages/components/components/threats-modal/threat-fix-confirmation.tsx
@@ -24,9 +24,10 @@ import { ThreatsModalContext } from '.';
 const ThreatFixConfirmation = () => {
 	const [ isSm ] = useBreakpointMatch( [ 'sm', 'lg' ], [ null, '<' ] );
 
-	const { threatsList, setThreatsList, isBulk, handleUpgradeClick } =
+	const { currentThreats, threatsList, setThreatsList, isBulk, handleUpgradeClick } =
 		useContext( ThreatsModalContext );
 
+	const [ canReturnToBulk, setCanReturnToBulk ] = useState( false );
 	const [ selectedThreats, setSelectedThreats ] = useState( threatsList );
 
 	const handleToggleThreat = useCallback( ( threat: Threat, isChecked: boolean ) => {
@@ -51,10 +52,17 @@ const ThreatFixConfirmation = () => {
 		return handlers;
 	}, [ threatsList, handleToggleThreat ] );
 
+	const viewBulkThreats = useCallback( () => {
+		setThreatsList( currentThreats );
+		setSelectedThreats( currentThreats );
+		setCanReturnToBulk( false );
+	}, [ currentThreats, setThreatsList, setSelectedThreats ] );
+
 	const viewIndividualThreat = useCallback(
 		( threat: Threat ) => {
 			setThreatsList( [ threat ] );
 			setSelectedThreats( [ threat ] );
+			setCanReturnToBulk( true );
 		},
 		[ setThreatsList, setSelectedThreats ]
 	);
@@ -153,7 +161,11 @@ const ThreatFixConfirmation = () => {
 					onClick={ handleUpgradeClick }
 				/>
 			) }
-			<ThreatActions selectedThreats={ selectedThreats } />
+			<ThreatActions
+				selectedThreats={ selectedThreats }
+				canReturnToBulk={ canReturnToBulk }
+				viewBulkThreats={ viewBulkThreats }
+			/>
 		</div>
 	);
 };