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

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjUtil;
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.bpm.api.task.BpmProcessInstanceApi;
import cn.iocoder.yudao.module.bpm.api.task.dto.BpmProcessInstanceCreateReqDTO;
import cn.iocoder.yudao.module.bpm.dal.dataobject.definition.BpmProcessDefinitionInfoDO;
import cn.iocoder.yudao.module.bpm.service.definition.BpmCategoryService;
import cn.iocoder.yudao.module.bpm.service.definition.BpmProcessDefinitionService;
import cn.iocoder.yudao.module.hrm.controller.admin.performanceappraise.vo.*;
import cn.iocoder.yudao.module.hrm.dal.dataobject.performanceappraise.*;
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 jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.flowable.engine.repository.ProcessDefinition;
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.time.LocalDateTime;
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;
import static cn.iocoder.yudao.module.hrm.util.HrmAuditStatusUtils.convertBpmResultToAuditStatus;

/**
 * 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;

    /**
     * BPM 绩效考核审批分类编码
     */
    private static final String APPRAISE_APPROVE_CATEGORY_CODE = "performance_appraise_approve";

    @Resource
    private HrmPerformanceAppraiseMapper appraiseMapper;
    @Resource
    private HrmPerformanceAppraiseItemMapper itemMapper;
    @Resource
    private HrmPerformanceAppraiseResultMapper resultMapper;
    @Resource
    private HrmPerformanceAppraiseResultItemMapper resultItemMapper;
    @Resource
    private BpmProcessInstanceApi bpmProcessInstanceApi;
    @Resource
    private BpmCategoryService bpmCategoryService;
    @Resource
    private BpmProcessDefinitionService bpmProcessDefinitionService;

    @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()));
        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())
                .setProcessInstanceId(oldAppraise.getProcessInstanceId()));
        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, String processDefinitionKey, Long userId) {
        // 1. 校验方案是否在草稿状态
        HrmPerformanceAppraiseDO appraise = appraiseMapper.selectById(id);
        if (appraise == null) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_NOT_EXISTS);
        }
        if (ObjUtil.notEqual(appraise.getAuditStatus(), HrmAuditStatusEnum.DRAFT.getStatus())) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_SUBMIT_FAIL_NOT_DRAFT);
        }
        if (CollUtil.isEmpty(itemMapper.selectListByAppraiseId(id))) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_ITEM_EMPTY);
        }

        // 2. 校验审批分类和流程定义是否存在，且传入的流程定义处于激活状态
        validateAppraiseApproveCategoryAndProcessDefinition();
        ProcessDefinition processDefinition = bpmProcessDefinitionService.getActiveProcessDefinition(processDefinitionKey);
        if (processDefinition == null) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_BPM_PROCESS_DEFINITION_NOT_EXISTS);
        }

        // 3. 创建审批流程实例
        String processInstanceId = bpmProcessInstanceApi.createProcessInstance(userId,
                new BpmProcessInstanceCreateReqDTO().setProcessDefinitionKey(processDefinitionKey)
                        .setBusinessKey(String.valueOf(id)));

        // 4. 更新方案流程编号与审批状态
        appraiseMapper.updateById(new HrmPerformanceAppraiseDO().setId(id).setProcessInstanceId(processInstanceId)
                .setAuditStatus(HrmAuditStatusEnum.PROCESS.getStatus()));
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void auditAppraise(HrmPerformanceAppraiseAuditReqVO auditReqVO) {
        HrmPerformanceAppraiseDO appraise = validateAppraiseExists(auditReqVO.getId());
        if (ObjUtil.notEqual(appraise.getAuditStatus(), HrmAuditStatusEnum.PROCESS.getStatus())) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_UPDATE_AUDIT_STATUS_FAIL_NOT_PROCESS);
        }
        Integer auditStatus = auditReqVO.getAuditStatus();
        if (!Objects.equals(auditStatus, HrmAuditStatusEnum.APPROVE.getStatus())
                && !Objects.equals(auditStatus, HrmAuditStatusEnum.REJECT.getStatus())) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_UPDATE_AUDIT_STATUS_FAIL_NOT_PROCESS);
        }
        HrmPerformanceAppraiseDO update = new HrmPerformanceAppraiseDO()
                .setId(appraise.getId())
                .setAuditStatus(auditStatus)
                .setAuditUserId(getLoginUserId())
                .setAuditTime(LocalDateTime.now());
        if (auditReqVO.getRemark() != null) {
            update.setAuditRemark(auditReqVO.getRemark());
        }
        // 审批通过 -> 方案发布
        if (Objects.equals(auditStatus, HrmAuditStatusEnum.APPROVE.getStatus())) {
            update.setStatus(STATUS_PUBLISHED);
        }
        appraiseMapper.updateById(update);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void updateAppraiseAuditStatus(Long id, Integer bpmResult) {
        // 1.1 校验方案是否存在
        HrmPerformanceAppraiseDO appraise = appraiseMapper.selectById(id);
        if (appraise == null) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_NOT_EXISTS);
        }
        // 1.2 只有审批中，可以更新审批结果
        if (ObjUtil.notEqual(appraise.getAuditStatus(), HrmAuditStatusEnum.PROCESS.getStatus())) {
            log.error("[updateAppraiseAuditStatus][appraise({}) 不处于审批中，无法更新审批结果({})]", appraise.getId(), bpmResult);
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_UPDATE_AUDIT_STATUS_FAIL_NOT_PROCESS);
        }

        // 2. 更新方案审批结果
        Integer auditStatus = convertBpmResultToAuditStatus(bpmResult);
        HrmPerformanceAppraiseDO update = new HrmPerformanceAppraiseDO().setId(id).setAuditStatus(auditStatus);
        if (Objects.equals(auditStatus, HrmAuditStatusEnum.APPROVE.getStatus())) {
            update.setStatus(STATUS_PUBLISHED);
        }
        appraiseMapper.updateById(update);
    }

    @Override
    public List<Map<String, Object>> getAppraiseApproveProcessDefinitionList() {
        // 1. 校验审批分类和流程定义是否存在
        validateAppraiseApproveCategoryAndProcessDefinition();

        // 2. 获取分类下的流程定义信息
        List<BpmProcessDefinitionInfoDO> definitionInfoList = bpmProcessDefinitionService
                .getProcessDefinitionInfoListByCategory(APPRAISE_APPROVE_CATEGORY_CODE);

        // 3. 获取流程定义详情
        Set<String> processDefinitionIds = convertSet(definitionInfoList, BpmProcessDefinitionInfoDO::getProcessDefinitionId);
        List<ProcessDefinition> processDefinitions = bpmProcessDefinitionService.getProcessDefinitionList(processDefinitionIds);

        // 4. 过滤：只保留激活状态，且相同 key 只保留最新版本
        Map<String, ProcessDefinition> latestVersionMap = new HashMap<>();
        for (ProcessDefinition pd : processDefinitions) {
            if (pd.isSuspended()) {
                continue;
            }
            ProcessDefinition existing = latestVersionMap.get(pd.getKey());
            if (existing == null || pd.getVersion() > existing.getVersion()) {
                latestVersionMap.put(pd.getKey(), pd);
            }
        }

        // 5. 返回流程定义列表
        List<Map<String, Object>> result = new ArrayList<>();
        for (ProcessDefinition pd : latestVersionMap.values()) {
            Map<String, Object> item = new HashMap<>();
            item.put("id", pd.getId());
            item.put("key", pd.getKey());
            item.put("name", pd.getName());
            item.put("version", pd.getVersion());
            result.add(item);
        }
        return result;
    }

    @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);
    }

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

    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;
    }

    /**
     * 校验 BPM 绩效考核审批分类和流程定义是否存在
     */
    private void validateAppraiseApproveCategoryAndProcessDefinition() {
        // 1. 校验分类是否存在
        Map<String, ?> categoryMap = bpmCategoryService.getCategoryMap(Collections.singletonList(APPRAISE_APPROVE_CATEGORY_CODE));
        if (!categoryMap.containsKey(APPRAISE_APPROVE_CATEGORY_CODE)) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_BPM_PROCESS_DEFINITION_NOT_EXISTS);
        }

        // 2. 校验分类下是否有可用的流程定义
        List<BpmProcessDefinitionInfoDO> definitionInfoList = bpmProcessDefinitionService
                .getProcessDefinitionInfoListByCategory(APPRAISE_APPROVE_CATEGORY_CODE);
        if (CollUtil.isEmpty(definitionInfoList)) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_BPM_PROCESS_DEFINITION_NOT_EXISTS);
        }

        // 3. 校验是否有激活状态的流程定义
        Set<String> processDefinitionIds = convertSet(definitionInfoList, BpmProcessDefinitionInfoDO::getProcessDefinitionId);
        List<ProcessDefinition> processDefinitions = bpmProcessDefinitionService.getProcessDefinitionList(processDefinitionIds);
        boolean hasActiveDefinition = processDefinitions.stream().anyMatch(pd -> !pd.isSuspended());
        if (!hasActiveDefinition) {
            throw exception(ErrorCodeConstants.PERFORMANCE_APPRAISE_BPM_PROCESS_DEFINITION_NOT_EXISTS);
        }
    }

}