package com.taobao.util;

/**
 * @Author TAO
 * @ClassName ExcelData
 * @Description TODO
 * @Date 2019/1/9 15:02
 * @Version 1.0
 */

import org.apache.poi.hssf.usermodel.HSSFDataFormat;
import org.apache.poi.hssf.usermodel.HSSFDateUtil;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.web.multipart.MultipartFile;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.List;

/**
 * 解析excel 上傳資料
 */
public class AnalysisExcelData {


    /**
     * @Author
     * @Description //TODO
     * @Date 2019/8/15 12:14
     * @Param file ：上傳的excel檔案
     * @return  * @param null
     */
    public static List<List> getExcelData(MultipartFile file) throws IOException {
        checkFile(file);
        //獲得Workbook工作薄物件
        Workbook workbook = getWorkBook(file);
        //建立返回物件，把每行中的值作為一個數組，所有行作為一個集合返回
        
        List<List> list = new ArrayList<>();
        if (workbook != null) {
            for (int sheetNum = 0; sheetNum < workbook.getNumberOfSheets(); sheetNum++) {
                //獲得當前sheet工作表
                Sheet sheet = workbook.getSheetAt(sheetNum);
                if (sheet == null) {
                    continue;
                }
                //獲得當前sheet的開始行
                int firstRowNum = sheet.getFirstRowNum();
                //獲得當前sheet的結束行
                int lastRowNum = sheet.getLastRowNum();
                //迴圈除了所有行,如果要迴圈除第一行以外的就firstRowNum+1
                for (int rowNum = firstRowNum + 1; rowNum <= lastRowNum; rowNum++) {
                    //獲得當前行
                    Row row = sheet.getRow(rowNum);
                    if (row == null) {
                        continue;
                    }
                    //獲得當前行的開始列
                    int firstCellNum = row.getFirstCellNum();
                    //獲得當前行的列數
                    int lastCellNum = row.getLastCellNum();
                    if (lastCellNum > 0) {
                        ArrayList<Object> cellValues = new ArrayList<>();
                        //迴圈當前行
                        for (int cellNum = firstCellNum; cellNum < lastCellNum; cellNum++) {
                            Cell cell = row.getCell(cellNum);
                            cellValues.add(getCellValue(cell));
                        }
                        list.add(cellValues);
                    }
                }
            }
        }
        
        return list;
    }


    /**
     * 檢查檔案
     *
     * @param file
     * @throws IOException
     */
    public static void checkFile(MultipartFile file) throws IOException {
        //判斷檔案是否存在
        if (null == file) {
            System.err.println("檔案不存在！");
        }
        //獲得檔名
        String fileName = file.getOriginalFilename();
        //判斷檔案是否是excel檔案
        if (!fileName.endsWith("xls") && !fileName.endsWith("xlsx")) {
            System.err.println("不是excel檔案");
        }
    }

    public static Workbook getWorkBook(MultipartFile file) {
        //獲得檔名
        String fileName = file.getOriginalFilename();
        //建立Workbook工作薄物件，表示整個excel
        Workbook workbook = null;
        try {
            //獲取excel檔案的io流
            InputStream is = file.getInputStream();
            //根據檔案字尾名不同(xls和xlsx)獲得不同的Workbook實現類物件
            if (fileName.endsWith("xls")) {
                //2003
                workbook = new HSSFWorkbook(is);
            } else if (fileName.endsWith("xlsx")) {
                //2007 及2007以上
                workbook = new XSSFWorkbook(is);
            }
        } catch (IOException e) {
            e.getMessage();
        }
        return workbook;
    }

    public static String getCellValue(Cell cell) {
        String cellValue = "";
        if (cell == null) {
            return cellValue;
        }
        //判斷資料的型別
        //判斷資料的型別
        switch (cell.getCellType()) {
            case Cell.CELL_TYPE_NUMERIC: //數字
                cellValue = stringDateProcess(cell);
                break;
            case Cell.CELL_TYPE_STRING: //字串
                cellValue = String.valueOf(cell.getStringCellValue());
                break;
            case Cell.CELL_TYPE_BOOLEAN: //Boolean
                cellValue = String.valueOf(cell.getBooleanCellValue());
                break;
            case Cell.CELL_TYPE_FORMULA: //公式
                cellValue = String.valueOf(cell.getCellFormula());
                break;
            case Cell.CELL_TYPE_BLANK: //空值
                cellValue = "";
                break;
            case Cell.CELL_TYPE_ERROR: //故障
                cellValue = "非法字元";
                break;
            default:
                cellValue = "未知型別";
                break;
        }
        return cellValue;
    }


    public static String stringDateProcess(Cell cell) {
        String result = new String();
        if (HSSFDateUtil.isCellDateFormatted(cell)) {// 處理日期格式、時間格式
            SimpleDateFormat sdf = null;
            if (cell.getCellStyle().getDataFormat() == HSSFDataFormat.getBuiltinFormat("h:mm")) {
                sdf = new SimpleDateFormat("HH:mm");
            } else {// 日期
                sdf = new SimpleDateFormat("yyyy-MM-dd hh:mm:ss");
            }
            Date date = cell.getDateCellValue();
            result = sdf.format(date);
        } else if (cell.getCellStyle().getDataFormat() == 58) {
            // 處理自定義日期格式：m月d日(通過判斷單元格的格式id解決，id的值是58)
            SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd hh:mm:ss");
            double value = cell.getNumericCellValue();
            Date date = org.apache.poi.ss.usermodel.DateUtil
                    .getJavaDate(value);
            result = sdf.format(date);
        } else {
            double value = cell.getNumericCellValue();
            CellStyle style = cell.getCellStyle();
            DecimalFormat format = new DecimalFormat();
            String temp = style.getDataFormatString();
            // 單元格設定成常規
            if (temp.equals("General")) {
                format.applyPattern("#");
            }
            result = format.format(value);
        }

        return result;
    }

    public static InputStream convertorStream(Workbook workbook) {
        InputStream in = null;
        try {
            //臨時緩衝區
            ByteArrayOutputStream out = new ByteArrayOutputStream();
            //建立臨時檔案
            workbook.write(out);
            byte[] bookByteAry = out.toByteArray();
            in = new ByteArrayInputStream(bookByteAry);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return in;
    }


}