/*
* Copyright 2009-2011 Prime Technology.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.primefaces.component.export;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.List;
import javax.el.MethodExpression;
import javax.faces.component.UIColumn;
import javax.faces.component.UIComponent;
import javax.faces.component.UIData;
import javax.faces.context.FacesContext;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import org.primefaces.component.datatable.DataTable;
import com.lowagie.text.Document;
import com.lowagie.text.DocumentException;
import com.lowagie.text.Font;
import com.lowagie.text.FontFactory;
import com.lowagie.text.Paragraph;
import com.lowagie.text.pdf.PdfPTable;
import com.lowagie.text.pdf.PdfWriter;
public class PDFExporter extends Exporter {
@Override
public void export(FacesContext context, DataTable table, String filename, boolean pageOnly, int[] excludeColumns, String encodingType, MethodExpression preProcessor, MethodExpression postProcessor) throws IOException {
try {
Document document = new Document();
ByteArrayOutputStream baos = new ByteArrayOutputStream();
PdfWriter.getInstance(document, baos);
if(preProcessor != null) {
preProcessor.invoke(context.getELContext(), new Object[]{document});
}
if(!document.isOpen()) {
document.open();
}
PdfPTable pdfTable = exportPDFTable(context, table, pageOnly,excludeColumns, encodingType);
document.add(pdfTable);
if(postProcessor != null) {
postProcessor.invoke(context.getELContext(), new Object[]{document});
}
document.close();
writePDFToResponse(((HttpServletResponse) context.getExternalContext().getResponse()), baos, filename);
} catch (DocumentException e) {
throw new IOException(e.getMessage());
}
}
private PdfPTable exportPDFTable(FacesContext context, DataTable table, boolean pageOnly, int[] excludeColumns, String encoding) {
List<UIColumn> columns = getColumnsToExport(table, excludeColumns);
int numberOfColumns = columns.size();
PdfPTable pdfTable = new PdfPTable(numberOfColumns);
Font font = FontFactory.getFont(FontFactory.TIMES, encoding);
Font headerFont = FontFactory.getFont(FontFactory.TIMES, encoding, Font.DEFAULTSIZE, Font.BOLD);
String rowIndexVar = table.getRowIndexVar();
int first = pageOnly ? table.getFirst() : 0;
int size = pageOnly ? (first + table.getRows()) : table.getRowCount();
addFacetColumns(pdfTable, columns, headerFont, ColumnType.HEADER);
for(int i = first; i < size; i++) {
table.setRowIndex(i);
if(rowIndexVar != null) {
context.getExternalContext().getRequestMap().put(rowIndexVar, i);
}
for(int j = 0; j < numberOfColumns; j++) {
addColumnValue(pdfTable, columns.get(j).getChildren(), j, font);
}
}
if(hasColumnFooter(columns)) {
addFacetColumns(pdfTable, columns, headerFont, ColumnType.FOOTER);
}
table.setRowIndex(-1);
if(rowIndexVar != null) {
context.getExternalContext().getRequestMap().remove(rowIndexVar);
}
return pdfTable;
}
private void addFacetColumns(PdfPTable pdfTable, List<UIColumn> columns, Font font, ColumnType columnType) {
for(int i = 0; i < columns.size(); i++) {
UIColumn column = (UIColumn) columns.get(i);
addColumnValue(pdfTable, column.getFacet(columnType.facet()), font);
}
}
private void addColumnValue(PdfPTable pdfTable, UIComponent component, Font font) {
String value = component == null ? "" : exportValue(FacesContext.getCurrentInstance(), component);
pdfTable.addCell(new Paragraph(value, font));
}
private void addColumnValue(PdfPTable pdfTable, List<UIComponent> components, int index, Font font) {
StringBuilder builder = new StringBuilder();
for(UIComponent component : components) {
if(component.isRendered() ) {
String value = exportValue(FacesContext.getCurrentInstance(), component);
if(value != null)
builder.append(value);
}
}
pdfTable.addCell(new Paragraph(builder.toString(), font));
}
private void writePDFToResponse(HttpServletResponse response, ByteArrayOutputStream baos, String fileName) throws IOException, DocumentException {
response.setContentType("application/pdf");
response.setHeader("Expires", "0");
response.setHeader("Cache-Control","must-revalidate, post-check=0, pre-check=0");
response.setHeader("Pragma", "public");
response.setHeader("Content-disposition", "attachment;filename="+ fileName + ".pdf");
response.setContentLength(baos.size());
ServletOutputStream out = response.getOutputStream();
baos.writeTo(out);
out.flush();
}
}