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

class RandSe {
  @prop({ required: true })
  seType: number;					// 属性类型
  @prop({ required: true })
  value: number;					// 属性数值
}

class Holes {
  @prop({ required: true })
  id: number;					// 宝石 id，待定，也可能是 type + lv
  @prop({ required: true })
  lv: number;					// 宝石等级
}
@index({ roleId: 1, hid: 1, id: 1 })
@index({ seqId: 1 })
export default class Equip extends BaseModel {
  @prop({ required: true })
  roleId: string;					// 角色 id
  @prop({ required: true })
  roleName: string;				// 角色名称

  @prop({ required: true })
  id: number;						// 装备 id
  @prop({ required: true })
  type: number;						// 装备类型
  @prop({ required: true })
  name: string;					// 装备名称
  @prop({ required: true })
  seqId: number;					// 装备表自增 id
  @prop({ required: false })
  hid: number;						// 装备此装备的武将 id
  @prop({ required: false, default: 1 })
  count: number;					// 装备数量

  @prop({ required: true, default: 1 })
  quality: number;				// 品质
  @prop({ required: true, default: 1 })
  lv: number;							// 强化等级
  @prop({ required: false, type: RandSe, default: [] })
  randSe: RandSe[];							// 强化随机属性

  @prop({ required: true, default: 0 })
  randRange: number;			// 物攻策攻数值浮动上下限变化率，浮点数
  @prop({ required: true, default: 0 })
  initHoleCnt: number;		// 初始珠宝孔位个数
  @prop({ required: true, type: Holes, default: [] })
  holes: Holes[];

  public static async findbyRole(roleId: string, lean = true) {
    const equips = await EquipModel.find({ roleId }).lean(lean);
    return equips;
  }
  
  public static async createEquip(equipInfo: {roleId: string, roleName: string, id: number, seqId: number, type: number, name: string, lv?: number}, lean = true) {
    const doc = new EquipModel();
    const update = Object.assign(doc.toJSON(), equipInfo);
    const equip = await EquipModel.findOneAndUpdate({ seqId: equipInfo.seqId }, update, {upsert: true, new: true}).lean(lean);
    return equip;
  }

  public static async putOn(hid: number, equipId: string, lean = true) {
    const equip = await EquipModel.findOneAndUpdate({ _id: equipId }, {hid}, {new: true}).lean(lean);
    return equip;
  }

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

export const EquipModel = getModelForClass(Equip);