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

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.equipmentreceipt.vo.MesWmEquipmentReceiptPageReqVO;
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.equipmentreceipt.MesWmEquipmentReceiptDO;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import org.apache.ibatis.annotations.Mapper;

import java.time.LocalDateTime;

/**
 * MES 设备入库单 Mapper
 */
@Mapper
public interface MesWmEquipmentReceiptMapper extends BaseMapperX<MesWmEquipmentReceiptDO> {

    default PageResult<MesWmEquipmentReceiptDO> selectPage(MesWmEquipmentReceiptPageReqVO reqVO) {
        return selectPage(reqVO, new LambdaQueryWrapperX<MesWmEquipmentReceiptDO>()
                .likeIfPresent(MesWmEquipmentReceiptDO::getCode, reqVO.getCode())
                .likeIfPresent(MesWmEquipmentReceiptDO::getName, reqVO.getName())
                .likeIfPresent(MesWmEquipmentReceiptDO::getSource, reqVO.getSource())
                .eqIfPresent(MesWmEquipmentReceiptDO::getStatus, reqVO.getStatus())
                .betweenIfPresent(MesWmEquipmentReceiptDO::getReceiptDate, reqVO.getReceiptDate())
                .orderByDesc(MesWmEquipmentReceiptDO::getId));
    }

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

    /**
     * 提交审批：置为审批中，并清空上一轮审核结果
     *
     * 用 UpdateWrapper 而非 updateById：MyBatis-Plus 默认更新策略忽略 null，
     * 直接 updateById 无法把审核不通过后残留的审核人 / 审核时间 / 审核意见清空。
     *
     * @param id     编号
     * @param status 目标状态（审批中）
     * @return 影响行数
     */
    default int submitAndResetAuditInfo(Long id, Integer status) {
        return update(null, new LambdaUpdateWrapper<MesWmEquipmentReceiptDO>()
                .eq(MesWmEquipmentReceiptDO::getId, id)
                .set(MesWmEquipmentReceiptDO::getStatus, status)
                .set(MesWmEquipmentReceiptDO::getReviewerId, null)
                .set(MesWmEquipmentReceiptDO::getReviewerName, null)
                .set(MesWmEquipmentReceiptDO::getReviewTime, null)
                .set(MesWmEquipmentReceiptDO::getReviewRemark, null));
    }

    /**
     * 审核：写入审核结果、审核人与审核意见
     *
     * @param id           编号
     * @param status       目标状态（已审批 / 草稿）
     * @param reviewerId   审核人编号
     * @param reviewerName 审核人姓名
     * @param reviewRemark 审核意见
     * @return 影响行数
     */
    default int auditEquipmentReceipt(Long id, Integer status, Long reviewerId,
                                      String reviewerName, String reviewRemark) {
        return update(null, new LambdaUpdateWrapper<MesWmEquipmentReceiptDO>()
                .eq(MesWmEquipmentReceiptDO::getId, id)
                .set(MesWmEquipmentReceiptDO::getStatus, status)
                .set(MesWmEquipmentReceiptDO::getReviewerId, reviewerId)
                .set(MesWmEquipmentReceiptDO::getReviewerName, reviewerName)
                .set(MesWmEquipmentReceiptDO::getReviewTime, LocalDateTime.now())
                .set(MesWmEquipmentReceiptDO::getReviewRemark, reviewRemark));
    }

}