package org.ctzk.jcpc.utils;

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.write.metadata.WriteSheet;
import com.alibaba.excel.write.metadata.style.WriteCellStyle;
import com.alibaba.excel.write.metadata.style.WriteFont;
import com.alibaba.excel.write.style.HorizontalCellStyleStrategy;
import org.apache.poi.ss.usermodel.BorderStyle;
import org.apache.poi.ss.usermodel.HorizontalAlignment;
import org.apache.poi.ss.usermodel.IndexedColors;
import org.apache.poi.ss.usermodel.VerticalAlignment;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.OutputStream;
import java.util.List;
import java.util.Map;

/**
 * org.ctzk.jcpc.utils
 *
 * @author luozhen
 * @version V1.0
 * @date 2020-2-12 05:15
 * @description
 */

public class EasyExcelUtil {

    private static final Logger log = LoggerFactory.getLogger(EasyExcelUtil.class);

    /**
     * 導出文件時為Writer生成OutputStream
     */
    private static OutputStream getOutputStream(HttpServletResponse response, String fileName) {
        OutputStream out = null;
        String encodedFileName = "";
        try {
            if (fileName.toLowerCase().endsWith(".xlsx")) {
                encodedFileName = new String(fileName.getBytes("UTF-8"), "ISO8859-1");
            } else {
                encodedFileName = new String(fileName.getBytes("UTF-8"), "ISO8859-1") + ".xlsx";
            }
            response.setHeader("Content-Disposition", "attachment;filename=\"" + encodedFileName + "\"");
            response.setCharacterEncoding("utf-8");
            response.setContentType("application/vnd.ms-excel");
            out = response.getOutputStream();
        } catch (IOException e) {
            e.printStackTrace();
            log.error("創建文件失??！");
        }
        return out;
    }

    /**
     * 設置樣式
     *
     * @return
     */
    private static HorizontalCellStyleStrategy cellStyle() {
        // 頭的策略
        WriteCellStyle headWriteCellStyle = new WriteCellStyle();
        // 背景設置為紅色
        headWriteCellStyle.setFillForegroundColor(IndexedColors.RED.getIndex());
        // 設置底邊框;
        headWriteCellStyle.setBorderBottom(BorderStyle.THIN);
        // 設置底邊框顏色;
        headWriteCellStyle.setBottomBorderColor(IndexedColors.BLACK.getIndex());
        // 設置左邊框;
        headWriteCellStyle.setBorderLeft(BorderStyle.THIN);
        // 設置左邊框顏色;
        headWriteCellStyle.setLeftBorderColor(IndexedColors.BLACK.getIndex());
        // 設置右邊框;
        headWriteCellStyle.setBorderRight(BorderStyle.THIN);
        // 設置右邊框顏色;
        headWriteCellStyle.setRightBorderColor(IndexedColors.BLACK.getIndex());
        // 設置頂邊框;
        headWriteCellStyle.setBorderTop(BorderStyle.THIN);
        // 設置頂邊框顏色;
        headWriteCellStyle.setTopBorderColor(IndexedColors.BLACK.getIndex());
        // 設置自動換行
        headWriteCellStyle.setWrapped(true);
        // 設置垂直居中
        headWriteCellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
        // 設置水平居中
        headWriteCellStyle.setHorizontalAlignment(HorizontalAlignment.CENTER);
        WriteFont headWriteFont = new WriteFont();
        headWriteFont.setFontHeightInPoints((short) 15);
        headWriteCellStyle.setWriteFont(headWriteFont);
        // 內容的策略
        WriteCellStyle contentWriteCellStyle = new WriteCellStyle();
        // 設置底邊框;
        contentWriteCellStyle.setBorderBottom(BorderStyle.THIN);
        // 設置底邊框顏色;
        contentWriteCellStyle.setBottomBorderColor(IndexedColors.BLACK.getIndex());
        // 設置左邊框;
        contentWriteCellStyle.setBorderLeft(BorderStyle.THIN);
        // 設置左邊框顏色;
        contentWriteCellStyle.setLeftBorderColor(IndexedColors.BLACK.getIndex());
        // 設置右邊框;
        contentWriteCellStyle.setBorderRight(BorderStyle.THIN);
        // 設置右邊框顏色;
        contentWriteCellStyle.setRightBorderColor(IndexedColors.BLACK.getIndex());
        // 設置頂邊框;
        contentWriteCellStyle.setBorderTop(BorderStyle.THIN);
        // 設置頂邊框顏色;
        contentWriteCellStyle.setTopBorderColor(IndexedColors.BLACK.getIndex());
        // 設置自動換行
        contentWriteCellStyle.setWrapped(true);
        // 設置垂直居中
        contentWriteCellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
        // 設置水平居中
        contentWriteCellStyle.setHorizontalAlignment(HorizontalAlignment.CENTER);
        WriteFont contentWriteFont = new WriteFont();
        // 字體大小
        contentWriteFont.setFontHeightInPoints((short) 12);
        contentWriteCellStyle.setWriteFont(contentWriteFont);
        // 這個策略是 頭是頭的樣式 內容是內容的樣式 其他的策略可以自己實現
        HorizontalCellStyleStrategy horizontalCellStyleStrategy = new HorizontalCellStyleStrategy(headWriteCellStyle, contentWriteCellStyle);
        return horizontalCellStyleStrategy;
    }

    /**
     * 通過模板,創建excel導出
     *
     * @param response
     * @param dataList
     * @param templateFileName
     * @param fileName
     */
    public static void writeExcelForTemplate(HttpServletResponse response, List dataList, Map map, String templateFileName, String fileName) {
        ExcelWriter excelWriter = EasyExcel.write(getOutputStream(response, fileName)).registerWriteHandler(cellStyle()).withTemplate(templateFileName).build();
        WriteSheet writeSheet = EasyExcel.writerSheet().build();
        excelWriter.fill(dataList, writeSheet);
        excelWriter.fill(map, writeSheet);
        excelWriter.finish();
    }

    /**
     * 自定義表頭導出excel
     *
     * @param response
     * @param headList
     * @param dataList
     * @param fileName
     */
    public static void writeExcel(HttpServletResponse response, List headList, List dataList, String fileName) {
        EasyExcel.write(getOutputStream(response, fileName)).head(headList).sheet("Sheet1").doWrite(dataList);
    }
}