From 85c81f5e2a8aa1a44ed52def42a8dc2f6628fdcf Mon Sep 17 00:00:00 2001
From: Sambo Chea <sombochea@cubetiqs.com>
Date: Sat, 12 Jun 2021 19:22:48 +0700
Subject: [PATCH] Task: Add auth context and provider and add routes and route
 wrapper and more configs and utils

---
 src/config/index.ts         |   4 +-
 src/context/AuthContext.tsx | 131 ++++++++++++++++++++++++++++++++++++
 src/pages/About/index.tsx   |   4 +-
 src/pages/Error/403.tsx     |   7 ++
 src/pages/Error/404.tsx     |   7 ++
 src/pages/Home/index.tsx    |   4 +-
 src/pages/Login/index.less  |   4 ++
 src/pages/Login/index.tsx   |   9 +++
 src/pages/Profile/index.tsx |   7 ++
 src/routes/AuthRoute.tsx    |  21 ++++++
 src/routes/index.tsx        |  31 ++++++---
 src/routes/interfaces.ts    |  27 ++++++++
 src/routes/routes.tsx       |  62 ++++++++---------
 src/routes/types.ts         |  11 ++-
 src/utils/log_util.ts       |   3 +
 src/utils/ls_util.ts        |   7 ++
 16 files changed, 289 insertions(+), 50 deletions(-)
 create mode 100644 src/context/AuthContext.tsx
 create mode 100644 src/pages/Error/403.tsx
 create mode 100644 src/pages/Error/404.tsx
 create mode 100644 src/pages/Login/index.less
 create mode 100644 src/pages/Login/index.tsx
 create mode 100644 src/pages/Profile/index.tsx
 create mode 100644 src/routes/AuthRoute.tsx
 create mode 100644 src/routes/interfaces.ts
 create mode 100644 src/utils/log_util.ts
 create mode 100644 src/utils/ls_util.ts

diff --git a/src/config/index.ts b/src/config/index.ts
index 1a617d9..97297e6 100644
--- a/src/config/index.ts
+++ b/src/config/index.ts
@@ -1,3 +1,5 @@
 export const AppConfig = {
-    APP_NAME: process.env.REACT_APP_NAME
+    APP_NAME: process.env.REACT_APP_NAME,
+
+    CLIENT_TOKEN_KEY: "client_token",
 }
\ No newline at end of file
diff --git a/src/context/AuthContext.tsx b/src/context/AuthContext.tsx
new file mode 100644
index 0000000..4d75790
--- /dev/null
+++ b/src/context/AuthContext.tsx
@@ -0,0 +1,131 @@
+import { AppConfig } from '@/config'
+import { printError, printInfo } from '@/utils/log_util'
+import { getStorage, setStorage } from '@/utils/ls_util'
+import React, { useContext, useCallback, useReducer, useEffect } from 'react'
+
+interface AuthContextState {
+    login: (args: { username: string; password: string }) => Promise<void>
+    state: {
+        user: {
+            username: string
+            userId: string
+        } | null
+    }
+    getToken: () => string | undefined | null
+    logout: () => void
+    isLogin: () => boolean
+}
+
+enum AuthActionType {
+    'LOGIN' = 'LOGIN',
+    'LOGOUT' = 'LOGOUT',
+}
+
+const AuthReducer: (
+    state: AuthContextState['state'],
+    action: {
+        type: AuthActionType
+        payload?: any
+    }
+) => AuthContextState['state'] = (state, action) => {
+    switch (action.type) {
+        case AuthActionType.LOGIN:
+            return {
+                ...state,
+                user: action.payload,
+            }
+        case AuthActionType.LOGOUT:
+            return {
+                ...state,
+                user: null,
+            }
+        default:
+            return state
+    }
+}
+
+const AuthContext = React.createContext<AuthContextState>({
+    login: async () => {},
+    state: {
+        user: null,
+    },
+    getToken: () => {
+        return undefined
+    },
+    logout: () => {},
+    isLogin: () => { return false },
+})
+
+const AuthProvider: React.FC = (props) => {
+    const [state, dispatch] = useReducer(AuthReducer, {
+        user: null,
+    })
+
+    const login = async (args: { username: string; password: string }) => {
+        const res = {
+            data: {
+                token: 'DEMO_HAH',
+            },
+        }
+
+        setStorage(AppConfig.CLIENT_TOKEN_KEY, res.data.token)
+        await verify()
+        dispatch({
+            type: AuthActionType.LOGIN,
+            payload: args,
+        })
+
+        return
+    }
+
+    useEffect(() => {
+        const doVerify = async () => {
+            const token = getStorage(AppConfig.CLIENT_TOKEN_KEY)
+            const userAuthInfo = {}
+
+            printInfo("User details =>", token, userAuthInfo)
+        }
+
+        doVerify()
+    }, [])
+
+    const verify = async () => {
+        const token = getStorage(AppConfig.CLIENT_TOKEN_KEY)
+        printError('Verify not implemented with token: ', token)
+        const userAuthInfo = {}
+        return userAuthInfo
+    }
+
+    const logout = useCallback(() => {
+        setStorage(AppConfig.CLIENT_TOKEN_KEY, undefined)
+        dispatch({
+            type: AuthActionType.LOGOUT,
+        })
+    }, [])
+
+    const getToken = () => getStorage(AppConfig.CLIENT_TOKEN_KEY)
+
+    const isLogin = (): boolean => {
+        return getToken() != null   
+    }
+
+    return (
+        <AuthContext.Provider
+            value={{
+                login,
+                state,
+                logout,
+                isLogin,
+                getToken,
+            }}
+        >
+            {props.children}
+        </AuthContext.Provider>
+    )
+}
+
+export const useAuthContext = () => {
+    return useContext(AuthContext)
+}
+
+export default AuthProvider
diff --git a/src/pages/About/index.tsx b/src/pages/About/index.tsx
index 6a9c4fb..409b531 100644
--- a/src/pages/About/index.tsx
+++ b/src/pages/About/index.tsx
@@ -2,8 +2,8 @@ import './index.less'
 
 export default function About() {
   return (
-    <div className="about-title">
-      <h1>About Us</h1>
+    <div>
+      <h1 className="about-title">About Us</h1>
     </div>
   )
 }
diff --git a/src/pages/Error/403.tsx b/src/pages/Error/403.tsx
new file mode 100644
index 0000000..9076439
--- /dev/null
+++ b/src/pages/Error/403.tsx
@@ -0,0 +1,7 @@
+export default function AccessDenied() {
+    return (
+        <div>
+            <h1>Access denied!</h1>
+        </div>
+    )
+}
diff --git a/src/pages/Error/404.tsx b/src/pages/Error/404.tsx
new file mode 100644
index 0000000..b75f137
--- /dev/null
+++ b/src/pages/Error/404.tsx
@@ -0,0 +1,7 @@
+export default function NotFound() {
+    return (
+        <div>
+            <h1>Not found!</h1>
+        </div>
+    )
+}
diff --git a/src/pages/Home/index.tsx b/src/pages/Home/index.tsx
index 72f6dfd..f454ede 100644
--- a/src/pages/Home/index.tsx
+++ b/src/pages/Home/index.tsx
@@ -2,8 +2,8 @@ import './index.less'
 
 export default function Home() {
   return (
-    <div className="home-title">
-      <h1>Home</h1>
+    <div>
+      <h1 className="home-title">Home</h1>
     </div>
   )
 }
diff --git a/src/pages/Login/index.less b/src/pages/Login/index.less
new file mode 100644
index 0000000..f0a9348
--- /dev/null
+++ b/src/pages/Login/index.less
@@ -0,0 +1,4 @@
+.login-title {
+  text-align: center;
+  font-weight: bold;
+}
diff --git a/src/pages/Login/index.tsx b/src/pages/Login/index.tsx
new file mode 100644
index 0000000..e4074c4
--- /dev/null
+++ b/src/pages/Login/index.tsx
@@ -0,0 +1,9 @@
+import './index.less'
+
+export default function Login() {
+  return (
+    <div>
+      <h1 className="login-title">Login</h1>
+    </div>
+  )
+}
diff --git a/src/pages/Profile/index.tsx b/src/pages/Profile/index.tsx
new file mode 100644
index 0000000..292bf42
--- /dev/null
+++ b/src/pages/Profile/index.tsx
@@ -0,0 +1,7 @@
+export default function Profile() {
+    return (
+        <div>
+            <h1>Profile</h1>
+        </div>
+    )
+}
diff --git a/src/routes/AuthRoute.tsx b/src/routes/AuthRoute.tsx
new file mode 100644
index 0000000..e1c2139
--- /dev/null
+++ b/src/routes/AuthRoute.tsx
@@ -0,0 +1,21 @@
+import { useAuthContext } from '@/context/AuthContext'
+import AccessDenied from '@/pages/Error/403'
+import { Route } from 'react-router-dom'
+
+const AuthRoute = (props: any) => {
+    const { component, ...rest } = props
+    const { isLogin } = useAuthContext()
+
+    if (!isLogin()) {
+      return (<AccessDenied />)
+    }
+
+    return (
+        <Route
+            {...rest}
+            render={component}
+        />
+    )
+}
+
+export default AuthRoute
diff --git a/src/routes/index.tsx b/src/routes/index.tsx
index 504534b..f3f0e86 100644
--- a/src/routes/index.tsx
+++ b/src/routes/index.tsx
@@ -1,4 +1,6 @@
+import AuthProvider from '@/context/AuthContext'
 import { BrowserRouter as Router, Switch, Route } from 'react-router-dom'
+import AuthRoute from './AuthRoute'
 import { routes } from './routes'
 
 const RouterView = () => {
@@ -6,16 +8,25 @@ const RouterView = () => {
         <Router>
             <Switch>
                 {routes.map((route) => {
-                    return (
-                        <Route
-                            key={route.key}
-                            exact={route.exact}
-                            component={route.component}
-                            children={route.children}
-                            location={route.location}
-                            path={route.path}
-                        />
-                    )
+                    const { withAuthority } = route
+                    if (withAuthority) {
+                        return (
+                            <AuthProvider key={route.key}>
+                                <AuthRoute {...route} />
+                            </AuthProvider>
+                        )
+                    } else {
+                        return (
+                            <Route
+                                key={route.key}
+                                exact={route.exact}
+                                component={route.component}
+                                children={route.children}
+                                location={route.location}
+                                path={route.path}
+                            />
+                        )
+                    }
                 })}
             </Switch>
         </Router>
diff --git a/src/routes/interfaces.ts b/src/routes/interfaces.ts
new file mode 100644
index 0000000..ce85d7a
--- /dev/null
+++ b/src/routes/interfaces.ts
@@ -0,0 +1,27 @@
+import { RouteProps } from 'react-router-dom'
+
+export interface Authority {
+    authority?: string | undefined
+    strict?: boolean | undefined
+    with?: Authority | undefined
+}
+
+export interface AuthorityProps {
+    withAuthority?: boolean | undefined
+    authorities?: Authority | string[] | string | undefined
+    strictAuthority?: boolean | undefined
+}
+
+export interface CustomRouteProps extends RouteProps, AuthorityProps {
+    key: string
+}
+
+export interface MenuProps {
+    icon: any
+    label?: string | undefined
+}
+
+export interface SideSubMenuProps extends MenuProps {
+    key?: string | undefined
+    subMenus: SideSubMenuProps[]
+}
diff --git a/src/routes/routes.tsx b/src/routes/routes.tsx
index f48616d..3fd5a69 100644
--- a/src/routes/routes.tsx
+++ b/src/routes/routes.tsx
@@ -1,36 +1,18 @@
-import { RouteProps } from 'react-router-dom'
-import About from '../pages/About'
-import Home from '../pages/Home'
-import RouteTypes from './types'
-
-interface Authority {
-    authority?: string | undefined
-    strict?: boolean | undefined
-    with?: Authority | undefined
-}
-
-interface AuthorityProps {
-    authorities?: Authority | string[] | string | undefined
-    strictAuthority?: boolean | undefined
-}
-
-interface CustomRouteProps extends RouteProps, AuthorityProps {
-    key: string
-}
-
-export interface MenuProps {
-    icon: any
-    label?: string | undefined
-}
-
-export interface SideMenuRouteProps extends CustomRouteProps, MenuProps {}
-
-export interface SideSubMenuProps extends MenuProps {
-    key?: string | undefined
-    subMenus: SideSubMenuProps[]
-}
+import About from '@/pages/About'
+import NotFound from '@/pages/Error/404'
+import Home from '@/pages/Home'
+import Login from '@/pages/Login'
+import Profile from '@/pages/Profile'
+import { CustomRouteProps } from '@/routes/interfaces'
+import { RouteTypes } from '@/routes/types'
 
 const routes: CustomRouteProps[] = [
+    // Auth
+    {
+        key: 'login',
+        path: RouteTypes.LOGIN,
+        component: () => <Login />,
+    },
     {
         key: 'home',
         exact: true,
@@ -42,8 +24,20 @@ const routes: CustomRouteProps[] = [
         path: RouteTypes.ABOUT,
         component: () => <About />,
     },
+    {
+        exact: true,
+        key: 'profile',
+        path: RouteTypes.PROFILE,
+        component: () => <Profile />,
+        withAuthority: true,
+    },
+
+    // Errors
+    {
+        key: 'notfound',
+        path: RouteTypes.ERROR_404,
+        component: () => <NotFound />,
+    },
 ]
 
-const menus: (SideMenuRouteProps | SideSubMenuProps)[] = []
-
-export { routes, menus }
+export { routes }
diff --git a/src/routes/types.ts b/src/routes/types.ts
index 715bd6a..ef8de34 100644
--- a/src/routes/types.ts
+++ b/src/routes/types.ts
@@ -1,6 +1,15 @@
 const RouteTypes = {
     HOME: "/",
     ABOUT: "/about",
+    PROFILE: "/profile",
+
+    // Auth
+    LOGIN: '/login',
+
+    // Errors
+    ERROR_404: "**",
 }
 
-export default RouteTypes
\ No newline at end of file
+export {
+    RouteTypes,
+}
\ No newline at end of file
diff --git a/src/utils/log_util.ts b/src/utils/log_util.ts
new file mode 100644
index 0000000..d3cc5db
--- /dev/null
+++ b/src/utils/log_util.ts
@@ -0,0 +1,3 @@
+export const printInfo = (message?: any, ...opts: any[]) => console.log(message, opts)
+export const printError = (message?: any, ...opts: any[]) => console.error(message, opts)
+export const printWarn = (message?: any, ...opts: any[]) => console.warn(message, opts)
\ No newline at end of file
diff --git a/src/utils/ls_util.ts b/src/utils/ls_util.ts
new file mode 100644
index 0000000..e112993
--- /dev/null
+++ b/src/utils/ls_util.ts
@@ -0,0 +1,7 @@
+export const setStorage = (key: string, value?: any) => {
+    localStorage.setItem(key, JSON.stringify(value))
+}
+
+export const getStorage = (key: string, defaultValue?: string): (string | undefined | null) => {
+    return localStorage.getItem(key) ?? defaultValue
+}
\ No newline at end of file