Added Home Router with middleware function

app.js

@@ -15,6 +15,7 @@ const USER_ROUTER = require('./controllers/Users');
 const SESSION_ROUTER = require('./controllers/Sessions');
 const HOME_ROUTER = require('./controllers/Home');
 
+// const checkSession = require('./lib/helpers/checkSession');
 
 //configure express and related packages
 app.engine('html', mustache());
@@ -38,10 +39,7 @@ app.use(session({
   }
 }));
 
-app.all('/', (req, res, next) => {
-  console.log('helloWorld');
-  next();
-})
+// app.all(['/'], checkSession)
 
 //start the server
 var PORT = process.env.PORT || 3000;

controllers/Home.js

@@ -2,10 +2,8 @@ const express = require('express');
 const router = express.Router();
 const { getAllDjs } = require('../models/Home');
 
-// Home#Show
-router.get('/', getAllDjs, (req, res, next) => {
+router.all('/', (req, res, next) => {
   const user = req.session.user;
-
   if (user) {
     if (user.type === "fan") {
       console.log(user.type);
@@ -14,9 +12,14 @@ router.get('/', getAllDjs, (req, res, next) => {
       res.redirect("/djboard");
     }
   } else {
+    next();
+  }
+})
+
+// Home#Show
+router.get('/', getAllDjs, (req, res, next) => {
     const djs = req.user;
     res.render('home/index', { djs });
-  }
 });
 
 module.exports = router;

lib/helpers/checkSession.js

@@ -0,0 +1,13 @@
+module.exports = (req, res, next) => {
+  const user = req.session.user;
+  if (user) {
+    if (user.type === "fan") {
+      console.log(user.type);
+      res.redirect('/userboard');
+    } else if (user.type === "dj") {
+      res.redirect("/djboard");
+    }
+  } else {
+    next();
+  }
+}