package com.gk.firework.Domain.Utils; import com.gk.firework.Domain.Exception.BusinessException; import org.apache.poi.ooxml.POIXMLDocumentPart; import org.apache.poi.openxml4j.exceptions.InvalidFormatException; import org.apache.poi.util.Units; import org.apache.poi.xwpf.usermodel.*; import org.apache.xmlbeans.XmlException; import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTRow; import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTSectPr; import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTTcBorders; import org.openxmlformats.schemas.wordprocessingml.x2006.main.STBorder; import javax.imageio.ImageIO; import java.awt.*; import java.awt.image.RenderedImage; import java.io.*; import java.util.*; import java.util.List; import java.util.Map.Entry; import java.util.regex.Matcher; import java.util.regex.Pattern; public class WordTemplate { private XWPFDocument document; public void loadTemplate(File file,InputStream in) throws IOException{ //File file = new File(fileName); InputStream target = null; if (file != null) { target = new FileInputStream(file); }else{ target = in; } try{ document = new XWPFDocument(target); target.close(); } catch(IOException ex){ if(target != null){ target.close(); } throw ex; } } public void saveFile(String fileName) throws IOException { // File outputFile=new File(fileName);//filename=D:\Users\inner_peace\Downloads // if(!outputFile.exists()){//判断outputFile这个文件夹是否存在,不存在,创建文件夹(Tips:使用mkdirs创建多层目录,而不是mkdir创建单一目录) // System.out.println("start"); // outputFile.mkdirs(); // System.out.println("end"); // } File f=new File(fileName); String a=""; if(!f.exists()){//判断outputFile文件夹下是否存在name文件,不存在,创建 f.createNewFile(); }/*else{//存在,需要重新创建一个不同名字的文件,这边暂时是在文件名称后不断加(01) String xx=name.substring(0,name.length()-5); a=xx+"(1)"; }*/ /*File ff=new File(fileName+a+".docx"); ff.createNewFile(); System.out.println(43);*/ FileOutputStream fos = new FileOutputStream(f); try{ document.write(fos); fos.close(); }catch (IOException ex){ if(fos != null){ fos.close(); } throw ex; } } public void saveFileAndQRCode(String fileName,String info) throws IOException { File f=new File(fileName); if(!f.exists()){ f.createNewFile(); } FileOutputStream fos = new FileOutputStream(f); try{ Image image = QRCode.generateQRCode(info); // String QRCodeUrl = Properties.transportCertificatePath + info + "_" + "QRCode.png"; File file = new File(QRCodeUrl); if (!file.exists()) {file.createNewFile();} ImageIO.write((RenderedImage) image, "png", file); List paragraphs = document.getParagraphs(); for (int i = 0; i < paragraphs.size(); i++) { XWPFParagraph paragraph = paragraphs.get(i); List runs = paragraph.getRuns(); for (int l = 0; l < runs.size(); l++) { XWPFRun run = runs.get(l); if (run.text().equals("") || run.text().equals("/n")) { continue; } if (run.text().equals("#QRCode#")) {//如果遍历到标记位置退出循环,并记录一下标记位置所在段落对象的下标 paragraph.removeRun(l);//同时把正文两字移除 XWPFRun xwpfRun = paragraph.insertNewRun(l); xwpfRun.addPicture( new FileInputStream(file),XWPFDocument.PICTURE_TYPE_PNG,"QRCode.png", Units.toEMU(40), Units.toEMU(40)); paragraph.addRun(xwpfRun); }else if (run.text().contains("第一联") || run.text().contains("第二联") ) { if (i + 1 < paragraphs.size()) { paragraphs.get(i + 1).setPageBreak(true ); } } } } document.write(fos); }catch (IOException ex){ ex.printStackTrace(); throw new BusinessException("生成文件发生错误"); } catch (InvalidFormatException e) { e.printStackTrace(); throw new BusinessException("格式不合法"); }finally { fos.close(); } } public static class Table{ private String[] fields; List data; HashMap index; public Table(String[] fields,List data){ this.fields = fields; this.data = data; this.index = new HashMap(); for(int i=0 ; i tableMap,Map paraMap) { List bodyElements = document.getBodyElements(); int templateBodySize = bodyElements.size(); int curT = 0; int curP = 0; for (int a = 0; a < templateBodySize; a++) { IBodyElement body = bodyElements.get(a); // System.out.println( body.getElementType()); if (BodyElementType.TABLE.equals(body.getElementType())) { // XWPFTable table = body.getBody().getTableArray(curT); List tables = body.getBody().getTables(); table = tables.get(curT); if (table != null) { List tableCells = table.getRows().get(0).getTableCells(); // 获取到模板表格第一行,用来判断表格类型 String tableText = table.getText(); if (tableText.indexOf("##{foreach") > -1) { if (tableCells.size() != 2 || tableCells.get(0).getText().indexOf("##{foreach") < 0 || tableCells.get(0).getText().trim().length() == 0) { throw new RuntimeException("need two cell in row."); } String tableType = tableCells.get(0).getText(); String dataSource = tableCells.get(1).getText(); if (!tableMap.containsKey(dataSource)) { throw new RuntimeException("need table:" + dataSource); } Table tableData = tableMap.get(dataSource); if ("##{foreachTable}##".equals(tableType)) { // System.out.println("循环生成表格"); addTableInDocFooterByTable(table, tableData); } else if ("##{foreachTableRow}##".equals(tableType)) { // System.out.println("循环生成表格内部的行"); addTableInDocFooterByRow(table, tableData, paraMap); } } else if (tableText.indexOf("{") > -1) { addTableInDocFooterByParas(table, paraMap); } else { // 没有查找到任何标签,该表格是一个静态表格,仅需要复制一个即可。 addTableInDocFooterByOrigin(table); } curT++; } } else if (BodyElementType.PARAGRAPH.equals(body.getElementType())) {// 处理段落 XWPFParagraph ph = body.getBody().getParagraphArray(curP); if (ph != null) { // htmlText = htmlText+readParagraphX(ph); addParagraphInDocFooter(ph, null, paraMap); curP++; } } else if (BodyElementType.CONTENTCONTROL.equals(body.getElementType())) { IBodyElement bodyElement = bodyElements.get(curP); IBody body1 = bodyElement.getBody(); } } for (int a = 0; a < templateBodySize; a++) { document.removeBodyElement(0); } } // public void replaceDocument2(Map cellMap,Map params) throws IOException, XmlException { List bodyElements = document.getBodyElements(); int templateBodySize = bodyElements.size(); int curT = 0; for (int a = 0; a < templateBodySize; a++) { IBodyElement body = bodyElements.get(a); // System.out.println( body.getElementType()); if (BodyElementType.TABLE.equals(body.getElementType())) { // XWPFTable table = body.getBody().getTableArray(curT); List tables = body.getBody().getTables(); POIXMLDocumentPart part = body.getBody().getPart(); table = tables.get(curT); Table mainTable = cellMap.get("mainTable"); List data = mainTable.data; int remove = 0; List rows = table.getRows(); for (int i = 0; i < rows.size(); i++) { XWPFTableRow row = rows.get(i); List tableCells = row.getTableCells(); for (int j = 0; j < tableCells.size(); j++) { XWPFTableCell cell = tableCells.get(j); String text = cell.getText(); if (params.get(text) != null) { cell.removeParagraph(0); cell.setText(params.get(text)); } //生成表内容 if ("##foreachRows##".equals(cell.getText())) { //从下面一行开始循环生成单元格 XWPFTableRow infoRow = rows.get(i + 1); for (int k = 0; k < mainTable.data.size(); k++) { CTRow ctrow = CTRow.Factory.parse(infoRow.getCtRow().newInputStream()); XWPFTableRow newRow = new XWPFTableRow(ctrow, table); String[] strings = data.get(k); for (int m = 0; m < strings.length; m++) { XWPFTableCell cellSS = newRow.getTableCells().get(m);// 在新增的行上面创建cell //cell.setText(fields.get(k));//不能使用该方法直接加内容,这样会在原内容后面追加,并且不能保证跟原字体样式一致 for (XWPFParagraph p : cellSS.getParagraphs()) { for (XWPFRun r : p.getRuns()) { r.setText(strings[m], 0);//要深入到原cell中的run替换内容才能保证样式一致 } } } table.addRow(newRow, i + k); } //删除####foreachRows##这一行 table.removeRow(i + mainTable.data.size()); //删除type specification num 这一行 table.removeRow(i + mainTable.data.size()); } } } } } } private void addTableInDocFooterByTable(XWPFTable templateTable, Table data) { for (String[] row: data.data) { List templateTableRows = templateTable.getRows();// 获取模板表格所有行 XWPFTable newCreateTable = document.createTable();// 创建新表格,默认一行一列 for (int i = 1; i < templateTableRows.size(); i++) { XWPFTableRow newCreateRow = newCreateTable.createRow(); CopyTableRow(newCreateRow, templateTableRows.get(i));// 复制模板行文本和样式到新行 } newCreateTable.removeRow(0);// 移除多出来的第一行 document.createParagraph();// 添加回车换行 replaceTable(newCreateTable, getMap(data.index,row));//替换标签 } } private HashMap getMap(HashMap key,String[] row){ HashMap r = new HashMap(); Iterator> it =key.entrySet().iterator(); while(it.hasNext()){ Entry en =it.next(); r.put(en.getKey(),row[en.getValue()]); } return r; } private void addTableInDocFooterByRow(XWPFTable templateTable, Table data , Map parametersMap) { XWPFTable newCreateTable = document.createTable();// 创建新表格,默认一行一列 List TempTableRows = templateTable.getRows();// 获取模板表格所有行 int tagRowsIndex = 0;// 标签行indexs for (int i = 0, size = TempTableRows.size(); i < size; i++) { String rowText = TempTableRows.get(i).getCell(0).getText();// 获取到表格行的第一个单元格 if (rowText.indexOf("##{foreachRows}##") > -1) { tagRowsIndex = i; break; } } /* 复制模板行和标签行之前的行 */ for (int i = 1; i < tagRowsIndex; i++) { XWPFTableRow newCreateRow = newCreateTable.createRow(); CopyTableRow(newCreateRow, TempTableRows.get(i));// 复制行 replaceTableRow(newCreateRow, parametersMap);// 处理不循环标签的替换 } /* 循环生成模板行 */ XWPFTableRow tempRow = TempTableRows.get(tagRowsIndex + 1);// 获取到模板行 for (String[] row: data.data) { XWPFTableRow newCreateRow = newCreateTable.createRow(); CopyTableRow(newCreateRow, tempRow);// 复制模板行 replaceTable(newCreateTable, getMap(data.index,row));//替换标签 } /* 复制模板行和标签行之后的行 */ for (int i = tagRowsIndex + 2; i < TempTableRows.size(); i++) { XWPFTableRow newCreateRow = newCreateTable.createRow(); CopyTableRow(newCreateRow, TempTableRows.get(i));// 复制行 replaceTableRow(newCreateRow, parametersMap);// 处理不循环标签的替换 } newCreateTable.removeRow(0);// 移除多出来的第一行 document.createParagraph();// 添加回车换行 } private void addTableInDocFooterByOrigin(XWPFTable templateTable ) { List templateTableRows = templateTable.getRows();// 获取模板表格所有行 XWPFTable newCreateTable = document.createTable();// 创建新表格,默认一行一列 for (int i = 0; i < templateTableRows.size(); i++) { XWPFTableRow newCreateRow = newCreateTable.createRow(); CopyTableRow(newCreateRow, templateTableRows.get(i));// 复制模板行文本和样式到新行 } newCreateTable.removeRow(0);// 移除多出来的第一行 document.createParagraph();// 添加回车换行 } private void addTableInDocFooterByParas(XWPFTable templateTable, Map parametersMap) { //表格不循环仅简单替换标签 List templateTableRows = templateTable.getRows();// 获取模板表格所有行 XWPFTable newCreateTable = document.createTable();// 创建新表格,默认一行一列 for (int i = 0; i < templateTableRows.size(); i++) { XWPFTableRow newCreateRow = newCreateTable.createRow(); CopyTableRow(newCreateRow, templateTableRows.get(i));// 复制模板行文本和样式到新行 } newCreateTable.removeRow(0);// 移除多出来的第一行 // document.createParagraph();// 添加回车换行 replaceTable(newCreateTable, parametersMap); } private void addParagraphInDocFooter(XWPFParagraph templateParagraph, List> list, Map parametersMap) { XWPFParagraph createParagraph = document.createParagraph(); // 设置段落样式 createParagraph.getCTP().setPPr(templateParagraph.getCTP().getPPr()); // 移除原始内容 for (int pos = 0; pos < createParagraph.getRuns().size(); pos++) { createParagraph.removeRun(pos); } // 添加Run标签 for (XWPFRun s : templateParagraph.getRuns()) { XWPFRun targetrun = createParagraph.createRun(); CopyRun(targetrun, s); } replaceParagraph(createParagraph, parametersMap); } private void replaceParagraph(XWPFParagraph xWPFParagraph, Map parametersMap) { List runs = xWPFParagraph.getRuns(); String xWPFParagraphText = xWPFParagraph.getText(); String regEx = "\\{.+?\\}"; Pattern pattern = Pattern.compile(regEx); Matcher matcher = pattern.matcher(xWPFParagraphText);//正则匹配字符串{****} if (matcher.find()) { // 查找到有标签才执行替换 int beginRunIndex = xWPFParagraph.searchText("{", new PositionInParagraph()).getBeginRun();// 标签开始run位置 int endRunIndex = xWPFParagraph.searchText("}", new PositionInParagraph()).getEndRun();// 结束标签 StringBuffer key = new StringBuffer(); if (beginRunIndex == endRunIndex) { // {**}在一个run标签内 XWPFRun beginRun = runs.get(beginRunIndex); String beginRunText = beginRun.text(); int beginIndex = beginRunText.indexOf("{"); int endIndex = beginRunText.indexOf("}"); int length = beginRunText.length(); if (beginIndex == 0 && endIndex == length - 1) { // 该run标签只有{**} XWPFRun insertNewRun = xWPFParagraph.insertNewRun(beginRunIndex); insertNewRun.getCTR().setRPr(beginRun.getCTR().getRPr()); // 设置文本 key.append(beginRunText.substring(1, endIndex)); insertNewRun.setText(getValueBykey(key.toString(),parametersMap)); xWPFParagraph.removeRun(beginRunIndex + 1); } else { // 该run标签为**{**}** 或者 **{**} 或者{**}**,替换key后,还需要加上原始key前后的文本 XWPFRun insertNewRun = xWPFParagraph.insertNewRun(beginRunIndex); insertNewRun.getCTR().setRPr(beginRun.getCTR().getRPr()); // 设置文本 key.append(beginRunText.substring(beginRunText.indexOf("{")+1, beginRunText.indexOf("}"))); String textString=beginRunText.substring(0, beginIndex) + getValueBykey(key.toString(),parametersMap) + beginRunText.substring(endIndex + 1); insertNewRun.setText(textString); xWPFParagraph.removeRun(beginRunIndex + 1); } }else { // {**}被分成多个run //先处理起始run标签,取得第一个{key}值 XWPFRun beginRun = runs.get(beginRunIndex); String beginRunText = beginRun.text(); int beginIndex = beginRunText.indexOf("{"); if (beginRunText.length()>1 ) { key.append(beginRunText.substring(beginIndex+1)); } ArrayList removeRunList = new ArrayList<>();//需要移除的run //处理中间的run for (int i = beginRunIndex + 1; i < endRunIndex; i++) { XWPFRun run = runs.get(i); String runText = run.text(); key.append(runText); removeRunList.add(i); } // 获取endRun中的key值 XWPFRun endRun = runs.get(endRunIndex); String endRunText = endRun.text(); int endIndex = endRunText.indexOf("}"); //run中**}或者**}** if (endRunText.length()>1 && endIndex!=0) { key.append(endRunText.substring(0,endIndex)); } //******************************************************************* //取得key值后替换标签 //先处理开始标签 if (beginRunText.length()==2 ) { // run标签内文本{ XWPFRun insertNewRun = xWPFParagraph.insertNewRun(beginRunIndex); insertNewRun.getCTR().setRPr(beginRun.getCTR().getRPr()); // 设置文本 insertNewRun.setText(getValueBykey(key.toString(),parametersMap)); xWPFParagraph.removeRun(beginRunIndex + 1);//移除原始的run }else { // 该run标签为**{**或者 {** ,替换key后,还需要加上原始key前的文本 XWPFRun insertNewRun = xWPFParagraph.insertNewRun(beginRunIndex); insertNewRun.getCTR().setRPr(beginRun.getCTR().getRPr()); // 设置文本 String textString=beginRunText.substring(0,beginRunText.indexOf("{"))+getValueBykey(key.toString(),parametersMap); insertNewRun.setText(textString); xWPFParagraph.removeRun(beginRunIndex + 1);//移除原始的run } //处理结束标签 if (endRunText.length()==1 ) { // run标签内文本只有} XWPFRun insertNewRun = xWPFParagraph.insertNewRun(endRunIndex); insertNewRun.getCTR().setRPr(endRun.getCTR().getRPr()); // 设置文本 insertNewRun.setText(""); xWPFParagraph.removeRun(endRunIndex + 1);//移除原始的run }else { // 该run标签为**}**或者 }** 或者**},替换key后,还需要加上原始key后的文本 XWPFRun insertNewRun = xWPFParagraph.insertNewRun(endRunIndex); insertNewRun.getCTR().setRPr(endRun.getCTR().getRPr()); // 设置文本 String textString=endRunText.substring(endRunText.indexOf("}")+1); insertNewRun.setText(textString); xWPFParagraph.removeRun(endRunIndex + 1);//移除原始的run } //处理中间的run标签 for (int i = 0; i < removeRunList.size(); i++) { XWPFRun xWPFRun = runs.get(removeRunList.get(i));//原始run XWPFRun insertNewRun = xWPFParagraph.insertNewRun(removeRunList.get(i)); insertNewRun.getCTR().setRPr(xWPFRun.getCTR().getRPr()); insertNewRun.setText(""); xWPFParagraph.removeRun(removeRunList.get(i) + 1);//移除原始的run } }// 处理${**}被分成多个run replaceParagraph( xWPFParagraph, parametersMap); }//if 有标签 } /** * 复制表格行XWPFTableRow格式 * * @param target * 待修改格式的XWPFTableRow * @param source * 模板XWPFTableRow */ private void CopyTableRow(XWPFTableRow target, XWPFTableRow source) { int tempRowCellsize = source.getTableCells().size();// 模板行的列数 for (int i = 0; i < tempRowCellsize - 1; i++) { target.addNewTableCell();// 为新添加的行添加与模板表格对应行行相同个数的单元格 } // 复制样式 target.getCtRow().setTrPr(source.getCtRow().getTrPr()); // 复制单元格 for (int i = 0; i < target.getTableCells().size(); i++) { copyTableCell(target.getCell(i), source.getCell(i)); } } /** * 复制单元格XWPFTableCell格式 * * @author Juveniless * @date 2017年11月27日 下午3:41:02 * @param newTableCell * 新创建的的单元格 * @param templateTableCell * 模板单元格 * */ private void copyTableCell(XWPFTableCell newTableCell, XWPFTableCell templateTableCell) { // 列属性 newTableCell.getCTTc().setTcPr(templateTableCell.getCTTc().getTcPr()); if(templateTableCell.getText().indexOf("{noborder}")>-1){ CTTcBorders tblBorders = newTableCell.getCTTc().getTcPr().addNewTcBorders(); tblBorders.addNewLeft().setVal(STBorder.NIL); tblBorders.addNewRight().setVal(STBorder.NIL); tblBorders.addNewTop().setVal(STBorder.NIL); newTableCell.getCTTc().getTcPr().setTcBorders(tblBorders); } // 删除目标 targetCell 所有文本段落 for (int pos = 0; pos < newTableCell.getParagraphs().size(); pos++) { newTableCell.removeParagraph(pos); } // 添加新文本段落 for (XWPFParagraph sp : templateTableCell.getParagraphs()) { XWPFParagraph targetP = newTableCell.addParagraph(); copyParagraph(targetP, sp); } } /** * 复制文本段落XWPFParagraph格式 * * @author Juveniless * @date 2017年11月27日 下午3:43:08 * @param newParagraph * 新创建的的段落 * @param templateParagraph * 模板段落 * */ private void copyParagraph(XWPFParagraph newParagraph, XWPFParagraph templateParagraph) { // 设置段落样式 newParagraph.getCTP().setPPr(templateParagraph.getCTP().getPPr()); // 添加Run标签 for (int pos = 0; pos < newParagraph.getRuns().size(); pos++) { newParagraph.removeRun(pos); } for (XWPFRun s : templateParagraph.getRuns()) { XWPFRun targetrun = newParagraph.createRun(); CopyRun(targetrun, s); } } /** * 复制文本节点run * @author Juveniless * @date 2017年11月27日 下午3:47:17 * @param newRun * 新创建的的文本节点 * @param templateRun * 模板文本节点 * */ private void CopyRun(XWPFRun newRun, XWPFRun templateRun) { newRun.getCTR().setRPr(templateRun.getCTR().getRPr()); // 设置文本 newRun.setText(templateRun.text()); } private void replaceTableRow(XWPFTableRow tableRow, Map parametersMap) { List tableCells = tableRow.getTableCells(); for (XWPFTableCell xWPFTableCell : tableCells) { List paragraphs = xWPFTableCell.getParagraphs(); for (XWPFParagraph xwpfParagraph : paragraphs) { replaceParagraph(xwpfParagraph, parametersMap); } } } private void replaceTable(XWPFTable xwpfTable, Map parametersMap){ List rows = xwpfTable.getRows(); for (XWPFTableRow xWPFTableRow : rows ) { List tableCells = xWPFTableRow.getTableCells(); for (XWPFTableCell xWPFTableCell : tableCells ) { List paragraphs2 = xWPFTableCell.getParagraphs(); for (XWPFParagraph xWPFParagraph : paragraphs2) { replaceParagraph(xWPFParagraph, parametersMap); } } } } private String getValueBykey(String key, Map map) { String returnValue=""; if (key != null) { key = key.trim(); try { returnValue=map.get(key)!=null ? map.get(key).toString() : ""; } catch (Exception e) { returnValue=""; } } // System.out.println(key+":" + returnValue); return returnValue; } }