Refactor to use action instead of util function

2 files changed, 32 insertions, 1 deletions

diff --git a/src/store/actions/user.ts b/src/store/actions/user.ts
index 941101df..b1cb8719 100644
--- a/src/store/actions/user.ts
+++ b/src/store/actions/user.ts
@@ -4,12 +4,14 @@ import {Action, ThunkAction} from '@reduxjs/toolkit';
 import {
   getProfilePic,
   loadProfileInfo,
+  removeBadgesService,
   sendSuggestedPeopleLinked,
 } from '../../services';
 import {UniversityBadge, UserType} from '../../types/types';
 import {getTokenOrLogout} from '../../utils';
 import {
   clearHeaderAndProfileImages,
+  profileBadgeRemoved,
   profileBadgesUpdated,
   profileCompletionStageUpdated,
   setIsOnboardedUser,
@@ -107,6 +109,27 @@ export const updateUserBadges =
     }
   };
 
+/**
+ * Removes a single badge from logged-in user by badge name.
+ * @param badgeName name of badge to be removed
+ * @param loggedInUserId userId of loggedInUser
+ */
+export const removeUserBadge =
+  (
+    badgeName: string,
+    loggedInUserId: string,
+  ): ThunkAction<Promise<void>, RootState, unknown, Action<string>> =>
+  async (dispatch) => {
+    try {
+      const success = await removeBadgesService([badgeName], loggedInUserId);
+      if (success) {
+        dispatch({type: profileBadgeRemoved.type, payload: {badge: badgeName}});
+      }
+    } catch (error) {
+      console.log(error);
+    }
+  };
+
 export const updateProfileCompletionStage =
   (
     stage: number,
diff --git a/src/store/reducers/userReducer.ts b/src/store/reducers/userReducer.ts
index 97bf845c..03fee112 100644
--- a/src/store/reducers/userReducer.ts
+++ b/src/store/reducers/userReducer.ts
@@ -1,4 +1,6 @@
 import {createSlice} from '@reduxjs/toolkit';
+import {ActionSheetIOS} from 'react-native';
+import {Badge} from 'react-native-elements';
 import {NO_USER_DATA} from '../initialStates';
 
 /**
@@ -46,6 +48,12 @@ const userDataSlice = createSlice({
       state.profile.badges = action.payload.badges;
     },
 
+    profileBadgeRemoved: (state, action) => {
+      state.profile.badges = state.profile.badges.filter(
+        (badge) => badge.name !== action.payload.badge,
+      );
+    },
+
     profileCompletionStageUpdated: (state, action) => {
       state.profile.profile_completion_stage = action.payload.stage;
     },
@@ -95,6 +103,6 @@ export const {
   setSuggestedPeopleImage,
   clearHeaderAndProfileImages,
   profileBadgesUpdated,
-  // setChatClientReady,
+  profileBadgeRemoved,
 } = userDataSlice.actions;
 export const userDataReducer = userDataSlice.reducer;