package br.com.ec.controller.util;
import java.io.ByteArrayOutputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import com.itextpdf.text.Document;
import com.itextpdf.text.DocumentException;
import com.itextpdf.text.pdf.PdfCopy;
import com.itextpdf.text.pdf.PdfImportedPage;
import com.itextpdf.text.pdf.PdfReader;
import com.itextpdf.text.pdf.parser.PdfReaderContentParser;
import com.itextpdf.text.pdf.parser.SimpleTextExtractionStrategy;
import com.itextpdf.text.pdf.parser.TextExtractionStrategy;
import br.com.ec.core.util.VerificadorUtil;
public class PdfUtil
{
public static void main
(String[] args
) throws IOException, DocumentException
{
String caminhoArquivo =
"D://recibo.pdf";
List<byte[]> arquivos = separarPDF
(caminhoArquivo
);
int i =
1;
for (byte[] arquivo : arquivos
) {
// System.out.println(extrairPDF(arquivo));
String[] linhas = extrairPDFEmLinhas
(arquivo
);
boolean capturar =
false;
String valorVencimentos =
"";
String codigo =
"";
String linhaAnterior =
"";
for (String linha : linhas
) {
if (linha.
contains("Referência")) {
valorVencimentos =
"" + linhaAnterior
;
}
if (capturar
) {
codigo =
"" + linha
;
capturar =
false;
System.
out.
println("CÓDIGO: " + codigo
);
System.
out.
println("VALOR: R$" + valorVencimentos
);
}
if (linha.
contains("CC:")) {
capturar =
true;
}
linhaAnterior =
"" + linha
;
}
// System.out.println(caminhoArquivo.substring(0, caminhoArquivo.indexOf(".pdf")) + "-" + String.format("%03d", i + 1) + ".pdf");
// gravarArquivo(caminhoArquivo.substring(0, caminhoArquivo.indexOf(".pdf")) + "-" + String.format("%03d", i + 1) + ".pdf", arquivo);
// i++;
//break;
}
}
public static void gravarArquivo
(String caminhoArquivo,
byte[] arquivo
) {
try {
FileOutputStream fos =
new FileOutputStream(caminhoArquivo
);
fos.
write(arquivo
);
fos.
flush();
fos.
close();
} catch (Exception e
) {
e.
printStackTrace();
}
}
public static String[] extrairPDFEmLinhas
(byte[] arquivo
) throws IOException {
StringBuilder texto =
new StringBuilder();
texto.
append(extrairPDF
(arquivo
));
return texto.
toString().
split("\n");
}
public static String extrairPDF
(byte[] arquivo
) throws IOException {
PdfReader reader =
new PdfReader
(arquivo
);
String textoExtraido =
null;
try {
PdfReaderContentParser parser =
new PdfReaderContentParser
(reader
);
int quantidadePaginas = reader.
getNumberOfPages();
TextExtractionStrategy strategy
;
for (int i =
1; i
<= quantidadePaginas
; i++
) {
strategy = parser.
processContent(i,
new SimpleTextExtractionStrategy
());
textoExtraido = strategy.
getResultantText().
toString();
}
} catch (Exception e
) {
e.
printStackTrace();
} finally {
reader.
close();
}
return textoExtraido
;
}
public static boolean verificarSeTemTexto
(String texto,
String textoExtraido
) {
if (VerificadorUtil.
naoEstaNuloOuVazio(textoExtraido
)) {
if (textoExtraido.
contains(texto
)) {
return true;
}
}
return false;
}
public static List<byte[]> separarPDF
(byte[] arquivo
) throws IOException {
return separarPDF
(new PdfReader
(arquivo
));
}
public static List<byte[]> separarPDF
(String caminhoArquivo
) throws IOException {
return separarPDF
(new PdfReader
(caminhoArquivo
));
}
private static List<byte[]> separarPDF
(PdfReader reader
) {
List<byte[]> arquivos =
new ArrayList<byte[]>();
try {
int quantidadePaginas = reader.
getNumberOfPages();
int i =
0;
while (i
< quantidadePaginas
) {
Document documento =
new Document(reader.
getPageSizeWithRotation(1));
ByteArrayOutputStream pdfOutputStream =
new ByteArrayOutputStream();
PdfCopy writer =
new PdfCopy
(documento, pdfOutputStream
);
documento.
open();
PdfImportedPage pagina = writer.
getImportedPage(reader, ++i
);
writer.
addPage(pagina
);
documento.
close();
writer.
close();
arquivos.
add(pdfOutputStream.
toByteArray());
pdfOutputStream.
close();
}
} catch (Exception e
) {
e.
printStackTrace();
} finally {
reader.
close();
}
return arquivos
;
}
public static void gerarPDF
(byte[] bytesPDF,
String destinoArquivo
) {
try {
PdfReader reader =
new PdfReader
(bytesPDF
);
Document documento =
new Document(reader.
getPageSizeWithRotation(1));
PdfCopy copy =
new PdfCopy
(documento,
new FileOutputStream(destinoArquivo
));
documento.
open();
PdfImportedPage page = copy.
getImportedPage(reader,
1);
copy.
addPage(page
);
documento.
close();
copy.
close();
} catch (Exception e
) {
e.
printStackTrace();
}
}
public static void separarSalvandoArquivoPDF
(String caminhoArquivo
) {
try {
PdfReader reader =
new PdfReader
(caminhoArquivo
);
int n = reader.
getNumberOfPages();
int i =
0;
while (i
< n
) {
String destinoArquivo = caminhoArquivo.
substring(0, caminhoArquivo.
indexOf(".pdf")) +
"-" +
String.
format("%03d", i +
1) +
".pdf";
Document document =
new Document(reader.
getPageSizeWithRotation(1));
PdfCopy writer =
new PdfCopy
(document,
new FileOutputStream(destinoArquivo
));
document.
open();
PdfImportedPage page = writer.
getImportedPage(reader, ++i
);
writer.
addPage(page
);
document.
close();
writer.
close();
}
} catch (Exception e
) {
e.
printStackTrace();
}
}
}