Please wait. This can take some minutes ...
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.primefaces.component.export.CSVExporter Maven / Gradle / Ivy
/*
* Copyright 2009-2014 PrimeTek.
*
* 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.IOException;
import java.io.PrintWriter;
import java.io.Writer;
import java.util.*;
import javax.el.MethodExpression;
import javax.faces.FacesException;
import javax.faces.component.UIComponent;
import javax.faces.context.ExternalContext;
import javax.faces.context.FacesContext;
import org.primefaces.component.api.DynamicColumn;
import org.primefaces.component.api.UIColumn;
import org.primefaces.component.datatable.DataTable;
import org.primefaces.util.Constants;
public class CSVExporter extends Exporter {
@Override
public void export(FacesContext context, DataTable table, String filename, boolean pageOnly, boolean selectionOnly, String encodingType, MethodExpression preProcessor, MethodExpression postProcessor) throws IOException {
ExternalContext externalContext = context.getExternalContext();
configureResponse(externalContext, filename, encodingType);
Writer writer = externalContext.getResponseOutputWriter();
addColumnFacets(writer, table, ColumnType.HEADER);
if (pageOnly) {
exportPageOnly(context, table, writer);
}
else if (selectionOnly) {
exportSelectionOnly(context, table, writer);
}
else {
exportAll(context, table, writer);
}
if (table.hasFooterColumn()) {
addColumnFacets(writer, table, ColumnType.FOOTER);
}
writer.flush();
writer.close();
externalContext.responseFlushBuffer();
}
@Override
public void export(FacesContext facesContext, List clientIds, String outputFileName, boolean pageOnly, boolean selectionOnly, String encodingType, MethodExpression preProcessor, MethodExpression postProcessor) throws IOException {
throw new UnsupportedOperationException("Not supported yet.");
}
@Override
public void export(FacesContext facesContext, String outputFileName, List tables, boolean pageOnly, boolean selectionOnly, String encodingType, MethodExpression preProcessor, MethodExpression postProcessor) throws IOException {
throw new UnsupportedOperationException("Not supported yet.");
}
protected void addColumnFacets(Writer writer, DataTable table, ColumnType columnType) throws IOException {
boolean firstCellWritten = false;
for (UIColumn col : table.getColumns()) {
if (col instanceof DynamicColumn) {
((DynamicColumn) col).applyStatelessModel();
}
if (col.isRendered() && col.isExportable()) {
if (firstCellWritten) {
writer.write(",");
}
UIComponent facet = col.getFacet(columnType.facet());
if(facet != null) {
addColumnValue(writer, facet);
}
else {
String textValue;
switch(columnType) {
case HEADER:
textValue = col.getHeaderText();
break;
case FOOTER:
textValue = col.getFooterText();
break;
default:
textValue = "";
break;
}
addColumnValue(writer, textValue);
}
firstCellWritten = true;
}
}
writer.write("\n");
}
@Override
protected void exportCells(DataTable table, Object document) {
PrintWriter writer = (PrintWriter) document;
boolean firstCellWritten = false;
for (UIColumn col : table.getColumns()) {
if (col instanceof DynamicColumn) {
((DynamicColumn) col).applyStatelessModel();
}
if (col.isRendered() && col.isExportable()) {
if (firstCellWritten) {
writer.write(",");
}
try {
addColumnValue(writer, col.getChildren(), col);
} catch (IOException ex) {
throw new FacesException(ex);
}
firstCellWritten = true;
}
}
}
protected void configureResponse(ExternalContext externalContext, String filename, String encodingType) {
externalContext.setResponseContentType("text/csv; charset=" + encodingType);
externalContext.setResponseHeader("Expires", "0");
externalContext.setResponseHeader("Cache-Control","must-revalidate, post-check=0, pre-check=0");
externalContext.setResponseHeader("Pragma", "public");
externalContext.setResponseHeader("Content-disposition", "attachment;filename="+ filename + ".csv");
externalContext.addResponseCookie(Constants.DOWNLOAD_COOKIE, "true", Collections.emptyMap());
}
protected void addColumnValues(Writer writer, List columns) throws IOException {
for(Iterator iterator = columns.iterator(); iterator.hasNext();) {
UIColumn col = iterator.next();
addColumnValue(writer, col.getChildren(), col);
if (iterator.hasNext())
writer.write(",");
}
}
protected void addColumnValue(Writer writer, UIComponent component) throws IOException {
String value = component == null ? "" : exportValue(FacesContext.getCurrentInstance(), component);
addColumnValue(writer, value);
}
protected void addColumnValue(Writer writer, String value) throws IOException {
value = (value == null) ? "" : value.replaceAll("\"", "\"\"");
writer.write("\"" + value + "\"");
}
protected void addColumnValue(Writer writer, List components, UIColumn column) throws IOException {
FacesContext context = FacesContext.getCurrentInstance();
writer.write("\"");
if(column.getExportFunction() != null) {
String value = exportColumnByFunction(context, column);
//escape double quotes
value = value == null ? "" : value.replaceAll("\"", "\"\"");
writer.write(value);
}
else {
for (UIComponent component : components) {
if (component.isRendered()) {
String value = exportValue(context, component);
//escape double quotes
value = value == null ? "" : value.replaceAll("\"", "\"\"");
writer.write(value);
}
}
}
writer.write("\"");
}
@Override
protected void postRowExport(DataTable table, Object document) {
((PrintWriter) document).write("\n");
}
}