From 0d2735fe0f41215e287eae7bd073620115428faa Mon Sep 17 00:00:00 2001
From: Michal Gold <mgold@redhat.com>
Date: Sun, 10 Mar 2024 16:20:46 +0200
Subject: [PATCH] V2Wizard: fix remove partition funtionallity (HMS-3753)

this commit add the remove partition functionlity to file system step
---
 .../steps/FileSystem/FileSystemConfiguration.tsx       | 10 ++++++++--
 src/store/wizardSlice.ts                               |  9 +++++++++
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/src/Components/CreateImageWizardV2/steps/FileSystem/FileSystemConfiguration.tsx b/src/Components/CreateImageWizardV2/steps/FileSystem/FileSystemConfiguration.tsx
index ea93ddae..9a1e6162 100644
--- a/src/Components/CreateImageWizardV2/steps/FileSystem/FileSystemConfiguration.tsx
+++ b/src/Components/CreateImageWizardV2/steps/FileSystem/FileSystemConfiguration.tsx
@@ -26,6 +26,7 @@ import {
   changePartitionMinSize,
   changePartitionMountpoint,
   selectImageTypes,
+  removePartition,
   selectPartitions,
 } from '../../../../store/wizardSlice';
 import UsrSubDirectoriesDisabled from '../../UsrSubDirectoriesDisabled';
@@ -163,6 +164,11 @@ const getSuffix = (mountpoint: string) => {
 
 const Row = ({ partition }: RowPropTypes) => {
   const [units, setUnits] = useState<Units>('MiB');
+  const dispatch = useAppDispatch();
+
+  const handleRemovePartition = (id: string) => {
+    dispatch(removePartition(id));
+  };
 
   return (
     <Tr>
@@ -184,9 +190,9 @@ const Row = ({ partition }: RowPropTypes) => {
         <Button
           variant="link"
           icon={<MinusCircleIcon />}
-          onClick={() => {}}
+          onClick={() => handleRemovePartition(partition.id)}
           data-testid="remove-mount-point"
-          isDisabled={true}
+          isDisabled={partition.mountpoint === '/'}
         />
       </Td>
     </Tr>
diff --git a/src/store/wizardSlice.ts b/src/store/wizardSlice.ts
index 97f4ecb4..c86adeda 100644
--- a/src/store/wizardSlice.ts
+++ b/src/store/wizardSlice.ts
@@ -383,6 +383,14 @@ export const wizardSlice = createSlice({
     addPartition: (state, action: PayloadAction<Partition>) => {
       state.fileSystem.partitions.push(action.payload);
     },
+    removePartition: (state, action: PayloadAction<Partition['id']>) => {
+      state.fileSystem.partitions.splice(
+        state.fileSystem.partitions.findIndex(
+          (partition) => partition.id === action.payload
+        ),
+        1
+      );
+    },
     changePartitionMountpoint: (
       state,
       action: PayloadAction<{ id: string; mountpoint: string }>
@@ -465,6 +473,7 @@ export const {
   changeEnabledServices,
   changeFileSystemPartitionMode,
   addPartition,
+  removePartition,
   changePartitionMountpoint,
   changePartitionMinSize,
   changeCustomRepositories,