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

function genCodeTmp(len) {
  const chars = '123456789ABCDEFGHJKLMNPQRSTWXYZabcdefghijklmnopqrstuvwxyz';
  const charArr = chars.split('');
  let code = '';
  for (let i = 0; i < len; i++) {
    code += charArr[Math.floor(Math.random() * charArr.length)];
  }
  return code;
}

function getRefTime(curTime = new Date()) {
  if (curTime.getHours() < 5) {
    curTime.setDate(curTime.getDate() - 1);
  }
  curTime.setHours(5, 0, 0, 0);
  return curTime;
}

/**
  * 天梯派遣记录表
  */
@index({ roleId: 1, status: 1 })
@index({ batchCode: 1 })
export default class TowerTaskRec extends BaseModel {
  @prop({ required: true })
  roleId: string; // 角色 id
  @prop({ required: true })
  roleName: string; // 角色名

  @prop({ required: true})
  batchCode: string; // 本批派遣任务唯一标识
  @prop({ required: true, default: 0})
  status: number; // 派遣任务当前状态，0-可派遣，1-已派遣，2-已完成，3-已领取
  @prop({ required: true, type: Number, default: [] })
  heroes: Array<number>; // 此批派遣使用的全部武将

  @prop({ required: true})
  taskId: number; // 任务唯一 Id，来自任务表
  @prop({ required: true })
  taskCode: string; // 服务器生成的任务唯一编号
  @prop({ required: false })
  sendTime: Date; // 派遣时间

  public static async getCurTasks(roleId: string, lean = true) {
    const refTime = getRefTime();
    const recs = await TowerTaskRecModel.find({roleId, createdAt: {$gte: refTime}}).lean(lean);
    return recs;
  }

  public static async createTasks(roleId: string, roleName: string, taskIds: Array<number>/*, lean = true*/) {
    const batchCode = genCodeTmp(8);
    const tasksData = taskIds.map(id => {
      const taskModel = new TowerTaskRecModel();
      const taskCode = genCodeTmp(6);
      const taskData = Object.assign(taskModel.toJSON(), {batchCode, taskId: id, taskCode, roleId, roleName});
      return taskData;
    });
    const recs = await TowerTaskRecModel.insertMany(tasksData, {ordered: false});
    return recs;
  }

  public static async getTasksByCode(batchCode: string, lean = true) {
    const recs = await TowerTaskRecModel.find({batchCode}).lean(lean);
    return recs;
  }

  public static async sendHeroes(_roleId: string, batchCode: string, tasks: Array<{taskCode:string, heroes: Array<number>}>, sendTime: Date, lean = true) {
    let recs = new Array();
    const refTime = getRefTime();
    for (let task of tasks) {
      const rec = await TowerTaskRecModel.findOneAndUpdate(
        {batchCode, taskCode: task.taskCode, createdAt: {$gte: refTime}},
        {heroes: task.heroes, status: 1, sendTime},
        {new: true}).lean(lean);

      recs.push(rec);
    }
    return recs;
  }

  public static async finishTask(batchCode: string, tasks: Array<string>, lean = true) {
    let recs = new Array();
    // const refTime = getRefTime();
    for (let task of tasks) {
      const rec = await TowerTaskRecModel.findOneAndUpdate(
        {batchCode, taskCode: task},
        {status: 3},
        {new: true}).lean(lean);

      recs.push(rec);
    }
    return recs;
  }

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

export const TowerTaskRecModel = getModelForClass(TowerTaskRec);