protected org.zkoss.poi.ss.usermodel.Font getPoiFontFromRichText(
Workbook book, Cell cell, RichTextString rstr, int run) {
if (run < 0) return null; // ZSS-1138
org.zkoss.poi.ss.usermodel.Font font =
rstr instanceof HSSFRichTextString
? book.getFontAt(((HSSFRichTextString) rstr).getFontOfFormattingRun(run))
: ((XSSFRichTextString) rstr).getFontOfFormattingRun((XSSFWorkbook) book, run);
if (font == null) {
CellStyle style = cell.getCellStyle();
short fontIndex = style != null ? style.getFontIndex() : (short) 0;
return book.getFontAt(fontIndex);
}
return font;
}
@Override
public void setDateTimePattern(String dateTimePattern) {
this.dateTimePattern = dateTimePattern;
DateTimeConverter dtConverter = (DateTimeConverter) converters.lookup(Date.class);
dtConverter.setDateTimePattern(dateTimePattern);
dateCellStyle = workbook.createCellStyle();
CreationHelper helper = workbook.getCreationHelper();
dateCellStyle.setDataFormat(helper.createDataFormat().getFormat(dateTimePattern));
}
protected SFont importFont(CellStyle poiCellStyle) {
SFont font = null;
final short fontIndex = poiCellStyle.getFontIndex();
if (importedFont.containsKey(fontIndex)) {
font = importedFont.get(fontIndex);
} else {
Font poiFont = workbook.getFontAt(fontIndex);
font = createZssFont(poiFont);
importedFont.put(fontIndex, font); // ZSS-677
}
return font;
}
protected SCellStyle importCellStyle(CellStyle poiCellStyle, boolean inStyleTable) {
SCellStyle cellStyle = null;
// short idx = poiCellStyle.getIndex(); // ZSS-685
if ((cellStyle = importedStyle.get(poiCellStyle)) == null) { // ZSS-685
cellStyle = book.createCellStyle(inStyleTable);
importedStyle.put(poiCellStyle, cellStyle); // ZSS-685
String dataFormat = poiCellStyle.getRawDataFormatString();
if (dataFormat == null) { // just in case
dataFormat = SCellStyle.FORMAT_GENERAL;
}
if (!poiCellStyle.isBuiltinDataFormat()) {
cellStyle.setDirectDataFormat(dataFormat);
} else {
cellStyle.setDataFormat(dataFormat);
}
cellStyle.setWrapText(poiCellStyle.getWrapText());
cellStyle.setLocked(poiCellStyle.getLocked());
cellStyle.setAlignment(PoiEnumConversion.toHorizontalAlignment(poiCellStyle.getAlignment()));
cellStyle.setVerticalAlignment(
PoiEnumConversion.toVerticalAlignment(poiCellStyle.getVerticalAlignment()));
cellStyle.setRotation(poiCellStyle.getRotation()); // ZSS-918
cellStyle.setIndention(poiCellStyle.getIndention()); // ZSS-915
Color fgColor = poiCellStyle.getFillForegroundColorColor();
Color bgColor = poiCellStyle.getFillBackgroundColorColor();
// if (fgColor == null && bgColor != null) { //ZSS-797
// fgColor = bgColor;
// }
// ZSS-857: SOLID pattern: switch fillColor and backColor
cellStyle.setFillPattern(PoiEnumConversion.toFillPattern(poiCellStyle.getFillPattern()));
SColor fgSColor = book.createColor(BookHelper.colorToForegroundHTML(workbook, fgColor));
SColor bgSColor = book.createColor(BookHelper.colorToBackgroundHTML(workbook, bgColor));
if (cellStyle.getFillPattern() == FillPattern.SOLID) {
SColor tmp = fgSColor;
fgSColor = bgSColor;
bgSColor = tmp;
}
cellStyle.setFillColor(fgSColor);
cellStyle.setBackColor(bgSColor); // ZSS-780
cellStyle.setBorderLeft(PoiEnumConversion.toBorderType(poiCellStyle.getBorderLeft()));
cellStyle.setBorderTop(PoiEnumConversion.toBorderType(poiCellStyle.getBorderTop()));
cellStyle.setBorderRight(PoiEnumConversion.toBorderType(poiCellStyle.getBorderRight()));
cellStyle.setBorderBottom(PoiEnumConversion.toBorderType(poiCellStyle.getBorderBottom()));
cellStyle.setBorderLeftColor(
book.createColor(
BookHelper.colorToBorderHTML(workbook, poiCellStyle.getLeftBorderColorColor())));
cellStyle.setBorderTopColor(
book.createColor(
BookHelper.colorToBorderHTML(workbook, poiCellStyle.getTopBorderColorColor())));
cellStyle.setBorderRightColor(
book.createColor(
BookHelper.colorToBorderHTML(workbook, poiCellStyle.getRightBorderColorColor())));
cellStyle.setBorderBottomColor(
book.createColor(
BookHelper.colorToBorderHTML(workbook, poiCellStyle.getBottomBorderColorColor())));
cellStyle.setHidden(poiCellStyle.getHidden());
// same style always use same font
cellStyle.setFont(importFont(poiCellStyle));
}
return cellStyle;
}
public File generateXLSResponse(QueryResult queryResult, Locale locale, String baseURL) {
File excelFile = new File("export_parts.xls");
// Blank workbook
XSSFWorkbook workbook = new XSSFWorkbook();
// Create a blank sheet
XSSFSheet sheet = workbook.createSheet("Parts Data");
String header = StringUtils.join(queryResult.getQuery().getSelects(), ";");
String[] columns = header.split(";");
Map<Integer, String[]> data = new HashMap<>();
String[] headerFormatted = createXLSHeaderRow(header, columns, locale);
data.put(1, headerFormatted);
Map<Integer, String[]> commentsData = new HashMap<>();
String[] headerComments = createXLSHeaderRowComments(header, columns);
commentsData.put(1, headerComments);
List<String> selects = queryResult.getQuery().getSelects();
int i = 1;
for (QueryResultRow row : queryResult.getRows()) {
i++;
data.put(i, createXLSRow(selects, row, baseURL));
commentsData.put(i, createXLSRowComments(selects, row));
}
// Iterate over data and write to sheet
Set<Integer> keyset = data.keySet();
int rownum = 0;
for (Integer key : keyset) {
Row row = sheet.createRow(rownum++);
String[] objArr = data.get(key);
int cellnum = 0;
for (String obj : objArr) {
Cell cell = row.createCell(cellnum++);
cell.setCellValue(obj);
}
CreationHelper factory = workbook.getCreationHelper();
Drawing drawing = sheet.createDrawingPatriarch();
String[] commentsObjArr = commentsData.get(key);
cellnum = 0;
for (String commentsObj : commentsObjArr) {
if (commentsObj.length() > 0) {
Cell cell = row.getCell(cellnum) != null ? row.getCell(cellnum) : row.createCell(cellnum);
// When the comment box is visible, have it show in a 1x3 space
ClientAnchor anchor = factory.createClientAnchor();
anchor.setCol1(cell.getColumnIndex());
anchor.setCol2(cell.getColumnIndex() + 1);
anchor.setRow1(row.getRowNum());
anchor.setRow2(row.getRowNum() + 1);
Comment comment = drawing.createCellComment(anchor);
RichTextString str = factory.createRichTextString(commentsObj);
comment.setString(str);
// Assign the comment to the cell
cell.setCellComment(comment);
}
cellnum++;
}
}
// Define header style
Font headerFont = workbook.createFont();
headerFont.setBoldweight(Font.BOLDWEIGHT_BOLD);
headerFont.setFontHeightInPoints((short) 10);
headerFont.setFontName("Courier New");
headerFont.setItalic(true);
headerFont.setColor(IndexedColors.WHITE.getIndex());
CellStyle headerStyle = workbook.createCellStyle();
headerStyle.setFont(headerFont);
headerStyle.setFillForegroundColor(IndexedColors.GREY_25_PERCENT.getIndex());
headerStyle.setFillPattern(CellStyle.SOLID_FOREGROUND);
// Set header style
for (int j = 0; j < columns.length; j++) {
Cell cell = sheet.getRow(0).getCell(j);
cell.setCellStyle(headerStyle);
if (cell.getCellComment() != null) {
String comment = cell.getCellComment().getString().toString();
if (comment.equals(QueryField.CTX_PRODUCT_ID)
|| comment.equals(QueryField.CTX_SERIAL_NUMBER)
|| comment.equals(QueryField.PART_MASTER_NUMBER)) {
for (int k = 0; k < queryResult.getRows().size(); k++) {
Cell grayCell =
sheet.getRow(k + 1).getCell(j) != null
? sheet.getRow(k + 1).getCell(j)
: sheet.getRow(k + 1).createCell(j);
grayCell.setCellStyle(headerStyle);
}
}
}
}
try {
// Write the workbook in file system
FileOutputStream out = new FileOutputStream(excelFile);
workbook.write(out);
out.close();
} catch (Exception e) {
LOGGER.log(Level.FINEST, null, e);
}
return excelFile;
}