thinkjs/lib/middleware/controller.js

const assert = require('assert');
const helper = require('think-helper');

function invokeController(options){
  return (ctx, next) => {
    const isMultiModule = think.data.modules.length;
    if(isMultiModule){
      assert(ctx.module, 'module can not be blank in multi module');
    }
    assert(ctx.controller, 'controller can not be blank');
    assert(ctx.action, 'action can not be blank');

    const controllers = think.data.controller;
    if(isMultiModule){
      controllers = controllers[ctx.module];
    }
    //controllers empty
    if(helper.isEmpty(controllers)){
      return next();
    }
    let controller = controllers[ctx.controller];
    // controller not exist
    if(helper.isEmpty(controller)){
      return next();
    }
    const instance = new controller(ctx);
    let promise = Promise.resolve();
    if(instance.__before){
      promise = Promise.resolve(instance.__before());
    }
    // if return false, it will be prevent next process
    return promise.then(data => {
      if(data === false) return false;
      let method = `${ctx.action}Action`;
      if(!instance[method]){
        method = '__call';
      }
      if(instance[method]){
        return Promise.resolve(instance[method]())
      }
    }).then(data => {
      if(data === false) return false;
      if(instance.__after){
        return Promise.resolve(instance.__after());
      }
    }).then(data => {
      if(data !== false){
        return next();
      }
    });
  }
}

module.exports = invokeController;