package per.qy.dexter.fileoperate;

import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Iterator;
import java.util.UUID;

import javax.imageio.ImageIO;

import org.apache.pdfbox.cos.COSName;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.pdmodel.PDPage;
import org.apache.pdfbox.pdmodel.PDResources;
import org.apache.pdfbox.pdmodel.encryption.InvalidPasswordException;
import org.apache.pdfbox.pdmodel.graphics.image.PDImageXObject;
import org.apache.pdfbox.text.PDFTextStripper;
import org.junit.Test;

public class PdfTest {

	@Test
	public void testPdf() {
		String path = "D:\\temp\\temp\\test.pdf";
		File file = new File(path);
		InputStream is = null;
		PDDocument document = null;
		try {
			if (path.endsWith(".pdf")) {
				document = PDDocument.load(file);
				int pageSize = document.getNumberOfPages();
				// 一頁一頁讀取
				for (int i = 0; i < pageSize; i++) {
					// 文字內容
					PDFTextStripper stripper = new PDFTextStripper();
					// 設定按順序輸出
					stripper.setSortByPosition(true);
					stripper.setStartPage(i + 1);
					stripper.setEndPage(i + 1);
					String text = stripper.getText(document);
					System.out.println(text.trim());
					System.out.println("-=-=-=-=-=-=-=-=-=-=-=-=-");

					// 圖片內容
					PDPage page = document.getPage(i);
					PDResources resources = page.getResources();
					Iterable<COSName> cosNames = resources.getXObjectNames();
					if (cosNames != null) {
						Iterator<COSName> cosNamesIter = cosNames.iterator();
						while (cosNamesIter.hasNext()) {
							COSName cosName = cosNamesIter.next();
							if (resources.isImageXObject(cosName)) {
								PDImageXObject Ipdmage = (PDImageXObject) resources.getXObject(cosName);
								BufferedImage image = Ipdmage.getImage();
								FileOutputStream out = new FileOutputStream("D:\\temp\\temp\\" + UUID.randomUUID() + ".png");
								try {
									ImageIO.write(image, "png", out);
								} catch (IOException e) {
								} finally {
									try {
										out.close();
									} catch (IOException e) {
									}
								}
							}
						}
					}
				}
			}
		} catch (InvalidPasswordException e) {
		} catch (IOException e) {
		} finally {
			try {
				if (document != null) {
					document.close();
				}
				if (is != null) {
					is.close();
				}
			} catch (IOException e) {
			}
		}
	}
	
}