package foundation.io.file;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.math.BigDecimal;
import java.text.SimpleDateFormat;
import java.util.Date;

import org.apache.poi.ss.usermodel.BorderStyle;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.FillPatternType;
import org.apache.poi.ss.usermodel.Font;
import org.apache.poi.ss.usermodel.IndexedColors;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import foundation.data.entity.Entity;
import foundation.data.entity.EntitySet;
import foundation.data.meta.field.FieldWriter;
import foundation.data.meta.field.FieldsRuntime;
import foundation.io.define.DataIO;
import foundation.io.engine.ISheetWriter;
import foundation.io.engine.IWorkBookWriter;
import foundation.io.mapping.MappingsRuntime;
import foundation.io.object.Headers;
import foundation.io.object.Titles;
import foundation.translator.ValueType;
import foundation.util.MapList;


public class BookStandardWriter extends IWorkBookWriter implements ISheetWriter {

	private File file;
	private FileInputStream inputStream;
	private XSSFWorkbook workbook;
	private MapList<String, Sheet> sheetList;
	
	private Sheet sheet;
	private int titleRowNo;
	private int dataRowNo;
	private Row firstDataRow;
	private CellStyle[] cellStyles;
	private CellStyle errorCellStyles;
	private int currentRowNo;
	private int columnCount;
	private MappingsRuntime mappingsRuntime;
	private SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd");
	private static String resultFieldName = "导入反馈";
	
	public BookStandardWriter(File file, int batchCount) throws IOException {
		this.file = file;
		
		inputStream = new FileInputStream(file);
		workbook = new XSSFWorkbook(inputStream);
		
		sheetList = new MapList<String, Sheet>();
		
		for (Sheet sheet: workbook) {
			sheetList.add(sheet.getSheetName(), sheet);
		}
	}

	@Override
	public ISheetWriter openSheetWriter(DataIO dataIO) {
		this.titleRowNo = dataIO.getToTitleRowNo();
		this.dataRowNo = dataIO.getToDataRowNo();
		
		String sheetName = dataIO.getToName();
		
		sheet = sheetList.get(sheetName);
		if (sheet == null) {
			sheet = sheetList.get(0);
		}
		
		if (sheet == null) {
			sheet = workbook.createSheet(sheetName);
		}

		currentRowNo = -1;
		
		firstDataRow = sheet.getRow(dataRowNo);
		if (firstDataRow == null) {
			firstDataRow = sheet.createRow(dataRowNo);
		}
		
		return this;
	}
	
	@Override
	public Headers readHeaders() {
		Headers headers = new Headers();
		
		Row titleRow = sheet.getRow(titleRowNo);
		
		columnCount = titleRow.getLastCellNum();
		Cell titleCell, valueCell;
		
		for (int i = 0; i < columnCount; i++) {
			//1. 读取标题列
			titleCell = (titleRow != null) ? titleRow.getCell(i) : null;
			String title = (titleCell != null) ? getCellStringValue(titleCell) : null;
			
			//2. 读取第一行数据
			valueCell = firstDataRow.getCell(i);
			String value = (valueCell != null) ? getCellStringValue(valueCell) : null;
			
			//3. 添加到  header 中
			headers.addOne(i, title, value);
		}
		
		return headers;
	}

	private String getCellStringValue(Cell cell) {
		try {
			return cell.getStringCellValue();
		}
		catch (Exception e) {
			return null;
		}
	}

	@Override
	public void writeTitles(Titles titles) {
		Row titleRow = sheet.getRow(titleRowNo);
				
		if (titleRow == null) {
			titleRow = sheet.createRow(titleRowNo);
		}
		
		Cell cell;
		
		for (int i = 0; i < columnCount; i++) {
			cell = titleRow.getCell(i);
			
			if (cell == null) {
				cell = titleRow.createCell(i);
			}
			
			String title = titles.get(i);
			cell.setCellValue(String.valueOf(title));
		}		
	}


	@Override
	public void writeErrorTitle() {
		Row titleRow = sheet.getRow(titleRowNo);
		columnCount = titleRow.getLastCellNum();
		
		Cell cell = titleRow.createCell(columnCount);

		createErrrorCellStyles();
		cell.setCellStyle(errorCellStyles);
		cell.setCellValue(resultFieldName);
	}

	@Override
	public void parepareWriteData() {
		//1. 创建格式
		createDataCellStyles();
		
		//2. 清空第一个数据行上的数据
		clearFirstDataRow();
	}
	
	private void createDataCellStyles() {
		cellStyles = new CellStyle[columnCount];
		for (int i = 0; i < columnCount; i++) {
			Cell cell = firstDataRow.getCell(i);
			
			if (cell == null) {
				continue;
			}
			
			cellStyles[i] = cell.getCellStyle();
		}
	}
	
	private void createErrrorCellStyles() {
		errorCellStyles = workbook.createCellStyle();
		
		errorCellStyles.setFillForegroundColor(IndexedColors.YELLOW.getIndex());
		errorCellStyles.setFillPattern(FillPatternType.SOLID_FOREGROUND);
		errorCellStyles.setBorderRight(BorderStyle.THIN);
		
		Font font = workbook.createFont();
		font.setFontHeightInPoints((short)11);
		errorCellStyles.setFont(font);
	}
	
	private void clearFirstDataRow() {
		Row dataRow = sheet.getRow(dataRowNo);
		
		if (dataRow == null) {
			return;
		}		
		
		Cell cell;
		for (int i = 0; i < columnCount; i++) {
			cell = dataRow.getCell(i);
			
			if (cell == null) {
				continue;
			}
			
			cell.setBlank();
		}
	}

	@Override
	public int writeData(EntitySet entitySet) {
		int result = 0;
		
		if (currentRowNo < 0) {
			currentRowNo = dataRowNo;
		}
		
		int[] fromIndexes = mappingsRuntime.getFromIndex();
		int[] toIndexes = mappingsRuntime.getToIndex();
		int dataSize = toIndexes.length;
		
		Row dataRow = null; Cell cell; 
		CellStyle cellStyle; int fromIndex, toIndex;
		FieldWriter fieldWriter; Object[] data;
		
		FieldsRuntime fields = entitySet.getEntityMeta();
		ValueType[] valueTypes = fields.getValueTypes();
		FieldWriter[] fieldWriters = fields.getWriters();
		
		for (Entity entity: entitySet) {
			data = entity.getDataArray();
			dataRow = sheet.getRow(currentRowNo);
			
			if (dataRow == null) {
				dataRow = sheet.createRow(currentRowNo);
			}
			
			dataRow.setRowStyle(firstDataRow.getRowStyle());
			dataRow.setHeight(firstDataRow.getHeight());
			dataRow.setHeightInPoints(firstDataRow.getHeightInPoints());
			
			for (int i = 0; i < columnCount; i++) {
				cellStyle = cellStyles[i]; 
				
				if (cellStyle == null) {
					continue;
				}
				
				//1. 单元格
				cell = dataRow.getCell(i);
				
				if (cell == null) {
					cell = dataRow.createCell(i);
				}
				
				//2. 设置格式
				cell.setCellStyle(cellStyle);
			}
			
			for (int i = 0; i < dataSize; i++) {
				toIndex = toIndexes[i];
				cell = dataRow.getCell(toIndex);
				
				//3. 获取数据
				fromIndex = fromIndexes[i]; 
				fieldWriter = fieldWriters[fromIndex];
				Object value = fieldWriter.getValue(data[fromIndex]);
				
				//4. 数据写入
				writeOneCell(cell, valueTypes[fromIndex], value);
			}
			
			currentRowNo++;
			result++;
		}

		return result;
	}
        
	@Override
	public int writeErrors(EntitySet entitySet) {
		int result = 0;
		if (currentRowNo < 0) {
			currentRowNo = dataRowNo;
		}
		
		Row dataRow = null; Cell cell;

		int[] fromIndexes = mappingsRuntime.getFromIndex();
		int dataSize = fromIndexes.length;
		
		CellStyle cellStyle = null; int fromIndex;
		FieldWriter fieldWriter; Object[] data;
		
		FieldsRuntime fields = entitySet.getEntityMeta();
		ValueType[] valueTypes = fields.getValueTypes();
		FieldWriter[] fieldWriters = fields.getWriters();
		
		for (Entity entity: entitySet) {
			data = entity.getDataArray();
			dataRow = sheet.getRow(currentRowNo);
			
			if (dataRow == null) {
				dataRow = sheet.createRow(currentRowNo);
			}
			
			dataRow.setRowStyle(firstDataRow.getRowStyle());
			dataRow.setHeight(firstDataRow.getHeight());
			dataRow.setHeightInPoints(firstDataRow.getHeightInPoints());
			
			//1. 创建样式
			for (int i = 0; i < columnCount; i++) {
				cellStyle = cellStyles[i]; 
				
				if (cellStyle == null) {
					continue;
				}
				
				cell = dataRow.getCell(i);
				
				if (cell == null) {
					cell = dataRow.createCell(i);
				}
				
				cell.setCellStyle(cellStyle);
			}
			
			cell = dataRow.createCell(columnCount);
			cell.setCellStyle(cellStyle);
			
			for (int i = 0; i < dataSize; i++) {
				cell = dataRow.getCell(i);
				
				//2. 获取数据
				fromIndex = fromIndexes[i]; 
				fieldWriter = fieldWriters[fromIndex];
				Object value = fieldWriter.getValue(data[fromIndex]);
				
				//4. 数据写入
				writeOneCell(cell, valueTypes[fromIndex], value);
			}
			
			currentRowNo++;
			result++;
		}
		
		return result;
	}
	
	protected void writeOneCell(Cell cell, ValueType valueType, Object value) {
		if (value == null) {
			return ;
		}

		if (ValueType.Decimal == valueType) {
			cell.setCellValue(((BigDecimal)value).doubleValue());
		}
		else if (ValueType.Double == valueType) {
			cell.setCellValue((Double)value);
		}
		else if (ValueType.Int == valueType) {
			cell.setCellValue((Integer)value);
		}   
		else if (ValueType.Long == valueType) {
			cell.setCellValue((Long)value);
		} 
		else if (ValueType.Float == valueType) {
			cell.setCellValue((Float)value);
		}
		else if (ValueType.Date == valueType) {
			String formatDate = dateFormat.format((Date)value);
			cell.setCellValue(formatDate);
		}
		else {
			cell.setCellValue(String.valueOf(value));
		}
	}
        
	@Override
	public void close() {
		try {
			FileOutputStream outputStream = new FileOutputStream(file);
			try {
				workbook.write(outputStream);
				workbook.close();
				close = true;
			}
			finally {
				outputStream.close();
			}
		}
		catch (Exception e) {
			logger.error("close work book error: {}", e.getMessage());
		}
	}

	@Override
	public void setMappingRuntime(MappingsRuntime mappingsRuntime) {
		this.mappingsRuntime = mappingsRuntime;
	}
	
}