ZYZ/shared/db/Hero.ts

import BaseModel, { CeAttr } from './BaseModel';
import { index, getModelForClass, prop, Ref, mongoose } from '@typegoose/typegoose';
import Equip from './Equip';
import { CounterModel } from './Counter';
import { COUNTER } from '../consts/consts';

/**
  * 英雄表
*/

class Connect {
    @prop({ required: true })
    shipId: number;
    @prop({ required: true })
    level: number;

}

class Skin {
    @prop({ required: true })
    id: number;
    @prop({ required: true })
    enable: boolean;
}

interface heroUpdate {
    exp?: number;						
    lv?: number;							
    ce?: number;	
    star?: number;						
    starStage?: number;		 
    colorStar?: number;
    colorStarStage?: number;
    quality?: number;
    job?:number;
    jobStage?:number;
    favour?:number;
    favourLv?:number;
    skins?: Skin[];
    conections?: Connect[];
    _id?:number;
}

@index({ roleId: 1, hid: 1 })
@index({ roleId: 1, seqId: 1 })
export default class Hero extends BaseModel {

    @prop({ required: true })
    roleId: string;					// 角色 id
    @prop({ required: true })
    roleName: string;				// 角色名称
    @prop({ required: true })
    serverId: number;               // 区服 id

    @prop({ required: true })
    hid: number;						// 武将 id
    @prop({ required: true })
    hName: string;					// 武将名
    @prop({ required: true })
    seqId: number;					// 武将表自增 id

    @prop({ required: true, default: 0 })
    exp: number;						// 经验值
    @prop({ required: true, default: 1 })
    lv: number;							// 武将等级
    @prop({ required: true, default: 0 })
    ce: number;							// 武将战力
    @prop({ required: true, default: 0 })
    historyCe: number;			        // 武将历史最高战力
    @prop({required: true, default: new CeAttr() })
    ceAttr: CeAttr;                     // 影响战力的属性

    @prop({ required: true, default: 1 })
    star: number;						// 星级
    @prop({ required: true, default: 0 })
    starStage: number;		  // 星级六维阶段
    @prop({ required: true, default: 0 })
    colorStar: number;						// 觉醒， 彩星
    @prop({ required: true, default: 0 })
    colorStarStage: number;			// 觉醒六维阶段
    
    @prop({ required: true, default: 0 })
    quality: number;			  // 品质

    @prop({ required: true, default: 0 })
    job: number;					  // 职业
    @prop({ required: true, default: 0 })
    jobStage: number;			  // 职阶

    @prop({ required: true, default: 0 })
    favour: number;					// 好感度
    @prop({ required: true, default: 1 })
    favourLv: number;				// 好感等级
    @prop({ required: true, type:Connect, default: [] })
    conections: Connect[];					// 羁绊
    @prop({ required: true, default: [] })
    skins: Skin[];								// 皮肤

    @prop({ ref: Equip, type: mongoose.Schema.Types.ObjectId })
    equips: Ref<Equip>[]; // 武将装备引用数组

    public static async findByRole(roleId: string, lean = true) {
        const heros = await HeroModel.find({ roleId }).populate('equips').lean(lean);
        return heros || [];
    }

    public static async findBySeqIdRange(seqIds: Array<number>, roleId: string, lean = true) {
        const hero = await HeroModel.find({ seqId: {$in: seqIds}, roleId }).lean(lean);
        return hero;
    }
    public static async findBySeqIdAndRole(seqId: number, roleId: string, lean = true) {
        const hero = await HeroModel.findOne({ seqId, roleId }).lean(lean);
        return hero;
    }
    public static async findByHidAndRole(hid: number, roleId: string, lean = true) {
        const hero = await HeroModel.findOne({ hid, roleId }).lean(lean);
        return hero;
    }

    public static async addEquip(roleId: string, hid: number, equipId: string, lean = true) {
        const hero = await HeroModel.findOneAndUpdate({ roleId, hid }, {$push: {equips: equipId}}, {new: true}).lean(lean);
        if (hero) {
        await Equip.putOn(hero.hid, equipId);
        }
        return hero;
    }

    public static async createHero(heroInfo: {roleId: string, serverId: number, roleName: string, hid: number, hName: string, star: number, quality: number, job: number, lv?: number }, lean = true) {
        const doc = new HeroModel();
        const seqId = await CounterModel.getNewCounter(COUNTER.HID)||-1;
        const update = Object.assign(doc.toJSON(), heroInfo, {seqId});
        delete update._id;
        const hero = await HeroModel.findOneAndUpdate({roleId: heroInfo.roleId, hid: heroInfo.hid}, update, {upsert: true, new: true}).select('hid hName lv exp star starStage colorStar colorStarStage quality job jobStage conections favour favourLv skins equips').lean(lean);
        return hero;
    }

    public static async sumTopHeroCe(roleId: string, num: number) {
        let ce = await HeroModel.aggregate([
        { $match : { roleId } },
        { $sort: {historyCe: -1} },
        { $limit: num },
        { $group: { _id: null, historyCe: { $sum: '$historyCe' } } }
        ]);
        return ce.length > 0?ce[0].historyCe:0;
    }

    public static async sumHeroCe(roleId: string) {
        let ce = await HeroModel.aggregate([
        { $match : { roleId } },
        { $group: { _id: null, ce: { $sum: '$ce' } } }
        ]);
        return ce.length > 0?ce[0].ce:0;
    }

    public static async getTopHero(roleId: string, num: number, lean = true) {
        const heroes = await HeroModel.find({roleId}).limit(num).sort({ce: -1}).lean(lean);
        return heroes;
    }

    public static async updateCe(roleId: string, hid: number, ce: number, oldCe: number, historyCe: number, lean = true) {
        let distance = ce - oldCe;
        let historyDistance = ce > historyCe ?ce - historyCe: 0;
        let result = await HeroModel.findOneAndUpdate({roleId, hid}, {$inc:{ce: distance, historyCe: historyDistance}}).lean(lean);
        return result||{};
    }

    public static async deleteAccount(roleId: string, lean = true) {
        let result = await HeroModel.deleteMany({roleId}).lean(lean);
        return result||{};
    }

    public static async updateHeroInfo(roleId: string, hid:number, heroUpdate:heroUpdate, lean = true) {
        delete heroUpdate._id;
        let result = await HeroModel.findOneAndUpdate({roleId, hid}, {$set:heroUpdate}).lean(lean);
        return result||{};
    }

}

export const HeroModel = getModelForClass(Hero);