Add context to pass the user token through components

.gitignore

@@ -1,3 +1,4 @@
+/server/craban
 /server/craban.sqlite
 /server/web/static/*
 /webapp/node_modules/

webapp/src/client.js

@@ -1,14 +1,13 @@
 export class Client {
   login(username, password) {
-    console.log('Logging with', username)
-
+    // ToDo: handle error
     return fetch("/api/login", {
       method: 'POST',
       headers: { 'Content-Type': 'application/json' },
       body: JSON.stringify({ username, password })
     }).then(r => r.text()).then(token => {
-      console.log("GOT TOKEN")
       localStorage.setItem('token', token)
+      return token
     })
   }
 }

webapp/src/index.js

@@ -1,3 +1,4 @@
+import { useState, useContext } from 'react'
 import ReactDOM from 'react-dom'
 import {
   BrowserRouter as Router,
@@ -9,12 +10,13 @@ import {
 
 import { CssBaseline } from '@material-ui/core';
 
+import UserContext from './user-context'
 import Login from './pages/login'
+import Home from './pages/home'
 
 
 const Secure = ({children}) => {
-  const token = localStorage.getItem('token')
-
+  const { token } = useContext(UserContext)
   if (!token) {
     return <Redirect to="/login" />
   }
@@ -22,12 +24,11 @@ const Secure = ({children}) => {
   return children
 }
 
-const Home = () => {
-  return <h1>Home</h1>
-}
-
 const App = () => {
+  const [token, setToken] = useState(localStorage.getItem('token'))
+
   return (
+    <UserContext.Provider value={{ token, setToken }}>
       <Router>
         <CssBaseline />
         <Switch>
@@ -42,6 +43,7 @@ const App = () => {
           </Route>
         </Switch>
       </Router>
+    </UserContext.Provider>
   )
 }
 

webapp/src/pages/home.js

@@ -0,0 +1,33 @@
+import { useContext } from 'react'
+import {
+  Box,
+  Button
+} from '@material-ui/core';
+
+import UserContext from '../user-context'
+
+
+const Home = () => {
+  const { token, setToken } = useContext(UserContext)
+
+  const handleLogout = (e) => {
+    e.preventDefault()
+    localStorage.removeItem('token')
+    setToken(null)
+  }
+
+  return (
+    <Box>
+      <h1>Home</h1>
+
+      <Button
+        variant="contained"
+        color="secondary"
+        onClick={handleLogout}>
+        Logout
+      </Button>
+    </Box>
+  )
+}
+
+export default Home

webapp/src/pages/login.js

@@ -1,4 +1,4 @@
-import { useState } from 'react'
+import { useState, useContext } from 'react'
 import { Redirect } from 'react-router-dom'
 import {
   Box,
@@ -7,9 +7,12 @@ import {
 } from '@material-ui/core';
 
 import client from '../client'
+import UserContext from '../user-context'
 
 const Login = () => {
-  if (localStorage.getItem('token')) {
+  const { token, setToken } = useContext(UserContext)
+
+  if (token) {
     return <Redirect to='/' />
   }
 
@@ -18,9 +21,8 @@ const Login = () => {
 
   const handleSubmit = (e) => {
     e.preventDefault()
-    client.login(username, password).then(() => {
-      setUsername('')
-      setPassword('')
+    client.login(username, password).then(token => {
+      setToken(token)
     })
   }
 

webapp/src/user-context.js

@@ -0,0 +1,5 @@
+import React from 'react'
+
+const UserContext = React.createContext()
+
+export default UserContext