123 lines
4.5 KiB
TypeScript
123 lines
4.5 KiB
TypeScript
import { COUNTER } from './../consts';
|
|
import { CounterAllModal } from './CounterAll';
|
|
import BaseModel from './BaseModel';
|
|
import { index, getModelForClass, prop, DocumentType, mongoose, ReturnModelType } from '@typegoose/typegoose';
|
|
import { nowSeconds } from '../pubUtils/timeUtil';
|
|
const bcrypt = require('bcrypt');
|
|
const SALT_WORK_FACTOR = 5
|
|
|
|
|
|
/**
|
|
* GM账号字段接口
|
|
*/
|
|
@index({ uid: 1 })
|
|
|
|
export default class GMUser extends BaseModel {
|
|
@prop({ required: true })
|
|
uid: number;
|
|
|
|
@prop({ required: true })
|
|
name: string;
|
|
|
|
@prop({ required: true })
|
|
username: string;
|
|
|
|
@prop({ required: true })
|
|
groupId: number;
|
|
|
|
@prop({ required: true, type: String })
|
|
envs: string[];
|
|
|
|
@prop({ required: true, default: true })
|
|
isEnable: boolean;
|
|
|
|
@prop({ required: true })
|
|
password: string;
|
|
|
|
@prop({ required: true })
|
|
salt: string;
|
|
|
|
@prop({ required: true })
|
|
token: string;
|
|
|
|
@prop({ required: true })
|
|
expireTime: number;
|
|
|
|
@prop({ required: true })
|
|
type: number; // 1-37账号登录 2-我方账号登录
|
|
|
|
public static async updateInfo(uid: number, params: GMUserUpdate) {
|
|
let user = await GMUserModel.findOneAndUpdate({ uid }, params, { new: true }).lean();
|
|
return user;
|
|
}
|
|
|
|
private static async encryptPass(password: string, salt?: string) {
|
|
if (!salt) {
|
|
salt = await bcrypt.genSalt(SALT_WORK_FACTOR);
|
|
}
|
|
let npassword = await bcrypt.hash(password, salt);
|
|
return { npassword, salt };
|
|
}
|
|
|
|
public static async getGmAccountList(lean = true) {
|
|
const user: GMUserType[] = await GMUserModel.find().select('uid username name token envs groupId isEnable type').lean(lean);
|
|
return user;
|
|
}
|
|
|
|
public static async getGmAccountByToken(token: string, lean = true) {
|
|
// console.log(token)
|
|
const user: GMUserType = await GMUserModel.findOne({ token }).select('uid username name token envs groupId isEnable expireTime').lean(lean);
|
|
return user;
|
|
}
|
|
|
|
public static async getGmAccountByUid(uid: number, lean = true) {
|
|
const user: GMUserType = await GMUserModel.findOne({ uid }).select('uid username name token envs groupId isEnable').lean(lean);
|
|
return user;
|
|
}
|
|
|
|
public static async getGmAccountByUsername(username: string, lean = true) {
|
|
const user: GMUserType = await GMUserModel.findOne({ username }).select('uid username name token envs groupId isEnable').lean(lean);
|
|
return user;
|
|
}
|
|
|
|
public static async login(username: string, password: string, token: string, lean = true) {
|
|
const user: GMUserType = await GMUserModel.findOne({ username }).select('salt').lean();
|
|
if (user) {
|
|
let { salt } = user;
|
|
let { npassword } = await this.encryptPass(password, salt);
|
|
const checkUser: GMUserType = await GMUserModel.findOneAndUpdate({ username, password: npassword }, { $set: { token, expireTime: nowSeconds() + 24 * 60 * 60 } }).select('uid username name token envs groupId isEnable type').lean(lean);
|
|
return checkUser;
|
|
} else {
|
|
return null
|
|
}
|
|
}
|
|
|
|
public static async channelLogin(username: string, token: string) {
|
|
const user: GMUserType = await GMUserModel.findOneAndUpdate({ username }, { $set: { token, expireTime: nowSeconds() + 1 * 60 * 60 } }, {new: true}).select('token type').lean();
|
|
return user;
|
|
}
|
|
|
|
public static async createGmAccount(username: string, password: string, name: string, token: string, lean = true) {
|
|
const uid = await CounterAllModal.getNewCounter(COUNTER.GMUID);
|
|
let r = await this.encryptPass(password);
|
|
const user: GMUserType = await GMUserModel.findOneAndUpdate({ username }, { uid, password: r.npassword, salt: r.salt, name, token }, { upsert: true, new: true }).select('uid username name').lean(lean);
|
|
return user;
|
|
}
|
|
|
|
public static async changePass(uid: number, password: string, lean = true) {
|
|
let r = await this.encryptPass(password);
|
|
const user: GMUserType = await GMUserModel.findOneAndUpdate({ uid }, { password: r.npassword, salt: r.salt }).select('uid username name').lean(lean);
|
|
return user;
|
|
}
|
|
}
|
|
|
|
export let GMUserModel: ReturnModelType<typeof GMUser, {}>;
|
|
export function loadGMUserModel(connect: mongoose.Connection) {
|
|
GMUserModel = getModelForClass(GMUser, {
|
|
existingConnection: connect
|
|
});
|
|
}
|
|
|
|
export interface GMUserType extends Pick<DocumentType<GMUser>, keyof GMUser> { };
|
|
export type GMUserUpdate = Partial<GMUserType>; // 将所有字段变成可选项
|