253 lines
8.8 KiB
TypeScript
253 lines
8.8 KiB
TypeScript
import { COUNTER, ENCRYPT_KEY, ENCRYPT_IV } from './../consts';
|
|
import { CounterModel } from './Counter';
|
|
import BaseModel from './BaseModel';
|
|
import { index, getModelForClass, prop, DocumentType } from '@typegoose/typegoose';
|
|
import { genCode, aesEncryptcfb, aesDecryptcfb } from '../pubUtils/util';
|
|
const bcrypt = require('bcrypt');
|
|
const SALT_WORK_FACTOR = 5
|
|
|
|
class PlatForm {
|
|
@prop({ required: true })
|
|
platform: string; // 平台：ios, android, web, pc
|
|
@prop({ required: true })
|
|
unionId: string; // 用户标识
|
|
}
|
|
|
|
/**
|
|
* 用户字段接口
|
|
*/
|
|
@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, set: (val: string) => aesEncryptcfb(val, ENCRYPT_KEY, ENCRYPT_IV), get: (val: string) => aesDecryptcfb(val, ENCRYPT_KEY, ENCRYPT_IV) })
|
|
tel: string; // 账号
|
|
|
|
@prop({ required: true, default: '' })
|
|
telHash: string;
|
|
|
|
@prop({ required: true })
|
|
password: string;
|
|
|
|
@prop({ required: true })
|
|
salt: string;
|
|
|
|
@prop({ required: true, default: '' })
|
|
channelId: string;
|
|
|
|
// 游客相关
|
|
@prop({ required: false, default: false })
|
|
isGuest: boolean; // 是否是游客
|
|
@prop({ required: false })
|
|
guestId: string; // 游客id
|
|
@prop({ required: false, default: 0 })
|
|
guestTime: number; // 游客体验时间
|
|
|
|
// 防沉迷相关
|
|
@prop({ required: false, default: false })
|
|
hasAuthenticated: boolean; // 是否是已认证
|
|
@prop({ required: false })
|
|
birthday: string; // 生日年月
|
|
@prop({ required: false })
|
|
pi: string; // 已通过实名认证用户唯一标识
|
|
@prop({ required: false, default: 0 })
|
|
todayPlayTime: number; // 今日游戏时间
|
|
@prop({ required: false })
|
|
playTime: number; // 总游戏时间
|
|
|
|
@prop({ required: false, default: false })
|
|
hasSetPw: boolean; // 是否设置过密码
|
|
|
|
// 最后登录 IP
|
|
@prop({ required: false })
|
|
ip: string;
|
|
|
|
@prop({ required: true })
|
|
lastLoginTime: Date;
|
|
|
|
@prop({ required: true })
|
|
createTime: Date;
|
|
|
|
@prop({ required: true })
|
|
platform: string;
|
|
@prop({ required: true, default: [], type: PlatForm })
|
|
platforms: PlatForm[];
|
|
|
|
@prop({ required: true, type: String, default: [], _id: false })
|
|
device: 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;
|
|
|
|
// 个推CID
|
|
@prop({ required: false, default: "" })
|
|
getuiCID: string;
|
|
|
|
public static async createUser(isGuest: boolean, tel: string, token: string, platform: string, pkgName: string, serverType: string, deviceId: string, guestTime: number = 0, ip: string = '') {
|
|
let _tel = aesEncryptcfb(tel, ENCRYPT_KEY, ENCRYPT_IV);
|
|
const curTime: Date = new Date();
|
|
|
|
const uid = await CounterModel.getNewCounter(COUNTER.UID);
|
|
const userCode = genCode(8);
|
|
const doc = new UserModel();
|
|
let update = {};
|
|
update = Object.assign(update, doc.toJSON(), { platform, pkgName, serverType, createTime: curTime, uid, userCode, username: `用户${uid}`, isGuest, token, lastLoginTime: curTime, guestTime, ip });
|
|
if (isGuest) update["guestId"] = _tel;
|
|
delete update["device"];
|
|
|
|
const user: UserType = await UserModel.findOneAndUpdate({ tel }, { $set: update, $addToSet: { device: deviceId } }, { upsert: true, new: true }).lean({ getters: true });
|
|
return user;
|
|
}
|
|
|
|
public static async getLastDeviceGuest(deviceId: string, token: string) {
|
|
const user: UserType = await UserModel.findOneAndUpdate({ device: { $elemMatch: { $eq: deviceId } }, isGuest: true }, { token }, { new: true }).sort({ createTime: -1 }).lean({ getters: true });
|
|
return user;
|
|
}
|
|
|
|
public static async updateToken(tel: string, token: string, deviceId: string, ip: string) {
|
|
|
|
const curTime: Date = new Date();
|
|
|
|
let user = await UserModel.findOneAndUpdate({ tel }, { $set: { token, lastLoginTime: curTime, ip }, $addToSet: { device: deviceId } }, { new: true }).lean({ getters: true });
|
|
return user;
|
|
}
|
|
|
|
public static async updateGetuiCID(tel: string, cid: string) {
|
|
let user = await UserModel.findOneAndUpdate({ tel }, { $set: { getuiCID: cid } }, { new: true }).lean({ getters: true });
|
|
return user;
|
|
}
|
|
|
|
public static async createOrUpdate(isGuest: boolean, tel: string, token: string, platform: string, pkgName: string, serverType: string, deviceId: string, ip: string) {
|
|
// console.log(tel);
|
|
|
|
let user: UserType = await UserModel.findOne({ tel }).lean({ getters: true });
|
|
if (!user) {
|
|
user = await UserModel.createUser(isGuest, tel, token, platform, pkgName, serverType, deviceId, 0, ip);
|
|
} else {
|
|
user = await UserModel.updateToken(tel, token, deviceId, ip);
|
|
}
|
|
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 setPass(uid: number, password: string) {
|
|
let r = await this.encryptPass(password);
|
|
const user: UserType = await UserModel.findOneAndUpdate({ uid }, { $set: { password: r.npassword, salt: r.salt, hasSetPw: true } }, { new: true }).lean({ getters: true });
|
|
return user;
|
|
}
|
|
|
|
public static async bindTel(uid: number, tel: string) {
|
|
const user: UserType = await UserModel.findOneAndUpdate({ uid, isGuest: true }, { $set: { tel, isGuest: false } }, { new: true }).lean({ getters: true });
|
|
return user;
|
|
}
|
|
|
|
public static async checkPass(tel: string, password: string, token: string, deviceId: string) {
|
|
|
|
const user: UserType = await UserModel.findOne({ tel }).select('salt').lean({ getters: true });
|
|
if (user) {
|
|
const curTime: Date = new Date();
|
|
let { salt } = user;
|
|
let { npassword } = await this.encryptPass(password, salt);
|
|
const checkUser: UserType = await UserModel.findOneAndUpdate({ tel, password: npassword }, { $set: { token, lastLoginTime: curTime }, $addToSet: { device: deviceId } }, { new: true }).lean({ getters: true });
|
|
return checkUser;
|
|
} else {
|
|
return null
|
|
}
|
|
}
|
|
|
|
public static async findUserByToken(token: string) {
|
|
const user: UserType = await UserModel.findOne({ token }).select('uid token serverType auth tel userCode pkgName ip').lean({ getters: true });
|
|
return user;
|
|
}
|
|
|
|
public static async findTokenByTel(tel: string) {
|
|
|
|
const user: UserType = await UserModel.findOne({ tel }).lean();
|
|
return user.token;
|
|
}
|
|
|
|
public static async findTokenByUid(uid: number) {
|
|
const user: UserType = await UserModel.findOne({ uid }).lean();
|
|
return user ? user.token : null;
|
|
}
|
|
|
|
public static async findUserByTel(tel: string) {
|
|
|
|
const user: UserType = await UserModel.findOne({ tel }).select('uid tel hasSetPw').lean({ getters: true });
|
|
return user;
|
|
}
|
|
|
|
public static async findUserByUid(uid: number) {
|
|
const user: UserType = await UserModel.findOne({ uid }).select('uid tel').lean({ getters: true });
|
|
return user;
|
|
}
|
|
|
|
public static async findUserByUserCode(userCode: string) {
|
|
const user: UserType = await UserModel.findOne({ userCode }).lean({ getters: true });
|
|
return user;
|
|
}
|
|
|
|
public static async addAuth(uid: number, auth: number) {
|
|
const user: UserType = await UserModel.findOneAndUpdate({ uid }, { auth }, { new: true }).select('uid tel').lean({ getters: true });
|
|
return user;
|
|
}
|
|
|
|
public static async authentication(uid: number, birthday: string, pi: string) {
|
|
const user: UserType = await UserModel.findOneAndUpdate({ uid }, { hasAuthenticated: true, birthday, pi }, { new: true }).lean({ getters: true });
|
|
return user;
|
|
}
|
|
|
|
public static async findUserByField(field: string, value?: Array<number | string>) {
|
|
let searchObj = {};
|
|
if (field != 'all') {
|
|
searchObj[field] = {
|
|
$in: value
|
|
};
|
|
}
|
|
const user: UserType[] = await UserModel.find(searchObj).select('uid tel username serverType auth').lean({ getters: true });
|
|
return user;
|
|
}
|
|
|
|
|
|
public static async updatePlayTime(userCode: string, isGuest: boolean, todayPlayTime: number, playTime: number) {
|
|
const user: UserType = await UserModel.findOneAndUpdate({ userCode }, { $set: { guestTime: isGuest?playTime: 0, todayPlayTime: todayPlayTime, playTime } }, { new: true }).lean({ getters: true });
|
|
return user;
|
|
}
|
|
}
|
|
|
|
|
|
export const UserModel = getModelForClass(User);
|
|
|
|
export interface UserType extends Pick<DocumentType<User>, keyof User> { };
|