package com.zy.asrs.task;

import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.core.common.Cools;
import com.core.common.R;
import com.zy.api.controller.params.WorkTaskParams;
import com.zy.api.service.WcsApiService;
import com.zy.asrs.entity.WrkMast;
import com.zy.asrs.service.WrkMastService;
import com.zy.asrs.task.core.ReturnT;
import com.zy.asrs.task.handler.WorkMastHandler;
import com.zy.asrs.task.support.OutboundBatchSeqReleaseGuard;
import com.zy.asrs.task.support.WorkPublishLockKeys;
import com.zy.asrs.utils.Utils;
import com.zy.common.properties.SlaveProperties;
import com.zy.common.utils.RedisUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import java.math.BigInteger;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedHashMap;
import java.util.Date;
import java.util.Map;
import java.util.List;
import java.util.Objects;

/**
 * Created by vincent on 2020/7/7
 */
@Component
public class WorkMastScheduler {

    private static final Logger log = LoggerFactory.getLogger(WorkMastScheduler.class);
    private static final int MAX_PUBLISH_TASKS_ONCE = 20;
    private static final long OUTBOUND_USER_NO_LOCK_SECONDS = 60L;
    private static final int DOUBLE_EXTENSION_OUTBOUND_CRN_NO = 19;

    @Autowired
    private WcsApiService wcsApiService;
    @Autowired
    private WrkMastService wrkMastService;
    @Autowired
    private WorkMastHandler workMastHandler;
    @Autowired
    private OutboundBatchSeqReleaseGuard outboundBatchSeqReleaseGuard;
    @Autowired
    private RedisUtil redisUtil;
    @Autowired
    private SlaveProperties slaveProperties;

    private enum LocDepthType {
        SHALLOW,
        DEEP,
        UNKNOWN
    }

    @Scheduled(cron = "0/3 * * * * ? ")
    private void execute(){
        List<WrkMast> wrkMasts = wrkMastService.selectToBeCompleteData();
        if (wrkMasts.isEmpty()) {
            return;
        }
        for (WrkMast wrkMast : wrkMasts) {
            ReturnT<String> returnT = workMastHandler.start(wrkMast);
            if (!returnT.isSuccess()) {
                wrkMast.setUpdMk("X");
                wrkMast.setErrorMemo(returnT.getMsg());
                wrkMast.setErrorTime(new Date());
                if (!wrkMastService.updateById(wrkMast)) {
                    log.error("工作档[workNo={}]标记待处理失败", wrkMast.getWrkNo());
                }
            }
        }
    }

    /**
     * 任务自动下发。
     * <p>
     * 调度器只负责从工作档中挑出“当前允许下发”的任务，并将其转换成 WCS 接口需要的报文结构；
     * 出库任务按 userNo -> batchSeq 分层汇总后串行下发，确保同一 userNo 下前一个 batchSeq 完成后再发下一个。
     * <p>
     * 当前下发规则是：
     * 1. 入库任务按 WCS 接口批量下发；
     * 2. 移库任务按单条任务下发；
     * 3. 出库任务按 work_mast.user_no -> work_mast.batch_seq 分层汇总，并按 batchSeq 自然升序下发。
     *
     * @author Ryan
     * @date 2026/1/10 14:42
     */
    @Scheduled(cron = "0/10 * * * * ? ")
    private synchronized void autoPubTasks() {
        // 仅处理待下发/已生成下发号的工作档。
        List<WrkMast> wrkMasts = wrkMastService.selectList(new EntityWrapper<WrkMast>().in("wrk_sts", Arrays.asList(1L, 11L))
                .orderBy("user_no", true)
                .orderBy("batch_seq", true)
                .orderBy("plt_type", true));
        if (wrkMasts.isEmpty()) {
            return;
        }

        List<WorkTaskParams> paramsList = new ArrayList<>();
        List<WorkTaskParams> moveParamsList = new ArrayList<>();
        Map<String, LinkedHashMap<String, List<WrkMast>>> outboundTasksByUserNo = new LinkedHashMap<>();
        for (WrkMast wrkMast : wrkMasts) {
            // 出库类任务（ioType > 100）默认需要 ERP 确认；未确认的任务在这里直接跳过。
            if (wrkMast.getIoType() > 100 && !"Y".equalsIgnoreCase(wrkMast.getPdcType())) {
                continue;
            }

            if (isOutboundPublishTask(wrkMast)) {
                if (Cools.isEmpty(wrkMast.getBatchSeq())) {
                    log.warn("出库进仓编号(batchSeq)为空，跳过下发, wrkNo={}, userNo={}",
                            wrkMast.getWrkNo(), wrkMast.getUserNo());
                    continue;
                }
                String userNo = normalizeGroupKey(wrkMast.getUserNo());
                String batchSeq = normalizeGroupKey(wrkMast.getBatchSeq());
                outboundTasksByUserNo
                        .computeIfAbsent(userNo, key -> new LinkedHashMap<>())
                        .computeIfAbsent(batchSeq, key -> new ArrayList<>())
                        .add(wrkMast);
            } else {
                WorkTaskParams params = buildWorkTaskParams(wrkMast);
                if (isMovePublishTask(params)) {
                    moveParamsList.add(params);
                } else {
                    paramsList.add(params);
                }
            }
        }

        if (publishTaskChunks(paramsList)) {
            return;
        }

        if (publishMoveTasksOneByOne(moveParamsList)) {
            return;
        }

        if (outboundTasksByUserNo.isEmpty()) {
            return;
        }

        for (Map.Entry<String, LinkedHashMap<String, List<WrkMast>>> userEntry : outboundTasksByUserNo.entrySet()) {
            String userNo = userEntry.getKey();
            List<String> batchSeqs = new ArrayList<>(userEntry.getValue().keySet());
            batchSeqs.sort(this::compareBatchSeqNatural);

            for (String batchSeq : batchSeqs) {
                String blockMsg = outboundBatchSeqReleaseGuard.validateReady(userNo, batchSeq);
                if (!Cools.isEmpty(blockMsg)) {
                    log.info(blockMsg);
                    break;
                }

                List<WrkMast> batchMasts = selectOutboundBatchMasts(userEntry.getValue().get(batchSeq));
                if (batchMasts == null || batchMasts.isEmpty()) {
                    continue;
                }
                List<WorkTaskParams> batchParams = buildWorkTaskParams(batchMasts);

                if (publishOutboundTaskChunks(userNo, batchSeq, batchParams)) {
                    return;
                }
            }
        }
    }

    private List<WorkTaskParams> buildWorkTaskParams(List<WrkMast> wrkMasts) {
        List<WorkTaskParams> paramsList = new ArrayList<>();
        if (wrkMasts == null || wrkMasts.isEmpty()) {
            return paramsList;
        }
        for (WrkMast wrkMast : wrkMasts) {
            paramsList.add(buildWorkTaskParams(wrkMast));
        }
        return paramsList;
    }

    private WorkTaskParams buildWorkTaskParams(WrkMast wrkMast) {
        // WMS 库位编码转换成 WCS 可识别的库位编码。
        String wcsSourceLocNo = Cools.isEmpty(wrkMast.getSourceLocNo()) ? "" : Utils.WMSLocToWCSLoc(wrkMast.getSourceLocNo());
        String wcsLocNo = Cools.isEmpty(wrkMast.getLocNo()) ? "" : Utils.WMSLocToWCSLoc(wrkMast.getLocNo());
        WorkTaskParams params = new WorkTaskParams();

        // 101: 出库。有序任务才向 WCS 传 batch/batchSeq；seq=0 表示无序，不传这两个字段。
        if (wrkMast.getIoType() == 101) {
            params.setType("out")
                    .setTaskNo(wrkMast.getWrkNo() + "")
                    .setLocNo(wcsSourceLocNo)
                    .setStaNo(String.valueOf(wrkMast.getStaNo()))
                    .setTaskPri(wrkMast.getIoPri().intValue())
                    .setBarcode(wrkMast.getBarcode());
            if (wrkMast.getPltType() != null && wrkMast.getPltType() > 0) {
                params.setBatch(wrkMast.getBatchSeq())
                        .setBatchSeq(wrkMast.getPltType());
            }
        // 2: 入库。入库接口使用 sourceStaNo + 目标库位。
        } else if (wrkMast.getIoType() == 2 && !Cools.isEmpty(wrkMast.getSourceStaNo())) {
            params.setType("in")
                    .setTaskNo(wrkMast.getWrkNo() + "")
                    .setSourceStaNo(String.valueOf(wrkMast.getSourceStaNo()))
                    .setLocNo(wcsLocNo)
                    .setTaskPri(wrkMast.getIoPri().intValue())
                    .setBarcode(wrkMast.getBarcode());
        // 其余走移库接口，源库位和目标库位都需要带给 WCS。
        } else {
            params.setType("move")
                    .setTaskNo(wrkMast.getWrkNo() + "")
                    .setSourceLocNo(wcsSourceLocNo)
                    .setLocNo(wcsLocNo)
                    .setBarcode(wrkMast.getBarcode());
        }
        return params;
    }

    private boolean isOutboundPublishTask(WrkMast wrkMast) {
        return wrkMast != null && Objects.equals(wrkMast.getIoType(), 101);
    }

    private boolean isMovePublishTask(WorkTaskParams params) {
        return params != null && "move".equalsIgnoreCase(params.getType());
    }

    private List<WrkMast> selectOutboundBatchMasts(List<WrkMast> batchMasts) {
        if (batchMasts == null || batchMasts.isEmpty()) {
            return new ArrayList<>();
        }
        List<WrkMast> ordered = new ArrayList<>(batchMasts);
        if (hasOrderedOutboundSeq(ordered)) {
            ordered.sort(this::compareOutboundSeq);
            return ordered;
        }
        if (!hasDoubleExtensionOutboundTask(ordered)) {
            return ordered;
        }

        List<WrkMast> shallowMasts = new ArrayList<>();
        List<WrkMast> deepMasts = new ArrayList<>();
        List<WrkMast> otherMasts = new ArrayList<>();
        for (WrkMast mast : ordered) {
            if (!isDoubleExtensionOutboundTask(mast)) {
                otherMasts.add(mast);
                continue;
            }
            LocDepthType locDepthType = resolveOutboundLocDepth(mast);
            if (locDepthType == LocDepthType.SHALLOW) {
                shallowMasts.add(mast);
            } else if (locDepthType == LocDepthType.DEEP) {
                deepMasts.add(mast);
            } else {
                otherMasts.add(mast);
            }
        }
        if (shallowMasts.isEmpty() && deepMasts.isEmpty()) {
            return ordered;
        }

        List<WrkMast> result = new ArrayList<>(ordered.size());
        result.addAll(shallowMasts);
        result.addAll(deepMasts);
        result.addAll(otherMasts);
        return result;
    }

    private boolean hasOrderedOutboundSeq(List<WrkMast> batchMasts) {
        for (WrkMast mast : batchMasts) {
            if (outboundSeq(mast) > 0) {
                return true;
            }
        }
        return false;
    }

    private int compareOutboundSeq(WrkMast left, WrkMast right) {
        int leftSeq = outboundSeq(left);
        int rightSeq = outboundSeq(right);
        boolean leftOrdered = leftSeq > 0;
        boolean rightOrdered = rightSeq > 0;
        if (leftOrdered && rightOrdered) {
            return Integer.compare(leftSeq, rightSeq);
        }
        if (leftOrdered) {
            return -1;
        }
        if (rightOrdered) {
            return 1;
        }
        return 0;
    }

    private int outboundSeq(WrkMast mast) {
        return mast == null || mast.getPltType() == null ? 0 : mast.getPltType();
    }

    private boolean hasDoubleExtensionOutboundTask(List<WrkMast> batchMasts) {
        for (WrkMast mast : batchMasts) {
            if (isDoubleExtensionOutboundTask(mast)) {
                return true;
            }
        }
        return false;
    }

    private boolean isDoubleExtensionOutboundTask(WrkMast mast) {
        return isOutboundPublishTask(mast)
                && mast.getCrnNo() != null
                && mast.getCrnNo() >= DOUBLE_EXTENSION_OUTBOUND_CRN_NO;
    }

    private LocDepthType resolveOutboundLocDepth(WrkMast mast) {
        if (mast == null || Cools.isEmpty(mast.getSourceLocNo()) || slaveProperties == null) {
            return LocDepthType.UNKNOWN;
        }
        try {
            if (Utils.isShallowLoc(slaveProperties, mast.getSourceLocNo())) {
                return LocDepthType.SHALLOW;
            }
            if (Utils.isDeepLoc(slaveProperties, mast.getSourceLocNo())) {
                return LocDepthType.DEEP;
            }
        } catch (Exception e) {
            log.warn("双伸出库库位深浅判断失败, wrkNo={}, sourceLocNo={}", mast.getWrkNo(), mast.getSourceLocNo(), e);
        }
        return LocDepthType.UNKNOWN;
    }

    private int compareBatchSeqNatural(String left, String right) {
        String safeLeft = Cools.isEmpty(left) ? "" : left;
        String safeRight = Cools.isEmpty(right) ? "" : right;
        int leftIndex = 0;
        int rightIndex = 0;
        while (leftIndex < safeLeft.length() && rightIndex < safeRight.length()) {
            char leftChar = safeLeft.charAt(leftIndex);
            char rightChar = safeRight.charAt(rightIndex);
            if (Character.isDigit(leftChar) && Character.isDigit(rightChar)) {
                int leftStart = leftIndex;
                int rightStart = rightIndex;
                while (leftIndex < safeLeft.length() && Character.isDigit(safeLeft.charAt(leftIndex))) {
                    leftIndex++;
                }
                while (rightIndex < safeRight.length() && Character.isDigit(safeRight.charAt(rightIndex))) {
                    rightIndex++;
                }
                String leftNumber = safeLeft.substring(leftStart, leftIndex);
                String rightNumber = safeRight.substring(rightStart, rightIndex);
                int compare = new BigInteger(leftNumber).compareTo(new BigInteger(rightNumber));
                if (compare != 0) {
                    return compare;
                }
                compare = Integer.compare(leftNumber.length(), rightNumber.length());
                if (compare != 0) {
                    return compare;
                }
                continue;
            }
            int compare = Character.compare(leftChar, rightChar);
            if (compare != 0) {
                return compare;
            }
            leftIndex++;
            rightIndex++;
        }
        return Integer.compare(safeLeft.length(), safeRight.length());
    }

    private String normalizeGroupKey(String value) {
        return Cools.isEmpty(value) ? "" : value;
    }

    private boolean publishTaskChunks(List<WorkTaskParams> paramsList) {
        if (paramsList == null || paramsList.isEmpty()) {
            return false;
        }
        for (int start = 0; start < paramsList.size(); start += MAX_PUBLISH_TASKS_ONCE) {
            int end = Math.min(start + MAX_PUBLISH_TASKS_ONCE, paramsList.size());
            List<WorkTaskParams> chunk = paramsList.subList(start, end);
            R r = wcsApiService.pubWrksToWcs(chunk);
            if (isWcsSuccess(r)) {
                return true;
            }
            log.warn("批量下发任务到WCS失败, start={}, size={}, result={}", start, chunk.size(), r);
        }
        return false;
    }

    private boolean publishMoveTasksOneByOne(List<WorkTaskParams> moveParamsList) {
        if (moveParamsList == null || moveParamsList.isEmpty()) {
            return false;
        }
        for (WorkTaskParams params : moveParamsList) {
            R r = wcsApiService.pubWrkToWcs(params);
            if (isWcsSuccess(r)) {
                return true;
            }
            log.warn("移库任务下发到WCS失败, taskNo={}, result={}",
                    params == null ? null : params.getTaskNo(), r);
        }
        return false;
    }

    private boolean publishOutboundTaskChunks(String userNo, String batchSeq, List<WorkTaskParams> batchParams) {
        if (batchParams == null || batchParams.isEmpty()) {
            return false;
        }
        for (int start = 0; start < batchParams.size(); start += MAX_PUBLISH_TASKS_ONCE) {
            int end = Math.min(start + MAX_PUBLISH_TASKS_ONCE, batchParams.size());
            List<WorkTaskParams> chunk = batchParams.subList(start, end);
            String lockKey = WorkPublishLockKeys.outboundUserNoLock(userNo);
            String lockValue = String.valueOf(System.currentTimeMillis());
            if (!redisUtil.setIfAbsent(lockKey, lockValue, OUTBOUND_USER_NO_LOCK_SECONDS)) {
                log.info("出库任务正在下发，跳过本轮, userNo={}, batchSeq={}, lockKey={}", userNo, batchSeq, lockKey);
                return false;
            }
            try {
                R r = wcsApiService.pubWrksToWcs(chunk);
                if (isWcsSuccess(r)) {
                    return true;
                }
                log.warn("批量下发出库任务到WCS失败, userNo={}, batchSeq={}, start={}, size={}, result={}",
                        userNo, batchSeq, start, chunk.size(), r);
            } finally {
                Object currentLockValue = redisUtil.get(lockKey);
                if (Objects.equals(currentLockValue, lockValue)) {
                    redisUtil.del(lockKey);
                }
            }
        }
        return false;
    }

    private boolean isWcsSuccess(R r) {
        return r != null && Objects.equals(r.get("code"), 200);
    }

}