package com.zy.acs.manager.core.service;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.zy.acs.common.utils.RedisSupport;
import com.zy.acs.framework.common.Cools;
import com.zy.acs.framework.common.SnowflakeIdWorker;
import com.zy.acs.framework.exception.CoolException;
import com.zy.acs.manager.common.utils.MapDataUtils;
import com.zy.acs.manager.core.domain.BlockVehicleDto;
import com.zy.acs.manager.core.domain.TaskPosDto;
import com.zy.acs.manager.core.service.astart.MapDataDispatcher;
import com.zy.acs.manager.core.service.astart.RetreatNavigateNode;
import com.zy.acs.manager.core.service.astart.WaveNodeType;
import com.zy.acs.manager.manager.entity.*;
import com.zy.acs.manager.manager.enums.JamStateType;
import com.zy.acs.manager.manager.enums.SegmentStateType;
import com.zy.acs.manager.manager.enums.TaskTypeType;
import com.zy.acs.manager.manager.service.*;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.transaction.interceptor.TransactionAspectSupport;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
import java.util.stream.Collectors;

/**
 * Wavefront
 * Created by vincent on 6/25/2024
 */
@Slf4j
@Component
public class TrafficService {

    public static final Integer MIN_SLICE_PATH_LENGTH = 3;

    public static final Integer MAX_JAM_TIMEOUT = 5 * 1000;

    private final RedisSupport redis = RedisSupport.defaultRedisSupport;

    @Autowired
    private AgvService agvService;
    @Autowired
    private AgvDetailService agvDetailService;
    @Autowired
    private CodeService codeService;
    @Autowired
    private TravelService travelService;
    @Autowired
    private SegmentService segmentService;
    @Autowired
    private MainService mainService;
    @Autowired
    private MapService mapService;
    @Autowired
    private MapDataDispatcher mapDataDispatcher;
    @Autowired
    private SnowflakeIdWorker snowflakeIdWorker;
    @Autowired
    private ActionService actionService;
    @Autowired
    private RetreatNavigateService retreatNavigateService;
    @Autowired
    private TaskService taskService;
    @Autowired
    private JamService jamService;
    @Autowired
    private AvoidWaveCalculator avoidWaveCalculator;

    @Transactional
    public synchronized void trigger(Segment segment) {
        try {
            Date now = new Date();
            Travel travel = travelService.getById(segment.getTravelId());
            Agv agv = agvService.getById(travel.getAgvId());
            AgvDetail agvDetail = agvDetailService.selectByAgvId(travel.getAgvId());
            long endNode = segment.getEndNode();

            // valid ----------------------------------------------------
            if (!agvService.judgeEnable(agv.getId())) {
                return;
            }
            if (!Cools.isEmpty(segmentService.getByAgvAndState(agv.getId(), SegmentStateType.RUNNING.toString()))) {
                return;
            }
            List<Segment> waitingSegList = segmentService.getByAgvAndState(agv.getId(), SegmentStateType.WAITING.toString());
            if (!Cools.isEmpty(waitingSegList)) {
                for (Segment waitingSeg : waitingSegList) {
                    if (!waitingSeg.getId().equals(segment.getId())) {
                        return;
                    }
                }
            }
            if (jamService.count(new LambdaQueryWrapper<Jam>()
                    .eq(Jam::getJamAgv, agv.getId())
                    .eq(Jam::getJamSeg, segment.getId())
                    .eq(Jam::getCycleAvo, 1)
                    .eq(Jam::getCycleCode, endNode)
                    .eq(Jam::getState, JamStateType.RUNNING.toString())
            ) > 0) {
                return;
            }

            // execute -----------------------------------------------
            //        ArrayList<List<TaskPosDto>> list = JSON.parseObject(travel.getTaskContent(), new TypeReference<ArrayList<List<TaskPosDto>>>() {});

            // get path list
            avoidWaveCalculator.calcWaveScope();    // * sync wave scope

            // checkout path
            Code startCode = codeService.getById(agvDetail.getRecentCode());
            Code endCode = codeService.getById(endNode);
            List<String> pathList = this.checkoutPath(agv, startCode, endCode, segment);
            if (Cools.isEmpty(pathList)) {
                return;
            }

            List<Segment> segmentList = new ArrayList<>();
            segmentList.add(segment);

            String lastCodeData = pathList.get(pathList.size() - 1);
            if (!endCode.getData().equals(lastCodeData)) {
                // slice
                Code lastCode = codeService.selectByData(lastCodeData);

//                if (pathList.size() <= MIN_SLICE_PATH_LENGTH) {
//                    return;
//                }

                // revert
                segment.setState(SegmentStateType.INIT.toString());
                segment.setUpdateTime(now);
                if (!segmentService.updateById(segment)) {
                    log.error("Segment [{}] failed to update ！！！", segment.getGroupId() + " - " + segment.getSerial());
                }
                segmentList.clear();

                // new move seg
                Segment insertSeg = new Segment();
                insertSeg.setUuid(String.valueOf(snowflakeIdWorker.nextId()).substring(3));
                insertSeg.setTravelId(travel.getId());
                insertSeg.setAgvId(agv.getId());
                insertSeg.setTaskId(segment.getTaskId());
                insertSeg.setSerial(segment.getSerial() - 1);
                insertSeg.setEndNode(lastCode.getId());
                insertSeg.setPosType(TaskPosDto.PosType.MOVE.toString());
                insertSeg.setState(SegmentStateType.WAITING.toString());
                if (!segmentService.save(insertSeg)) {
                    log.error("Segment [{}] failed to save ！！！", segment.getTravelId() + " - " + segment.getSerial());
                }
                segmentList.add(insertSeg);

            } else {

                // complete first segment then merge behind segment
                int serial = segment.getSerial();
                boolean interrupt = false;
                while (!interrupt) {

                    Segment nextStep = segmentService.getNextStepOfInit(travel.getId(), serial);
                    serial ++;
                    if (null == nextStep) {
                        interrupt = true;
                    } else {
                        if (nextStep.getEndNode() == endNode) {

                            segmentList.add(nextStep);
                        } else {
                            interrupt = true;
                        }
                    }

                }
            }

            mapService.lockPath(null, pathList, agv.getUuid());

            mainService.generateAction(segment.getAgvId(), segmentList, pathList);

        } catch (Exception e) {
            log.error("TrafficService.trigger", e);
            TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();

        }
    }

    private List<String> checkoutPath(Agv agv, Code startCode, Code endCode, Segment segment) {
        Integer lev = null;
        String agvNo = agv.getUuid();
        Date now = new Date();
        Jam jam = jamService.getJam(agv.getId(), startCode.getId(), segment.getId());

        // 结果集
        List<String> pathList = new ArrayList<>();

        // 无障碍解
        List<String> unlockPathList = mapService.checkoutPath(agvNo, startCode, endCode, false);
        // 避让解
        List<String> lockPathList = mapService.checkoutPath(agvNo, startCode, endCode, true);

        if (!Cools.isEmpty(lockPathList) && // 存在避让解
            Math.abs(lockPathList.size() - unlockPathList.size()) <= Arrays.stream(mapDataDispatcher.getCodeMatrix(lev)).mapToInt(row -> row.length).sum() / 10
        ) {

            pathList = lockPathList;
        } else {

            // 可走行路径集合计算
            List<BlockVehicleDto> blockVehicleList = this.slicePathAndReturnBlockVehicleList(lev, unlockPathList, agvNo, pathList);    // jamAgvNo may was wave

            if (!Cools.isEmpty(pathList)) {

                if (!pathList.get(pathList.size() - 1).equals(endCode.getData())) {
                    assert !Cools.isEmpty(blockVehicleList);

                    boolean hasUnavoidableBlocks  = blockVehicleList.stream().anyMatch(blockVehicleDto -> !blockVehicleDto.isAvoidable());
                    if (hasUnavoidableBlocks && pathList.size() <= MIN_SLICE_PATH_LENGTH) {
                        log.info("AGV[{}] waiting in place, because the path list is too short...", agvNo);
                        pathList.clear();
                    }
                }

            // 无可走行路径
            } else {

                assert !Cools.isEmpty(blockVehicleList);
                Integer maxJamTimeoutFactor = null;

                // persist jam data
                jam = this.createOrUpdateJam(agv, startCode, segment, jam);

                // ? has unAvoidable block vehicles
                if (blockVehicleList.stream().anyMatch(blockVehicleDto -> !blockVehicleDto.isAvoidable())) {

                    // set factor of jam timeout
                    maxJamTimeoutFactor = 1;
                } else {

                    // ? already do notify to avoid
                    if (!Cools.isEmpty(jam.getAvoAgv())
                            && BlockVehicleDto.customContain(blockVehicleList, agvService.getById(jam.getAvoAgv()).getUuid())) {

                        maxJamTimeoutFactor = 10;

                    } else {

                        // select optimal block vehicle
                        String blockAgvNo = blockVehicleList.stream()
                                .filter(BlockVehicleDto::isAvoidable)
                                .map(BlockVehicleDto::getVehicle)
                                .findFirst().orElse(null);

                        // block vehicle info
                        Agv blockAgv = agvService.selectByUuid(blockAgvNo);
                        String blockAgvCode = codeService.getById(agvDetailService.selectByAgvId(blockAgv.getId()).getRecentCode()).getData();

                        do {

                            // 阻塞车辆正在原地作业，等待 ===>> 超过等待时间，绕路
                            List<Segment> runningSegList = segmentService.getByAgvAndState(blockAgv.getId(), SegmentStateType.RUNNING.toString());
                            if (!Cools.isEmpty(runningSegList)) {
                                maxJamTimeoutFactor = 1;
                                break;
                            }

                            // 判断下个任务是否为原地任务，如果是则等待 ===>> 超过等待时间，绕路；如果不是，让阻塞车辆避让
                            Segment waitingSeg = segmentService.getJustWaitingSeg(blockAgv.getId());
                            if (null != waitingSeg && waitingSeg.getEndNode().equals(codeService.selectByData(blockAgvCode).getId())) {
                                maxJamTimeoutFactor = 1;
                                break;
                            }

                            // notify block vehicle to avoid
                            if (this.notifyVehicleAvoid(blockAgvNo, blockAgvCode, unlockPathList, agvNo, jam)) {
                                if (jam.getCycleAvo() == 1) {
                                    jam.setCycleCode(endCode.getId());
                                }
                                jam.setAvoAgv(blockAgv.getId());
                                jam.setNotifyTime(new Date());
                                if (!jamService.updateById(jam)) {
                                    throw new CoolException(jam.getUuid() + "-jam failed to update！！！");
                                }
                            } else {

                                maxJamTimeoutFactor = 1;
                            }

                        } while (false);
                    }

                }

                // handle jam timeout
                if (null != maxJamTimeoutFactor) {
                    if (System.currentTimeMillis() - jam.getStartTime().getTime() > MAX_JAM_TIMEOUT * maxJamTimeoutFactor) {

                        if (!Cools.isEmpty(lockPathList)) {

                            pathList = lockPathList;
                        } else {
                            log.error("{}号车辆检索[{}] ===>> [{}]路径失败，原因：{}"
                                    , agvNo, startCode.getData(), endCode.getData(), "路径阻塞超时");
                        }
                    } else {
                        log.warn("{}号车辆正在等待交通堵塞，阻塞车辆：【{}】"
                                , agvNo
                                , blockVehicleList.stream().map(BlockVehicleDto::getVehicle).collect(Collectors.toList()).toString()
                        );
                    }
                }

            }

        }

        if (!Cools.isEmpty(pathList)) {

            if (null != jam) {
                jam.setEndTime(now);
                jam.setUpdateTime(now);
                jam.setState(JamStateType.FINISH.toString());
                if (!jamService.updateById(jam)) {
                    log.error("Jam[{}] failed to update！！！", jam.getUuid());
                }
            }
            // deal expired jam
            for (Jam expiredJam : jamService.list(new LambdaQueryWrapper<Jam>()
                    .eq(Jam::getJamAgv, agv.getId())
                    .eq(Jam::getState, JamStateType.RUNNING.toString()))) {
                expiredJam.setEndTime(now);
                expiredJam.setUpdateTime(now);
                expiredJam.setState(JamStateType.FINISH.toString());
                if (!jamService.updateById(expiredJam)) {
                    log.error("Jam[{}] failed to update！！！", expiredJam.getUuid());
                }
            }

        }

        return pathList;
    }

    private List<BlockVehicleDto> slicePathAndReturnBlockVehicleList(Integer lev, List<String> fullPathList, String agvNo, List<String> pathList) {
        List<BlockVehicleDto> blockVehicleList = new ArrayList<>();

//        DynamicNode[][] dynamicMatrix = mapDataDispatcher.getDynamicMatrix(lev);
        String[][] waveMatrix = mapDataDispatcher.getWaveMatrix(lev);
        for (String code : fullPathList) {
            int[] node = mapDataDispatcher.getCodeMatrixIdx(lev, code);

            String waveNode = waveMatrix[node[0]][node[1]];
            assert !waveNode.equals(WaveNodeType.DISABLE.val);
            if (!waveNode.equals(WaveNodeType.ENABLE.val)) {
                List<String> waveNodeList = MapDataUtils.parseWaveNode(waveNode);
                List<String> otherWaveList = MapDataUtils.hasOtherWave(waveNodeList, agvNo);
                if (!Cools.isEmpty(otherWaveList)) {
                    for (String otherWave : otherWaveList) {
                        if (1 < mapDataDispatcher.queryCodeListFromDynamicNode(lev, otherWave).size()) {
                            blockVehicleList.add(new BlockVehicleDto(otherWave, false));
                        } else {
                            blockVehicleList.add(new BlockVehicleDto(otherWave, true));
                        }
                    }
                    break;
                }
            }

//            DynamicNode dynamicNode = dynamicMatrix[node[0]][node[1]];
//            String vehicle = dynamicNode.getVehicle();
//            assert !vehicle.equals(DynamicNodeType.BLOCK.val);
//            if (!vehicle.equals(DynamicNodeType.ACCESS.val) && !vehicle.equals(agvNo)) {
//                blockVehicleList.add(vehicle);
//                break;
//            }

            pathList.add(code);
        }

        if (pathList.size() <= 1) {
            pathList.clear();
        }

        return blockVehicleList.stream().distinct().collect(Collectors.toList());
    }

    /**
     * avoidPathList include wave node and dynamic node
     */
    private boolean notifyVehicleAvoid(String agvNo, String agvPosCode, List<String> avoidPathList, String sponsor, Jam jam) {
        Agv agv = agvService.selectByUuid(agvNo);
        if (!Cools.isEmpty(segmentService.getByAgvAndState(agv.getId(), SegmentStateType.RUNNING.toString()))) {
            log.warn("{}号车辆避让失败，存在进行中任务！！！", agvNo);
            return false;
        }

        int[] startMapIdx = mapDataDispatcher.getCodeMatrixIdx(null, agvPosCode);
        RetreatNavigateNode startNode = new RetreatNavigateNode(startMapIdx[0], startMapIdx[1], agvPosCode);

        assert avoidPathList.size() >= 2;
        RetreatNavigateNode finalNode = retreatNavigateService.execute(agvNo, startNode, avoidPathList, sponsor, jam);
        if (null == finalNode) {
            log.warn("{}号车辆避让失败，检索避让点失败！！！", agvNo);
            return false;
        }

        String endCodeData = finalNode.getCodeData();
        Code endCode = codeService.selectByData(endCodeData);
        jam.setAvoCode(endCode.getId());

        List<Segment> waitingSegList = segmentService.getByAgvAndState(agv.getId(), SegmentStateType.WAITING.toString());
        if (!Cools.isEmpty(waitingSegList)) {

            // revert
            Date now = new Date();
            for (Segment seg : waitingSegList) {
                seg.setState(SegmentStateType.INIT.toString());
                seg.setUpdateTime(now);
                if (!segmentService.updateById(seg)) {
                    log.error("Segment [{}] 更新失败 ！！！", seg.getTravelId() + " - " + seg.getSerial());
                }
            }
            Segment segment = waitingSegList.get(0);

            Segment insertSeg = new Segment();
            insertSeg.setUuid(String.valueOf(snowflakeIdWorker.nextId()).substring(3));
            insertSeg.setTravelId(segment.getTravelId());
            insertSeg.setAgvId(agv.getId());
            insertSeg.setTaskId(segment.getTaskId());
            insertSeg.setSerial(segment.getSerial() - 1);
            insertSeg.setEndNode(endCode.getId());
            insertSeg.setPosType(TaskPosDto.PosType.MOVE.toString());
            insertSeg.setState(SegmentStateType.WAITING.toString());
            if (!segmentService.save(insertSeg)) {
                log.error("Segment [{}] 保存失败 ！！！", segment.getTravelId() + " - " + segment.getSerial());
                return false;
            }

        } else {

            return mainService.buildMinorTask(agv, null, TaskTypeType.MOVE, endCodeData);
        }

        return true;
    }

    private Jam createOrUpdateJam(Agv agv, Code startCode, Segment segment, Jam jam) {
        if (jam == null) {
            jam = new Jam();
            jam.setUuid(String.valueOf(snowflakeIdWorker.nextId()).substring(3));
            jam.setJamAgv(agv.getId());
            jam.setJamCode(startCode.getId());
            jam.setJamSeg(segment.getId());
            jam.setStartTime(new Date());
            jam.setState(JamStateType.RUNNING.toString());
            if (!jamService.save(jam)) {
                log.error("AGV[{}] failed to save jam", agv.getUuid());
                throw new CoolException("failed to save jam");
            }
        } else {
            jam.setDuration(System.currentTimeMillis() - jam.getStartTime().getTime());
            if (!jamService.updateById(jam)) {
                log.error("AGV[{}] failed to update jam", agv.getUuid());
            }
        }
        return jam;
    }

}