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

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
}

interface equipUpdate {
    hid?: number;						
    quality?: number;							
    wearLv?: number;	
    _id?:number;
    holes?:Array<Holes>
}

@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(ids: Array<number>) {
        let result = await EquipModel.deleteMany({ seqId: { $in: ids } });
        return result;
    }

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

    public static async getEquip(seqId: number) {
        let equip: EquipType = await EquipModel.findOne({ seqId });
        return equip;
    }

    public static async updateEquipInfo(seqId:number, equipUpdate: equipUpdate, lean = true) {
        delete equipUpdate._id;
        let result: EquipType = await EquipModel.findOneAndUpdate({seqId}, {$set: equipUpdate}).lean(lean);
        return result;
    }
}

export const EquipModel = getModelForClass(Equip);

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