add google oauth

Author: lassejlv

index.js

@@ -5,10 +5,19 @@ const express = require("express");
 const path = require("path");
 const app = express();
 const chalk = require("chalk");
-const passport = require("passport");
 const session = require("express-session");
+const passport = require("passport");
+const mongoose = require("mongoose");
+const MongoStore = require("connect-mongo")(session);
 const PORT = process.env.PORT || 5000;
 
+// Passport
+require("./src/utils/passport")(passport);
+
+// Passport Middlewares
+app.use(passport.initialize());
+app.use(passport.session());
+
 // Routes
 const apiRouter = require("./src/routes/api");
 const authRouter = require("./src/routes/auth");
@@ -22,9 +31,10 @@ app.use("/", indexRouter);
 // Sessions
 app.use(
     session({
-        secret: process.env.SESSION_SECRET,
+        secret: process.env.COOKIE_SESSION,
         resave: false,
         saveUninitialized: false,
+        store: new MongoStore({ mongooseConnection: mongoose.connection }),
     })
 );
 

public/style.css

@@ -64,6 +64,7 @@ nav {
 .main__whyComponent {
     width: 100%;
     margin-bottom: 2rem;
+    font-weight: 500;
     font-size: larger;
 }
 

public/uploads/V8gpDEN79BbaIjd-billede-8.JPG

@@ -5,29 +5,24 @@ const UserSchema = new mongoose.Schema({
         type: String,
         required: true,
     },
-
     displayName: {
         type: String,
         required: true,
     },
-
     firstName: {
         type: String,
         required: true,
     },
-
     lastName: {
         type: String,
         required: true,
     },
-
     image: {
         type: String,
     },
-
     createdAt: {
         type: Date,
-        default: Date.now(),
+        default: Date.now,
     },
 });
 

public/uploads/dx95Uc5PnzOuW4s-billede.PNG

@@ -0,0 +1,22 @@
+const yourid = require("yourid");
+module.exports = {
+    ensureAuth: function (req, res, next) {
+        if (req.isAuthenticated()) {
+            return next();
+        } else {
+            res.redirect(
+                "/?redirect=" +
+                    req.originalUrl +
+                    "&error=You must be logged in to view this page." +
+                    `&error_id=${yourid.generate(15)}`
+            );
+        }
+    },
+    ensureGuest: function (req, res, next) {
+        if (!req.isAuthenticated()) {
+            return next();
+        } else {
+            res.redirect("/dashboard");
+        }
+    },
+};

src/auth.js

@@ -5,4 +5,19 @@ router.get("/", (req, res) => {
     res.send(200);
 });
 
+router.get("/google", passport.authenticate("google", { scope: ["profile"] }));
+
+router.get(
+    "/google/callback",
+    passport.authenticate("google", {
+        failureRedirect: "/",
+        successRedirect: "/dashboard",
+    })
+);
+
+router.get("/logout", (req, res) => {
+    req.logout();
+    res.redirect("/");
+});
+
 module.exports = router;

src/database/models/user.js

@@ -1,20 +1,20 @@
 const router = require("express").Router();
 const Image = require("../database/models/image");
+const { ensureAuth, ensureGuest } = require("../middleware/requireAuth");
 
-router.get("/", (req, res) => {
+router.get("/", ensureGuest, (req, res) => {
     res.render("index");
 });
 
 router.get("/upload", (req, res) => {
     res.render("upload");
 });
 
-router.get("/dashboard", (req, res) => {
-    // res.render("dashboard");
-    res.send(200);
+router.get("/dashboard", ensureAuth, (req, res) => {
+    res.render("dashboard");
 });
 
-router.get("/view/:id", (req, res, next) => {
+router.get("/view/:id", ensureGuest, (req, res, next) => {
     Image.findOne({ imageId: req.params.id }, (err, image) => {
         if (err) {
             res.send(err);

src/middleware/requireAuth.js

@@ -0,0 +1,46 @@
+const Google = require("passport-google-oauth20").Strategy;
+const mongoose = require("mongoose");
+const User = require("../database/models/User");
+
+module.exports = function (passport) {
+    passport.use(
+        new Google(
+            {
+                clientID: process.env.GOOGLE_CLIENT_ID,
+                clientSecret: process.env.GOOGLE_CLIENT_SECRET,
+                callbackURL: "/auth/google/callback",
+            },
+
+            async (accessToken, refreshToken, profile, done) => {
+                const newUser = {
+                    googleId: profile.id,
+                    displayName: profile.displayName,
+                    firstName: profile.name.givenName,
+                    lastName: profile.name.familyName,
+                    image: profile.photos[0].value,
+                };
+
+                try {
+                    let user = await User.findOne({ googleId: profile.id });
+
+                    if (user) {
+                        done(null, user);
+                    } else {
+                        user = await User.create(newUser);
+                        done(null, user);
+                    }
+                } catch (err) {
+                    console.log(err);
+                }
+            }
+        )
+    );
+
+    passport.serializeUser((user, done) => {
+        done(null, user.id);
+    });
+
+    passport.deserializeUser((id, done) => {
+        User.findById(id, (err, user) => done(err, user));
+    });
+};