package br.com.caelum.stella.boleto.transformer;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashMap;
import java.util.Locale;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import javax.imageio.ImageIO;
import net.sf.jasperreports.engine.JRException;
import net.sf.jasperreports.engine.JRParameter;
import net.sf.jasperreports.engine.JasperExportManager;
import net.sf.jasperreports.engine.JasperFillManager;
import net.sf.jasperreports.engine.JasperPrint;
import net.sf.jasperreports.engine.JasperPrintManager;
import net.sf.jasperreports.engine.data.JRBeanCollectionDataSource;
import net.sf.jasperreports.engine.util.JRLoader;
import br.com.caelum.stella.boleto.Boleto;
import br.com.caelum.stella.boleto.exception.GeracaoBoletoException;
/**
* Geração do boleto em arquivos. Essa é uma classe utilitária, dessa maneira
* você não precisa trabalhar com InputStream, byte[], etc.
*
* Basta passar um Boleto para o construtor e usar o método adequado para gerar
* um PDF, PNG, etc e grava-lo como o arquivo desejado.
*
*
*/
public class GeradorDeBoleto {
protected final Collection<Boleto> boletos;
protected InputStream templateJasper;
protected JasperPrint relatorio;
protected Map<String,Object> parametros = new HashMap<String,Object>();
/**
* Cria um gerador de boletos com o template padrão.
*
* @param boletos boletos a serem gerados.
*/
public GeradorDeBoleto(Collection<Boleto> boletos) {
this.boletos = boletos;
try {
templateJasper = GeradorDeBoleto.class.getResourceAsStream("/br/com/caelum/stella/boleto/templates/boleto-default.jasper");
parametros.put(JRParameter.REPORT_LOCALE, new Locale("pt", "BR"));
InputStream template_sub = GeradorDeBoleto.class.getResourceAsStream("/br/com/caelum/stella/boleto/templates/boleto-default_instrucoes.jasper");
parametros.put("SUB_INSTRUCOES", JRLoader.loadObject(template_sub));
} catch (JRException e) {
throw new GeracaoBoletoException(e);
}
}
/**
* Cria um gerador de boletos com o template padrão.
*
* @param boletos boletos a serem gerados.
*/
public GeradorDeBoleto(Boleto... boletos) {
this(Arrays.asList(boletos));
}
/**
* Cria um gerador de boletos que usa um template customizado.
*
* @param template o template (.jasper) a ser usado (obrigatório).
* @param parametros parametros extras para o relatório( opcional).
* @param boletos boletos.
*/
public GeradorDeBoleto(InputStream template, Map<String,Object> parametros, Boleto... boletos) {
this(boletos);
this.templateJasper = template;
if(parametros != null){
Set<Entry<String,Object>> entrySet = parametros.entrySet();
for (Entry<String,Object> entry : entrySet) {
this.parametros.put(entry.getKey(), entry.getValue());
}
}
}
/**
* Cria um gerador de boletos que usa um template customizado.
*
* @param template o template (.jasper) a ser usado (obrigatório).
* @param parametros parametros extras para o relatório( opcional).
* @param boletos boletos.
*/
public GeradorDeBoleto(InputStream template, Map<String,Object> parametros, Collection<Boleto> boletos) {
this(boletos);
this.templateJasper = template;
if(parametros != null){
Set<Entry<String,Object>> entrySet = parametros.entrySet();
for (Entry<String,Object> entry : entrySet) {
this.parametros.put(entry.getKey(), entry.getValue());
}
}
}
/**
* Gera um boleto em PDF, e grava no caminho indicado.
*
* @param arquivo.
*/
public void geraPDF(String arquivo) {
geraPDF(new File(arquivo));
}
protected JasperPrint geraRelatorio(){
try{
if(relatorio == null){
JRBeanCollectionDataSource dataSource = new JRBeanCollectionDataSource(boletos);
relatorio = JasperFillManager.fillReport(templateJasper,parametros, dataSource);
}
return relatorio;
}catch(Exception e){
throw new GeracaoBoletoException(e);
}
}
/**
* Gera um boleto em PDF, e grava no arquivo indicado.
*
* @param arquivo arquivo para gravar o PDF.
*/
public void geraPDF(File arquivo) {
try {
OutputStream out = new FileOutputStream(arquivo);
geraPDF(out);
} catch (FileNotFoundException e) {
throw new GeracaoBoletoException(e);
}
}
/**
* Gera um boleto em PDF, e grava no OutputStream passado.
*
* @param out Local para gravação do PDF.
*/
public void geraPDF(OutputStream out){
try {
JasperPrint relatorio = geraRelatorio();
JasperExportManager.exportReportToPdfStream(relatorio, out);
} catch (Exception e) {
throw new GeracaoBoletoException(e);
}
}
/**
* Gera um boleto em PNG, e grava no caminho indicado.
*
* @param arquivo caminho para o aquivo onde será gravado o PNG.
*/
public void geraPNG(String arquivo) {
geraPNG(new File(arquivo));
}
/**
* Gera um boleto em PNG, e grava no arquivo indicado.
*
* @param arquivo caminho para o aquivo onde será gravado o PNG.
*/
public void geraPNG(File arquivo) {
try {
geraPNG(new FileOutputStream(arquivo));
} catch (FileNotFoundException e) {
throw new GeracaoBoletoException(e);
}
}
/**
* Gera um boleto em PNG, e grava no OutputStream indicado.
*
* @param out local para gravação.
*/
public void geraPNG(OutputStream out) {
try {
JasperPrint relatorio = geraRelatorio();
BufferedImage image = (BufferedImage) JasperPrintManager.printPageToImage(relatorio, 0, 2);
ImageIO.write(image,"png",out);
} catch (Exception e) {
throw new GeracaoBoletoException(e);
}
}
/**
* Gera o boleto no formato PDF.
* @return array de bytes representando o PDF desse boleto ja gerado.
*/
public byte[] geraPDF() {
ByteArrayOutputStream stream = new ByteArrayOutputStream();
geraPDF(stream);
return stream.toByteArray();
}
/**
* Gera o boleto no formato PNG.
* @return array de bytes representando o PNG desse boleto ja gerado.
*/
public byte[] geraPNG() {
ByteArrayOutputStream stream = new ByteArrayOutputStream();
geraPNG(stream);
return stream.toByteArray();
}
/**
* Gera o boleto no formato PDF.
* @return inputStream com o conteúdo do arquivo.
*/
public InputStream geraPDFStream() {
return new ByteArrayInputStream(geraPDF());
}
/**
* Gera o boleto no formato PNG.
* @return inputStream com o conteúdo do arquivo
*/
public InputStream geraPNGStream() {
return new ByteArrayInputStream(geraPNG());
}
}