package com.lxyer.excel.poi;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.poifs.filesystem.OfficeXmlFileException;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.lang.reflect.Field;
import java.text.SimpleDateFormat;
import java.util.*;

/**
 * 使用poi报表导出工具类
 * 把poi的一个调用接口抽出来，便于导出功能的管理
 * @author LiangXianYou lxy208@126.com
 * @param 
 *
 */
public class ExcelKit {
	
	/**
	 * <b>导出list中map做载体的数据到excel</b><br>
	 * 参数说明：<br>
	 * 	list:存放了Map数据的集合<br>
	 * 	hdNames:表头列名<br>
	 * 	hds:对应表头的数据KEY<br>
	 *  xlsName:导出文件名<br>
	 * @author LiangXianYou
	 * @date 2015-6-25 上午10:22:51
	 */
	public static <T> boolean exportByMap(List<Map<String, Object>> list, String[] hdNames, String[] hds, String xlsName, HttpServletRequest request, HttpServletResponse response) throws Exception{
		
		Workbook wb = new XSSFWorkbook();			//创建工作薄
		Sheet sheet = wb.createSheet();				//创建工作表
		sheet.autoSizeColumn(( short ) 0 );			//自适应宽度
		//写入表头---Excel的第一行数据
		Row nRow = sheet.createRow(0);				//创建行
		for(int i=0;i<hdNames.length;i++){
			Cell nCell = nRow.createCell(i);			//创建单元格
			nCell.setCellValue(hdNames[i]);
		}  
		
		//写入每一行数据---一条记录就是一行数据
		for(int i=0;i<list.size();i++){
			for(int j=0;j<hds.length;j++){
				Object o = list.get(i).get(hds[j]);		//得到列的值
				data2Excel(sheet, o, i+1, j);			//将值写入Excel
			}
		}
		return downloadExcel(wb, xlsName, request, response);
	}

	/**
	 * Excels导出多个sheet
	 * @author LiangXianYou
	 * @date 2015-6-16 下午5:56:56
	 */
	//map:data,sheetName,hds,hdNames,xlsName;request,response
	public static <T> boolean exportExcels(List<Map<String, Object>> list,String xlsName, HttpServletRequest request, HttpServletResponse response) throws Exception{

		Workbook wb = new XSSFWorkbook();			//创建工作薄
		for(int i=0;i<list.size();i++){
			Map<String, Object> map = list.get(i);
			Sheet sheet = wb.createSheet();				//创建工作表
			String sheetName = (String) map.get("sheetName");
			wb.setSheetName(i, sheetName);
			//写入表头---Excel的第一行数据
			Row nRow = sheet.createRow(0);				//创建行
			String[] hdNames = (String[]) map.get("hdNames");
			for(int j=0;j<hdNames.length;j++){
				Cell nCell = nRow.createCell(j);			//创建单元格
				nCell.setCellValue(hdNames[j]);
			}
			
			//写入每一行数据---一条记录就是一行数据
			@SuppressWarnings("unchecked")
			List<T> data = (List<T>) map.get("data");
			String[] hds = (String[]) map.get("hds");
			for(int j=0;j<data.size();j++){
				for(int k=0;k<hds.length;k++){
					Object o = getFieldValue(data.get(j), hds[k]);	//得到列的值
					data2Excel(sheet, o, j+1, k);					//将值写入Excel
				}
			}
		}
		
		return downloadExcel(wb, xlsName, request, response);
	}
	
	/**
	 * 参数说明：
	 * list：list数据集合
	 * hdNames：表头需要显示的名称
	 * hds：表头对应的对象属性名称，和 hdNames一一对应
	 * xlsName：导出Excel的预定义名称
	 * request:	HttpServletRequest
	 * response：HttpServletResponse
	 * 通过数据构建Excel
	 * @author LiangXianYou
	 * @throws Exception 
	 * @date 2015-3-13 上午11:00:30
	 */
	public static <T> boolean exportExcel(List<T> list, String[] hdNames, String[] hds, String xlsName, HttpServletRequest request, HttpServletResponse response) throws Exception{

		Workbook wb = new XSSFWorkbook();			//创建工作薄
		Sheet sheet = wb.createSheet();				//创建工作表
		sheet.autoSizeColumn(( short ) 0 );			//自适应宽度
		//写入表头---Excel的第一行数据
		Row nRow = sheet.createRow(0);				//创建行
		for(int i=0;i<hdNames.length;i++){
			Cell nCell = nRow.createCell(i);			//创建单元格
			nCell.setCellValue(hdNames[i]);
		}
		
		//写入每一行数据---一条记录就是一行数据
		for(int i=0;i<list.size();i++){
			for(int j=0;j<hds.length;j++){
				Object o = getFieldValue(list.get(i), hds[j]);	//得到列的值
				data2Excel(sheet, o, i+1, j);						//将值写入Excel
			}
		}
		return downloadExcel(wb, xlsName, request, response);
	}
	
	/**
	 * 传递一个Wookbook，给定文件名，以及request和response下载Excel文档 
	 * @author LiangXianYou
	 * @throws IOException
	 * @date 2015-3-13 上午9:00:36
	 */
	@SuppressWarnings("all")
	private static boolean downloadExcel(Workbook wb, String xlsName, HttpServletRequest request, HttpServletResponse response) throws IOException{
			if(request.getHeader("user-agent").indexOf("MSIE") != -1) {   
				xlsName = java.net.URLEncoder.encode(xlsName,"utf-8") + ".xls"; 
			} else {
			    xlsName = new String(xlsName.getBytes("utf-8"),"iso-8859-1")+ ".xls";   
			}
			OutputStream os = response.getOutputStream();
			response.setContentType("application/vnd.ms-excel");
			response.setHeader("Content-disposition", "attachment;filename="+xlsName);    
			
			wb.write(os);
			return true;
	}
	
	/**
	 * 通过泛型实例对象得到某一字段值
	 * @author LiangXianYou
	 * @date 2015-3-13 上午10:53:32
	 */
	private static <T> Object  getFieldValue(T t, String fieldName) throws Exception{
		if(t == null){
			return null;
		}
		Field field = t.getClass().getDeclaredField(fieldName);
		field.setAccessible(true);//暴力反射
		Object o = field.get(t);//得到字段数据的值
		return o;
	}
	
	/**
	 * 将数据写到Excel中
	 * @author LiangXianYou
	 * @date 2015-3-13 上午10:37:55
	 */
	private static void data2Excel(Sheet sheet, Object o, Integer r, Integer c){
		//通过获得sheet中的某一列，有得到，没有创建
		Row nRow = sheet.getRow(r);
		if(nRow == null){
			nRow = sheet.createRow(r);
		}
//		nRow.setColumnWidth(r, arg1);
		
		Cell nCell = nRow.createCell(c);
		
		
		//根据不同类型进行转化，如有其它类型没有考虑周全的，使用发现的时候添加
		char type = 'x';
		if(o instanceof Integer){
			type = 1;
		}else if(o instanceof Double){
			type = 2;
		}else if(o instanceof Float){
			type = 3;
		}else if(o instanceof String){
			type = 4;
		}else if(o instanceof Date){
			type = 5;
		}else if(o instanceof Calendar){
			type = 6;
		}else if(o instanceof Boolean){
			type = 7;
		}else if(o == null) {
			type = 8;
		}
		
		switch (type) {
			case 1:nCell.setCellValue((Integer)o);break;
			case 2:nCell.setCellValue((Double)o);break;
			case 3:nCell.setCellValue((Float)o);break;
			case 4:nCell.setCellValue((String)o);break;
			case 5:nCell.setCellValue(new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(o));break;
			case 6:nCell.setCellValue((Calendar)o);break;
			case 7:nCell.setCellValue((Boolean)o);break;
			case 8:nCell.setCellValue("");break;
			default:nCell.setCellValue(o + "");break;
		}
	}

	//======================= 读取excel ===============================

	/**
	 * read excel head
	 * @param file
	 * @param fields
	 * @param lastRowNum hope get row count
	 * @return
	 * @throws IOException
	 */
	public static List<Map> readExcelHead(File file, String[] fields, int lastRowNum) throws IOException {
		return readExcel(file, fields, lastRowNum, null);
	}

	//read excel sheet[0]
	public static List<Map> readExcel(File file, String[] fields) throws IOException {
		return readExcel(file, fields, -1, null);
	}

	public static List<Map> readExcel(File file, String[] fields, String sheetName) throws IOException {
		return readExcel(file, fields, -1, sheetName);
	}

	//read excel all sheet
	public static Map<String, List<Map>> readExcelAll(File file, String[] fields) throws IOException {
		return readExcelAll(file, fields, -1);
	}

	//read excel sheet[0]
	private static List<Map> readExcel(File file, String[] fields, int lastRowNum, String sheetName) throws IOException {
		Workbook wk;
		try {
			wk = new HSSFWorkbook(new FileInputStream(file));//if excel version 2007+ will throws OfficeXmlFileException
		}catch (OfficeXmlFileException e){
			wk = new XSSFWorkbook(new FileInputStream(file));
		}
		Sheet sheet = sheetName == null ? wk.getSheetAt(0) : wk.getSheet(sheetName);
		if (sheet == null) throw new OfficeXmlFileException("sheet["+sheetName+"] can't find");
		return readExcel(sheet, fields, lastRowNum);
	}

	/**
	 * read excel all sheet,
	 * can read excel 2007+ and 2007-
	 * @param file
	 * @param fields
	 * @param lastRowNum
	 * @return
	 * @throws IOException
	 */
	private static Map<String, List<Map>> readExcelAll(File file, String[] fields, int lastRowNum) throws IOException {
		Workbook wk;
		try {
			wk = new HSSFWorkbook(new FileInputStream(file));//if excel version 2007+ will throws OfficeXmlFileException
		}catch (OfficeXmlFileException e){
			wk = new XSSFWorkbook(new FileInputStream(file));
		}

		Map<String, List<Map>> data = new LinkedHashMap<>();
		//reading all sheet
		for (int i = 0; i < wk.getNumberOfSheets(); i++) {
			Sheet sheet = wk.getSheetAt(i);
			List<Map> maps = readExcel(sheet, fields, lastRowNum);

			//move head
			if (lastRowNum < 0 && maps.size() > 0){
				maps.remove(0);
			}
			data.put(sheet.getSheetName(), maps);
		}
		return data;
	}

	/**
	 * read excel
	 * @author Lxyer 2016/8/1 10:32.
	 */
	private static List<Map> readExcel(Sheet sheet, String[] fields, int lastRowNum) throws OfficeXmlFileException {
		List<Map> list = new ArrayList<>();
		if (lastRowNum < 0 || lastRowNum > sheet.getLastRowNum()){
			lastRowNum = sheet.getLastRowNum();
		}

		int t = 0;
		r:for (int i=0; i<=lastRowNum; i++){
			Row row = sheet.getRow(i);
			if (row == null) continue ;
			short cellNum = row.getLastCellNum();
			//空跳过/连续三行为空结束
			if (isEmptyRow(row, fields.length)) {
				if (t++ > 3) break;
				continue ;
			}

			Map map = new HashMap();
			for (int j=0; j<cellNum && j<fields.length; j++){
				Cell cell = row.getCell(j);
				if (cell == null){
					map.put(fields[j], "");
					continue;
				}

				int cellType = cell.getCellType();
				if (cellType == 0){
					map.put(fields[j], (long)cell.getNumericCellValue()+"");
				}else {
					map.put(fields[j], cell.getStringCellValue());
				}
			}
			list.add(map);
		}
		return list;
	}

	//空跳过/连续三行为空结束
	private static boolean isEmptyRow(Row row, int len){
		for (int i = 0; i< row.getLastCellNum() && i < len; i++) {
			Cell cell = row.getCell(i);//列
			if (cell != null){
				if (cell.getCellTypeEnum() != CellType.NUMERIC && cell.getStringCellValue() != null && !cell.getStringCellValue().isEmpty()){
					return false;
				}else if (cell.getCellTypeEnum() == CellType.NUMERIC && cell.getNumericCellValue() != 0){
					return false;
				}
			}
		}
		return true;
	}
	
}