core/middleware/router.js

/**
 * 路由中间件
 * @author yutent<yutent.io@gmail.com>
 * @date 2020/09/18 15:16:29
 */

export default function(req, res, next) {
  var debug = this.get('debug')

  // 1. 先判断控制器是否存在
  if (!this.__MODULES__[req.app]) {
    if (this.__MODULES__.__error__) {
      res.error(
        debug ? this.__MODULES__.__error__.stack : this.__MODULES__.__error__,
        500
      )
    } else {
      res.error(`Controller [${req.app}] not found`, 404)
    }
    return
  }

  // 2. 默认二级路由为index
  if (req.path.length < 1) {
    req.path.push('index')
  }

  // 3. 实例化控制器
  this.__MODULES__[req.app]
    .then(({ default: Mod }) => {
      var app = new Mod({ ctx: this, req, res })
      var err = ''

      // action模式, 则路由自动调用对应的action方法
      // __main__模式, 则路由全部走__main__方法
      if (this.get('routeMode') === 'action') {
        var route = req.path.shift()
        var act = route + 'Action'

        if (app[act]) {
          return app[act].apply(app, req.path)
        } else {
          err = new Error(`Route [${route}] not found`)
        }
      } else {
        if (app.__main__) {
          return app.__main__.apply(app, req.path)
        } else {
          err = new Error('__main__() not found')
        }
      }

      err.status = 404
      return Promise.reject(err)
    })
    .catch(err => {
      res.error(debug ? err.stack || err : err, err.status || 500)
    })
}