[后端] 生命周期添加mountingRoute，并增加对应的挂载路由插件

Author: smallpath

server/app.js

@@ -27,14 +27,15 @@ const router = koaRouter()
   const beforeRestfulRoutes = lifecycle.getBeforeRestfulRoutes()
   const afterRestfulRoutes = lifecycle.getAfterRestfulRoutes()
 
-  router.post(
-    '/admin/qiniu',
-    ...beforeRestfulRoutes,
-    require('./service/qiniu'),
-    ...afterRestfulRoutes
-  )
-  router.post('/admin/login', require('./auth/login'))
-  router.post('/admin/logout', require('./auth/logout'))
+  const middlewareRoutes = lifecycle.getMiddlewareRoutes()
+
+  for (const item of middlewareRoutes) {
+    const middlewares = [...item.middleware]
+    item.needBeforeRoutes && middlewares.unshift(...beforeRestfulRoutes)
+    item.needAfterRoutes && middlewares.push(...afterRestfulRoutes)
+
+    router[item.method](item.path, ...middlewares)
+  }
 
   Object.keys(models).map(name => models[name]).forEach(model => {
     mongoRest(router, model, '/api', {
@@ -53,7 +54,7 @@ const router = koaRouter()
     }
 
     app.listen(config.serverPort, () => {
-      log.debug(`koa2 is running at ${config.serverPort}`)
+      log.info(`Koa2 is running at ${config.serverPort}`)
     })
   } catch (err) {
     log.error(err)

server/blogpack.js

@@ -17,22 +17,33 @@ class blogpack {
     }
   }
 
+  getMiddlewareRoutes(...args) {
+    return this.plugins
+            .filter(plugin => plugin['mountingRoute'])
+            .map(plugin => {
+              return Object.assign({}, plugin.mountingRoute(), {
+                needBeforeRoutes: plugin.needBeforeRoutes || false,
+                needAfterRoutes: plugin.needAfterRoutes || false
+              })
+            })
+  }
+
   getBeforeRestfulRoutes() {
     return this.plugins
-                .filter(plugin => plugin['beforeRestful'])
-                .map(plugin => plugin['beforeRestful'])
+            .filter(plugin => plugin['beforeRestful'])
+            .map(plugin => plugin['beforeRestful'])
   }
 
   getAfterRestfulRoutes() {
     return this.plugins
-                .filter(plugin => plugin['afterRestful'])
-                .map(plugin => plugin['afterRestful'])
+            .filter(plugin => plugin['afterRestful'])
+            .map(plugin => plugin['afterRestful'])
   }
 
   getBeforeServerStartFuncs() {
     return this.plugins
-                .filter(plugin => plugin['beforeServerStart'])
-                .map(plugin => plugin['beforeServerStart'])
+            .filter(plugin => plugin['beforeServerStart'])
+            .map(plugin => plugin['beforeServerStart'])
   }
 }
 

server/build/blogpack.prod.config.js

@@ -2,6 +2,7 @@ const blogpack = require('../blogpack')
 const base = require('./blogpack.base.config')
 const useRoutesPrefix = '../plugins/beforeUseRoutes'
 const serverStartPrefix = '../plugins/beforeServerStart'
+const env = process.env
 
 const config = Object.assign({}, base, {
   plugins: base.plugins.concat([
@@ -17,12 +18,17 @@ const RatelimitPlugin = require(`${useRoutesPrefix}/ratelimit`)
 const BodyParserPlugin = require(`${useRoutesPrefix}/bodyParser`)
 const LogTimePlugin = require(`${useRoutesPrefix}/logTime`)
 const RestcPlugin = require(`${useRoutesPrefix}/restc`)
+
 const InitOptionPlugin = require(`${serverStartPrefix}/initOption`)
 const InstallThemePlugin = require(`${serverStartPrefix}/installTheme`)
 const InitUserPlugin = require(`${serverStartPrefix}/initUser`)
 
 const CheckAuthPlugin = require('../plugins/beforeRestful/checkAuth')
 
+const QiniuUploadPlugin = require('../plugins/mountingRoute/qiniu')
+const LoginPlugin = require('../plugins/mountingRoute/login')
+const LogoutPlugin = require('../plugins/mountingRoute/logout')
+
 config.plugins.push(
   // beforeUseRoutes
   new RatelimitPlugin({
@@ -35,13 +41,24 @@ config.plugins.push(
   new LogTimePlugin(),
   new RestcPlugin(),
 
+  // beforeRestful
+  new CheckAuthPlugin(),
+
+  // moutingRoute
+  new QiniuUploadPlugin({
+    qiniuAccessKey: env.qiniuAccessKey || '',
+    qiniuSecretKey: env.qiniuSecretKey || '',
+    qiniuBucketHost: env.qiniuBucketHost || '',
+    qiniuBucketName: env.qiniuBucketName || '',
+    qiniuPipeline: env.qiniuPipeline || ''
+  }),
+  new LoginPlugin(),
+  new LogoutPlugin(),
+
   // beforeServerStart
   new InitUserPlugin(),
   new InstallThemePlugin(),
-  new InitOptionPlugin(),
-
-  // beforeRestful
-  new CheckAuthPlugin()
+  new InitOptionPlugin()
 )
 
 module.exports = config

server/conf/config.tpl

@@ -15,11 +15,5 @@ module.exports = {
   tokenExpiresIn: env.tokenExpiresIn || 3600,
 
   defaultAdminName: env.defaultAdminName || 'admin',
-  defaultAdminPassword: env.defaultAdminPassword || 'admin',
-
-  qiniuAccessKey: env.qiniuAccessKey || '',
-  qiniuSecretKey: env.qiniuSecretKey || '',
-  qiniuBucketHost: env.qiniuBucketHost || '',
-  qiniuBucketName: env.qiniuBucketName || '',
-  qiniuPipeline: env.qiniuPipeline || ''
+  defaultAdminPassword: env.defaultAdminPassword || 'admin'
 }

server/lifecycle/index.js

@@ -15,7 +15,7 @@ db.on('error', (err) => {
 })
 
 db.once('open', () => {
-  log.debug('MongoDB is ready')
+  log.info('MongoDB is ready')
 })
 
 const Schema = mongoose.Schema

server/model/mongo.js

@@ -18,7 +18,7 @@ client.on('error', function(err) {
 })
 
 client.on('connect', function() {
-  log.debug('Redis is ready')
+  log.info('Redis is ready')
 })
 
 module.exports = client

server/model/redis.js

@@ -1,8 +1,18 @@
-const redis = require('../model/redis')
-const tokenService = require('../service/token')
-const { user: model } = require('../model/mongo')
+const redis = require('../../model/redis')
+const tokenService = require('../../service/token')
+const { user: model } = require('../../model/mongo')
 
-module.exports = async function(ctx, next) {
+module.exports = class {
+  mountingRoute() {
+    return {
+      method: 'post',
+      path: '/admin/login',
+      middleware: [middleware]
+    }
+  }
+}
+
+async function middleware(ctx, next) {
   let users, user
 
   try {

server/auth/login.js renamed to server/plugins/mountingRoute/login.js

@@ -1,7 +1,17 @@
-const redis = require('../model/redis')
-const tokenService = require('../service/token')
+const redis = require('../../model/redis')
+const tokenService = require('../../service/token')
 
-module.exports = async function(ctx, next) {
+module.exports = class {
+  mountingRoute() {
+    return {
+      method: 'post',
+      path: '/admin/logout',
+      middleware: [middleware]
+    }
+  }
+}
+
+async function middleware(ctx, next) {
   const headers = ctx.request.headers
   let token
   try {

server/auth/logout.js renamed to server/plugins/mountingRoute/logout.js

@@ -5,7 +5,7 @@ let {
     qiniuSecretKey,
     qiniuBucketName,
     qiniuPipeline
-} = require('../conf/config')
+} = require('../../conf/config')
 
 qiniu.conf.ACCESS_KEY = qiniuAccessKey
 qiniu.conf.SECRET_KEY = qiniuSecretKey
@@ -43,6 +43,18 @@ const getQiniuTokenFromFileName = (fileName) => {
   }
 }
 
-module.exports = ({ request, response }, next) => {
-  return response.body = getQiniuTokenFromFileName(request.body.key)
+module.exports = class {
+  mountingRoute() {
+    return {
+      method: 'post',
+      path: '/admin/qiniu',
+      needBeforeRoutes: true,
+      middleware: [
+        function({ request, response }, next) {
+          return response.body = getQiniuTokenFromFileName(request.body.key)
+        }
+      ],
+      needAfterRoutes: false
+    }
+  }
 }