yapi/server/controllers/base.js

94 lines
2.5 KiB
JavaScript
Raw Normal View History

2017-07-05 17:47:51 +08:00
import yapi from '../yapi.js'
2017-07-10 11:56:53 +08:00
import projectModel from '../models/project.js'
2017-07-11 12:12:43 +08:00
import userModel from '../models/user.js'
const jwt = require('jsonwebtoken');
2017-07-05 17:47:51 +08:00
class baseController{
constructor(ctx){
2017-07-12 17:01:27 +08:00
this.ctx = ctx;
2017-07-11 16:50:17 +08:00
//网站上线后role对象key是不能修改的value可以修改
this.roles = {
admin: 'Admin',
member: '网站会员'
}
2017-07-11 12:12:43 +08:00
}
2017-07-10 11:11:35 +08:00
2017-07-11 12:12:43 +08:00
async init(ctx){
this.$user = null;
2017-07-12 17:01:27 +08:00
let ignoreRouter = [
'/user/login_by_token',
'/user/login',
'/user/reg',
'/user/status',
'/user/logout'
]
if(ignoreRouter.indexOf(ctx.path) > -1){
2017-07-11 12:12:43 +08:00
this.$auth = true;
}else{
await this.checkLogin(ctx)
}
}
2017-07-10 11:11:35 +08:00
2017-07-11 12:12:43 +08:00
getUid(ctx){
return parseInt(this.$uid, 10);
2017-07-05 17:47:51 +08:00
}
2017-07-11 12:12:43 +08:00
async checkLogin(ctx){
let token = ctx.cookies.get('_yapi_token');
let uid = ctx.cookies.get('_yapi_uid');
try{
if(!token || !uid) return false;
let userInst = yapi.getInst(userModel); //创建user实体
let result = await userInst.findById(uid);
let decoded = jwt.verify(token, result.passsalt)
if(decoded.uid == uid){
this.$uid = uid;
this.$auth = true;
this.$user = result;
return true;
}
return false;
}catch(e){
return false;
}
2017-07-05 17:47:51 +08:00
}
2017-07-11 12:12:43 +08:00
async getLoginStatus(ctx){
2017-07-12 17:01:27 +08:00
if(await this.checkLogin(ctx) === true){
2017-07-11 12:12:43 +08:00
return ctx.body = yapi.commons.resReturn(yapi.commons.fieldSelect(this.$user,['_id','username','email', 'up_time', 'add_time']));
}
2017-07-11 16:50:17 +08:00
return ctx.body = yapi.commons.resReturn(null, 300 , 'Please login.');
2017-07-06 20:55:02 +08:00
2017-07-05 17:47:51 +08:00
}
2017-07-06 19:21:54 +08:00
getRole(){
2017-07-11 16:50:17 +08:00
return this.$user.role;
2017-07-06 19:21:54 +08:00
}
2017-07-10 11:56:53 +08:00
async jungeProjectAuth(id){
let model = yapi.getInst(projectModel);
if(this.getRole() === 'admin') return true;
if(!id) return false;
let result = await model.get(id);
if(result.uid === this.getUid()){
return true;
}
return false;
}
async jungeMemberAuth(id, member_uid){
let model = yapi.getInst(projectModel);
if(this.getRole() === 'admin') return true;
if(!id || !member_uid) return false;
let result = await model.checkMemberRepeat(id, member_uid);
if(result > 0){
return true;
}
return false;
}
2017-07-05 17:47:51 +08:00
}
module.exports = baseController