import BaseModel from './BaseModel';
import { index, getModelForClass, prop, Ref, mongoose } from '@typegoose/typegoose';
import Equip from './Equip';

/**
  * 英雄表
*/
@index({ roleId: 1, hid: 1 })
export default class Hero extends BaseModel {

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

  @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: 100 })
  ce: number;							// 武将战力

  @prop({ required: true })
  star: number;						// 星级
  @prop({ required: true, default: 1 })
  rank: number;						// 阶数
  @prop({ required: true, default: 0 })
  favour: number;					// 好感度
  @prop({ required: true, default: 1 })
  favourLv: number;				// 好感等级
  @prop({ required: true, default: [] })
  conections: [{					// 羁绊
    id: number;						// 羁绊编号
    name: string;					// 羁绊名称
    valid: boolean;				// 是否开启
  }];
  @prop({ required: true, default: [] })
  skins: [{								// 皮肤
    id: number;						// id
    enable: boolean;			// 是否装备
  }]

  @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 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, roleName: string, hid: number, hName: string, seqId: number, lv?:number}, lean = true) {
    const doc = new HeroModel();
    const update = Object.assign(heroInfo, doc.toJSON());
    const hero = await HeroModel.findOneAndUpdate({roleId: heroInfo.roleId, hid: heroInfo.hid}, update, {upsert: true, new: true}).lean(lean);
    return hero;
  }

}

export const HeroModel = getModelForClass(Hero);