ZYZ/game-server/app/services/battleService.ts

import { HeroModel } from './../db/Hero';
import { HangUpRecordModel } from './../db/HangUpRecord';
import { pinus } from 'pinus';
import { HANG_UP_CONSTS, TOWER_TASK_CONST, REDIS_KEY, TASK_TYPE, TIME_OUTPUT_TYPE } from './../consts';
import { BattleRecordModel } from './../db/BattleRecord';
import { TowerRecordModel } from './../db/TowerRecord';
import { RoleModel } from './../db/Role';
import { shouldRefresh, resResult, cal, getRandEelmWithWeight } from '../pubUtils/util';
import { STATUS } from '../consts/statusCode';
import { HangUpSpdUpRecModel } from '../db/HangUpSpdUpRec';
import { TowerTaskRecModel } from '../db/TowerTaskRec';
import { cloneDeep } from 'lodash';
import { Rank } from './rankService';
import { checkTask } from './taskService';
import { getRandExpedition, gameData } from '../pubUtils/data';
import { ItemInter, RewardInter } from '../pubUtils/interface';
import { getTimeFunM } from '../pubUtils/timeUtil';

export async function checkTowerWar(roleId: string, battleId: number, heroes: Array<number>) {
    let { towerLv } = await RoleModel.findByRoleId(roleId);
    const towerInfo = gameData.tower.get(towerLv);
    if (!towerInfo) {
        console.error(`天梯层数异常，lv ${towerLv} by ${roleId}`);
        return { status: -1, resResult: resResult(STATUS.TOWER_INFO_NOT_FOUND) };
    }
    const warIds = towerInfo.warArray;

    if (warIds.indexOf(battleId) === -1) {
        return { status: -1, resResult: resResult(STATUS.TOWER_WRONG_BATTLE_ID) };
    }

    const tower = await TowerRecordModel.getRecordByLv(roleId, towerLv);
    if(!tower) {
        return { status: -1, resResult: resResult(STATUS.TOWER_NOT_FOUND) };
    }
    let { heroes: recHeroes = [], warStatus = [] } = tower;
    for (let hid of heroes) {
        if (recHeroes.indexOf(hid) !== -1) {
            return { status: -1, resResult: resResult(STATUS.TOWER_DUPLICATE_HERO) };
        }
    }
    const curWarStatus = warStatus.find(elem => elem.warId == battleId);
    if (curWarStatus && curWarStatus.status) {
        return { status: -1, resResult: resResult(STATUS.TOWER_DUPLICATE_CHALLENGE) };
    }

    return { status: 0, data: {
        heroes: recHeroes, warStatus: curWarStatus
    }};
}

export async function towerBattleEnd(sid: string, funcs: number[], roleId: string, serverId: number, battleCode: string, battleId: number, succeed: boolean, heroes: Array<number>) {
    if (succeed) {
        let battleRec = await BattleRecordModel.getBattleRecordByCode(battleCode);
        if (battleRec.battleId != battleId) {
            return { status: -1, resResult: resResult(STATUS.BATTLE_ID_NOT_MATCH) };
        }
        let { towerLv, roleName } = await RoleModel.findByRoleId(roleId);
        let { warStatus, heroes: recHeroes } = await TowerRecordModel.getRecordByLv(roleId, towerLv);
        for (let hid of heroes) {
            if (recHeroes.indexOf(hid) !== -1) {
                return { status: -1, resResult: resResult(STATUS.TOWER_DUPLICATE_CHALLENGE) };
            }
        }

        let inc = 1;
        warStatus.forEach(st => {
            if (st.warId !== battleId && st.status === false) {
                inc = 0;
            }
        })
        let newRec = await TowerRecordModel.updateRecord(roleId, towerLv, battleCode, battleId, heroes, inc);
        let towerReward: RewardInter[] = null;
        if (inc === 1) {
            let role = await RoleModel.towerLvUp(roleId);
            // 更新redis
            let r = new Rank(REDIS_KEY.TOWER_RANK, { serverId });
            await r.setRankWithRoleInfo(roleId, towerLv + 1, role.towerUpTime.getTime(), role);

            const nextTowerInfo = gameData.tower.get(towerLv + 1);
            if (nextTowerInfo) {
                const { warArray } = nextTowerInfo;
                const sts = warArray.map(id => {
                    return {warId: id, status: false};
                });

                await TowerRecordModel.createRecord({roleId, lv: towerLv + 1, warStatus: sts});

            }
            const { reward } = gameData.tower.get(towerLv);
            if (reward) towerReward = reward;
            if (towerLv + 1 == HANG_UP_CONSTS.ENABLE_LV) {
                await startHangUp(roleId, roleName);
                pinus.app.get('channelService').pushMessageByUids('hangUpEnable', resResult(STATUS.SUCCESS, {enable: true}), [{uid: roleId, sid}]);
            }
            // 任务
            await checkTask(roleId, sid, funcs, TASK_TYPE.BATTLE_TOWER_LV, role.towerLv, false, {});
        }
        return {
            status: 0,
            data: {
                newRec,
                towerReward,
                towerStatus: !!inc
            }
        };
    }
}

async function startHangUp(roleId: string, roleName: string) {
    await HangUpRecordModel.initRecord(roleId, roleName);
}

export async function checkHangUpSpdUpCnt(roleId: string, cnt: number, curTime: Date) {
    const role = await RoleModel.findByRoleId(roleId);
    let {hangUpSpdUpCnt, gold, lastSpdUpTime} = role;
    if (!lastSpdUpTime || (shouldRefresh(lastSpdUpTime, curTime) && hangUpSpdUpCnt <= HANG_UP_CONSTS.MAX_SPD_UP_CNT)) {
        hangUpSpdUpCnt = HANG_UP_CONSTS.MAX_SPD_UP_CNT;
    }        
    if (cnt > hangUpSpdUpCnt) {
        return {status: -1, resResult:resResult(STATUS.TOWER_NOT_ENOUGH_HANG_UP_TIME)}
    }
    return {status: 0, data: {hangUpSpdUpCnt, gold, lastSpdUpTime}}
}

export async function calcuHangUpReward(roleId: string, speedUp = false, speedUpCnt = 1, curTime = new Date()) {
    let { towerLv = 1, hangUpSpdUpCnt = 0, lastSpdUpTime } = await RoleModel.findByRoleId(roleId);
    if(towerLv < HANG_UP_CONSTS.ENABLE_LV) {
        return {status: -1, resResult: resResult(STATUS.TOWER_HANG_UP_NOT_START)}
    }

    let towerInfo = gameData.tower.get(towerLv - 1); // towerLv 是当前层，奖励计算按照已经通过的层，即上一层
    let timeReward = []; // 奖励
    let needReceiveGoods = []; // 由于小数，未能领取的奖励
    let baseReward = towerInfo.rewardOfcollect;

    let { startTime } = await HangUpRecordModel.getCurRec(roleId);
    let endTime = curTime; // 挂机结束时间，现在到开始时间，经历了10的倍数的时间
    let deltaTime = curTime.getTime() - startTime.getTime(); // 累计的挂机时间，受最大时间限制
    if (deltaTime > HANG_UP_CONSTS.MAX_TIME) {
        deltaTime = HANG_UP_CONSTS.MAX_TIME;
        endTime = curTime; // 累积到超过24小时，那么结束时间和下一次开启时间就取整了
    } else {
        let multiReal = Math.floor(deltaTime / HANG_UP_CONSTS.UNIT_TIME);// 距开始挂机实际过去的时间单位
        endTime = new Date(startTime.getTime() + multiReal * HANG_UP_CONSTS.UNIT_TIME)
    }

    if (speedUp) { // 加速，直接收取6小时收益，小数的累积和普通收取独立
        if (hangUpSpdUpCnt >= speedUpCnt || !lastSpdUpTime || shouldRefresh(lastSpdUpTime, new Date)) { // 可加速
            let multi = Math.floor(HANG_UP_CONSTS.SPD_UP_REC_TIME / HANG_UP_CONSTS.UNIT_TIME);
            let spdUpRec = await HangUpSpdUpRecModel.getSpdUpRec(roleId, towerLv - 1);
            let goods = [];
            if(spdUpRec) {
                let { notReceivedGoodsList = [], cnt } = spdUpRec;
                let notReceivedGoods = notReceivedGoodsList.find(cur => cur.cnt == cnt );
                goods = notReceivedGoods?notReceivedGoods.goods:[];
            }
            for(let {id, count} of baseReward) {
                let newCount = cal.mul(count, multi);
                let oldGoods = goods.find(cur => cur.gid == id);
                if(oldGoods) newCount = cal.add(newCount, oldGoods.count);
                let roundCount = Math.floor(newCount);
                if(newCount > roundCount) {
                    needReceiveGoods.push({gid: id, count: cal.sub(newCount, roundCount)});
                }
                if(roundCount > 0) {
                    timeReward.push({id, count: roundCount})
                }
            }
        }
    } else {
        let lastRec = await HangUpRecordModel.getLastRec(roleId);
        let notReceivedGoods = lastRec?lastRec.notReceivedGoods: [];

        let multi = Math.floor(deltaTime / HANG_UP_CONSTS.UNIT_TIME); // 结算奖励的倍数，受最大时间限制

        // console.log(deltaTime, multi, baseReward);
        for(let {id, count} of baseReward) {
            let newCount = cal.mul(count, multi);
            let oldGoods = notReceivedGoods.find(cur => cur.gid == id);
            if(oldGoods) newCount = cal.add(newCount, oldGoods.count);
            let roundCount = Math.floor(newCount);
            if(newCount > roundCount) {
                needReceiveGoods.push({gid: id, count: cal.sub(newCount, roundCount)});
            }
            if(roundCount > 0) {
                timeReward.push({id, count: roundCount})
            }
        }
    }
    return {
        status: 0,
        data: {
            endLv: towerLv - 1,
            startTime,
            deltaTime,
            endTime,
            timeReward,
            needReceiveGoods
        }
    };
}

async function checkCond(roleId: string, heroes, type: number, param: number, cnt: number) {

    let heroCnt = 0;

    switch (type) {
        case 1:
            for(let seqId of heroes) {
                const {star, colorStar} = await HeroModel.findBySeqIdAndRole(seqId, roleId);
                if (star + colorStar >= param) {
                    heroCnt++;
                }
            }
            break;
        case 2:
            for(let seqId of heroes) {
                let {hid} = await HeroModel.findBySeqIdAndRole(seqId, roleId);
                const hInfo = gameData.hero.get(hid);
                if (hInfo.camp === param) {
                    heroCnt++;
                }
            }
            break;
        case 3:
            for(let seqId of heroes) {
                let {hid} = await HeroModel.findBySeqIdAndRole(seqId, roleId);
                const hInfo = gameData.hero.get(hid);
                if (gameData.job.get(hInfo.jobid).job_class === param) {
                    heroCnt++;
                }
            }
            break;
        default:
            break;
    }
    if (heroCnt >= cnt) {
        return true;
    }
    return false;
}

export async function checkTaskConditions(roleId: string, heroes: number[], condition: { type: number, param: number, count: number }[]) {

    let res = true;
    for (let {type, param, count} of condition) {
        const checkRes = await checkCond(roleId, heroes, type, param, count);
        if (!checkRes) {
            res = false;
            break;
        }
    }
    return res;
}

/**
 * 创建新的派遣任务
 * @param towerLv 玩家层数
 * @param roleId 玩家id
 * @param roleName 玩家名
 * @param hideNum 隐藏的任务数量
 */
export async function createCurTasks(towerLv: number, batchCode: string, roleId: string, roleName: string, hideNum: number, showNum: number, rids: Array<number>) {
    let randomList = [];
    let dicTask = gameData.towerTask;
    for(let [ _id, task] of dicTask) {
        let {suitFloor: {min, max} } = task;
        let flag = max?(towerLv >= min && towerLv <= max): (towerLv >= min);
        if(flag) {
            randomList.push(task);
        }
    }
    let returnTasks = new Array();
    const arr = [{refreshStatus: 0, num: hideNum}, {refreshStatus: 1, num: showNum}];
    for(let {refreshStatus, num} of arr) {
        let taskIds = [];
        for(let i = 0; i < num; i++) {
            let list = randomList.filter(cur => !rids.includes(cur.taskId));
            let {dic: tmp} = getRandEelmWithWeight(list);
            if(tmp) {
                taskIds.push(tmp.taskId);
                rids.push(tmp.taskId)
            }
        }
        const curTasks = await TowerTaskRecModel.createTasks(roleId, roleName, batchCode, taskIds, refreshStatus);

        if(refreshStatus == 1) {
            returnTasks = curTasks;
        }
    }

    return returnTasks;
}

export function getRemainTime(curTime: Date) {
    let nextTime = <number>getTimeFunM(curTime).getAfterDayWithHour(1);
    return Math.floor((nextTime - curTime.getTime())/1000);
}

export function treatTask(recs: Array<any>, curTime: Date) {
    return recs.map(cur => {
        let { heroes, batchCode, taskId, taskCode, status, completeTime} = cur;

        let getStatusResult = getTaskStatus(status, completeTime, curTime);
        return { heroes, batchCode, taskId, taskCode, completeTime, ...getStatusResult }
      });
}

// 刷新任务，正在进行中的部分保留，没有进行中的刷掉
export function getDoingOrWaitingTasks (taskList: Array<any>, curTime: Date) {
    let doingTaskCode = new Array<string>(), waitingTaskCode = new Array<string>(), doingIds = new Array<number>();
    for(let task of taskList) {
        let {status} = getTaskStatus(task.status, task.completeTime, curTime);
        if(status == 1||status == 2) {
            doingTaskCode.push(task.taskCode);
            doingIds.push(task.taskId);
        } else if(status == 0|| status == 3) {
            waitingTaskCode.push(task.taskCode);
        }
    }
    return {doingTaskCode, waitingTaskCode, doingIds}
}

export function getTaskStatus(status: number, completeTime: Date, curTime: Date) {
    let remainTime = 0;
    if(status == 1) {
        // console.log(curTime.getTime(), completeTime.getTime())
      remainTime = Math.floor((completeTime.getTime() - curTime.getTime())/1000);
      if(remainTime < 0) {
        status = 2; remainTime = 0;
      }
    }

    return {status, remainTime}
}

export async function getTasksReward(roleId: string, curTime: Date) {
    let oldTasks = await TowerTaskRecModel.getCurTasks(roleId);
    let goods: ItemInter[] = [];
    for(let task of oldTasks) {
        let {completeTime, reward, termsForAdd, additionalReward} = gameData.towerTask.get(task.taskId);
        if (task.sendTime && task.sendTime.getTime() + completeTime * 1000 < curTime.getTime()) {
            goods = goods.concat(reward);
            if(termsForAdd) {
                const result = checkTaskConditions(roleId, task.heroes, termsForAdd);
                if(result) {
                    goods = goods.concat(additionalReward);
                }
            }
        }
    }
    return goods
}

/**
 * 获取随机机器人数据
 * @param cnt 机器人数量
 * @param withAttr 是否返回属性，false: 仅返回阵容
 */
export function getRandRobot(cnt = 1, withAttr = false) {
    let setInfo = getRandExpedition(cnt);
    if (!setInfo || setInfo.length !== cnt) return null;

    let robots = [];
    for (let info of setInfo) {
        if (!info || !info.warId) continue;
        const warInfo = gameData.warJson.get(info.warId);
        if (!warInfo || !warInfo.length) continue;

        let heroes = [];
        for(let hero of warInfo) {
            if (hero && hero.relation === 2 && hero.actorId) {
                heroes.push(hero.actorId);
            }
        }
        robots.push(heroes);
    }
    return robots;
}

/**
 * 拷贝敌军数组并添加当前血量字段
 * @param source 原敌军数组，不包含当前血量字段
 */
export function transBossHpArr(source: Array<{dataId: number, hp: number, actorId: number}>): Array<{dataId: number, hp: number, curHp: number, actorId: number}> {
    let desArr = [];
    source.forEach(elem => {
        let { hp } = elem;
        desArr.push(Object.assign(elem, {curHp: hp}));
    });
    return cloneDeep(desArr);
}