2 files changed, 54 insertions, 5 deletions
diff --git a/src/store/actions/momentCategories.tsx b/src/store/actions/momentCategories.tsx
index 987fc9e5..c91e9ec8 100644
--- a/src/store/actions/momentCategories.tsx
+++ b/src/store/actions/momentCategories.tsx
@@ -1,7 +1,10 @@
 import {RootState} from '../rootReducer';
 import {loadMomentCategories, postMomentCategories} from '../../services';
 import {Action, ThunkAction} from '@reduxjs/toolkit';
-import {momentCategoriesFetched} from '../reducers';
+import {
+  momentCategoriesFetched,
+  profileCompletionStageUpdated,
+} from '../reducers';
 import {getTokenOrLogout} from '../../utils';
 
 /**
@@ -28,21 +31,32 @@ export const loadUserMomentCategories = (
 /**
  * Handle addition / deletion of categories for a user
  * @param categories List of categories
- * @param userId id of the user for whom categories should be updated
+ * @param add true if the call to his function is to add categories
  */
 export const updateMomentCategories = (
   categories: string[],
+  add: boolean,
 ): ThunkAction<Promise<void>, RootState, unknown, Action<string>> => async (
   dispatch,
 ) => {
   try {
     const token = await getTokenOrLogout(dispatch);
-    const success = await postMomentCategories(categories, token);
+    let success = false;
+    let stage: number | undefined = 1;
+
+    stage = await postMomentCategories(categories, token);
+    success = stage ? true : false;
     if (success) {
       dispatch({
         type: momentCategoriesFetched.type,
         payload: {categories},
       });
+      if (add) {
+        dispatch({
+          type: profileCompletionStageUpdated.type,
+          payload: {stage},
+        });
+      }
     }
   } catch (error) {
     console.log(error);
diff --git a/src/store/actions/user.ts b/src/store/actions/user.ts
index eee5fcde..8550f3bd 100644
--- a/src/store/actions/user.ts
+++ b/src/store/actions/user.ts
@@ -2,7 +2,13 @@ import {RootState} from '../rootReducer';
 import {UserType} from '../../types/types';
 import {loadProfileInfo, loadAvatar, loadCover} from '../../services';
 import {Action, ThunkAction} from '@reduxjs/toolkit';
-import {userLoggedIn, userDetailsFetched, socialEdited} from '../reducers';
+import {
+  userLoggedIn,
+  userDetailsFetched,
+  socialEdited,
+  profileCompletionStageUpdated,
+  setIsOnboardedUser,
+} from '../reducers';
 import {getTokenOrLogout} from '../../utils';
 
 /**
@@ -50,7 +56,6 @@ export const updateSocial = (
   dispatch,
 ) => {
   try {
-    console.log(social);
     dispatch({
       type: socialEdited.type,
       payload: {social, value},
@@ -60,6 +65,36 @@ export const updateSocial = (
   }
 };
 
+export const updateProfileCompletionStage = (
+  stage: number,
+): ThunkAction<Promise<void>, RootState, unknown, Action<string>> => async (
+  dispatch,
+) => {
+  try {
+    dispatch({
+      type: profileCompletionStageUpdated.type,
+      payload: {stage},
+    });
+  } catch (error) {
+    console.log(error);
+  }
+};
+
+export const updateIsOnboardedUser = (
+  isOnboardedUser: boolean,
+): ThunkAction<Promise<void>, RootState, unknown, Action<string>> => async (
+  dispatch,
+) => {
+  try {
+    dispatch({
+      type: setIsOnboardedUser.type,
+      payload: {isOnboardedUser},
+    });
+  } catch (error) {
+    console.log(error);
+  }
+};
+
 export const logout = (): ThunkAction<
   Promise<void>,
   RootState,