import BaseModel from './BaseModel';
import { index, getModelForClass, prop } from '@typegoose/typegoose';
import { getRefTime } from '../pubUtils/util';
import { TOWER_TASK_CONST } from '../consts/consts';

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;
}



/**
  * 天梯派遣记录表
  */
@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, default: 1})
  refreshStatus: number; // 派遣任务是否显示 0-不显示，被刷新掉了 1-显示
  @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; // 派遣时间
  @prop({ required: false })
  completeTime: Date; // 派遣结束时间

  public static async getCurTasks(roleId: string, lean = true) {
    let curTime = new Date();
    const refTime = getRefTime(curTime, TOWER_TASK_CONST.REFRESH_TIME);
    const recs = await TowerTaskRecModel.find({roleId, createdAt: {$gte: refTime}, refreshStatus: 1}).sort({createdAt: 1}).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>, completeTime: number}>, sendTime: Date, lean = true) {
    let recs = new Array();
    let curTime = new Date();
    const refTime = getRefTime(curTime, TOWER_TASK_CONST.REFRESH_TIME);
    for (let {taskCode, heroes, completeTime} of tasks) {
      const rec = await TowerTaskRecModel.findOneAndUpdate(
        {batchCode, taskCode: taskCode, createdAt: {$gte: refTime}},
        {heroes: heroes, status: 1, sendTime, completeTime: new Date(sendTime.getTime() + completeTime * 1000)},
        {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 refreshTask(roleId: string, lean = true) {
    let curTime = new Date();
    const refTime = getRefTime(curTime, TOWER_TASK_CONST.REFRESH_TIME);
    console.log(refTime)
    const recs = await TowerTaskRecModel.updateMany({roleId, createdAt: {$gte: refTime}, refreshStatus: 1}, {refreshStatus: 0}, {new: true}).lean(lean);
    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);