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

import cn.idev.excel.FastExcelFactory;
import cn.iocoder.yudao.framework.common.exception.ErrorCode;
import cn.iocoder.yudao.framework.common.util.http.HttpUtils;
import cn.iocoder.yudao.framework.excel.core.handler.ColumnWidthMatchStyleStrategy;
import cn.iocoder.yudao.module.mes.controller.admin.energy.vo.*;
import jakarta.annotation.Resource;
import jakarta.servlet.http.HttpServletResponse;
import org.springframework.stereotype.Service;
import java.io.IOException;
import java.util.*;
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;

@Service
public class EnergyExportService {
    @Resource private EnergyRecordService energyRecordService;
    public void records(EnergyRecordPageReqVO req, HttpServletResponse response) throws IOException {
        req.setPageNo(1); req.setPageSize(50_001);
        var page = energyRecordService.page(req);
        if (page.getTotal() > 50_000) throw exception(new ErrorCode(1_040_952_000, "单次最多导出50000条，请缩小日期范围"));
        List<List<Object>> rows = page.getList().stream().map(r -> Arrays.<Object>asList(r.getRecordDate().toString(),
                EnergyAlertService.energyName(r.getEnergyType()), r.getUsageArea(), r.getProductCategory(), r.getUsagePurpose(),
                EnergyAlertService.unit(r.getEnergyType()), r.getQuantity(), r.getUnitPrice(), r.getAmount(), r.getRecorder(), r.getRemark())).toList();
        write(response, "能耗台账.xlsx", List.of(new Sheet("能耗台账",
                List.of("日期", "能源", "使用区域", "产品类别", "用途", "用量单位", "用量", "单价（元/单位）", "费用（元）", "录入人", "备注"), rows)));
    }
    public void analysis(EnergyAnalysisReqVO req, HttpServletResponse response) throws IOException {
        var data = energyRecordService.analysis(req);
        var head = List.of("统计项", "能源", "用量单位", "用量", "费用（元）", "录入条数");
        List<List<Object>> summary = data.getSummary().stream().map(r -> row("合计", r)).toList();
        List<List<Object>> trend = data.getTrend().stream().map(r -> row(r.getLabel(), r)).toList();
        List<List<Object>> areas = data.getAreas().stream().map(r -> row(r.getLabel(), r)).toList();
        List<List<Object>> scope = List.of(Arrays.asList("开始日期", req.getStartDate().toString()),
                Arrays.asList("结束日期", req.getEndDate().toString()), Arrays.asList("汇总方式", "MONTH".equals(req.getGroupBy()) ? "按月" : "按日"),
                Arrays.asList("能源", req.getEnergyType() == null || req.getEnergyType().isEmpty() ? "全部" : EnergyAlertService.energyName(req.getEnergyType())),
                Arrays.asList("区域关键词", Objects.toString(req.getUsageArea(), "")), Arrays.asList("产品类别", Objects.toString(req.getProductCategory(), "")),
                Arrays.asList("总费用（元）", data.getTotalAmount()), Arrays.asList("录入条数", data.getRecordCount()));
        write(response, "能耗分析.xlsx", List.of(new Sheet("统计口径", List.of("项目", "内容"), scope),
                new Sheet("能源汇总", head, summary), new Sheet("日期趋势", head, trend), new Sheet("区域汇总", head, areas)));
    }
    private List<Object> row(String label, EnergyAnalysisRespVO.Summary r) {
        return Arrays.asList(label, EnergyAlertService.energyName(r.getEnergyType()), EnergyAlertService.unit(r.getEnergyType()), r.getQuantity(), r.getAmount(), r.getRecordCount());
    }
    private record Sheet(String name, List<String> headers, List<List<Object>> rows) {}
    private void write(HttpServletResponse response, String filename, List<Sheet> sheets) throws IOException {
        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
        response.setHeader("Content-Disposition", "attachment;filename=" + HttpUtils.encodeUtf8(filename));
        try (var writer = FastExcelFactory.write(response.getOutputStream()).autoCloseStream(false)
                .registerWriteHandler(new ColumnWidthMatchStyleStrategy()).build()) {
            for (int i = 0; i < sheets.size(); i++) {
                var sheet = sheets.get(i);
                writer.write(sheet.rows(), FastExcelFactory.writerSheet(i, sheet.name())
                        .head(sheet.headers().stream().map(List::of).toList()).build());
            }
        }
    }
}