package com.lc.ibps.components.upload.util; import java.awt.FontMetrics; import java.io.BufferedOutputStream; import java.io.DataOutputStream; import java.io.File; import java.io.FileInputStream; import java.io.FileOutputStream; import java.io.IOException; import java.io.InputStream; import java.io.OutputStreamWriter; import java.text.SimpleDateFormat; import java.util.ArrayList; import java.util.Date; import java.util.List; import java.util.Map; import java.util.UUID; import java.util.zip.ZipEntry; import java.util.zip.ZipOutputStream; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import javax.swing.JLabel; import org.apache.pdfbox.pdfparser.PDFParser; import org.apache.pdfbox.pdmodel.PDDocument; import org.apache.pdfbox.util.PDFTextStripper; import com.aspose.words.Document; import com.aspose.words.License; import com.aspose.words.SaveFormat; import com.itextpdf.text.Element; import com.itextpdf.text.Rectangle; import com.itextpdf.text.pdf.BaseFont; import com.itextpdf.text.pdf.PdfContentByte; import com.itextpdf.text.pdf.PdfCopy; import com.itextpdf.text.pdf.PdfGState; import com.itextpdf.text.pdf.PdfReader; import com.itextpdf.text.pdf.PdfStamper; import com.lc.ibps.base.core.util.AppUtil; import freemarker.template.Configuration; import freemarker.template.Template; import net.sf.json.JSONArray; import net.sf.json.JSONObject; /** * * @项目名称: * @类名称: DocumentUtils * @类描述: 文档工具类 * @修改备注: * @version: 1.0 */ public class DocumentUtils { private static int interval = -5; /** * 导出word * * @方法名:exportWord * @参数 @param templateName .xml格式word模板 * @参数 @param dataModel 数据 * @参数 @return * @返回类型 String */ public static String exportWord(String templateName, Object dataModel) { FileOutputStream fos = null; OutputStreamWriter writer = null; String outFile = null; try { // 导出路径 String outPath = AppUtil.getProperty("file.upload"); // 导出文件名 String outName = templateName + "_output.doc"; // 导出doc全路径 outFile = outPath + File.separator + outName; // 临时目录 String exceltemplates = outPath + File.separator + "exceltemplates"; fos = new FileOutputStream(outFile); writer = new OutputStreamWriter(fos, "UTF-8"); Configuration config = new Configuration(Configuration.VERSION_2_3_23); config.setDefaultEncoding("UTF-8"); config.setDirectoryForTemplateLoading(new File(exceltemplates)); // 模板路径 Template template = config.getTemplate(templateName + ".xml", "UTF-8"); template.process(dataModel, writer); } catch (Exception e) { e.printStackTrace(); } finally { try { writer.close(); } catch (IOException e) { e.printStackTrace(); } } return outFile; } /** * 模板、数据、生成word * * @方法名:exportWord * @参数 @param templateName 模板名称 * @参数 @param dataModel 数据 * @参数 @param target 目标doc路径 * @返回类型 void */ public static void exportWord(String templateName, Object dataModel, String target) { FileOutputStream fos = null; OutputStreamWriter writer = null; // 临时目录 String exceltemplates = AppUtil.getProperty("file.upload") + File.separator + "exceltemplates"; try { fos = new FileOutputStream(target); writer = new OutputStreamWriter(fos, "UTF-8"); Configuration config = new Configuration(Configuration.VERSION_2_3_23); config.setDefaultEncoding("UTF-8"); config.setDirectoryForTemplateLoading(new File(exceltemplates)); // 模板路径 Template template = config.getTemplate(templateName + ".xml", "UTF-8"); template.process(dataModel, writer); } catch (Exception e) { e.printStackTrace(); } finally { try { writer.close(); } catch (IOException e) { e.printStackTrace(); } } } /** * Word转PDF * * @方法名:convertPdf * @参数 @param templateName * @参数 @param wordPath * @参数 @return * @返回类型 String */ public static String convertPdf(String templateName, String wordPath) { String outFile = null; FileOutputStream fos = null; try { // 导出文件名 String outName = templateName + "_output.pdf"; // 导出路径 String outPath = AppUtil.getProperty("file.upload"); // 导出doc全路径 outFile = outPath + File.separator + outName; fos = new FileOutputStream(outFile); // 注册 License aposeLic = new License(); aposeLic.setLicense( new FileInputStream(DocumentUtils.class.getClassLoader().getResource("license.xml").getPath())); // 转换 Document doc = new Document(new FileInputStream(wordPath)); doc.save(fos, SaveFormat.PDF); } catch (Exception e) { e.printStackTrace(); } finally { try { fos.flush(); fos.close(); } catch (IOException e) { e.printStackTrace(); } } return outFile; } /** * word转成pdf * * @方法名:wordToPdf * @参数 @param source doc路径 * @参数 @param target 目标pdf路径 * @返回类型 void */ public static void wordToPdf(String source, String target) { FileInputStream fis = null; FileOutputStream fos = null; try { fis = new FileInputStream(source); fos = new FileOutputStream(target); // 注册 License aposeLic = new License(); String path = AppUtil.getProperty("file.upload") + File.separator + "exceltemplates" + File.separator + "license.xml"; aposeLic.setLicense(new FileInputStream(path)); // 转换 Document doc = new Document(fis); doc.save(fos, SaveFormat.PDF); } catch (Exception e) { e.printStackTrace(); } finally { try { fos.flush(); fos.close(); fis.close(); } catch (IOException e) { e.printStackTrace(); } } } /** * word转成pdf * * @方法名:wordToPdf * @参数 @param wordList 源WORD文件信息 * @返回类型 List<String> */ public static List<String> wordToPdf(JSONArray wordList) { FileInputStream fis = null; FileOutputStream fos = null; String root = AppUtil.getProperty("file.upload");// 上传根目录 String target = root + File.separator + "pdf";// 临时PDF存放目录 File pdf = new File(target); if (!pdf.exists()) { pdf.mkdir(); } List<String> pdfList = new ArrayList<>(); try { String name = UUID.randomUUID().toString(); // 注册 License aposeLic = new License();// 注册license String path = root + File.separator + "exceltemplates" + File.separator + "license.xml";// 获取license模板 aposeLic.setLicense(new FileInputStream(path)); for (Object object : wordList) { JSONObject o = JSONObject.fromObject(object); String targetPath = target + File.separator + name + "_word_tmp" + ".pdf"; fis = new FileInputStream(root + File.separator + o.getString("filePath")); fos = new FileOutputStream(targetPath); pdfList.add(targetPath); // 转换 Document doc = new Document(fis); doc.save(fos, SaveFormat.PDF); // 流处理 fos.flush(); fos.close(); fis.close(); } } catch (Exception e) { e.printStackTrace(); return null; } finally { try { fos.flush(); fos.close(); fis.close(); } catch (IOException e) { e.printStackTrace(); return null; } } return pdfList; } /** * 读取word文本 * * @方法名:readWordToString * @参数 @param path * @参数 @return * @返回类型 String */ /* * public static String readWordToString(String path) { String result = ""; try * { if (null != path && path.indexOf("doc") > 0) { if (path.endsWith(".doc")) { * InputStream is = new FileInputStream(path); WordExtractor ex = new * WordExtractor(is); result = ex.getText(); is.close(); } else if * (path.endsWith(".docx")) { OPCPackage o = POIXMLDocument.openPackage(path); * POIXMLTextExtractor extractor = new XWPFWordExtractor(o); result = * ((XWPFWordExtractor) extractor).getText(); o.close(); } } } catch (Exception * e) { e.printStackTrace(); result = ""; } return result; } */ /** * 读取pdf文本 * * @方法名:readPdfToString * @参数 @param path * @参数 @return * @返回类型 String */ public static String readPdfToString(String path) { try { FileInputStream fis = new FileInputStream(path); PDFParser p = new PDFParser(fis); p.parse(); PDDocument pdd = p.getPDDocument(); PDFTextStripper ts = new PDFTextStripper(); String result = ts.getText(pdd); pdd.close(); fis.close(); return result; } catch (Exception e) { e.printStackTrace(); return ""; } } /** * PDF合并 * * @param pdfFilenames * @param targetFilename * @throws Exception */ public static void combinPdf(List<String> pdfFilenames, String targetFilename) throws Exception { PdfReader reader = null; com.itextpdf.text.Document doc = new com.itextpdf.text.Document(); PdfCopy pdfCopy = new PdfCopy(doc, new FileOutputStream(targetFilename)); int pageCount = 0; doc.open(); for (int i = 0; i < pdfFilenames.size(); ++i) { reader = new PdfReader(pdfFilenames.get(i)); pageCount = reader.getNumberOfPages(); for (int j = 1; j <= pageCount; ++j) { pdfCopy.addPage(pdfCopy.getImportedPage(reader, j)); } } doc.close(); pdfCopy.close(); reader.close(); } public static void waterMark(String inputFile, String outputFile, String waterMarkName) { try { // 读取需要添加水印的PDF文件 PdfReader reader = new PdfReader(inputFile); PdfStamper stamper = new PdfStamper(reader, new FileOutputStream(outputFile)); // 设置水印字体 BaseFont base = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.EMBEDDED); Rectangle pageRect = null; PdfGState gs = new PdfGState(); // 设置水印透明度 gs.setFillOpacity(0.1f); gs.setStrokeOpacity(0.2f); // 计算PDF的页数 int total = reader.getNumberOfPages() + 1; JLabel label = new JLabel(); FontMetrics metrics; int textH = 0; int textW = 0; label.setText(waterMarkName); metrics = label.getFontMetrics(label.getFont()); textH = metrics.getHeight(); textW = metrics.stringWidth(label.getText()); PdfContentByte under; // 給没一页加上水印，可以随便指定页 for (int i = 1; i < total; i++) { pageRect = reader.getPageSizeWithRotation(i); under = stamper.getOverContent(i); under.saveState(); under.setGState(gs); under.beginText(); under.setFontAndSize(base, 40); under.showTextAligned(Element.ALIGN_LEFT, waterMarkName, textW, textH, 30); // 水印文字成30度角倾斜 /*for (int height = interval + textH; height < pageRect.getHeight(); height = height + textH * 3) { for (int width = interval + textW; width < pageRect.getWidth() + textW; width = width + textW * 2) { under.showTextAligned(Element.ALIGN_LEFT, waterMarkName, width - textW, height - textH, 30); } }*/ // 添加水印文字 under.endText(); } // 关闭流 stamper.close(); reader.close(); } catch (Exception e) { e.printStackTrace(); } } /** * 批量打包 * * @param jsonString * json格式字符串数据 * @param fileSaveRootPath * 项目根目录 * @return zip文件保存绝对路径 */ public static String createZipAndReturnPath(List<Map<String, String>> filePathList, String fileSaveRootPath) { // 生成jsonArray列表 SimpleDateFormat sdf = new SimpleDateFormat("YYYYMMDD"); String time = sdf.format(new Date()); try { // 生成打包下载后的zip文件 String papersZipName = "文件包"+time+".zip"; // zip文件保存路径 String zipPath = fileSaveRootPath + papersZipName; ZipOutputStream out = new ZipOutputStream(new FileOutputStream(zipPath)); // 遍历jsonArray列表获取所有JSONObject对象 for (int i = 0; i < filePathList.size(); i++) { Map<String, String> fileMap = filePathList.get(i); String fileName = fileMap.get("FILE_NAME"); String relativePath = fileMap.get("FILE_PATH"); // 获得下载文件完整路径 String downloadPath = fileSaveRootPath + relativePath; FileInputStream fis = new FileInputStream(downloadPath); out.putNextEntry(new ZipEntry(fileName)); // 写入压缩包 int len; byte[] buffer = new byte[1024]; while ((len = fis.read(buffer)) > 0) { out.write(buffer, 0, len); } out.closeEntry(); fis.close(); } out.close(); out.flush(); return zipPath; } catch (Exception e) { e.printStackTrace(); } return null; } public static void main(String[] args) { waterMark("d:\12.pdf", "D:\456.pdf", "新化啊"); } }