/*

* @Author: csxiaoyao

* @Date: 2018-04-19 10:43:31

* @Last Modified by: csxiaoyao

* @Last Modified time: 2018-04-19 10:43:31

*/

module.exports = {

APIError: function (code, message) {

this.code = code || 'internal:unknown_error';

this.message = message || '';

},

restify: (pathPrefix) => {

// REST API前缀，默认为/api/

pathPrefix = pathPrefix || '/api/';

return async (ctx, next) => {

// 是否是REST API前缀

if (ctx.request.path.startsWith(pathPrefix)) {

console.log(`Process API ${ctx.request.method} ${ctx.request.url}...`);

// 绑定rest()方法

ctx.rest = (data) => {

ctx.response.type = 'application/json';

ctx.response.body = data;

}

try {

await next();

} catch (e) {

console.log('Process API error...');

ctx.response.status = 400;

ctx.response.type = 'application/json';

ctx.response.body = {

code: e.code || 'internal:unknown_error',

message: e.message || ''

};

}

} else {

await next();

}

};

}

};