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

import cn.hutool.core.collection.CollUtil;
import cn.iocoder.yudao.module.hrm.api.employee.HrmEmployeeApi;
import cn.iocoder.yudao.module.hrm.api.employee.dto.HrmEmployeeSimpleDTO;
import cn.iocoder.yudao.module.mes.dal.dataobject.md.workstation.MesMdWorkshopDO;
import cn.iocoder.yudao.module.mes.dal.mysql.md.workstation.MesMdWorkshopMapper;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Component;

import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.stream.Collectors;

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

/**
 * 安全生产 - 主数据引用解析器
 *
 * <p>模块内所有「人员」与「车间」字段统一走这里解析，避免每个 Service 重复写一遍。</p>
 *
 * <p>统一约定：业务表存 {@code xxxId} 关联主数据，同时保留 {@code xxxName} 作为快照。
 * 保存时从主数据把名称带出写快照；查询时用主数据最新值覆盖快照，
 * 这样员工调岗、车间改名后台账会自动跟着变，而主数据被删时老记录仍能靠快照展示。</p>
 *
 * @author 超级管理员
 */
@Component
public class SafetyRefResolver {

    @Resource
    private HrmEmployeeApi hrmEmployeeApi;
    @Resource
    private MesMdWorkshopMapper workshopMapper;

    // ==================== 员工 ====================

    /**
     * 按员工编号取员工，必须存在
     *
     * @throws cn.iocoder.yudao.framework.common.exception.ServiceException 员工不存在时抛出
     */
    public HrmEmployeeSimpleDTO requireEmployee(Long employeeId, cn.iocoder.yudao.framework.common.exception.ErrorCode errorCode) {
        if (employeeId == null) {
            throw exception(errorCode);
        }
        HrmEmployeeSimpleDTO employee = hrmEmployeeApi.getEmployee(employeeId);
        if (employee == null) {
            throw exception(errorCode);
        }
        return employee;
    }

    /**
     * 可空的员工解析：id 为空返回 null，不校验存在性（用于非必填字段）
     */
    public HrmEmployeeSimpleDTO findEmployee(Long employeeId) {
        return employeeId == null ? null : hrmEmployeeApi.getEmployee(employeeId);
    }

    /**
     * 批量取员工，返回 id -> 员工 的 Map
     */
    public Map<Long, HrmEmployeeSimpleDTO> employeeMap(Collection<Long> employeeIds) {
        // 注意：这里必须返回可接受 null key 的空 Map。
        // 调用方会直接 map.get(记录里的 xxxId)，而 xxxId 允许为 null（例如组织未设负责人、
        // 证件未关联持证人），Map.of() 的 get(null) 会抛 NPE，导致列表接口 500。
        if (CollUtil.isEmpty(employeeIds)) {
            return new HashMap<>();
        }
        List<Long> ids = employeeIds.stream().filter(Objects::nonNull).distinct().toList();
        if (ids.isEmpty()) {
            return new HashMap<>();
        }
        return hrmEmployeeApi.getEmployeeList(ids).stream()
                .collect(Collectors.toMap(HrmEmployeeSimpleDTO::getId, e -> e, (a, b) -> a));
    }

    // ==================== 车间 ====================

    /**
     * 按车间编号取车间，必须存在
     */
    public MesMdWorkshopDO requireWorkshop(Long workshopId) {
        if (workshopId == null) {
            throw exception(SAFETY_WORKSHOP_NOT_EXISTS);
        }
        MesMdWorkshopDO workshop = workshopMapper.selectById(workshopId);
        if (workshop == null) {
            throw exception(SAFETY_WORKSHOP_NOT_EXISTS);
        }
        return workshop;
    }

    /**
     * 可空的车间解析
     */
    public MesMdWorkshopDO findWorkshop(Long workshopId) {
        return workshopId == null ? null : workshopMapper.selectById(workshopId);
    }

    /**
     * 批量取车间，返回 id -> 车间 的 Map
     */
    public Map<Long, MesMdWorkshopDO> workshopMap(Collection<Long> workshopIds) {
        // 同上：车间编号允许为 null，空集合必须返回可接受 null key 的 Map
        if (CollUtil.isEmpty(workshopIds)) {
            return new HashMap<>();
        }
        List<Long> ids = workshopIds.stream().filter(Objects::nonNull).distinct().toList();
        if (ids.isEmpty()) {
            return new HashMap<>();
        }
        return workshopMapper.selectBatchIds(ids).stream()
                .collect(Collectors.toMap(MesMdWorkshopDO::getId, w -> w, (a, b) -> a));
    }

}