fix redirect after login

Author: Matt

client/src/App.js

@@ -1,20 +1,16 @@
 import React from 'react'
-import { BrowserRouter as Router, Route } from 'react-router-dom'
+import { Switch, Route, withRouter } from 'react-router-dom'
 import LoginContainer from 'components/Login/LoginContainer'
 import ProtectedContainer from 'components/Protected/ProtectedContainer'
 import HomeContainer from 'components/Home/HomeContainer'
-import Header from 'components/Header'
 import 'App.css'
 
 const App = () => (
-  <Router>
-    <div>
-      <Header />
+  <Switch>
       <Route exact path="/" component={HomeContainer} />
       <Route path="/login" component={LoginContainer} />
       <Route path="/protected" component={ProtectedContainer} />
-    </div>
-  </Router>
+  </Switch>
 )
 
-export default App
+export default withRouter(App)

client/src/components/Login/LoginContainer.js

@@ -16,7 +16,7 @@ class LoginContainer extends React.Component {
     return (
       <div>
         {loggedIn ? (
-          <Redirect to="/home" />
+          <Redirect to="/" />
         ) : (
           <LoginForm
             handleSubmit={handleSubmit}

client/src/index.js

@@ -6,6 +6,8 @@ import { Provider } from 'react-redux'
 import { homeReducer } from './reducers'
 import { createStore, applyMiddleware } from 'redux'
 import thunk from 'redux-thunk'
+import { BrowserRouter as Router } from 'react-router-dom'
+import Header from 'components/Header'
 
 const store = createStore(
   homeReducer,
@@ -15,7 +17,12 @@ const store = createStore(
 
 ReactDOM.render(
   <Provider store={store}>
-    <App />
+      <Router>
+        <div>
+          <Header />
+          <App />
+        </div>
+    </Router>
   </Provider>,
   document.getElementById('root')
 )