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

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

  @prop({ required: true, default: '' })
  id: number;				  // 道具 id
  @prop({ required: true, default: '' })
  type: number;						// 道具类型
  @prop({ required: true, default: '' })
  itemName: string;			  // 道具名称
  @prop({ required: true, default: 0 })
  hid: number;			      // 将魂：武将id，其他：0
  @prop({ required: true })
  count: number;			    // 道具数量


  public static async findbyRole(roleId: string, lean = true) {
    const items = await ItemModel.find({ roleId }).select('id count type').lean(lean);
    return items;
  }

  public static async findbyRoleAndIds(roleId: string, ids: Array<number>, lean = true) {
    const items = await ItemModel.find({ roleId, id: {$in: ids} }).select('id count type').lean(lean);
    return items;
  }

  public static async findByRoleAndType(roleId: string, type: number, lean = true) {
    const items = await ItemModel.find({ roleId, type }).select('id count type').lean(lean);
    return items;
  }

  public static async findbyRoleAndGidAndCount(roleId: string, id: number, count: number, lean = true) {
    const items = await ItemModel.findOne({ roleId, id, count: {$gte: count} }).select('id count type').lean(lean);
    return items;
  }

  public static async increaseItem(roleId: string, id: number, count: number, itemInfo: {roleId: string, roleName: string, id: number, itemName: string, type: number, hid?: number }, lean = true) {
    const doc = new ItemModel();
    const setOnInsert = Object.assign(doc.toJSON(), itemInfo);
    const equips = await ItemModel.findOneAndUpdate({roleId, id },{$setOnInsert: setOnInsert, $inc: { count }}, {new: true, upsert: true}).lean(lean);
    return equips;
  }

  public static async decreaseItems(roleId: string, items: Array<{id: number, count: number}>, lean = true) {
    let updateItems = new Array<{id: number, count: number}>(), hasError: boolean = false, result = new Array();
    for(let {id, count} of items) {    
      const rec = await ItemModel.findOneAndUpdate({roleId, id, count: {$gte: count} },{ $inc: { count: -1 * count }}, {new: true}).lean(lean);
      if(rec) {
        result.push(rec);
        updateItems.push({id, count});
      } else {
        hasError = true; break;
      }
    }
    if(hasError) { // 数量不足
      for(let {id, count} of updateItems) { 
        await ItemModel.findOneAndUpdate({roleId, id },{ $inc: { count }}, {new: true}).lean(lean);
      } 
      return {hasError: true}
    } else {
      return {hasError: false, result}
    }
  }

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

export const ItemModel = getModelForClass(Item);