package cn.iocoder.yudao.module.mes.service.em.meter;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjUtil;
import cn.hutool.core.util.StrUtil;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.collection.MapUtils;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.mes.controller.admin.em.meter.vo.MesEmMeterPageReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.em.meter.vo.MesEmMeterRespVO;
import cn.iocoder.yudao.module.mes.controller.admin.em.meter.vo.MesEmMeterSaveReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.em.meter.vo.MesEmOptionRespVO;
import cn.iocoder.yudao.module.mes.dal.dataobject.dv.machinery.MesDvMachineryDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.em.meter.MesEmMeterDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.md.workstation.MesMdWorkshopDO;
import cn.iocoder.yudao.module.mes.dal.mysql.dv.machinery.MesDvMachineryMapper;
import cn.iocoder.yudao.module.mes.dal.mysql.em.meter.MesEmMeterMapper;
import cn.iocoder.yudao.module.mes.dal.mysql.em.record.MesEmRecordMapper;
import cn.iocoder.yudao.module.mes.dal.mysql.md.workstation.MesMdWorkshopMapper;
import cn.iocoder.yudao.module.mes.enums.em.MesEmEnergyTypeEnum;
import cn.iocoder.yudao.module.mes.enums.em.MesEmMeterStatusEnum;
import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Service;
import org.springframework.validation.annotation.Validated;

import java.math.BigDecimal;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Map;
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.framework.common.util.collection.CollectionUtils.convertSet;
import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.*;

/**
 * MES 能耗计量点 Service 实现类
 *
 * @author 超级管理员
 */
@Service
@Validated
public class MesEmMeterServiceImpl implements MesEmMeterService {

    @Resource
    private MesEmMeterMapper meterMapper;
    @Resource
    private MesEmRecordMapper recordMapper;
    @Resource
    private MesMdWorkshopMapper workshopMapper;
    @Resource
    private MesDvMachineryMapper machineryMapper;
    @Resource
    private AdminUserApi adminUserApi;

    @Override
    public Long createMeter(MesEmMeterSaveReqVO createReqVO) {
        validateMeterCodeUnique(null, createReqVO.getCode());
        MesEmMeterDO meter = BeanUtils.toBean(createReqVO, MesEmMeterDO.class);
        fillMeterDefaultsAndNames(meter);
        meterMapper.insert(meter);
        return meter.getId();
    }

    @Override
    public void updateMeter(MesEmMeterSaveReqVO updateReqVO) {
        validateMeterExists(updateReqVO.getId());
        validateMeterCodeUnique(updateReqVO.getId(), updateReqVO.getCode());
        MesEmMeterDO meter = BeanUtils.toBean(updateReqVO, MesEmMeterDO.class);
        fillMeterDefaultsAndNames(meter);
        meterMapper.updateById(meter);
    }

    @Override
    public void deleteMeter(Long id) {
        validateMeterExists(id);
        if (recordMapper.selectCountByMeterId(id) > 0) {
            throw exception(EM_METER_HAS_RECORD);
        }
        meterMapper.deleteById(id);
    }

    @Override
    public MesEmMeterDO getMeter(Long id) {
        return meterMapper.selectById(id);
    }

    @Override
    public void validateMeterExists(Long id) {
        if (meterMapper.selectById(id) == null) {
            throw exception(EM_METER_NOT_EXISTS);
        }
    }

    private void validateMeterCodeUnique(Long id, String code) {
        if (StrUtil.isBlank(code)) {
            return;
        }
        MesEmMeterDO meter = meterMapper.selectByCode(code);
        if (meter == null) {
            return;
        }
        if (ObjUtil.notEqual(meter.getId(), id)) {
            throw exception(EM_METER_CODE_DUPLICATE);
        }
    }

    /**
     * 补全单位、倍率、折标煤系数，并冗余车间/产线/设备名称
     */
    private void fillMeterDefaultsAndNames(MesEmMeterDO meter) {
        MesEmEnergyTypeEnum energyType = MesEmEnergyTypeEnum.getByType(meter.getEnergyType());
        if (StrUtil.isBlank(meter.getUnit()) && energyType != null) {
            meter.setUnit(energyType.getUnit());
        }
        if (meter.getRatio() == null) {
            meter.setRatio(BigDecimal.ONE);
        }
        if (meter.getInitialReading() == null) {
            meter.setInitialReading(BigDecimal.ZERO);
        }
        if (meter.getCoalFactor() == null && energyType != null) {
            meter.setCoalFactor(energyType.getCoalFactor());
        }
        if (meter.getStatus() == null) {
            meter.setStatus(MesEmMeterStatusEnum.ENABLE.getStatus());
        }
        // 车间
        if (meter.getWorkshopId() != null) {
            MesMdWorkshopDO workshop = workshopMapper.selectById(meter.getWorkshopId());
            if (workshop == null) {
                throw exception(EM_METER_OBJECT_NOT_EXISTS);
            }
            meter.setWorkshopName(workshop.getName());
        } else {
            meter.setWorkshopName(null);
        }
        // 产线
        if (meter.getLineId() != null) {
            MesEmOptionRespVO line = meterMapper.selectLineOptionById(meter.getLineId());
            if (line == null) {
                throw exception(EM_METER_OBJECT_NOT_EXISTS);
            }
            meter.setLineName(line.getName());
        } else {
            meter.setLineName(null);
        }
        // 设备；未选择车间时自动带入设备所属车间
        if (meter.getMachineryId() != null) {
            MesDvMachineryDO machinery = machineryMapper.selectById(meter.getMachineryId());
            if (machinery == null) {
                throw exception(EM_METER_OBJECT_NOT_EXISTS);
            }
            meter.setMachineryName(machinery.getName());
            if (meter.getWorkshopId() == null && machinery.getWorkshopId() != null) {
                MesMdWorkshopDO workshop = workshopMapper.selectById(machinery.getWorkshopId());
                if (workshop != null) {
                    meter.setWorkshopId(workshop.getId());
                    meter.setWorkshopName(workshop.getName());
                }
            }
        } else {
            meter.setMachineryName(null);
        }
    }

    @Override
    public PageResult<MesEmMeterDO> getMeterPage(MesEmMeterPageReqVO pageReqVO) {
        return meterMapper.selectPage(pageReqVO);
    }

    @Override
    public List<MesEmMeterDO> getMeterList() {
        return meterMapper.selectList();
    }

    @Override
    public List<MesEmMeterDO> getMeterList(Collection<Long> ids) {
        if (CollUtil.isEmpty(ids)) {
            return Collections.emptyList();
        }
        return meterMapper.selectByIds(ids);
    }

    @Override
    public List<MesEmMeterRespVO> getMeterRespVOList(List<MesEmMeterDO> list) {
        if (CollUtil.isEmpty(list)) {
            return Collections.emptyList();
        }
        Map<Long, AdminUserRespDTO> userMap = CollUtil.isNotEmpty(convertSet(list, MesEmMeterDO::getChargeUserId))
                ? adminUserApi.getUserMap(convertSet(list, MesEmMeterDO::getChargeUserId))
                : Collections.emptyMap();
        return BeanUtils.toBean(list, MesEmMeterRespVO.class, vo -> {
            vo.setEnergyTypeName(MesEmEnergyTypeEnum.getName(vo.getEnergyType()));
            vo.setStatusName(MesEmMeterStatusEnum.getName(vo.getStatus()));
            MapUtils.findAndThen(userMap, vo.getChargeUserId(), user -> vo.setChargeUserName(user.getNickname()));
        });
    }

    @Override
    public List<MesEmOptionRespVO> getMeterSimpleList() {
        List<MesEmMeterDO> list = meterMapper.selectListByStatus(MesEmMeterStatusEnum.ENABLE.getStatus());
        return list.stream().map(meter -> {
            MesEmOptionRespVO option = new MesEmOptionRespVO();
            option.setId(meter.getId());
            option.setCode(meter.getCode());
            option.setName(meter.getName());
            option.setEnergyType(meter.getEnergyType());
            option.setUnit(meter.getUnit());
            option.setWorkshopId(meter.getWorkshopId());
            option.setWorkshopName(meter.getWorkshopName());
            option.setLineId(meter.getLineId());
            option.setLineName(meter.getLineName());
            return option;
        }).collect(Collectors.toList());
    }

    @Override
    public List<MesEmOptionRespVO> getWorkshopOptions() {
        return meterMapper.selectWorkshopOptions();
    }

    @Override
    public List<MesEmOptionRespVO> getLineOptions() {
        return meterMapper.selectLineOptions();
    }

    @Override
    public List<MesEmOptionRespVO> getMachineryOptions() {
        List<MesEmOptionRespVO> options = meterMapper.selectMachineryOptions();
        if (CollUtil.isEmpty(options)) {
            return options;
        }
        Map<Long, MesMdWorkshopDO> workshopMap = workshopMapper.selectList().stream()
                .collect(Collectors.toMap(MesMdWorkshopDO::getId, Function.identity(), (a, b) -> a));
        options.forEach(option -> {
            MesMdWorkshopDO workshop = workshopMap.get(option.getWorkshopId());
            if (workshop != null) {
                option.setWorkshopName(workshop.getName());
            }
        });
        return options;
    }

}