package com.product.print.util;
|
|
import com.alibaba.fastjson.JSONArray;
|
import com.alibaba.fastjson.JSONObject;
|
import com.deepoove.poi.data.TextRenderData;
|
import com.deepoove.poi.policy.AbstractRenderPolicy;
|
import com.deepoove.poi.render.RenderContext;
|
import com.deepoove.poi.xwpf.BodyContainer;
|
import com.deepoove.poi.xwpf.BodyContainerFactory;
|
import com.product.common.lang.StringUtils;
|
import org.apache.poi.util.Units;
|
import org.apache.poi.xwpf.usermodel.*;
|
|
import java.io.ByteArrayInputStream;
|
import java.io.InputStream;
|
import java.util.Base64;
|
import java.util.Optional;
|
|
/**
|
* 自定义解析{{@var}}标签
|
*
|
* @Author cheng
|
* @Date 2023/6/27 15:50
|
* @Desc
|
*/
|
public class FlowOpinionRenderPolicy extends AbstractRenderPolicy<TextRenderData> {
|
/**
|
* 意见列表
|
*/
|
private JSONArray opinionArray;
|
|
public FlowOpinionRenderPolicy(JSONArray opinionArray) {
|
this.opinionArray = opinionArray;
|
}
|
|
|
@Override
|
public void doRender(RenderContext<TextRenderData> context) throws Exception {
|
XWPFRun run = context.getRun();
|
//标签名称
|
String tagName = context.getEleTemplate().getTagName();
|
JSONObject opinion=null;
|
//支持多个标签名称，用|分隔 例如：{{@var1|var2|var3...}} 按照顺序查找意见列表
|
for (String name : tagName.split("\\|")) {
|
opinion = getOpinion(name);
|
if(opinion!=null){
|
break;
|
}
|
}
|
if (opinion == null) {
|
//如果意见列表中没有找到，则直接替换为空
|
run.setText("", 0);
|
return;
|
}
|
XWPFParagraph parent = (XWPFParagraph) run.getParent();
|
XWPFRun newRun = run;
|
//判断是否有签名
|
if (!StringUtils.isEmpty(opinion.getString("sign_attachment_uuid"))) {
|
XWPFParagraph paragraph = createParagraph(run);
|
//设置签名的段落居右
|
paragraph.setAlignment(ParagraphAlignment.RIGHT);
|
//插入一个新的run
|
newRun = paragraph.createRun();
|
String signAttachmentBase64 = opinion.getString("sign_attachment_uuid");
|
String[] split = signAttachmentBase64.split(",");
|
//将base64转换字节流
|
byte[] bytes = Base64.getDecoder().decode(split[split.length - 1]);
|
//将字节流转为输入流
|
InputStream byteArrayInputStream = new ByteArrayInputStream(bytes);
|
//插入图片
|
newRun.addPicture(byteArrayInputStream, XWPFDocument.PICTURE_TYPE_PNG, "sign.png", Units.toEMU(50), Units.toEMU(20));
|
byteArrayInputStream.close();
|
newRun.setText("", 0);
|
}
|
XWPFParagraph paragraph = createParagraph(newRun);
|
//段落样式与原始段落一致
|
paragraph.getCTP().setPPr(parent.getCTP().getPPr());
|
//设置创建新的run
|
XWPFRun textRun = paragraph.createRun();
|
//复制原始run
|
PrintPoiUtil.copyRun(textRun, run);
|
//设置run的内容
|
textRun.setText(Optional.ofNullable(opinion.getString("opinion")).orElse("未填写办理意见"), 0);
|
//删除段落parent，如果只删除run，则会导致段落中还有一个空的run
|
parent.getCTP().newCursor().removeXml();
|
}
|
|
public XWPFParagraph createParagraph(XWPFRun run) {
|
//获取body容器
|
BodyContainer bodyContainer = BodyContainerFactory.getBodyContainer(run);
|
//插入一个新的段落
|
XWPFParagraph paragraph = bodyContainer.insertNewParagraph(run);
|
return paragraph;
|
}
|
|
public JSONObject getOpinion(String tagName) {
|
//先用tagName去意见列表中查找node_uuid是否存在，不存在则使用node_title去查找
|
for (int i = 0; i < opinionArray.size(); i++) {
|
JSONObject opinion = opinionArray.getJSONObject(i);
|
if (tagName.equals(opinion.getString("node_uuid"))) {
|
return opinion;
|
}
|
}
|
for (int i = 0; i < opinionArray.size(); i++) {
|
JSONObject opinion = opinionArray.getJSONObject(i);
|
if (tagName.equals(opinion.getString("node_title"))) {
|
return opinion;
|
}
|
}
|
return null;
|
}
|
}
|
