package com.estate.expert.admin.stat.report.builder;

import com.alibaba.fastjson.JSONObject;
import com.deepoove.poi.XWPFTemplate;
import com.deepoove.poi.config.Configure;
import com.deepoove.poi.config.ConfigureBuilder;
import com.deepoove.poi.plugin.table.LoopRowTableRenderPolicy;
import com.estate.expert.admin.service.UserFileService;
import com.estate.expert.admin.stat.report.render.CustomPictureRenderPolicy;
import com.estate.expert.admin.stat.report.render.DeviceTablePolicy;
import lombok.AllArgsConstructor;
import org.springframework.core.io.ClassPathResource;

import java.io.IOException;
import java.util.Map;


/**
 * 设备风险评估报告构建器
 */
@AllArgsConstructor
public class DeviceReportBuilder implements ReportBuilderBase {

    protected UserFileService userFileService;

    /**
     * 构建设备风险评估报告
     * @param data 数据
     * @param templateConfig 模板配置
     * @return 设备风险评估报告
     */
    @Override
    public XWPFTemplate buildReport(JSONObject data, Map<String, Integer> templateConfig) {
        var res = new ClassPathResource("templates/设备风险评估报告.docx");

        // 将data.评估结果.分类评估.详情 复制
        // 避免渲染的时候的策略冲突
        if (data.containsKey("评估结果") && data.getJSONObject("评估结果").containsKey("分类评估")) {
            var evalResult = data.getJSONObject("评估结果").getJSONObject("分类评估");
            var details = evalResult.getJSONArray("详情");
            evalResult.put("详情2", details);
        }

        ConfigureBuilder builder = Configure.builder();
        builder.bind("data.设备信息", new LoopRowTableRenderPolicy());
        builder.bind("data.评估结果.分类评估", new LoopRowTableRenderPolicy());
        builder.bind("data.评估结果.分类评估.详情2", new LoopRowTableRenderPolicy());
        builder.bind("data.评估结果.隐患等级分布汇总.分项", new LoopRowTableRenderPolicy());
        builder.bind("分项隐患", new LoopRowTableRenderPolicy());
        builder.bind("分项问题", new DeviceTablePolicy());
//        builder.bind("data.设备信息", new DeviceInfoTablePolicy());

        builder.addPlugin('%', new CustomPictureRenderPolicy(userFileService));

        try {
            return XWPFTemplate.compile(res.getInputStream(), builder.build()).render(
                    Map.of( "data", data
                    ));
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }
}