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

import cn.hutool.core.collection.CollUtil;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.hrm.controller.admin.attendance.vo.HrmEmployeeScheduleListReqVO;
import cn.iocoder.yudao.module.hrm.controller.admin.attendance.vo.HrmEmployeeScheduleRespVO;
import cn.iocoder.yudao.module.hrm.controller.admin.attendance.vo.HrmEmployeeScheduleSaveReqVO;
import cn.iocoder.yudao.module.hrm.dal.dataobject.attendance.HrmEmployeeScheduleDO;
import cn.iocoder.yudao.module.hrm.dal.dataobject.attendance.HrmShiftDO;
import cn.iocoder.yudao.module.hrm.dal.dataobject.employee.HrmEmployeeDO;
import cn.iocoder.yudao.module.hrm.dal.mysql.attendance.HrmEmployeeScheduleMapper;
import cn.iocoder.yudao.module.hrm.dal.mysql.attendance.HrmShiftMapper;
import cn.iocoder.yudao.module.hrm.dal.mysql.employee.HrmEmployeeMapper;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.validation.annotation.Validated;

import java.time.LocalDate;
import java.time.YearMonth;
import java.util.*;
import java.util.function.Function;
import java.util.stream.Collectors;

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

/**
 * 员工排班 Service 实现类
 *
 * @author 超级管理员
 */
@Service
@Validated
public class HrmEmployeeScheduleServiceImpl implements HrmEmployeeScheduleService {

    @Resource
    private HrmEmployeeScheduleMapper employeeScheduleMapper;
    @Resource
    private HrmEmployeeMapper employeeMapper;
    @Resource
    private HrmShiftMapper shiftMapper;

    @Override
    public List<HrmEmployeeScheduleRespVO> getScheduleList(HrmEmployeeScheduleListReqVO reqVO) {
        YearMonth yearMonth = YearMonth.parse(reqVO.getYearMonth());
        LocalDate startDate = yearMonth.atDay(1);
        LocalDate endDate = yearMonth.atEndOfMonth();

        // 1. 按条件查询排班记录
        List<HrmEmployeeScheduleDO> list;
        if (reqVO.getEmployeeId() != null) {
            list = employeeScheduleMapper.selectListByEmployeeIdAndPeriod(reqVO.getEmployeeId(), startDate, endDate);
        } else if (reqVO.getDeptId() != null) {
            List<HrmEmployeeDO> employees = employeeMapper.selectByDeptId(reqVO.getDeptId());
            Set<Long> employeeIds = employees.stream().map(HrmEmployeeDO::getId).collect(Collectors.toSet());
            list = CollUtil.isEmpty(employeeIds) ? Collections.emptyList()
                    : employeeScheduleMapper.selectListByEmployeeIdsAndPeriod(employeeIds, startDate, endDate);
        } else {
            list = employeeScheduleMapper.selectListByPeriod(startDate, endDate);
        }
        if (CollUtil.isEmpty(list)) {
            return Collections.emptyList();
        }

        // 2. 拼接员工信息与班次信息
        Set<Long> employeeIds = list.stream().map(HrmEmployeeScheduleDO::getEmployeeId)
                .filter(Objects::nonNull).collect(Collectors.toSet());
        Set<Long> shiftIds = list.stream().map(HrmEmployeeScheduleDO::getShiftId)
                .filter(Objects::nonNull).collect(Collectors.toSet());
        Map<Long, HrmEmployeeDO> employeeMap = CollUtil.isEmpty(employeeIds) ? Collections.emptyMap()
                : employeeMapper.selectBatchIds(employeeIds).stream()
                        .collect(Collectors.toMap(HrmEmployeeDO::getId, Function.identity(), (a, b) -> a));
        Map<Long, HrmShiftDO> shiftMap = CollUtil.isEmpty(shiftIds) ? Collections.emptyMap()
                : shiftMapper.selectBatchIds(shiftIds).stream()
                        .collect(Collectors.toMap(HrmShiftDO::getId, Function.identity(), (a, b) -> a));

        return list.stream().map(schedule -> {
            HrmEmployeeScheduleRespVO respVO = new HrmEmployeeScheduleRespVO();
            respVO.setId(schedule.getId());
            respVO.setEmployeeId(schedule.getEmployeeId());
            HrmEmployeeDO employee = schedule.getEmployeeId() == null ? null : employeeMap.get(schedule.getEmployeeId());
            if (employee != null) {
                respVO.setEmployeeName(employee.getName());
                respVO.setEmployeeNo(employee.getEmployeeNo());
            }
            respVO.setDate(schedule.getDate());
            respVO.setShiftId(schedule.getShiftId());
            HrmShiftDO shift = schedule.getShiftId() == null ? null : shiftMap.get(schedule.getShiftId());
            if (shift != null) {
                respVO.setShiftName(shift.getName());
                respVO.setShiftColor(shift.getColor());
            }
            respVO.setRemark(schedule.getRemark());
            return respVO;
        }).collect(Collectors.toList());
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void saveSchedule(List<HrmEmployeeScheduleSaveReqVO> saveReqVOList) {
        if (CollUtil.isEmpty(saveReqVOList)) {
            throw exception(SCHEDULE_ITEM_LIST_EMPTY);
        }
        // 1. 校验员工存在
        Set<Long> employeeIds = saveReqVOList.stream().map(HrmEmployeeScheduleSaveReqVO::getEmployeeId)
                .collect(Collectors.toSet());
        if (CollUtil.isNotEmpty(employeeIds)) {
            List<HrmEmployeeDO> employees = employeeMapper.selectBatchIds(employeeIds);
            if (employees.size() != employeeIds.size()) {
                throw exception(SCHEDULE_EMPLOYEE_NOT_EXISTS);
            }
        }
        // 2. 校验班次存在
        Set<Long> shiftIds = saveReqVOList.stream().map(HrmEmployeeScheduleSaveReqVO::getShiftId)
                .filter(Objects::nonNull).collect(Collectors.toSet());
        if (CollUtil.isNotEmpty(shiftIds)) {
            List<HrmShiftDO> shifts = shiftMapper.selectBatchIds(shiftIds);
            if (shifts.size() != shiftIds.size()) {
                throw exception(SCHEDULE_SHIFT_NOT_EXISTS);
            }
        }
        // 3. 逐条 upsert：有班次则保存，无班次则删除
        for (HrmEmployeeScheduleSaveReqVO reqVO : saveReqVOList) {
            HrmEmployeeScheduleDO existing = employeeScheduleMapper.selectByEmployeeIdAndDate(
                    reqVO.getEmployeeId(), reqVO.getDate());
            if (reqVO.getShiftId() == null) {
                if (existing != null) {
                    employeeScheduleMapper.deleteById(existing.getId());
                }
            } else {
                if (existing == null) {
                    employeeScheduleMapper.insert(BeanUtils.toBean(reqVO, HrmEmployeeScheduleDO.class));
                } else {
                    existing.setShiftId(reqVO.getShiftId());
                    existing.setRemark(reqVO.getRemark());
                    employeeScheduleMapper.updateById(existing);
                }
            }
        }
    }

}