package UnitTest.ImportExcel;

import com.dsideal.QingLong.Util.AsposeUtil;
import com.dsideal.QingLong.Util.ImportUtil;
import com.jfinal.kit.PropKit;
import com.jfinal.kit.StrKit;
import com.jfinal.plugin.activerecord.ActiveRecordPlugin;
import com.jfinal.plugin.activerecord.Record;
import com.jfinal.plugin.activerecord.CaseInsensitiveContainerFactory;
import com.jfinal.plugin.activerecord.dialect.PostgreSqlDialect;
import com.jfinal.plugin.hikaricp.HikariCpPlugin;
import org.apache.poi.ss.usermodel.IndexedColors;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import java.io.*;
import java.util.List;
import java.util.Map;

public class ImportExcelData {
    public static String path = "D:\\dsWork\\QingLong\\src\\main\\resource\\Excel\\";

    //模拟上传文件的文件名称
    public static String upload_excel_filename = "b61a2af2-223f-4058-a675-b212e4dd9487" + ".xlsx";


    public static void main(String[] args) throws IOException {
        //加载License
        AsposeUtil.getLicense();

        //告之配置文件位置
        PropKit.use("application.properties");
        HikariCpPlugin hp = new HikariCpPlugin(PropKit.get("jdbcUrl"), PropKit.get("user"),
                PropKit.get("password").trim(), PropKit.get("driverClassName"));
        hp.start();
        // 配置ActiveRecord插件
        ActiveRecordPlugin arp = new ActiveRecordPlugin(hp);
        //配置默认小写
        arp.setContainerFactory(new CaseInsensitiveContainerFactory(true));
        arp.setDialect(new PostgreSqlDialect());
        arp.start();

        //模板文件
        String f1 = path + File.separator + upload_excel_filename;
        //用户上传的填充完的EXCEL文件
        String f2 = path + File.separator + "school.xlsx";
        //对比两个EXCEL文件 是不是格式一致，也就是是不是上传了正确的模板文件
        int sheetCnt = ImportUtil.getExcelSheetCount(upload_excel_filename);
        for (int i = 0; i < sheetCnt; i++) {
            List<Map.Entry<Integer, Integer>> chayi = ImportUtil.checkYiZhi(f1, f2, i);
            if (chayi.size() > 0) {
                System.out.println("第" + (i + 1) + "个Sheet表检查过程中发现与要求上传的模板不一致，请重新下载模板填写完成后再次上传！");
                return;
            }
        }
        System.out.println("OK,是我下发的模板！");


        //打开EXCEL，进行检查
        InputStream is = new FileInputStream(f2);
        XSSFWorkbook wb = new XSSFWorkbook(is);

        //遍历每个Sheet注册好的信息，对用户上传的数据表进行检查
        boolean flag = true;
        for (int i = 0; i < sheetCnt; i++) {//表
            //Sheet表
            XSSFSheet sheet = wb.getSheetAt(i);
            //数据起始行
            Record r = ImportUtil.getExcelSheetInfo(upload_excel_filename, i);
            int data_start_row = r.getInt("data_start_row");
            //数据有效行数
            int lastRowNum = sheet.getLastRowNum();

            //遍历每一列
            List<Record> list = ImportUtil.getExcelSheetColumnsInfo(upload_excel_filename, i);
            int colIdx = 0;


            for (Record record : list) {//列
                int excel_column_idx = record.getInt("excel_column_idx");//第几列
                String column_type = record.getStr("column_type");//类型
                boolean allow_blank = record.getBoolean("allow_blank");//是不是允许为空

                if (!allow_blank) {
                    // 非空项目未填写
                    for (int j = data_start_row; j <= lastRowNum; j++) {//行
                        XSSFCell cell = sheet.getRow(j).getCell(colIdx);
                        if (StrKit.isBlank(ImportUtil.getCellValue(cell).toString())) {
                            ImportUtil.addComment(wb, cell, "此处不能为空！");
                            ImportUtil.fillYellow(wb,cell, IndexedColors.YELLOW.getIndex());
                            flag = false;
                        }
                    }
                }

                // 下拉列表不是从下拉列表中选择的

                // 检查数据类型是不是和规定的不兼容

                colIdx++;
            }
        }

        // 取消每一个表单的激活状态
        for (int i = 0; i < sheetCnt; i++) wb.getSheetAt(i).setSelected(false);
        // 设置活动表单为第一个表单
        wb.setActiveSheet(0);
        //保存文件
        FileOutputStream fileOut = new FileOutputStream(f2);
        wb.write(fileOut);
        //关闭Excel
        wb.close();

        if (!flag) {
            System.out.println("检查到输入的文件中存在不合法情况，请下载文件后查看处理后，重新上传！");
            return;
        }
        // 全都通过，导入数据

    }
}