R04-leo

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+  "emmet.optimizeStylesheetParsing": false
+}

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,41 @@
 const express = require('express')
+const flash = require('connect-flash')
+const session = require('express-session')
+const passport = require('./config/passport')
 const app = express()
-const port = 3000
+const { engine } = require('express-handlebars')
+const db = require('./models') // 引入資料庫
+const methodOverride = require('method-override')
+const helpers = require('./_helpers')
+const port = process.env.PORT || 3000
 
+if (process.env.NODE_ENV !== 'production') {
+  require('dotenv').config()
+}
+
+app.use(express.urlencoded({ extended: true }))
+app.use(session({ secret: 'secret', resave: false, saveUninitialized: false }))
+app.use(flash())
+app.use(methodOverride('_method'))
+app.use('/upload', express.static(__dirname + '/upload'))
+// setup passport
+app.use(passport.initialize())
+app.use(passport.session())
+
+app.engine('handlebars', engine({ defaultLayout: 'main', helpers: require('./config/handlebars-helpers') }));
+app.set('view engine', 'handlebars')
+
+// 把 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
+  next()
+})
 app.listen(port, () => {
   console.log(`Example app listening at http://localhost:${port}`)
 })
 
-require('./routes')(app)
+require('./routes')(app, passport)
 
-module.exports = app
+module.exports = app

config/config.json

@@ -9,11 +9,21 @@
   "production": {
     "username": "root",
     "password": null,
-    "database": "database_production",
+    "database": "database_test",
     "host": "127.0.0.1",
     "dialect": "mysql"
   },
-  "travis": {
+  "production": {
+    "use_env_variable": "CLEARDB_DATABASE_URL",
+    "dialect": "postgres",
+    "protocol": "postgres",
+    "dialectOptions": {
+      "ssl": {
+        "rejectUnauthorized": false
+      }
+    }
+  },
+    "travis": {
     "username": "travis",
     "database": "forum",
     "host": "127.0.0.1",

config/handlebars-helpers.js

@@ -0,0 +1,12 @@
+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,43 @@
+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
+const Like = db.Like
+
+// 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: 'LikedRestaurants' }
+    ]
+  }).then(user => {
+    user = user.toJSON()
+    return cb(null, user)
+  })
+})
+
+module.exports = passport

controllers/adminController.js

@@ -0,0 +1,169 @@
+const db = require('../models')
+const Restaurant = db.Restaurant
+const User = db.User
+const fs = require('fs')
+const imgur = require('imgur-node-api')
+const IMGUR_CLIENT_ID = process.env.IMGUR_CLIENT_ID
+const Category = db.Category
+
+const adminController = {
+  //getRestaurants
+  getRestaurants: (req, res) => {
+    return Restaurant.findAll({
+      raw: true,
+      nest: true,
+      include: [Category]
+    }).then(restaurants => {
+      return res.render('admin/restaurants', { restaurants })
+    })
+  },
+  //getRestaurant
+  getRestaurant: (req, res) => {
+    return Restaurant.findByPk(req.params.id, {
+      include: [Category]
+    }).then(restaurant => {
+      return res.render('admin/restaurant', {
+        restaurant: restaurant.toJSON()
+      })
+    })
+  },
+  /* create */
+  // createPage
+  createRestaurant: (req, res) => {
+    Category.findAll({
+      raw: true,
+      nest: true
+    }).then(categories => {
+      return res.render('admin/create', { categories })
+    })
+  },
+  // POST to create
+  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')
+            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')
+      })
+    }
+  },
+  editRestaurant: async (req, res) => {
+    const categories = await Category.findAll({
+      raw: true,
+      nest: true
+    })
+    const restaurant = await Restaurant.findByPk(req.params.id)
+    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 })
+    })
+  },
+  toggleAdmin: (req, res) => {
+    return User.findByPk(req.params.id)
+      .then(
+        (user) => {
+          if (user.email !== 'root@example.com') {
+            user.update({
+              isAdmin: !user.isAdmin
+            })
+              .then(() => {
+                req.flash('success_messages', '使用者權限變更成功')
+                res.redirect('/admin/users')
+              })
+          } else {
+            req.flash('error_messages', '禁止變更管理者權限')
+            return res.redirect('back')
+          }
+        })
+  }
+}
+
+module.exports = adminController

controllers/categoryController.js

@@ -0,0 +1,54 @@
+const db = require('../models')
+const Category = db.Category
+let categoryController = {
+  getCategories: (req, res) => {
+    return Category.findAll({
+      raw: true,
+      nest: true
+    }).then(categories => {
+      return res.render('admin/categories', { categories })
+    })
+  },
+  postCategory: (req, res) => {
+    if (!req.body.name) {
+      req.flash('error_messages', 'name didn\'t exist')
+      return res.redirect('back')
+    } else {
+      return Category.create({
+        name: req.body.name
+      })
+        .then((category) => {
+          res.redirect('/admin/categories')
+        })
+    }
+  },
+  getCategories: async (req, res) => {
+    const categories = await Category.findAll({
+      raw: true,
+      nest: true
+    })
+    if (req.params.id) {
+      const category = await Category.findByPk(req.params.id)
+      return res.render('admin/categories', { categories, category: category.toJSON() })
+    } else {
+      return res.render('admin/categories', { categories })
+    }
+  },
+  putCategory: async (req, res) => {
+    if (!req.body.name) {
+      req.flash('error_messages', 'name didn\'t exist')
+      return res.redirect('back')
+    } else {
+      const category = await Category.findByPk(req.params.id)
+      await category.update(req.body)
+      res.redirect('/admin/categories')
+    }
+  },
+  deleteCategory: async (req, res) => {
+
+    const category = await Category.findByPk(req.params.id)
+    await category.destroy()
+    res.redirect('/admin/categories')
+  }
+}
+module.exports = categoryController