aswi2022code-of-duty-gitlab/frontend/src/features/Auth/userReducer.ts @ a69da1e3

import { AnyAction } from 'redux'
import persistReducer from 'redux-persist/es/persistReducer'
import storage from 'redux-persist/lib/storage'

export interface UserState {
    accessToken?: string
    refreshToken?: string
    username: string
    roles: string[]
    isLoggedIn: boolean
}

const initialState: UserState = {
    roles: [],
    isLoggedIn: false,
    username: '',
}

// All possible actions
export enum AuthStateActions {
    LOG_IN = 'LOG_IN',
    LOG_OUT = 'LOG_OUT',
    UPDATE_ACCESS_TOKEN = 'REFRESH_ACCESS_TOKEN',
    UPDATE_REFRESH_TOKEN = 'UPDATE_REFRESH_TOKEN',
    UPDATE_TOKENS = 'UPDATE_TOKENS',
}

// Only needed if the state is to be persisted
const persistConfig = {
    key: 'auth',
    storage
}

const _authReducer = (
    state: UserState = initialState,
    action: AnyAction
): UserState => {
    switch (action.type) {
        case AuthStateActions.LOG_IN:
            return {
                ...action.payload,
                isAuthenticated: true,
            }
        case AuthStateActions.LOG_OUT:
            return initialState

        case AuthStateActions.UPDATE_ACCESS_TOKEN:
            return {
                ...state,
                accessToken: action.payload,
            }

        case AuthStateActions.UPDATE_REFRESH_TOKEN:
            return {
                ...state,
                refreshToken: action.payload,
            }

        case AuthStateActions.UPDATE_TOKENS:
            return { ...state, ...action.payload }

        default:
            return state
    }
}

const authReducer = persistReducer(persistConfig, _authReducer)

export default authReducer