package poi.excel;


import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.List;


import javax.servlet.http.HttpServletRequest;


import org.apache.log4j.Logger;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFFont;
import org.apache.poi.hssf.usermodel.HSSFRichTextString;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.hssf.util.HSSFColor;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;






/** 
* @ClassName: ExcelUtil 
* @Description: Excel匯入匯出工具類
* @author 周宣
* @date 2016-11-8 下午7:16:11 
*  
*/
public class ExcelUtil {
    private static final Logger logger = Logger.getLogger(ExcelUtil.class);
    
    /** 
    * @Title: createWorkbook 
    * @Description: 判斷excel檔案字尾名，生成不同的workbook 
    * @param @param is
    * @param @param excelFileName
    * @param @return
    * @param @throws IOException
    * @return Workbook
    * @throws 
    */
    public Workbook createWorkbook(InputStream is,String excelFileName) throws IOException{
        if (excelFileName.endsWith(".xls")) {
            return new HSSFWorkbook(is);
        }else if (excelFileName.endsWith(".xlsx")) {
            return new XSSFWorkbook(is);
        }
        return null;
    }


    /** 
    * @Title: getSheet 
    * @Description: 根據sheet索引號獲取對應的sheet
    * @param @param workbook
    * @param @param sheetIndex
    * @param @return
    * @return Sheet
    * @throws 
    */
    public Sheet getSheet(Workbook workbook,int sheetIndex){
        return workbook.getSheetAt(0);        
    }
    
    /** 
    * @Title: importDataFromExcel 
    * @Description: 將sheet中的資料儲存到list中，
    * 1、呼叫此方法時，vo的屬性個數必須和excel檔案每行資料的列數相同且一一對應，vo的所有屬性都為String
    * 2、在action呼叫此方法時，需宣告 
    *     private File excelFile;上傳的檔案
    *     private String excelFileName;原始檔案的檔名
    * 3、頁面的file控制元件name需對應File的檔名
    * @param @param vo javaBean
    * @param @param is 輸入流
    * @param @param excelFileName
    * @param @return
    * @return List<Object>
    * @throws 
    */
    public List<Object> importDataFromExcel(Object vo,InputStream is,String excelFileName){
        List<Object> list = new ArrayList<Object>();
        try {
            //建立工作簿
            Workbook workbook = this.createWorkbook(is, excelFileName);
            //建立工作表sheet
            Sheet sheet = this.getSheet(workbook, 0);
            //獲取sheet中資料的行數
            int rows = sheet.getPhysicalNumberOfRows();
            //獲取表頭單元格個數
            int cells = sheet.getRow(0).getPhysicalNumberOfCells();
            //利用反射，給JavaBean的屬性進行賦值
            Field[] fields = vo.getClass().getDeclaredFields();
            for (int i = 1; i < rows; i++) {//第一行為標題欄，從第二行開始取資料
                Row row = sheet.getRow(i);
                int index = 0;
                while (index < cells) {
                    Cell cell = row.getCell(index);
                    if (null == cell) {
                        cell = row.createCell(index);
                    }
                    cell.setCellType(Cell.CELL_TYPE_STRING);
                    String value = null == cell.getStringCellValue()?"":cell.getStringCellValue();
                    
                    Field field = fields[index];
                    String fieldName = field.getName();
                    String methodName = "set"+fieldName.substring(0,1).toUpperCase()+fieldName.substring(1);
                    Method setMethod = vo.getClass().getMethod(methodName, new Class[]{String.class});
                    setMethod.invoke(vo, new Object[]{value});
                    index++;
                }
                if (isHasValues(vo)) {//判斷物件屬性是否有值
                    list.add(vo);
                    vo.getClass().getConstructor(new Class[]{}).newInstance(new Object[]{});//重新建立一個vo物件
                }
                
            }
        } catch (Exception e) {
            logger.error(e);
        }finally{
            try {
                is.close();//關閉流
            } catch (Exception e2) {
                logger.error(e2);
            }
        }
        return list;
        
    }
    
    /** 
    * @Title: isHasValues 
    * @Description: 判斷一個物件所有屬性是否有值，如果一個屬性有值(分空)，則返回true
    * @param @param object
    * @param @return
    * @return boolean
    * @throws 
    */
    public boolean isHasValues(Object object){
        Field[] fields = object.getClass().getDeclaredFields();
        boolean flag = false;
        for (int i = 0; i < fields.length; i++) {
            String fieldName = fields[i].getName();
            String methodName = "get"+fieldName.substring(0, 1).toUpperCase()+fieldName.substring(1);
            Method getMethod;
            try {
                getMethod = object.getClass().getMethod(methodName);
                Object obj = getMethod.invoke(object);
                if (null != obj && "".equals(obj)) {
                    flag = true;
                    break;
                }
            } catch (Exception e) {
                logger.error(e);
            }
            
        }
        return flag;
        
    }
    
    public <T> void exportDataToExcel(List<T> list,String[] headers,String title,OutputStream os){
        HSSFWorkbook workbook = new HSSFWorkbook();
        //生成一個表格
        HSSFSheet sheet = workbook.createSheet(title);
        //設定表格預設列寬15個位元組
        sheet.setDefaultColumnWidth(15);
        //生成一個樣式
        HSSFCellStyle style = this.getCellStyle(workbook);
        //生成一個字型
        HSSFFont font = this.getFont(workbook);
        //把字型應用到當前樣式
        style.setFont(font);
        
        //生成表格標題
        HSSFRow row = sheet.createRow(0);
        row.setHeight((short)300);
        HSSFCell cell = null;
        
        for (int i = 0; i < headers.length; i++) {        
            cell = row.createCell(i);
            cell.setCellStyle(style);
            HSSFRichTextString text = new HSSFRichTextString(headers[i]);
            cell.setCellValue(text);
        }
        
        //將資料放入sheet中
        for (int i = 0; i < list.size(); i++) {
            row = sheet.createRow(i+1);
            T t = list.get(i);
            //利用反射，根據JavaBean屬性的先後順序，動態呼叫get方法得到屬性的值
            Field[] fields = t.getClass().getFields();
            try {
                for (int j = 0; j < fields.length; j++) {
                    cell = row.createCell(j);
                    Field field = fields[j];
                    String fieldName = field.getName();
                    String methodName = "get"+fieldName.substring(0, 1).toUpperCase()+fieldName.substring(1);
                    Method getMethod = t.getClass().getMethod(methodName,new Class[]{});
                    Object value = getMethod.invoke(t, new Object[]{});
                    
                    if(null == value)
                        value ="";
                    cell.setCellValue(value.toString());
                    
                }
            } catch (Exception e) {
                logger.error(e);
            }
        }
        
        try {
            workbook.write(os);
        } catch (Exception e) {
            logger.error(e);
        }finally{
            try {
                os.flush();
                os.close();
            } catch (IOException e) {
                logger.error(e);
            }
        }
        
    }
    
    /** 
    * @Title: getCellStyle 
    * @Description: 獲取單元格格式
    * @param @param workbook
    * @param @return
    * @return HSSFCellStyle
    * @throws 
    */
    public HSSFCellStyle getCellStyle(HSSFWorkbook workbook){
        HSSFCellStyle style = workbook.createCellStyle();
        style.setFillForegroundColor(HSSFColor.GREY_25_PERCENT.index);
        style.setFillPattern(HSSFCellStyle.SOLID_FOREGROUND);
        style.setBorderBottom(HSSFCellStyle.BORDER_THIN);
        style.setBorderTop(HSSFCellStyle.BORDER_THIN);
        style.setLeftBorderColor(HSSFCellStyle.BORDER_THIN);
        style.setRightBorderColor(HSSFCellStyle.BORDER_THIN);
        style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        
        return style;
    }


    /** 
    * @Title: getFont 
    * @Description: 生成字型樣式
    * @param @param workbook
    * @param @return
    * @return HSSFFont
    * @throws 
    */
    public HSSFFont getFont(HSSFWorkbook workbook){
        HSSFFont font = workbook.createFont();
        font.setColor(HSSFColor.WHITE.index);
        font.setFontHeightInPoints((short)12);
        font.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);
        return font;
    }
    
    public boolean isIE(HttpServletRequest request){
        return request.getHeader("USER-AGENT").toLowerCase().indexOf("msie")>0?true:false;    
    }
}