package cn.iocoder.yudao.module.mes.dal.mysql.wm.batch;


import cn.hutool.core.util.ObjUtil;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.module.mes.controller.admin.wm.batch.vo.MesWmBatchPageReqVO;
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.batch.MesWmBatchDO;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import org.apache.ibatis.annotations.Mapper;

import java.util.List;

/**
 * 批次管理 Mapper
 *
 * @author 芋道源码
 */
@Mapper
public interface MesWmBatchMapper extends BaseMapperX<MesWmBatchDO> {

    default MesWmBatchDO selectByCode(String code) {
        return selectOne(MesWmBatchDO:: getCode, code);
    }

    default Long selectCountByToolId(Long toolId) {
        return selectCount(MesWmBatchDO::getToolId, toolId);
    }

    default PageResult<MesWmBatchDO> selectPage(MesWmBatchPageReqVO reqVO) {
        return selectPage(reqVO, new LambdaQueryWrapperX<MesWmBatchDO>()
                .likeIfPresent(MesWmBatchDO::getCode, reqVO.getCode())
                .eqIfPresent(MesWmBatchDO::getItemId, reqVO.getItemId())
                .eqIfPresent(MesWmBatchDO::getVendorId, reqVO.getVendorId())
                .eqIfPresent(MesWmBatchDO::getClientId, reqVO.getClientId())
                .likeIfPresent(MesWmBatchDO::getSalesOrderCode, reqVO.getSalesOrderCode())
                .likeIfPresent(MesWmBatchDO::getPurchaseOrderCode, reqVO.getPurchaseOrderCode())
                .eqIfPresent(MesWmBatchDO::getWorkOrderId, reqVO.getWorkOrderId())
                .eqIfPresent(MesWmBatchDO::getTaskId, reqVO.getTaskId())
                .eqIfPresent(MesWmBatchDO::getWorkstationId, reqVO.getWorkstationId())
                .eqIfPresent(MesWmBatchDO::getToolId, reqVO.getToolId())
                .eqIfPresent(MesWmBatchDO::getMoldId, reqVO.getMoldId())
                .likeIfPresent(MesWmBatchDO::getLotNumber, reqVO.getLotNumber())
                .eqIfPresent(MesWmBatchDO::getQualityStatus, reqVO.getQualityStatus())
                .betweenIfPresent(MesWmBatchDO::getProduceDate, reqVO.getProduceDate())
                .betweenIfPresent(MesWmBatchDO::getExpireDate, reqVO.getExpireDate())
                .betweenIfPresent(MesWmBatchDO::getReceiptDate, reqVO.getReceiptDate())
                .orderByDesc(MesWmBatchDO::getId));
    }

    /**
     * 根据参数查询匹配的第一条批次记录
     * <p>
     * 使用 NULL 值精确匹配，返回 ID 最小的批次
     *
     * @param batch 批次参数
     * @return 匹配的批次记录
     */
    default MesWmBatchDO selectFirst(MesWmBatchDO batch) {
        LambdaQueryWrapper<MesWmBatchDO> query = new LambdaQueryWrapper<>();
        query.eq(MesWmBatchDO::getItemId, batch.getItemId());
        if (ObjUtil.isNull(batch.getVendorId())) {
            query.isNull(MesWmBatchDO::getVendorId);
        } else {
            query.eq(MesWmBatchDO::getVendorId, batch.getVendorId());
        }
        if (ObjUtil.isNull(batch.getClientId())) {
            query.isNull(MesWmBatchDO::getClientId);
        } else {
            query.eq(MesWmBatchDO::getClientId, batch.getClientId());
        }
        if (ObjUtil.isNull(batch.getSalesOrderCode())) {
            query.isNull(MesWmBatchDO::getSalesOrderCode);
        } else {
            query.eq(MesWmBatchDO::getSalesOrderCode, batch.getSalesOrderCode());
        }
        if (ObjUtil.isNull(batch.getPurchaseOrderCode())) {
            query.isNull(MesWmBatchDO::getPurchaseOrderCode);
        } else {
            query.eq(MesWmBatchDO::getPurchaseOrderCode, batch.getPurchaseOrderCode());
        }
        if (ObjUtil.isNull(batch.getWorkOrderId())) {
            query.isNull(MesWmBatchDO::getWorkOrderId);
        } else {
            query.eq(MesWmBatchDO::getWorkOrderId, batch.getWorkOrderId());
        }
        if (ObjUtil.isNull(batch.getTaskId())) {
            query.isNull(MesWmBatchDO::getTaskId);
        } else {
            query.eq(MesWmBatchDO::getTaskId, batch.getTaskId());
        }
        if (ObjUtil.isNull(batch.getWorkstationId())) {
            query.isNull(MesWmBatchDO::getWorkstationId);
        } else {
            query.eq(MesWmBatchDO::getWorkstationId, batch.getWorkstationId());
        }
        if (ObjUtil.isNull(batch.getToolId())) {
            query.isNull(MesWmBatchDO::getToolId);
        } else {
            query.eq(MesWmBatchDO::getToolId, batch.getToolId());
        }
        if (ObjUtil.isNull(batch.getMoldId())) {
            query.isNull(MesWmBatchDO::getMoldId);
        } else {
            query.eq(MesWmBatchDO::getMoldId, batch.getMoldId());
        }
        if (ObjUtil.isNull(batch.getLotNumber())) {
            query.isNull(MesWmBatchDO::getLotNumber);
        } else {
            query.eq(MesWmBatchDO::getLotNumber, batch.getLotNumber());
        }
        if (ObjUtil.isNull(batch.getQualityStatus())) {
            query.isNull(MesWmBatchDO::getQualityStatus);
        } else {
            query.eq(MesWmBatchDO::getQualityStatus, batch.getQualityStatus());
        }
        if (ObjUtil.isNull(batch.getProduceDate())) {
            query.isNull(MesWmBatchDO::getProduceDate);
        } else {
            query.eq(MesWmBatchDO::getProduceDate, batch.getProduceDate());
        }
        if (ObjUtil.isNull(batch.getExpireDate())) {
            query.isNull(MesWmBatchDO::getExpireDate);
        } else {
            query.eq(MesWmBatchDO::getExpireDate, batch.getExpireDate());
        }
        if (ObjUtil.isNull(batch.getReceiptDate())) {
            query.isNull(MesWmBatchDO::getReceiptDate);
        } else {
            query.eq(MesWmBatchDO::getReceiptDate, batch.getReceiptDate());
        }

        // 返回 ID 最小的批次
        query.orderByAsc(MesWmBatchDO::getId);
        query.last("LIMIT 1");
        return selectOne(query);
    }

    /**
     * 查询向前追溯批次列表
     * <p>
     * 查询当前批次被哪些工单的哪些批次产品消耗
     * SQL 逻辑：从领料明细 -> 领料单 -> 报工记录 -> 生产入库单 -> 生产入库行
     */
    List<MesWmBatchDO> selectListByForward(String batchCode);

    /**
     * 查询向后追溯批次列表
     * <p>
     * 查询当前批次的产品使用了哪些批次的物资
     * SQL 逻辑：从生产入库明细 -> 生产入库单 -> 领料单 -> 领料明细
     */
    List<MesWmBatchDO> selectListByBackward(String batchCode);

}