package cn.iocoder.yudao.module.mes.service.safety;

import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.mes.controller.admin.safety.vo.training.MesSafetyTrainingAttendeeSaveReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.safety.vo.training.MesSafetyTrainingPageReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.safety.vo.training.MesSafetyTrainingSaveReqVO;
import cn.iocoder.yudao.module.mes.dal.dataobject.safety.MesSafetyTrainingAttendeeDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.safety.MesSafetyTrainingDO;
import cn.iocoder.yudao.module.mes.dal.mysql.safety.MesSafetyTrainingAttendeeMapper;
import cn.iocoder.yudao.module.mes.dal.mysql.safety.MesSafetyTrainingMapper;
import jakarta.annotation.Resource;
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.ArrayList;
import java.util.List;
import java.util.Objects;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.*;

/**
 * 安全生产 - 培训 Service 实现
 *
 * @author 超级管理员
 */
@Service
@Validated
public class MesSafetyTrainingServiceImpl implements MesSafetyTrainingService {

    @Resource
    private MesSafetyTrainingMapper trainingMapper;

    @Resource
    private MesSafetyTrainingAttendeeMapper attendeeMapper;

    @Override
    public PageResult<MesSafetyTrainingDO> getTrainingPage(MesSafetyTrainingPageReqVO pageReqVO) {
        return trainingMapper.selectPage(pageReqVO);
    }

    @Override
    public MesSafetyTrainingDO getTraining(Long id) {
        return trainingMapper.selectById(id);
    }

    @Override
    public Long createTraining(MesSafetyTrainingSaveReqVO createReqVO) {
        validateCodeUnique(null, createReqVO.getCode());
        validateTimeRange(createReqVO);
        MesSafetyTrainingDO training = BeanUtils.toBean(createReqVO, MesSafetyTrainingDO.class);
        training.setAttendeeCount(0);
        training.setSignedCount(0);
        trainingMapper.insert(training);
        return training.getId();
    }

    @Override
    public void updateTraining(MesSafetyTrainingSaveReqVO updateReqVO) {
        validateTrainingExists(updateReqVO.getId());
        validateCodeUnique(updateReqVO.getId(), updateReqVO.getCode());
        validateTimeRange(updateReqVO);
        MesSafetyTrainingDO updateObj = BeanUtils.toBean(updateReqVO, MesSafetyTrainingDO.class);
        // 统计字段由参训人员维护，不通过此接口修改
        updateObj.setAttendeeCount(null);
        updateObj.setSignedCount(null);
        updateObj.setAvgScore(null);
        trainingMapper.updateById(updateObj);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void deleteTraining(Long id) {
        validateTrainingExists(id);
        // 级联删除参训记录，避免孤儿数据
        attendeeMapper.deleteByTrainingId(id);
        trainingMapper.deleteById(id);
    }

    @Override
    public MesSafetyTrainingDO validateTrainingExists(Long id) {
        if (id == null) {
            throw exception(MES_SAFETY_TRAINING_NOT_EXISTS);
        }
        MesSafetyTrainingDO training = trainingMapper.selectById(id);
        if (training == null) {
            throw exception(MES_SAFETY_TRAINING_NOT_EXISTS);
        }
        return training;
    }

    @Override
    public List<MesSafetyTrainingAttendeeDO> getAttendeeList(Long trainingId) {
        return attendeeMapper.selectListByTrainingId(trainingId);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void saveAttendees(Long trainingId, List<MesSafetyTrainingAttendeeSaveReqVO> attendees) {
        validateTrainingExists(trainingId);
        // 全量覆盖：先清空再插入，保证与前端列表一致
        attendeeMapper.deleteByTrainingId(trainingId);
        if (attendees != null && !attendees.isEmpty()) {
            List<MesSafetyTrainingAttendeeDO> list = new ArrayList<>(attendees.size());
            for (MesSafetyTrainingAttendeeSaveReqVO vo : attendees) {
                MesSafetyTrainingAttendeeDO attendee = BeanUtils.toBean(vo, MesSafetyTrainingAttendeeDO.class);
                attendee.setId(null);
                attendee.setTrainingId(trainingId);
                if (attendee.getSignStatus() == null) {
                    attendee.setSignStatus(MesSafetyTrainingAttendeeDO.SIGN_NONE);
                }
                if (attendee.getPassed() == null) {
                    attendee.setPassed(MesSafetyTrainingAttendeeDO.PASS_NONE);
                }
                list.add(attendee);
            }
            attendeeMapper.insertBatch(list);
        }
        refreshStatistics(trainingId);
    }

    @Override
    public void updateAttendee(MesSafetyTrainingAttendeeSaveReqVO reqVO) {
        if (reqVO.getId() == null) {
            throw exception(MES_SAFETY_TRAINING_ATTENDEE_NOT_EXISTS);
        }
        MesSafetyTrainingAttendeeDO exists = attendeeMapper.selectById(reqVO.getId());
        if (exists == null) {
            throw exception(MES_SAFETY_TRAINING_ATTENDEE_NOT_EXISTS);
        }
        MesSafetyTrainingAttendeeDO updateObj = new MesSafetyTrainingAttendeeDO();
        updateObj.setId(reqVO.getId());
        updateObj.setSignStatus(reqVO.getSignStatus());
        updateObj.setSignTime(reqVO.getSignTime());
        updateObj.setScore(reqVO.getScore());
        updateObj.setPassed(reqVO.getPassed());
        updateObj.setRemark(reqVO.getRemark());
        attendeeMapper.updateById(updateObj);
        refreshStatistics(exists.getTrainingId());
    }

    @Override
    public void deleteAttendee(Long id) {
        MesSafetyTrainingAttendeeDO exists = attendeeMapper.selectById(id);
        if (exists == null) {
            throw exception(MES_SAFETY_TRAINING_ATTENDEE_NOT_EXISTS);
        }
        attendeeMapper.deleteById(id);
        refreshStatistics(exists.getTrainingId());
    }

    @Override
    public void refreshStatistics(Long trainingId) {
        Long total = attendeeMapper.selectCountByTrainingId(trainingId);
        Long signed = attendeeMapper.selectCountSignedByTrainingId(trainingId);
        List<MesSafetyTrainingAttendeeDO> list = attendeeMapper.selectListByTrainingId(trainingId);
        // 平均分只统计有成绩的
        BigDecimal sum = BigDecimal.ZERO;
        int scored = 0;
        for (MesSafetyTrainingAttendeeDO a : list) {
            if (a.getScore() != null) {
                sum = sum.add(a.getScore());
                scored++;
            }
        }
        BigDecimal avg = scored > 0
                ? sum.divide(BigDecimal.valueOf(scored), 2, RoundingMode.HALF_UP)
                : null;
        MesSafetyTrainingDO updateObj = new MesSafetyTrainingDO();
        updateObj.setId(trainingId);
        updateObj.setAttendeeCount(total == null ? 0 : total.intValue());
        updateObj.setSignedCount(signed == null ? 0 : signed.intValue());
        updateObj.setAvgScore(avg);
        trainingMapper.updateById(updateObj);
    }

    /**
     * 校验培训时间区间
     */
    private void validateTimeRange(MesSafetyTrainingSaveReqVO reqVO) {
        LocalDateTime start = reqVO.getStartTime();
        LocalDateTime end = reqVO.getEndTime();
        if (start != null && end != null && !end.isAfter(start)) {
            throw exception(MES_SAFETY_TRAINING_TIME_INVALID);
        }
    }

    private void validateCodeUnique(Long id, String code) {
        MesSafetyTrainingDO exists = trainingMapper.selectByCode(code);
        if (exists == null) {
            return;
        }
        if (id == null || !Objects.equals(exists.getId(), id)) {
            throw exception(MES_SAFETY_TRAINING_CODE_DUPLICATE);
        }
    }

}