package cn.iocoder.yudao.module.hrm.dal.mysql.approval;

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 com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import cn.iocoder.yudao.module.hrm.controller.admin.approval.vo.HrmLeaveApplicationPageReqVO;
import cn.iocoder.yudao.module.hrm.dal.dataobject.approval.HrmLeaveApplicationDO;
import cn.iocoder.yudao.module.hrm.enums.HrmAuditStatusEnum;
import org.apache.ibatis.annotations.Mapper;

import java.time.LocalDateTime;
import java.util.List;

/**
 * 请假申请 Mapper
 *
 * @author 超级管理员
 */
@Mapper
public interface HrmLeaveApplicationMapper extends BaseMapperX<HrmLeaveApplicationDO> {

    default PageResult<HrmLeaveApplicationDO> selectPage(HrmLeaveApplicationPageReqVO reqVO) {
        return selectPage(reqVO, new LambdaQueryWrapperX<HrmLeaveApplicationDO>()
                .likeIfPresent(HrmLeaveApplicationDO::getNo, reqVO.getNo())
                .eqIfPresent(HrmLeaveApplicationDO::getUserId, reqVO.getUserId())
                .eqIfPresent(HrmLeaveApplicationDO::getDeptId, reqVO.getDeptId())
                .eqIfPresent(HrmLeaveApplicationDO::getLeaveType, reqVO.getLeaveType())
                .eqIfPresent(HrmLeaveApplicationDO::getStatus, reqVO.getStatus())
                .betweenIfPresent(HrmLeaveApplicationDO::getCreateTime, reqVO.getCreateTime())
                .orderByDesc(HrmLeaveApplicationDO::getId));
    }

    default HrmLeaveApplicationDO selectByNo(String no) {
        return selectOne(HrmLeaveApplicationDO::getNo, no);
    }

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

    /**
     * 审核：写入审核结果、审核人与审核意见
     *
     * reviewRemark 允许为 null（审核通过时意见可选），故同样使用 UpdateWrapper 直写。
     *
     * @param id           申请ID
     * @param status       目标状态（审核通过 / 审核不通过）
     * @param reviewerId   审核人编号
     * @param reviewerName 审核人姓名
     * @param reviewRemark 审核意见
     * @return 影响行数
     */
    default int auditLeaveApplication(Long id, Integer status, Long reviewerId,
                                      String reviewerName, String reviewRemark) {
        return update(null, new LambdaUpdateWrapper<HrmLeaveApplicationDO>()
                .eq(HrmLeaveApplicationDO::getId, id)
                .set(HrmLeaveApplicationDO::getStatus, status)
                .set(HrmLeaveApplicationDO::getReviewerId, reviewerId)
                .set(HrmLeaveApplicationDO::getReviewerName, reviewerName)
                .set(HrmLeaveApplicationDO::getReviewTime, LocalDateTime.now())
                .set(HrmLeaveApplicationDO::getReviewRemark, reviewRemark));
    }

    /**
     * 查询员工在指定时间范围内已审批通过的请假记录
     *
     * @param userId    员工ID（系统用户ID）
     * @param startTime 开始时间
     * @param endTime   结束时间
     * @return 请假记录列表
     */
    default List<HrmLeaveApplicationDO> selectApprovedListByUserIdAndTimeRange(Long userId, LocalDateTime startTime, LocalDateTime endTime) {
        return selectList(new LambdaQueryWrapperX<HrmLeaveApplicationDO>()
                .eq(HrmLeaveApplicationDO::getUserId, userId)
                .eq(HrmLeaveApplicationDO::getStatus, HrmAuditStatusEnum.APPROVE.getStatus())
                .ge(HrmLeaveApplicationDO::getStartTime, startTime)
                .le(HrmLeaveApplicationDO::getEndTime, endTime));
    }

}