From 686c8fe0089c37ac0146aef3d42a8768b61e45eb Mon Sep 17 00:00:00 2001
From: CodeServer <CodeServer@jlu42.com>
Date: Fri, 11 Mar 2022 08:45:17 +0000
Subject: [PATCH] finished Protected Routes

---
 src/App.tsx                    | 11 ++++++---
 src/ProtectedRoute.tsx         | 12 ++++++++++
 src/components/login/Login.tsx | 41 +++++++++++++++++++++++++++++++---
 src/context/AuthProvider.tsx   | 15 +++++++++++++
 src/hooks/useAuth.tsx          |  8 +++++++
 src/index.tsx                  |  7 +++++-
 6 files changed, 87 insertions(+), 7 deletions(-)
 create mode 100644 src/ProtectedRoute.tsx
 create mode 100644 src/context/AuthProvider.tsx
 create mode 100644 src/hooks/useAuth.tsx

diff --git a/src/App.tsx b/src/App.tsx
index 638d685..1151917 100644
--- a/src/App.tsx
+++ b/src/App.tsx
@@ -2,11 +2,13 @@ import React, { useState } from "react";
 import { BrowserRouter as Router, Routes, Route } from "react-router-dom";
 import "./styles.css";
 
+import Login from "./components/login/Login";
 import Home from "./components/home/Home";
 import Contacts from "./components/contacts/Contacts";
 import Calendar from "./components/calendar/Calendar";
 import Navbar from "./components/navbar/Navbar";
 import { ThemeProvider } from "@emotion/react";
+import ProtectedRoute from "./ProtectedRoute";
 
 import Theme from "./Theme";
 
@@ -47,9 +49,12 @@ const App: React.FC = () => {
       <Router>
         <Navbar />
         <Routes>
-          <Route path="/" element={<Home />} />
-          <Route path="/contacts" element={<Contacts />} />
-          <Route path="/calendar" element={<Calendar />} />
+          <Route path="/login" element={<Login />} />
+          <Route element={<ProtectedRoute />}>
+            <Route path="/" element={<Home />} />
+            <Route path="/contacts" element={<Contacts />} />
+            <Route path="/calendar" element={<Calendar />} />
+          </Route>
         </Routes>
       </Router>
     </ThemeProvider>
diff --git a/src/ProtectedRoute.tsx b/src/ProtectedRoute.tsx
new file mode 100644
index 0000000..927eb13
--- /dev/null
+++ b/src/ProtectedRoute.tsx
@@ -0,0 +1,12 @@
+import { useLocation, Outlet, Navigate } from "react-router-dom";
+import useAuth from "./hooks/useAuth";
+
+const ProtectedRoute = () => {
+    const { auth }: any = useAuth();
+    const location = useLocation();
+    return (
+        auth?.isLoggedIn ? <Outlet /> : <Navigate to="/login" state={{ from: location }} replace />
+    );
+}
+
+export default ProtectedRoute;
\ No newline at end of file
diff --git a/src/components/login/Login.tsx b/src/components/login/Login.tsx
index cb3f046..c63160c 100644
--- a/src/components/login/Login.tsx
+++ b/src/components/login/Login.tsx
@@ -1,3 +1,5 @@
+import { useRef, useState, useEffect, useContext } from 'react';
+import { useLocation, Link, useNavigate } from "react-router-dom";
 import { Stack } from "@mui/material";
 import Container from "@mui/material/Container";
 import TextField from "@mui/material/TextField";
@@ -5,17 +7,50 @@ import Button from "@mui/material/Button";
 import hsbcLogo from "../../assets/logo-png.png";
 import zoomLogo from "../../assets/zoom.png";
 import LoginIcon from '@mui/icons-material/Login';
+import useAuth from "../../hooks/useAuth";
 
 const Login: React.FC = () => {
+
+    const { setAuth }: any = useAuth();
+
+    const navigate = useNavigate();
+    const location: any = useLocation();
+    const from = location.state?.from?.pathname || "/";
+
+    const [username, setUsername] = useState('');
+    const [password, setPassword] = useState('');
+    // const [errMsg, setErrMsg] = useState('');
+    
+    // const userRef = useRef();
+    // const errRef = useRef();
+
+    // useEffect(() => {
+    //     userRef.current.focus();
+    // }, [])
+
+    // useEffect(() => {
+    //     setErrMsg('');
+    // }, [user, pwd])
+
+    const handleLogin = () => {
+        if (username === "" && password === "") {
+            setAuth({username: username, isLoggedIn: true});
+            navigate(from, {replace: true});
+        }
+        setUsername("");
+        setPassword("");
+    };
+
     return (
         <Container className="login">
             <Stack className="grid-container" spacing={2}>
                 <img className="login-logo" src={hsbcLogo} alt="HSBC Logo"/>
-                <TextField className="username-input" id="outlined-basic" label="Username" variant="outlined" placeholder="Username"/>
-                <TextField className="password-input" id="outlined-basic" label="Password" variant="outlined" placeholder="Password"/>
+                {/* <p ref={errRef} className={errMsg ? "errmsg" : "offscreen"} aria-live="assertive">{errMsg}</p> */}
+                <TextField className="username-input" id="outlined-basic" label="Username" variant="outlined" placeholder="Username" onChange={(event) => {setUsername(event.target.value)}}/>
+                <TextField className="password-input" id="outlined-basic" label="Password" variant="outlined" placeholder="Password" type="password" onChange={(event) => {setPassword(event.target.value)}}/>
                 <Stack direction="row" justifyContent="space-between" spacing={2}>
                     <a className="register-btn" href="#">Forgotten Your Password?</a>
-                    <Button endIcon={<LoginIcon />} className="login-btn" href="#" variant="contained">Login</Button>
+                    <Button endIcon={<LoginIcon />} className="login-btn" variant="contained" type="submit" onClick={handleLogin}>Login</Button>
                 </Stack>
                 <Stack direction="row" justifyContent="space-between" spacing={2}>
                     <label>Login with</label>
diff --git a/src/context/AuthProvider.tsx b/src/context/AuthProvider.tsx
new file mode 100644
index 0000000..cb661da
--- /dev/null
+++ b/src/context/AuthProvider.tsx
@@ -0,0 +1,15 @@
+import { createContext, useState } from "react";
+
+const AuthContext = createContext({});
+
+export const AuthProvider = ({ children }: {children: any}) => {
+    const [auth, setAuth] = useState({});
+
+    return (
+        <AuthContext.Provider value={{ auth, setAuth }}>
+            {children}
+        </AuthContext.Provider>
+    )
+}
+
+export default AuthContext;
\ No newline at end of file
diff --git a/src/hooks/useAuth.tsx b/src/hooks/useAuth.tsx
new file mode 100644
index 0000000..d1fb072
--- /dev/null
+++ b/src/hooks/useAuth.tsx
@@ -0,0 +1,8 @@
+import { useContext } from "react";
+import AuthContext from "../context/AuthProvider";
+
+const useAuth = () => {
+    return useContext(AuthContext);
+}
+
+export default useAuth;
\ No newline at end of file
diff --git a/src/index.tsx b/src/index.tsx
index 615fc79..d8f9587 100644
--- a/src/index.tsx
+++ b/src/index.tsx
@@ -1,9 +1,14 @@
 import ReactDOM from "react-dom";
 import App from "./App";
 import "./style/App.css";
+import { AuthProvider } from './context/AuthProvider';
 
 const Index:React.FC = () => {
-  return <App />;
+  return (
+    <AuthProvider>
+      <App />
+    </AuthProvider>
+  );
 };
 
 ReactDOM.render(<Index />, document.getElementById("root"));