R04-翁國翔

.env.example

@@ -0,0 +1 @@
+IMGUR_CLIENT_ID=SKIP

Procfile

@@ -0,0 +1 @@
+web: NODE_ENV=production node app.js

_helpers.js

@@ -1,12 +1,12 @@
 function ensureAuthenticated(req) {
-  return req.isAuthenticated();
+  return req.isAuthenticated()
 }
 
 function getUser(req) {
-  return req.user;
+  return req.user
 }
 
 module.exports = {
   ensureAuthenticated,
   getUser,
-};
+}

app.js

@@ -1,11 +1,51 @@
 const express = require('express')
+const handlebars = require('express-handlebars') // 引入 handlebars
+const bodyParser = require('body-parser')// Express 升級到了 4.17.1 之後，不用額外載入 body-parser，這行程式碼便不用寫
+const flash = require('connect-flash')
+const helpers = require('./_helpers')
+if (process.env.NODE_ENV !== 'production') {
+  require('dotenv').config()
+}
+
+const session = require('express-session')
+const passport = require('./config/passport')
+const methodOverride = require('method-override')
+
+const db = require('./models') // 引入資料庫
 const app = express()
-const port = 3000
+const port = process.env.PORT || 3000
+
+
+app.engine('handlebars', handlebars({
+  defaultLayout: 'main',
+  helpers: require('./config/handlebars-helpers')
+})) // Handlebars 註冊樣板引擎
+app.set('view engine', 'handlebars') // 設定使用 Handlebars 做為樣板引擎
+app.use(express.urlencoded({ extended: true }))
+
+app.use(session({ secret: 'secret', resave: false, saveUninitialized: false }))
+app.use(passport.initialize())
+app.use(passport.session())
+app.use(flash())
+
+// 把 req.flash 放到 res.locals 裡面
+app.use((req, res, next) => {
+  res.locals.success_messages = req.flash('success_messages')
+  res.locals.error_messages = req.flash('error_messages')
+  //res.locals.user = req.user //讓views能存取user
+  res.locals.user = helpers.getUser(req) // 取代 req.user
+  next()
+})
+
+app.use(methodOverride('_method'))
+app.use('/upload', express.static(__dirname + '/upload'))
+
 
 app.listen(port, () => {
   console.log(`Example app listening at http://localhost:${port}`)
 })
+// 引入 routes 並將 app 傳進去，讓 routes 可以用 app 這個物件來指定路由
 
-require('./routes')(app)
+require('./routes')(app, passport)
 
 module.exports = app

config/config.json

@@ -7,11 +7,7 @@
     "dialect": "mysql"
   },
   "production": {
-    "username": "root",
-    "password": null,
-    "database": "database_production",
-    "host": "127.0.0.1",
-    "dialect": "mysql"
+    "use_env_variable": "CLEARDB_DATABASE_URL"
   },
   "travis": {
     "username": "travis",
@@ -20,4 +16,4 @@
     "dialect": "mysql",
     "logging": false
   }
-}
+}

config/handlebars-helpers.js

@@ -0,0 +1,14 @@
+const moment = require('moment')
+
+
+module.exports = {
+  ifCond: function (a, b, options) {
+    if (a === b) {
+      return options.fn(this)
+    }
+    return options.inverse(this)
+  },
+  moment: function (a) {
+    return moment(a).fromNow()
+  }
+}

config/passport.js

@@ -0,0 +1,45 @@
+const passport = require('passport')
+const LocalStrategy = require('passport-local')
+const bcrypt = require('bcryptjs')
+const db = require('../models')
+const User = db.User
+const Restaurant = db.Restaurant
+
+
+
+// setup passport strategy
+passport.use(new LocalStrategy(
+  // customize user field
+  {
+    usernameField: 'email',
+    passwordField: 'password',
+    passReqToCallback: true
+  },
+  // authenticate user
+  (req, username, password, cb) => {
+    User.findOne({ where: { email: username } }).then(user => {
+      if (!user) return cb(null, false, req.flash('error_messages', '帳號或密碼輸入錯誤'))
+      if (!bcrypt.compareSync(password, user.password)) return cb(null, false, req.flash('error_messages', '帳號或密碼輸入錯誤！'))
+      return cb(null, user)
+    })
+  }
+))
+
+// serialize and deserialize user
+passport.serializeUser((user, cb) => {
+  cb(null, user.id)
+})
+passport.deserializeUser((id, cb) => {
+  User.findByPk(id, {
+    include: [
+      { model: Restaurant, as: 'FavoritedRestaurants' },
+      { model: Restaurant, as: 'LikeRestaurants' }
+
+    ]
+  }).then(user => {
+    user = user.toJSON()
+    return cb(null, user)
+  })
+})
+
+module.exports = passport

controllers/adminController.js

@@ -0,0 +1,183 @@
+const db = require('../models')
+const Restaurant = db.Restaurant
+const fs = require('fs')
+const imgur = require('imgur-node-api')
+const category = require('../models/category')
+const IMGUR_CLIENT_ID = process.env.IMGUR_CLIENT_ID
+const User = db.User
+const Category = db.Category
+
+
+const adminController = {
+  getRestaurants: (req, res) => {
+    return Restaurant.findAll({
+      raw: true,
+      nest: true,
+      include: [Category]
+    }).then(restaurants => {
+      // console.log(restaurants)
+      return res.render('admin/restaurants', { restaurants: restaurants })
+    })
+  },
+
+  createRestaurant: (req, res) => {
+    Category.findAll({
+      raw: true,
+      nest: true
+    }).then(categories => {
+      return res.render('admin/create', {
+        categories: categories
+      })
+    })
+  },
+
+  postRestaurant: (req, res) => {
+    if (!req.body.name) {
+      req.flash('error_messages', "name didn't exist")
+      return res.redirect('back')
+    }
+
+    const { file } = req
+    if (file) {
+      imgur.setClientID(IMGUR_CLIENT_ID);
+      imgur.upload(file.path, (err, img) => {
+        return Restaurant.create({
+          name: req.body.name,
+          tel: req.body.tel,
+          address: req.body.address,
+          opening_hours: req.body.opening_hours,
+          description: req.body.description,
+          image: file ? img.data.link : null,
+          CategoryId: req.body.categoryId
+
+        }).then((restaurant) => {
+          req.flash('success_messages', 'restaurant was successfully created')
+          return res.redirect('/admin/restaurants')
+        })
+      })
+    }
+    else {
+      return Restaurant.create({
+        name: req.body.name,
+        tel: req.body.tel,
+        address: req.body.address,
+        opening_hours: req.body.opening_hours,
+        description: req.body.description,
+        image: null,
+        CategoryId: req.body.categoryId
+
+      }).then((restaurant) => {
+        req.flash('success_messages', 'restaurant was successfully created')
+        return res.redirect('/admin/restaurants')
+      })
+    }
+  },
+  getRestaurant: (req, res) => {
+    return Restaurant.findByPk(req.params.id, { include: [Category] }).then(
+      restaurant => {
+        console.log(restaurant)
+        return res.render('admin/restaurant', {
+          restaurant: restaurant.toJSON()
+        })
+      }
+    )
+  },
+
+  editRestaurant: (req, res) => {
+    Category.findAll({
+      raw: true,
+      nest: true
+    }).then(categories => {
+      return Restaurant.findByPk(req.params.id).then(restaurant => {
+        return res.render('admin/create', {
+          categories: categories,
+          restaurant: restaurant.toJSON()
+        })
+      })
+
+    })
+  },
+
+  putRestaurant: (req, res) => {
+    if (!req.body.name) {
+      req.flash('error_messages', "name didn't exist")
+      return res.redirect('back')
+    }
+
+    const { file } = req
+    if (file) {
+      imgur.setClientID(IMGUR_CLIENT_ID);
+      imgur.upload(file.path, (err, img) => {
+        return Restaurant.findByPk(req.params.id)
+          .then((restaurant) => {
+            restaurant.update({
+              name: req.body.name,
+              tel: req.body.tel,
+              address: req.body.address,
+              opening_hours: req.body.opening_hours,
+              description: req.body.description,
+              image: file ? img.data.link : restaurant.image,
+              CategoryId: req.body.categoryId
+
+            })
+              .then((restaurant) => {
+                req.flash('success_messages', 'restaurant was successfully to update')
+                res.redirect('/admin/restaurants')
+              })
+          })
+      })
+    }
+    else {
+      return Restaurant.findByPk(req.params.id)
+        .then((restaurant) => {
+          restaurant.update({
+            name: req.body.name,
+            tel: req.body.tel,
+            address: req.body.address,
+            opening_hours: req.body.opening_hours,
+            description: req.body.description,
+            image: restaurant.image,
+            CategoryId: req.body.categoryId
+
+          })
+            .then((restaurant) => {
+              req.flash('success_messages', 'restaurant was successfully to update')
+              res.redirect('/admin/restaurants')
+            })
+        })
+    }
+  },
+
+  deleteRestaurant: (req, res) => {
+    return Restaurant.findByPk(req.params.id)
+      .then((restaurant) => {
+        restaurant.destroy()
+          .then((restaurant) => {
+            res.redirect('/admin/restaurants')
+          })
+      })
+  },
+
+  getUsers: (req, res) => {
+    return User.findAll({ raw: true }).then(users => {
+      return res.render('admin/users', { users: users })
+    })
+
+  },
+  toggleAdmin: (req, res) => {
+    return User.findByPk(req.params.id).then((user) => {
+      const { isAdmin, email } = user.toJSON();
+      if (email === "root@example.com") {
+        req.flash("error_messages", "禁止變更管理者權限");
+        return res.redirect("back");
+      } else {
+        user.update({ isAdmin: !isAdmin }).then((user) => {
+          req.flash("success_messages", "使用者權限變更成功");
+          res.redirect("/admin/users");
+        });
+      }
+    });
+  },
+}
+
+module.exports = adminController