package com.wb.excel.api;

import com.wb.excel.api.annotation.ColumnDescription;
import com.wb.excel.api.datatable.WCell;
import com.wb.excel.api.datatable.WColumn;
import com.wb.excel.api.datatable.WSheet;
import com.wb.excel.api.enumeration.DataType;
import com.wb.excel.api.enumeration.Status;
import com.wb.excel.api.style.*;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.Font;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.*;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.lang.reflect.Field;
import java.util.HashSet;
import java.util.Set;

/**
 * Excel帮助类.
 * <p/>
 * Created on 2014/9/27.
 *
 * @author
 * @since 0.1.0
 */
public class WExcel {
    /**
     * Excel表格
     */
    private XSSFWorkbook workbook;

    /**
     * 表头样式
     */
    private CellStyle headCellStyle;

    /**
     * 错误单元格的样式
     */
    private CellStyle errorCellStyle;

    /**
     * 错误的数字单元格的样式(数字靠右)
     */
    private CellStyle errorNumberCellStyle;

    /**
     * 普通单元格的样式
     */
    private CellStyle normalCellStyle;

    /**
     * 普通数字单元格的样式
     */
    private CellStyle normalNumberCellStyle;

    /**
     * 检查消息单元格的样式(最右边一列)
     */
    private CellStyle checkMessageCellStyle;

    /**
     * 检查结果失败的单元格的样式
     */
    private CellStyle checkFailureCellStyle;

    /**
     * 检查结果通过的单元格的样式
     */
    private CellStyle checkSuccessCellStyle;

    /**
     * 正常的文字
     */
    private Font normalFont;

    /**
     * 红色文字
     */
    private Font redFont;

    /**
     * 初始化一个Excel文件.
     */
    public void init() {
        workbook = new XSSFWorkbook();

        //---------- 创建样式 ------------------
        headCellStyle = new HeadCellStyle(workbook).getStyle();
        errorCellStyle = new ErrorCellStyle(workbook).getStyle();
        errorNumberCellStyle = new ErrorNumberCellStyle(workbook).getStyle();
        normalCellStyle = new NormalCellStyle(workbook).getStyle();
        normalNumberCellStyle = new NormalNumberCellStyle(workbook).getStyle();
        checkMessageCellStyle = new CheckMessageCellStyle(workbook).getStyle();
        checkFailureCellStyle = new CheckFailureCellStyle(workbook).getStyle();
        checkSuccessCellStyle = new CheckSuccessCellStyle(workbook).getStyle();

        //----------- 创建字体 ----------------
        normalFont = new NormalFont(workbook).getFont();
        redFont = new RedFont(workbook).getFont();
    }

    /**
     * 默认无参构造方法.
     */
    public WExcel() {
        init();
    }

    /**
     * 将一个DataTable转换为Excel对象,不显示检查结果和错误消息栏.<br/>
     * 通常用于导出报表.
     * {@link WExcel#WExcel(boolean flag, WSheet... dataTable)}
     *
     * @param tables 要导出的DataTable集合
     */
    public WExcel(WSheet... tables) {
        workbook = this.initExcel(false, tables);
    }

    /**
     * 将若干DataTable转换为Excel对象.
     *
     * @param flag   是否需要添加检查结果和错误消息栏（true，会添加；false，不会添加）
     * @param tables DataTable列表
     */
    public WExcel(boolean flag, WSheet... tables) {
        workbook = this.initExcel(flag, tables);
    }

    public XSSFWorkbook initExcel(boolean flag, WSheet... tables) {
        init();
        Set<String> nameSet = new HashSet<>(tables.length);
        for (WSheet table : tables) {
            String name = table.getName();
            int index = 1;
            while (nameSet.contains(name)) {
                name = table.getName() + "(" + index + ")";
                index++;
            }
            nameSet.add(name);

            //创建一个Sheet表
            XSSFSheet sheet = workbook.createSheet(name);
            sheet.setDefaultRowHeightInPoints(20);
            Row firstRow = sheet.createRow(0); // 下标为0的行开始
            XSSFDrawing drawing = sheet.createDrawingPatriarch();

            //----------------表头--------------------
            if (flag) {
                // 检查结果栏
                org.apache.poi.ss.usermodel.Cell firstCell = firstRow.createCell(0);
                String columnName = WSheet.CHECK_STATUS_NAME;
                firstCell.setCellStyle(headCellStyle);
                firstCell.setCellValue(new XSSFRichTextString(columnName));
                sheet.setColumnWidth(0, (4 + 8) * 256);
            }
            // 数据栏
            int hiddenNumber = 0;
            for (int j = 0; j < table.getColumnIndex(); j++) {
                WColumn WColumn = table.getWColumns()[j];
                if (WColumn.isHidden()) {
                    hiddenNumber++;
                    continue;
                }
                int k = j - hiddenNumber;
                if (flag) {
                    k++;
                }
                org.apache.poi.ss.usermodel.Cell firstCell = firstRow.createCell(k);
                String columnName = WColumn.getName();
                XSSFRichTextString textString;
                if (WColumn.isRequired()) {
                    textString = new XSSFRichTextString("*" + columnName);
                    textString.applyFont(0, 1, redFont);
                    textString.applyFont(1, textString.length(), normalFont);
                } else {
                    textString = new XSSFRichTextString(columnName);
                    textString.applyFont(normalFont);
                }
                StringBuilder sb = new StringBuilder();
                sb.append(WColumn.getDescription()).append("\n");
                if (WColumn.getDataType() != DataType.STRING) {
                    // 如果数据类型不是字符串类型，添加特殊数据类型的说明信息。
                    Field[] fields = DataType.class.getDeclaredFields();
                    for (Field field : fields) {
                        if (field.getName().equals(WColumn.getDataType().name())) {
                            if (field.isAnnotationPresent(ColumnDescription.class)) {

                                // 获取声明字段上的Description信息。
                                ColumnDescription columnDescription = field.getAnnotation(ColumnDescription.class);
                                if (columnDescription.value() != null) {
                                    sb.append(columnDescription.value());
                                }
                            }
                        }
                    }
                }

                // 如果填写了注释信息
                if (sb.length() > 1) {
                    XSSFClientAnchor anchor = new XSSFClientAnchor(0, 0, 0, 0, (short) 3, 3, (short) 5, 6);
                    XSSFComment comment = drawing.createCellComment(anchor);
                    comment.setString(sb.toString());
                    firstCell.setCellComment(comment);
                }
                firstCell.setCellValue(textString);
                firstCell.setCellStyle(headCellStyle);
                sheet.setColumnWidth(k, (4 + WColumn.getCellWidth()) * 256);
            }

            // 错误消息栏
            if (flag) {
                org.apache.poi.ss.usermodel.Cell firstCell = firstRow.createCell(table.getColumnIndex() + 1 - hiddenNumber);
                String columnName = WSheet.CHECK_STATUS_RESULT;
                firstCell.setCellStyle(headCellStyle);
                firstCell.setCellValue(new XSSFRichTextString(columnName));
                sheet.setColumnWidth(table.getColumnIndex() + 1, (4 + 10) * 256);
            }

            // 冻结第一行
            //sheet.createFreezePane(255,1);

            //------------------数据--------------------
            for (int i = 0; i < table.getRowIndex(); i++) {
                Row row = sheet.createRow(i + 1);
                boolean rowFlag = true;

                //如果该行有错误
                if (flag) {
                    org.apache.poi.ss.usermodel.Cell xssfCell = row.createCell(0);
                    if (table.getRowError(i) != null) {
                        rowFlag = false;
                        xssfCell.setCellValue("不通过");
                        xssfCell.setCellStyle(checkFailureCellStyle);
                    } else {
                        xssfCell.setCellValue("通过");
                        xssfCell.setCellStyle(checkSuccessCellStyle);
                    }
                }

                int j = 0;
                hiddenNumber = 0;
                for (; j < table.getColumnIndex(); j++) {
                    WColumn WColumn = table.getWColumns()[j];
                    if (WColumn.isHidden()) {
                        hiddenNumber++;
                        continue;
                    }
                    int k = j - hiddenNumber;
                    if (flag) {
                        k++;
                    }

                    org.apache.poi.ss.usermodel.Cell xssfCell = row.createCell(k);
                    WCell WCell = table.getCell(i, j);
                    if (null == WCell) {
                        continue;
                    }
                    String value = WCell.getValue();
                    xssfCell.setCellValue(value);

                    // 如果该列是数字类型,则靠右
                    if (table.getWColumns()[j].getDataType() == DataType.DECIMAL
                            || table.getWColumns()[j].getDataType() == DataType.NUMBER) {
                        if (flag && !WCell.getStatus().equals(Status.PASS)) {
                            xssfCell.setCellStyle(errorNumberCellStyle);
                        } else {
                            xssfCell.setCellStyle(normalNumberCellStyle);
                        }
                    } else {
                        if (flag && !WCell.getStatus().equals(Status.PASS)) {
                            xssfCell.setCellStyle(errorCellStyle);
                        } else {
                            xssfCell.setCellStyle(normalCellStyle);
                        }
                    }
                }

                if (flag && !rowFlag) {
                    org.apache.poi.ss.usermodel.Cell xssfCell = row.createCell(j + 1 - hiddenNumber);
                    xssfCell.setCellValue(table.getRowError(i));
                    xssfCell.setCellStyle(checkMessageCellStyle);
                }
            }
        }
        return workbook;
    }

    /**
     * 得到已生成好的Excel文件的字节流信息
     *
     * @return 字节流信息
     * @throws IOException
     */
    public byte[] getBytes() throws IOException {
        if (workbook != null) {
            ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
            workbook.write(outputStream);
            return outputStream.toByteArray();
        }
        return null;
    }

    //------------ getter & setter --------------------
    public Workbook getWorkbook() {
        return workbook;
    }

    public void setWorkbook(XSSFWorkbook workbook) {
        this.workbook = workbook;
    }
}