package cn.iocoder.yudao.module.hrm.service.performanceappraise;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONUtil;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.module.hrm.controller.admin.performanceappraise.vo.*;
import cn.iocoder.yudao.module.hrm.dal.dataobject.employee.HrmEmployeeDO;
import cn.iocoder.yudao.module.hrm.dal.dataobject.performanceappraise.*;
import cn.iocoder.yudao.module.hrm.dal.mysql.employee.HrmEmployeeMapper;
import cn.iocoder.yudao.module.hrm.dal.mysql.performanceappraise.*;
import cn.iocoder.yudao.module.hrm.enums.HrmAuditStatusEnum;
import cn.iocoder.yudao.module.hrm.enums.ErrorCodeConstants;
import cn.iocoder.yudao.module.system.api.approval.ApprovalConfigApi;
import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.validation.annotation.Validated;

import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.*;
import java.util.stream.Collectors;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertMap;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertSet;
import static cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId;

/**
 * HRM 绩效考核方案 Service 实现类
 */
@Service
@Validated
@Slf4j
public class HrmPerformanceAppraiseServiceImpl implements HrmPerformanceAppraiseService {

    /**
     * 方案状态：0 草稿 / 10 已发布 / 20 进行中 / 30 已完成
     */
    private static final Integer STATUS_DRAFT = 0;
    private static final Integer STATUS_PUBLISHED = 10;
    private static final Integer STATUS_PROCESSING = 20;

    /**
     * 档级：1 优秀(≥90) / 2 良好(≥80) / 3 合格(≥60) / 4 待改进(<60)
     */
    private static final BigDecimal GRADE_EXCELLENT = new BigDecimal("90");
    private static final BigDecimal GRADE_GOOD = new BigDecimal("80");
    private static final BigDecimal GRADE_PASS = new BigDecimal("60");
    private static final Integer GRADE_EXCELLENT_CODE = 1;
    private static final Integer GRADE_GOOD_CODE = 2;
    private static final Integer GRADE_PASS_CODE = 3;
    private static final Integer GRADE_IMPROVE_CODE = 4;

    /**
     * 评分人类型：1 自评 / 2 上级评
     */
    private static final Integer SCORER_SELF = 1;
    private static final Integer SCORER_SUPERIOR = 2;

    /**
     * 考核范围类型：1 全部员工 / 2 指定班组
     */
    private static final Integer SCOPE_ALL = 1;
    private static final Integer SCOPE_TEAM = 2;

    /**
     * 在职员工状态
     */
    private static final Integer EMPLOYEE_STATUS_ACTIVE = 1;

    /**
     * 绩效考核方案审批的业务类型编码
     *
     * 与 system_approval_config.biz_type 对应，审批人在「系统管理 - 审批配置」中维护。
     */
    private static final String APPRAISE_APPROVE_BIZ_TYPE = "performance_appraise_approve";

    @Resource
    private HrmPerformanceAppraiseMapper appraiseMapper;
    @Resource
    private HrmPerformanceAppraiseItemMapper itemMapper;
    @Resource
    private HrmPerformanceAppraiseResultMapper resultMapper;
    @Resource
    private HrmPerformanceAppraiseResultItemMapper resultItemMapper;
    @Resource
    private HrmEmployeeMapper employeeMapper;
    @Resource
    private ApprovalConfigApi approvalConfigApi;
    @Resource
    private AdminUserApi adminUserApi;

    @Override
    @Transactional(rollbackFor = Exception.class)
    public Long createAppraise(HrmPerformanceAppraiseSaveReqVO createReqVO) {
        HrmPerformanceAppraiseDO appraise = BeanUtils.toBean(createReqVO, HrmPerformanceAppraiseDO.class, req -> req
                .setStatus(STATUS_DRAFT)
                .setAuditStatus(HrmAuditStatusEnum.DRAFT.getStatus()));
        appraise.setScopeTeamIds(CollUtil.isEmpty(createReqVO.getScopeTeamIds())
                ? null : JSONUtil.toJsonStr(createReqVO.getScopeTeamIds()));
        appraiseMapper.insert(appraise);
        return appraise.getId();
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void updateAppraise(HrmPerformanceAppraiseSaveReqVO updateReqVO) {
        HrmPerformanceAppraiseDO oldAppraise = validateAppraiseCanModify(updateReqVO.getId());
        HrmPerformanceAppraiseDO updateObj = BeanUtils.toBean(updateReqVO, HrmPerformanceAppraiseDO.class, req -> req
                .setStatus(oldAppraise.getStatus())
                .setAuditStatus(oldAppraise.getAuditStatus()));
        updateObj.setScopeTeamIds(CollUtil.isEmpty(updateReqVO.getScopeTeamIds())
                ? null : JSONUtil.toJsonStr(updateReqVO.getScopeTeamIds()));
        appraiseMapper.updateById(updateObj);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void deleteAppraise(Collection<Long> ids) {
        if (CollUtil.isEmpty(ids)) {
            return;
        }
        for (Long id : ids) {
            validateAppraiseCanModify(id);
            // 删除指标与结果
            itemMapper.deleteByAppraiseId(id);
            List<HrmPerformanceAppraiseResultDO> results = resultMapper.selectList(
                    new LambdaQueryWrapperX<HrmPerformanceAppraiseResultDO>()
                            .eq(HrmPerformanceAppraiseResultDO::getAppraiseId, id));
            for (HrmPerformanceAppraiseResultDO result : results) {
                resultItemMapper.deleteByResultId(result.getId());
            }
            resultMapper.deleteByAppraiseId(id);
            appraiseMapper.deleteById(id);
        }
    }

    @Override
    public HrmPerformanceAppraiseDO getAppraise(Long id) {
        return appraiseMapper.selectById(id);
    }

    @Override
    public PageResult<HrmPerformanceAppraiseDO> getAppraisePage(HrmPerformanceAppraisePageReqVO pageReqVO) {
        return appraiseMapper.selectPage(pageReqVO);
    }

    @Override
    public List<HrmPerformanceAppraiseItemDO> getAppraiseItemList(Long appraiseId) {
        return itemMapper.selectListByAppraiseId(appraiseId);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void saveAppraiseItems(HrmPerformanceAppraiseItemSaveReqVO saveReqVO) {
        HrmPerformanceAppraiseDO appraise = validateAppraiseCanModify(saveReqVO.getAppraiseId());
        // 整表替换
        itemMapper.deleteByAppraiseId(appraise.getId());
        if (CollUtil.isEmpty(saveReqVO.getItems())) {
            return;
        }
        int sort = 0;
        for (HrmPerformanceAppraiseItemSaveReqVO.Item item : saveReqVO.getItems()) {
            Integer itemSort = item.getSort() != null ? item.getSort() : sort++;
            HrmPerformanceAppraiseItemDO itemDO = BeanUtils.toBean(item, HrmPerformanceAppraiseItemDO.class, req -> req
                    .setAppraiseId(appraise.getId())
                    .setSort(itemSort));
            itemMapper.insert(itemDO);
        }
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void submitAppraise(Long id) {
        // 1. 校验方案存在 + 可提交状态（草稿 / 审核不通过）
        HrmPerformanceAppraiseDO appraise = appraiseMapper.selectById(id);
        if (appraise == null) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_NOT_EXISTS);
        }
        if (ObjUtil.notEqual(appraise.getAuditStatus(), HrmAuditStatusEnum.DRAFT.getStatus())
                && ObjUtil.notEqual(appraise.getAuditStatus(), HrmAuditStatusEnum.REJECT.getStatus())) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_SUBMIT_FAIL_NOT_EDITABLE);
        }
        if (CollUtil.isEmpty(itemMapper.selectListByAppraiseId(id))) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_ITEM_EMPTY);
        }

        // 2. 校验审批配置已启用且配置了有效审批人（未配置时抛出带明确提示的业务异常）
        approvalConfigApi.validateApprovalEnabledAndGetApprovers(APPRAISE_APPROVE_BIZ_TYPE);

        // 3. 审批状态置为审批中，并清空上一轮审核结果，避免残留「审核不通过」的原因
        appraiseMapper.submitAndResetAuditInfo(id, HrmAuditStatusEnum.PROCESS.getStatus());
    }

    @Override
    public Set<Long> getAppraiseApproverUserIds() {
        return approvalConfigApi.getApproverUserIds(APPRAISE_APPROVE_BIZ_TYPE);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void auditAppraise(Long id, Boolean pass, String reviewRemark) {
        // 1. 校验存在
        HrmPerformanceAppraiseDO appraise = validateAppraiseExists(id);
        // 2. 校验处于审批中状态
        if (ObjUtil.notEqual(appraise.getAuditStatus(), HrmAuditStatusEnum.PROCESS.getStatus())) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_AUDIT_FAIL_NOT_PROCESS);
        }
        // 3. 校验当前登录用户是该业务类型的审批人（或签：任一人均可审核）
        Long userId = getLoginUserId();
        approvalConfigApi.validateApprover(APPRAISE_APPROVE_BIZ_TYPE, userId);
        // 4. 审核不通过：必须填写原因，审批状态置为审核不通过，退回提交人修改后可重新提交
        if (!Boolean.TRUE.equals(pass)) {
            if (StrUtil.isBlank(reviewRemark)) {
                throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_AUDIT_REJECT_REASON_REQUIRED);
            }
            // 方案状态保持不变
            appraiseMapper.auditAppraise(id, HrmAuditStatusEnum.REJECT.getStatus(), appraise.getStatus(),
                    userId, getUserNickname(userId), reviewRemark);
            return;
        }
        // 5. 审核通过：记录审核人与审核意见，审批状态置为审核通过
        // 6. 审核通过后置业务逻辑：方案状态流转为「已发布」
        appraiseMapper.auditAppraise(id, HrmAuditStatusEnum.APPROVE.getStatus(), STATUS_PUBLISHED,
                userId, getUserNickname(userId), reviewRemark);
    }

    /**
     * 获取用户昵称
     *
     * @param userId 用户编号
     * @return 昵称，用户不存在时返回 null
     */
    private String getUserNickname(Long userId) {
        if (userId == null) {
            return null;
        }
        AdminUserRespDTO user = adminUserApi.getUser(userId);
        return user == null ? null : user.getNickname();
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void saveAppraiseResult(HrmPerformanceAppraiseResultSaveReqVO saveReqVO) {
        // 1. 校验方案状态允许评分
        HrmPerformanceAppraiseDO appraise = appraiseMapper.selectById(saveReqVO.getAppraiseId());
        if (appraise == null) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_NOT_EXISTS);
        }
        if (!Objects.equals(appraise.getStatus(), STATUS_PUBLISHED)
                && !Objects.equals(appraise.getStatus(), STATUS_PROCESSING)) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_SCORE_FAIL_STATUS);
        }

        // 2. 加载指标集合，校验指标评分是否完整
        List<HrmPerformanceAppraiseItemDO> items = itemMapper.selectListByAppraiseId(appraise.getId());
        if (CollUtil.isEmpty(items)) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_ITEM_EMPTY);
        }
        Map<Long, HrmPerformanceAppraiseItemDO> itemMap = convertMap(items, HrmPerformanceAppraiseItemDO::getId);
        if (CollUtil.isEmpty(saveReqVO.getItems()) || saveReqVO.getItems().size() != items.size()) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_RESULT_ITEM_NOT_EXISTS);
        }
        for (HrmPerformanceAppraiseResultSaveReqVO.Item score : saveReqVO.getItems()) {
            if (!itemMap.containsKey(score.getItemId())) {
                throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_RESULT_ITEM_NOT_EXISTS);
            }
        }

        // 3. 按员工 upsert 结果
        Long resultId;
        HrmPerformanceAppraiseResultDO oldResult = resultMapper.selectByAppraiseIdAndEmployeeId(appraise.getId(), saveReqVO.getEmployeeId());
        if (oldResult != null) {
            resultId = oldResult.getId();
        } else {
            HrmPerformanceAppraiseResultDO newResult = new HrmPerformanceAppraiseResultDO()
                    .setAppraiseId(appraise.getId())
                    .setEmployeeId(saveReqVO.getEmployeeId());
            resultMapper.insert(newResult);
            resultId = newResult.getId();
        }
        resultItemMapper.deleteByResultId(resultId);

        // 4. 计算单指标得分、总分与档级
        BigDecimal sumWeight = BigDecimal.ZERO;
        BigDecimal weightedFinal = BigDecimal.ZERO;
        BigDecimal weightedSelf = BigDecimal.ZERO;
        BigDecimal weightedSuperior = BigDecimal.ZERO;
        for (HrmPerformanceAppraiseResultSaveReqVO.Item score : saveReqVO.getItems()) {
            HrmPerformanceAppraiseItemDO item = itemMap.get(score.getItemId());
            BigDecimal weight = item.getWeight() == null ? BigDecimal.ONE : item.getWeight();
            BigDecimal selfScore = score.getSelfScore() == null ? BigDecimal.ZERO : score.getSelfScore();
            BigDecimal superiorScore = score.getSuperiorScore() == null ? BigDecimal.ZERO : score.getSuperiorScore();
            // 单指标分：1 自评 / 2 上级评 / 其他 两者平均
            BigDecimal itemScore;
            if (Objects.equals(item.getScorerType(), SCORER_SELF)) {
                itemScore = selfScore;
            } else if (Objects.equals(item.getScorerType(), SCORER_SUPERIOR)) {
                itemScore = superiorScore;
            } else {
                itemScore = selfScore.add(superiorScore).divide(BigDecimal.valueOf(2), 2, RoundingMode.HALF_UP);
            }
            sumWeight = sumWeight.add(weight);
            weightedFinal = weightedFinal.add(weight.multiply(itemScore));
            weightedSelf = weightedSelf.add(weight.multiply(selfScore));
            weightedSuperior = weightedSuperior.add(weight.multiply(superiorScore));

            // 保存明细
            resultItemMapper.insert(new HrmPerformanceAppraiseResultItemDO()
                    .setResultId(resultId)
                    .setItemId(item.getId())
                    .setItemName(item.getItemName())
                    .setWeight(weight)
                    .setSelfScore(selfScore)
                    .setSuperiorScore(superiorScore));
        }
        BigDecimal selfTotal = weightedSelf.divide(sumWeight, 1, RoundingMode.HALF_UP);
        BigDecimal superiorTotal = weightedSuperior.divide(sumWeight, 1, RoundingMode.HALF_UP);
        BigDecimal finalScore = weightedFinal.divide(sumWeight, 1, RoundingMode.HALF_UP);

        // 5. 更新结果
        resultMapper.updateById(new HrmPerformanceAppraiseResultDO()
                .setId(resultId)
                .setSelfScore(selfTotal)
                .setSuperiorScore(superiorTotal)
                .setFinalScore(finalScore)
                .setGrade(calcGrade(finalScore))
                .setComment(saveReqVO.getComment()));

        // 6. 方案状态流转：已发布 -> 进行中
        if (Objects.equals(appraise.getStatus(), STATUS_PUBLISHED)) {
            appraiseMapper.updateById(new HrmPerformanceAppraiseDO().setId(appraise.getId()).setStatus(STATUS_PROCESSING));
        }
    }

    @Override
    public PageResult<HrmPerformanceAppraiseResultDO> getAppraiseResultPage(HrmPerformanceAppraiseResultPageReqVO pageReqVO) {
        return resultMapper.selectPage(pageReqVO);
    }

    @Override
    public HrmPerformanceAppraiseResultDO getAppraiseResult(Long appraiseId, Long employeeId) {
        return resultMapper.selectByAppraiseIdAndEmployeeId(appraiseId, employeeId);
    }

    @Override
    public List<HrmPerformanceAppraiseResultItemDO> getAppraiseResultItems(Long resultId) {
        return resultItemMapper.selectListByResultId(resultId);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public Integer generateAppraisees(Long appraiseId) {
        HrmPerformanceAppraiseDO appraise = validateAppraiseExists(appraiseId);
        // 1. 按考核范围筛选员工
        List<HrmEmployeeDO> employees;
        if (Objects.equals(appraise.getScopeType(), SCOPE_TEAM)) {
            List<Long> teamIds = StrUtil.isEmpty(appraise.getScopeTeamIds())
                    ? Collections.emptyList() : JSONUtil.toList(appraise.getScopeTeamIds(), Long.class);
            if (CollUtil.isEmpty(teamIds)) {
                throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_SCOPE_TEAM_EMPTY);
            }
            employees = employeeMapper.selectListByTeamIds(teamIds);
        } else {
            employees = employeeMapper.selectListByStatus(EMPLOYEE_STATUS_ACTIVE);
        }
        if (CollUtil.isEmpty(employees)) {
            return 0;
        }
        // 2. 已存在结果的不再生成（幂等）
        List<HrmPerformanceAppraiseResultDO> existing = resultMapper.selectList(
                new LambdaQueryWrapperX<HrmPerformanceAppraiseResultDO>()
                        .eq(HrmPerformanceAppraiseResultDO::getAppraiseId, appraiseId));
        Set<Long> existingEmployeeIds = convertSet(existing, HrmPerformanceAppraiseResultDO::getEmployeeId);
        // 3. 生成空白结果行
        int count = 0;
        for (HrmEmployeeDO employee : employees) {
            if (existingEmployeeIds.contains(employee.getId())) {
                continue;
            }
            resultMapper.insert(new HrmPerformanceAppraiseResultDO()
                    .setAppraiseId(appraiseId)
                    .setEmployeeId(employee.getId()));
            count++;
        }
        return count;
    }

    // ========== 私有方法 ==========

    private Integer calcGrade(BigDecimal finalScore) {
        if (finalScore == null) {
            return null;
        }
        if (finalScore.compareTo(GRADE_EXCELLENT) >= 0) {
            return GRADE_EXCELLENT_CODE;
        }
        if (finalScore.compareTo(GRADE_GOOD) >= 0) {
            return GRADE_GOOD_CODE;
        }
        if (finalScore.compareTo(GRADE_PASS) >= 0) {
            return GRADE_PASS_CODE;
        }
        return GRADE_IMPROVE_CODE;
    }

    private HrmPerformanceAppraiseDO validateAppraiseExists(Long id) {
        HrmPerformanceAppraiseDO appraise = appraiseMapper.selectById(id);
        if (appraise == null) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_NOT_EXISTS);
        }
        return appraise;
    }

    private HrmPerformanceAppraiseDO validateAppraiseCanModify(Long id) {
        HrmPerformanceAppraiseDO appraise = validateAppraiseExists(id);
        if (ObjUtil.notEqual(appraise.getStatus(), STATUS_DRAFT)) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_UPDATE_FAIL_NOT_DRAFT);
        }
        return appraise;
    }

}