package com.estate.expert.admin.service;

import com.alibaba.fastjson.JSONObject;
import com.deepoove.poi.XWPFTemplate;
import com.deepoove.poi.config.Configure;
import com.deepoove.poi.config.ConfigureBuilder;
import com.estate.expert.admin.domain.report.ReportTemplateType;
import com.estate.expert.admin.stat.report.builder.DeviceReportBuilder;
import com.estate.expert.admin.stat.report.builder.SystemReportBuilder;
import com.estate.expert.admin.stat.report.render.CheckInfoTablePolicy;
import lombok.extern.slf4j.Slf4j;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit.jupiter.SpringExtension;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.file.Files;
import java.util.Map;

@Slf4j
@SpringBootTest
@ExtendWith(SpringExtension.class)
@Transactional
public class ReportBuilderTest {

    @Resource
    UserFileService userFileService;

    @Test
    public void testDoc() throws IOException {
        // load file from resource
        var resource = getClass().getClassLoader().getResourceAsStream("templates/test.docx");

        ConfigureBuilder builder = Configure.builder();
        builder.bind("detail_table", new CheckInfoTablePolicy());

        var data = JSONObject.parseObject(Files.readString(new File("/Volumes/Works/tmp/system_report.json").toPath()));
        var checkInfos = data.getJSONObject("评估结果").getJSONObject("安全检查").getJSONArray("分项评估");

        XWPFTemplate template = XWPFTemplate.compile(resource, builder.build()).render(
                Map.of("detail_table", checkInfos)
        );
        template.writeAndClose(new FileOutputStream("/Volumes/Works/tmp/output.docx"));
    }

    @Test
    public void testSystemReport() throws IOException {
        var data = JSONObject.parseObject(Files.readString(new File("/Volumes/Works/tmp/system_report.json").toPath()));
        XWPFTemplate template = new SystemReportBuilder(userFileService).buildReport(data, ReportTemplateType.ThirdPartyReport.getConfig());;
        template.writeAndClose(new FileOutputStream("/Volumes/Works/tmp/system_report.docx"));
    }

    @Test
    public void testDeviceReport() throws IOException {
        var data = JSONObject.parseObject(Files.readString(new File("/Volumes/Works/tmp/device_report.json").toPath()));
        XWPFTemplate template = new DeviceReportBuilder(userFileService).buildReport(data, null);
        template.writeAndClose(new FileOutputStream("/Volumes/Works/tmp/device_report.docx"));
    }

    @Test
    public void testEmptyDeviceReport() throws IOException {
        XWPFTemplate template = new DeviceReportBuilder(userFileService).buildReport(new JSONObject(), null);
        template.writeAndClose(new FileOutputStream("/Volumes/Works/tmp/empty_device_report.docx"));
    }
}