package com.ruoyi.energy.service.impl;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.energy.mapper.EnergyConsumptionDetailMapper;
import com.ruoyi.energy.pojo.Energy;
import com.ruoyi.energy.mapper.EnergyMapper;
import com.ruoyi.energy.pojo.EnergyConsumptionDetail;
import com.ruoyi.energy.service.EnergyService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.framework.web.domain.R;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.util.List;

/**
 * <p>
 * 能源类型 服务实现类
 * </p>
 *
 * @author 芯导软件（江苏）有限公司
 * @since 2026-03-10 10:50:18
 */
@Service
public class EnergyServiceImpl extends ServiceImpl<EnergyMapper, Energy> implements EnergyService {

    @Autowired
    private EnergyMapper energyMapper;

    @Autowired
    private EnergyConsumptionDetailMapper energyConsumptionDetailMapper;

    @Override
    public IPage<Energy> pageEnergy(Page<Energy> page, Energy energy) {
        return energyMapper.pageEnergy(page,energy);
    }

    @Override
    public boolean batchDelete(List<Long> ids) {
        //先判断是否已经有填写的能耗数据
        List<EnergyConsumptionDetail> energyConsumptionDetails = energyConsumptionDetailMapper.selectList(Wrappers.<EnergyConsumptionDetail>lambdaQuery().in(EnergyConsumptionDetail::getEnergyId, ids));
        if (CollectionUtils.isNotEmpty(energyConsumptionDetails)){
            throw new RuntimeException("选中的能源类型已有对应的能耗数据,无法进行删除!");
        }
        return removeBatchByIds(ids);
    }

    @Override
    public R importData(MultipartFile file) {
        try {
            ExcelUtil<Energy> util = new ExcelUtil<Energy>(Energy.class);
            List<Energy> energyList = util.importExcel(file.getInputStream());
            if(CollectionUtils.isEmpty(energyList)){
                return R.fail("模板错误或导入数据为空");
            }
            energyList.forEach(energy -> {
                if (ObjectUtils.isEmpty(energy)) {
                    throw new RuntimeException("使用模板进行导入");
                }
                if (ObjectUtils.isEmpty(energy.getEnergyTyep())) {
                    throw new RuntimeException("能源类型不能为空");
                }
            });
            this.saveOrUpdateBatch(energyList);
            return R.ok(true);
        }catch (Exception e){
            e.printStackTrace();
            return R.fail(e.getMessage());
        }
    }

    @Override
    public void export(HttpServletResponse response) {
        List<Energy> list = energyMapper.selectList(null);
        if(CollectionUtils.isEmpty(list)){
            throw new RuntimeException("无导出数据");
        }
        ExcelUtil<Energy> util = new ExcelUtil<>(Energy.class);
        util.exportExcel(response, list, "能源类型");
    }
}