125 lines
3.6 KiB
TypeScript
125 lines
3.6 KiB
TypeScript
import { COUNTER } from './../consts';
|
|
import { CounterModel } from './Counter';
|
|
import BaseModel from './BaseModel';
|
|
import { index, getModelForClass, prop, DocumentType } from '@typegoose/typegoose';
|
|
import { genCode } from '../pubUtils/util';
|
|
|
|
/**
|
|
* 用户字段接口
|
|
*/
|
|
@index({ tel: 1 })
|
|
@index({ uid: 1 })
|
|
export default class User extends BaseModel {
|
|
|
|
@prop({ required: true })
|
|
uid: number;
|
|
|
|
@prop({ required: true })
|
|
username: string;
|
|
|
|
@prop({ required: true })
|
|
userCode: string; // 用户唯一字符串标识
|
|
|
|
@prop({ required: true })
|
|
token: string;
|
|
|
|
@prop({ required: true })
|
|
tel: string;
|
|
|
|
@prop({ required: true, default: '' })
|
|
telHash: string;
|
|
|
|
@prop({ required: true, default: '' })
|
|
channelId: string;
|
|
|
|
@prop({ required: false })
|
|
guestId: string;
|
|
|
|
// 最后登录 IP
|
|
@prop({ required: false })
|
|
ip: string;
|
|
|
|
@prop({ required: true })
|
|
lastLoginTime: Date;
|
|
|
|
@prop({ required: true })
|
|
createTime: Date;
|
|
|
|
platform: string;
|
|
@prop({ required: true, default: [] })
|
|
platforms: [{
|
|
platform: string; // 平台：ios, android, web, pc
|
|
unionId: string; // 用户标识
|
|
}];
|
|
|
|
@prop({ required: true, default: 'com.bantu.zyz' })
|
|
pkgName: string;
|
|
|
|
// 服务器类型：official, channel, ios, oversea
|
|
@prop({ required: true, default: 'official' })
|
|
serverType: string;
|
|
|
|
// 账号是否被屏蔽
|
|
@prop({ required: true, default: false })
|
|
blocked: boolean;
|
|
|
|
// 用户权限：0-普通用户，1-测试用户
|
|
@prop({ required: true, default: 0 })
|
|
auth: number;
|
|
|
|
public static async updateToken(tel: string, token: string, platform: string, pkgName: string, serverType: string, lean = true) {
|
|
let user: UserType = await UserModel.findOne({ tel }).lean();
|
|
const curTime: Date = new Date();
|
|
let update = {};
|
|
if (!user) {
|
|
const uid = await CounterModel.getNewCounter(COUNTER.UID);
|
|
const userCode = genCode(8);
|
|
const doc = new UserModel();
|
|
update = Object.assign(update, { platform, pkgName, serverType, createTime: curTime, uid, userCode, username: `用户${uid}` }, doc.toJSON());
|
|
}
|
|
update = Object.assign(update, { token, lastLoginTime: curTime });
|
|
user = await UserModel.findOneAndUpdate({ tel }, update, { upsert: true, new: true }).lean(lean);
|
|
return user;
|
|
}
|
|
|
|
public static async findUserByToken(token: string, lean = true) {
|
|
const user: UserType = await UserModel.findOne({ token }).select('uid token serverType auth').lean(lean);
|
|
return user;
|
|
}
|
|
|
|
public static async findTokenByTel(tel: string, lean = true) {
|
|
const { token } = await UserModel.findOne({ tel }).select('token').lean(lean);
|
|
return token;
|
|
}
|
|
|
|
public static async findUserByTel(tel: string, lean = true) {
|
|
const user: UserType = await UserModel.findOne({ tel }).select('uid tel').lean(lean);
|
|
return user;
|
|
}
|
|
|
|
public static async findUserByUid(uid: number, lean = true) {
|
|
const user: UserType = await UserModel.findOne({ uid }).select('uid tel').lean(lean);
|
|
return user;
|
|
}
|
|
|
|
public static async addAuth(uid: number, auth: number, lean = true) {
|
|
const user: UserType = await UserModel.findOneAndUpdate({ uid }, { auth }).select('uid tel').lean(lean);
|
|
return user;
|
|
}
|
|
|
|
public static async findUserByField(field: string, value?: Array<number|string>, lean = true) {
|
|
let searchObj = {};
|
|
if(field != 'all') {
|
|
searchObj[field] = {
|
|
$in: value
|
|
};
|
|
}
|
|
const user: UserType[] = await UserModel.find(searchObj).select('uid tel username serverType auth').lean(lean);
|
|
return user;
|
|
}
|
|
}
|
|
|
|
export const UserModel = getModelForClass(User);
|
|
|
|
export interface UserType extends Pick<DocumentType<User>, keyof User>{};
|