package com.qianwen.smartman.modules.mdc.utils;

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.metadata.CellData;
import com.alibaba.excel.metadata.Head;
import com.alibaba.excel.write.metadata.WriteSheet;
import com.alibaba.excel.write.metadata.WriteTable;
import com.alibaba.excel.write.metadata.holder.WriteSheetHolder;
import com.alibaba.excel.write.metadata.style.WriteCellStyle;
import com.alibaba.excel.write.style.HorizontalCellStyleStrategy;
import com.alibaba.excel.write.style.column.AbstractColumnWidthStyleStrategy;
import java.io.File;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.VerticalAlignment;
import com.qianwen.core.tool.utils.Charsets;
import com.qianwen.smartman.modules.mdc.dto.RowRangeDTO;
import com.qianwen.smartman.modules.mdc.strategy.BizMergeStrategy;
import com.qianwen.smartman.modules.mdc.strategy.CustomCellWriteHandler;
import com.qianwen.smartman.modules.mdc.strategy.ExcelWidthStyleStrategy;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.commons.CommonsMultipartFile;

/* loaded from: blade-api.jar:BOOT-INF/classes/org/springblade/modules/mdc/utils/ExcelStrategyUtil.class */
public final class ExcelStrategyUtil {
    private ExcelStrategyUtil() {
        throw new UnsupportedOperationException("This is a utility class and cannot be instantiated");
    }

    public static Map<String, List<RowRangeDTO>> addMerStrategy(List<List<Object>> data) {
        Map<String, List<RowRangeDTO>> strategyMap = new HashMap<>(16);
        List<Object> preExcelDTO = null;
        int i = 1;
        while (i < data.size()) {
            List<Object> currDTO = data.get(i);
            if (preExcelDTO != null && currDTO.get(0).equals(preExcelDTO.get(0))) {
                fillStrategyMap(strategyMap, i - 1);
            }
            preExcelDTO = currDTO;
            i++;
        }
        fillStrategyMap(strategyMap, i - 1);
        return strategyMap;
    }

    public static List<Integer> addColumnMerStrategy(List<List<Object>> data) {
        List<Integer> strategyList = new ArrayList<>();
        for (int i = 0; i < data.size(); i++) {
            List<Object> currDTO = data.get(i);
            if (currDTO.get(0).equals(currDTO.get(1))) {
                strategyList.add(Integer.valueOf(i));
            }
        }
        return strategyList;
    }

    private static void fillStrategyMap(Map<String, List<RowRangeDTO>> strategyMap, int index) {
        List<RowRangeDTO> rowRangeDtoList = strategyMap.get("0") == null ? new ArrayList<>() : strategyMap.get("0");
        boolean flag = false;
        for (RowRangeDTO dto : rowRangeDtoList) {
            if (dto.getEnd().intValue() == index) {
                dto.setEnd(Integer.valueOf(index + 1));
                flag = true;
            }
        }
        if (!flag) {
            rowRangeDtoList.add(new RowRangeDTO(Integer.valueOf(index), Integer.valueOf(index + 1)));
        }
        strategyMap.put("0", rowRangeDtoList);
    }

    public static MultipartFile export(String excelFileName, String sheetName, List<List<String>> head, List<List<Object>> data) throws Exception {
        Map<String, List<RowRangeDTO>> strategyMap = addMerStrategy(data);
        DiskFileItemFactory factory = new DiskFileItemFactory();
        factory.setRepository(new File(System.getProperty("java.io.tmpdir")));
        BizMergeStrategy bizMergeStrategy = new BizMergeStrategy(strategyMap);
        WriteCellStyle contentWriteCellStyle = new WriteCellStyle();
        WriteCellStyle headWriteCellStyle = new WriteCellStyle();
        contentWriteCellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
        HorizontalCellStyleStrategy horizontalCellStyleStrategy = new HorizontalCellStyleStrategy(headWriteCellStyle, contentWriteCellStyle);
        String fileName = URLEncoder.encode(excelFileName, Charsets.UTF_8.name());
        FileItem fileItem = factory.createItem("excel", "application/vnd.ms-excel", true, fileName);
        OutputStream os = fileItem.getOutputStream();
        EasyExcel.write(os).sheet(sheetName).head(head).registerWriteHandler(bizMergeStrategy).registerWriteHandler(horizontalCellStyleStrategy).registerWriteHandler(new CustomCellWriteHandler()).doWrite(data);
        return new CommonsMultipartFile(fileItem);
    }

    public static MultipartFile customerExport(String excelFileName, String sheetName, List<List<String>> head, List<List<Object>> data) throws Exception {
        String fileName = new String(excelFileName.getBytes(), StandardCharsets.UTF_8);
        DiskFileItemFactory factory = new DiskFileItemFactory();
        factory.setRepository(new File(System.getProperty("java.io.tmpdir")));
        FileItem fileItem = factory.createItem("excel", "application/vnd.ms-excel", true, fileName);
        OutputStream os = fileItem.getOutputStream();
        EasyExcel.write(os).sheet(sheetName).head(head).registerWriteHandler(new ExcelWidthStyleStrategy()).doWrite(data);
        return new CommonsMultipartFile(fileItem);
    }

    public static MultipartFile customerStyleExport(String excelFileName, String sheetName, List<List<String>> head, List<List<Object>> data) throws Exception {
        String fileName = new String(excelFileName.getBytes(), StandardCharsets.UTF_8);
        DiskFileItemFactory factory = new DiskFileItemFactory();
        factory.setRepository(new File(System.getProperty("java.io.tmpdir")));
        FileItem fileItem = factory.createItem("excel", "application/vnd.ms-excel", true, fileName);
        OutputStream os = fileItem.getOutputStream();
        EasyExcel.write(os).sheet(sheetName).head(head).registerWriteHandler(new AbstractColumnWidthStyleStrategy() { // from class: org.springblade.modules.mdc.utils.ExcelStrategyUtil.1
            protected void setColumnWidth(WriteSheetHolder writeSheetHolder, List<CellData> cellDataList, Cell cell, Head head2, Integer relativeRowIndex, Boolean isHead) {
                Sheet sheet = writeSheetHolder.getSheet();
                int columnIndex = cell.getColumnIndex();
                sheet.setColumnWidth(columnIndex, 4000);
                sheet.setDefaultRowHeight((short) 600);
            }
        }).doWrite(data);
        return new CommonsMultipartFile(fileItem);
    }

    public static MultipartFile customerTwoHeadExport(String excelFileName, String sheetName, List<List<String>> head0, List<List<String>> head1, List<List<Object>> data0, List<List<Object>> data1) throws Exception {
        String fileName = new String(excelFileName.getBytes(), StandardCharsets.UTF_8);
        DiskFileItemFactory factory = new DiskFileItemFactory();
        factory.setRepository(new File(System.getProperty("java.io.tmpdir")));
        FileItem fileItem = factory.createItem("excel", "application/vnd.ms-excel", true, fileName);
        OutputStream os = fileItem.getOutputStream();
        ExcelWriter excelWriter = EasyExcel.write(os).build();
        WriteSheet writeSheet = EasyExcel.writerSheet(sheetName).needHead(Boolean.FALSE).build();
        WriteTable writeTable0 = EasyExcel.writerTable(0).head(head0).needHead(true).build();
        WriteTable writeTable1 = EasyExcel.writerTable(1).head(head1).needHead(true).build();
        excelWriter.write(data0, writeSheet, writeTable0);
        excelWriter.write(data1, writeSheet, writeTable1);
        excelWriter.finish();
        return new CommonsMultipartFile(fileItem);
    }
}