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

class RandSe {
  @prop({ required: true })
  id: number;					// 随机属性位置id
  @prop({ required: true })
  seid: number;					// 随机属性池id
  @prop({ required: true })
  rand: number;					// 随机属性内需要随机的值
  @prop({ required: true })
  locked: boolean;				// 洗炼是否锁定
}

class Holes {
  @prop({ required: true })
  id: number;					  // 孔的id
  @prop({ required: true })
  isOpen: boolean;	  	// 是否开孔
  @prop({ required: true })
  jewel: number;				// 装备的宝石id
}

@index({ roleId: 1, hid: 1, id: 1 })
@index({ seqId: 1 })
@modelOptions({schemaOptions: {id: false}})
export default class Equip extends BaseModel {
  @prop({ required: true })
  roleId: string;					// 角色 id
  @prop({ required: true })
  roleName: string;				// 角色名称

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

  @prop({ required: true, default: 1 })
  quality: number;				// 品质
  @prop({ required: true, default: 0 })
  type: number;						// 装备类型
  @prop({ required: true, default: 1 })
  wearLv: number;							// 穿戴等级
  @prop({ required: true, default: 0 })
  suitId: number;							// 套装id

  @prop({ required: true, default: 0 })
  randRange: number;			// 固定属性随机值
  @prop({ required: false, type: RandSe, default: [] })
  randSe: RandSe[];							// 强化随机属性
  @prop({ required: true, type: Holes, default: [] })
  holes: Holes[];

  public static async findbyRole(roleId: string, lean = true) {
    const equips: EquipType[] = 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: EquipType = 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: EquipType = await EquipModel.findOneAndUpdate({ _id: equipId }, {hid}, {new: true}).lean(lean);
    return equip;
  }

  public static async deleteAccount(roleId: string) {
    let result = await EquipModel.deleteMany({roleId});
    return result;
  }
    public static async deleteEquips(roleId: string, ids:Array<number>) {
        let result = await EquipModel.deleteMany({roleId, seqId:{$in: ids}});
        return result;
    }

    public static async getEquips(roleId: string, ids:Array<number>) {
        let result = await EquipModel.find({roleId, seqId:{$in: ids}});
        return result;
    }
}

export const EquipModel = getModelForClass(Equip);

export interface EquipType extends Pick<DocumentType<Equip>, keyof Equip>{
  id: number;
};