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

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.zy.acs.common.enums.AgvStatusType;
import com.zy.acs.manager.core.service.MainService;
import com.zy.acs.manager.manager.entity.*;
import com.zy.acs.manager.manager.enums.FuncStaStateType;
import com.zy.acs.manager.manager.enums.FuncStaType;
import com.zy.acs.manager.manager.enums.TaskStsType;
import com.zy.acs.manager.manager.enums.TaskTypeType;
import com.zy.acs.manager.manager.service.*;
import com.zy.acs.manager.system.service.ConfigService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import java.util.Date;
import java.util.List;
import java.util.Objects;

/**
 * to judgement agv idle status when generate action !!!
 * Created by vincent on 2023/9/9
 */
@Slf4j
@Component
public class MaintainScheduler {

    @Autowired
    private AgvService agvService;
    @Autowired
    private AgvDetailService agvDetailService;
    @Autowired
    private FuncStaService funcStaService;
    @Autowired
    private MainService mainService;
    @Autowired
    private TaskService taskService;
    @Autowired
    private ConfigService configService;
    @Autowired
    private AgvModelService agvModelService;

    @Scheduled(cron = "0/5 * * * * ? ")
    private synchronized void autoCharge(){
        List<Agv> agvList = agvService.list(new LambdaQueryWrapper<Agv>().eq(Agv::getStatus, 1));
        for (Agv agv : agvList) {
            AgvDetail agvDetail = agvDetailService.selectByAgvId(agv.getId());
            if (null == agvDetail || null == agvDetail.getVol() || null == agvDetail.getAgvStatus()) {
                continue;
            }
            if (agvDetail.getVol() <= agv.getChargeLine() && !agvDetail.getAgvStatus().equals(AgvStatusType.CHARGE)) {
                if (taskService.count(new LambdaQueryWrapper<Task>()
                        .eq(Task::getAgvId, agv.getId())
                        .and(i -> {
//                            i.eq(Task::getTaskSts, TaskStsType.WAITING.val())
                            i.eq(Task::getTaskSts, TaskStsType.ASSIGN.val())
                                    .or().eq(Task::getTaskSts, TaskStsType.PROGRESS.val());
                        })) > 0) {
                    return;
                }
                mainService.buildMinorTask(agv, agvDetail, TaskTypeType.TO_CHARGE, null);
            }
        }
    }


//    @Scheduled(cron = "0/30 * * * * ? ")
    @Scheduled(cron = "0 */2 * * * ? ")
    private void autoStandby(){
        // 1.agv充电到一定电压后，回到待机位
        // 2.agv没有任务时，回到待机位
        List<Agv> agvList = agvService.list(new LambdaQueryWrapper<Agv>().eq(Agv::getStatus, 1));
        label: for (Agv agv : agvList) {
            AgvDetail agvDetail = agvDetailService.selectByAgvId(agv.getId());
            if (null == agvDetail || null == agvDetail.getVol() || null == agvDetail.getAgvStatus() || null == agvDetail.getCode()) {
                continue;
            }
            List<FuncSta> funcStaList = funcStaService.list(new LambdaQueryWrapper<FuncSta>()
                    .eq(FuncSta::getAgvId, agv.getId()).eq(FuncSta::getType, FuncStaType.STANDBY.toString()));
            for (FuncSta funcSta : funcStaList) {
                if (agvDetail.getRecentCode().equals(funcSta.getCode())) {
                    continue label;
                }
            }
            if (agvDetail.getVol() <= agv.getChargeLine()) {
                continue;
            }
            if (agvDetail.getAgvStatus().equals(AgvStatusType.CHARGE)) {
                AgvModel agvModel = agvModelService.getById(agv.getAgvModel());
                if (agvDetail.getVol() <= agvModel.getQuaBattery()) {
                    continue;
                }
            }
            if (taskService.count(new LambdaQueryWrapper<Task>()
                    .eq(Task::getAgvId, agv.getId())
                    .and(i -> {
                        i.eq(Task::getTaskSts, TaskStsType.WAITING.val())
                                .or().eq(Task::getTaskSts, TaskStsType.ASSIGN.val())
                                .or().eq(Task::getTaskSts, TaskStsType.PROGRESS.val());
                    })) > 0) {
                continue;
            }
            mainService.buildMinorTask(agv, agvDetail, TaskTypeType.TO_STANDBY, null);
        }
    }

    @Scheduled(cron = "0/5 * * * * ? ")
    private synchronized void releaseFuncSta(){
        List<FuncSta> funcStaList = funcStaService.list(new LambdaQueryWrapper<FuncSta>().eq(FuncSta::getState, FuncStaStateType.OCCUPIED.toString()));
        for (FuncSta funcSta : funcStaList) {
            Agv agv = agvService.getById(funcSta.getAgvId());
            boolean beIdle = true;
            switch (Objects.requireNonNull(FuncStaType.query(funcSta.getType()))) {
                case CHARGE: // 充电
                    for (AgvDetail agvDetail : agvDetailService.list(new LambdaQueryWrapper<AgvDetail>().eq(AgvDetail::getLastCode, funcSta.getCode()))) {
                        if (agvDetail.getAgvStatus().equals(AgvStatusType.CHARGE)) {
                            beIdle = false;
                            break;
                        }
                    }
                    if (taskService.count(new LambdaQueryWrapper<Task>()
                            .eq(Task::getTaskType, TaskTypeType.TO_CHARGE.val())
                            .eq(Task::getDestCode, funcSta.getCode())
                            .eq(Task::getAgvId, agv.getId())
                            .and(i -> {
                                i.eq(Task::getTaskSts, TaskStsType.INIT.val()).or()
                                 .eq(Task::getTaskSts, TaskStsType.WAITING.val()).or()
                                 .eq(Task::getTaskSts, TaskStsType.ASSIGN.val()).or()
                                 .eq(Task::getTaskSts, TaskStsType.PROGRESS.val());
                            })) > 0) {
                        beIdle = false;
                    }
                    break;
                case STANDBY: // 待机
                    if (agvDetailService.count(new LambdaQueryWrapper<AgvDetail>().eq(AgvDetail::getCode, funcSta.getCode())) > 0) {
                        beIdle = false;
                    }
                    if (taskService.count(new LambdaQueryWrapper<Task>()
                            .eq(Task::getTaskType, TaskTypeType.TO_STANDBY.val())
                            .eq(Task::getDestCode, funcSta.getCode())
                            .eq(Task::getAgvId, agv.getId())
                            .and(i -> {
                                i.eq(Task::getTaskSts, TaskStsType.INIT.val()).or()
                                .eq(Task::getTaskSts, TaskStsType.WAITING.val()).or()
                                .eq(Task::getTaskSts, TaskStsType.ASSIGN.val()).or()
                                .eq(Task::getTaskSts, TaskStsType.PROGRESS.val());
                            })) > 0) {
                        beIdle = false;
                    }
                    break;
                default:
                    break;
            }
            if (beIdle) {
                funcSta.setState(FuncStaStateType.IDLE.toString());
                funcSta.setUpdateTime(new Date());
                if (!funcStaService.updateById(funcSta)) {
                    log.error("FuncSta [{}] 更新状态失败 ！！！", funcSta.getName());
                }
            }
        }
    }

}