Fix save profile ui bug

src/actions/session.js

@@ -2,15 +2,12 @@ import * as firebase from 'firebase'
 import FetchClient from './../utils/fetchClient'
 import actionCreator from './../utils/actionCreator'
 import * as Constants from './../constants'
+import { setUser } from './user'
 
 export function setAccessToken(token) {
   return actionCreator(Constants.SET_AUTH_TOKEN, { token })
 }
 
-export function setUser(user) {
-  return actionCreator(Constants.SET_USER, { user })
-}
-
 export function setPreviousRoute(previousRoute) {
   return actionCreator(Constants.SET_PREVIOUS_ROUTE, { previousRoute })
 }

src/actions/user.js

@@ -1,5 +1,11 @@
 /* eslint import/prefer-default-export: 0 */
 import FetchClient from '../utils/fetchClient'
+import actionCreator from '../utils/actionCreator'
+import * as Constants from '../constants'
+
+export function setUser(user) {
+  return actionCreator(Constants.SET_USER, { user })
+}
 
 export function getUser(token, user) {
   return async () => {
@@ -22,7 +28,7 @@ export function getUser(token, user) {
 }
 
 export function updateUser(token, userId, user) {
-  return async () => {
+  return async (dispatch) => {
     try {
       const fetchClient = new FetchClient()
       fetchClient.setAuthToken(token)
@@ -31,6 +37,7 @@ export function updateUser(token, userId, user) {
         body: user
       })
       if (result) {
+        dispatch(setUser(result))
         return { result, error }
       }
       console.error(error)

src/containers/UserSettings.js

@@ -16,7 +16,6 @@ const tabValues = ['profile', 'invites', 'accounts', 'notification']
 
 const mapStateToProps = state => ({
   session: state.session,
-  user: state.user,
   invites: state.invites
 })