Guoyan
/
new_zaojia_backend
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

pdf 下载带图片内容

ai_300
zhouhaibin 2 weeks ago
parent
72a01f3da6
commit
a278a75f8d
3 changed files with 142 additions and 72 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 1
      zaojiaManagement/zaojia-productManagement/src/main/java/org/dromara/productManagement/service/impl/ContractualTasksServiceImpl.java
  2. 125
      zaojiaManagement/zaojia-productManagement/src/main/java/org/dromara/productManagement/service/impl/DocumentTaskResultsServiceImpl.java
  3. 28
      zaojiaManagement/zaojia-productManagement/src/main/java/org/dromara/productManagement/service/impl/DocumentTasksServiceImpl.java

1
zaojiaManagement/zaojia-productManagement/src/main/java/org/dromara/productManagement/service/impl/ContractualTasksServiceImpl.java
View File

@ -96,7 +96,6 @@ public class ContractualTasksServiceImpl extends AbstractTaskProcessor<Contractu
);
}
contractualTasksVo.setTaskDuration(formattedDuration);
return contractualTasksVo;
})
.collect(java.util.stream.Collectors.toList());

125
zaojiaManagement/zaojia-productManagement/src/main/java/org/dromara/productManagement/service/impl/DocumentTaskResultsServiceImpl.java
View File

@ -494,12 +494,6 @@ public class DocumentTaskResultsServiceImpl implements IDocumentTaskResultsServi
// Markdown转PDF方法
private void convertMarkdownToPdf(String markdown, String outputPath) throws Exception {
// 使用flexmark-java解析Markdown
// MutableDataSet options = new MutableDataSet();
// Parser parser = Parser.builder(options).build();
// HtmlRenderer renderer = HtmlRenderer.builder(options).build();
// Node document = parser.parse(markdown);
// String htmlContent = renderer.render(document);
MutableDataSet options = new MutableDataSet();
// 启用表格解析
options.set(Parser.EXTENSIONS, Arrays.asList(
@ -523,14 +517,12 @@ public class DocumentTaskResultsServiceImpl implements IDocumentTaskResultsServi
// 配置允许的标签
Safelist safelist = Safelist.relaxed()
// .addTags("em", "strong", "div", "span","br")
// .addAttributes(":all", "style", "class");
.addTags("em", "strong", "div", "span", "br", "table", "thead", "tbody", "tr", "th", "td")
.addTags("em", "strong", "div", "span", "br", "table", "thead", "tbody", "tr", "th", "td", "img")
.addAttributes(":all", "style", "class")
.addAttributes("table", "border", "cellpadding", "cellspacing");
.addAttributes("table", "border", "cellpadding", "cellspacing")
.addAttributes("img", "src", "alt", "width", "height"); // 允许img标签及其属性
htmlContent = Jsoup.clean(htmlContent, "", safelist, settings);
// 使用 NekoHTML 进一步处理
// 创建完整的HTML文档，使用微软雅黑
String html = String.format("""
<!DOCTYPE html>
@ -573,6 +565,14 @@ public class DocumentTaskResultsServiceImpl implements IDocumentTaskResultsServi
tr {
page-break-inside: avoid;
}
img {
width: 600px;
max-width: 100%%;
height: auto;
display: block;
margin: 20px auto;
page-break-inside: avoid;
}
</style>
</head>
<body>
@ -581,9 +581,8 @@ public class DocumentTaskResultsServiceImpl implements IDocumentTaskResultsServi
</html>
""",
htmlContent.replace("<br>", "\n"));
// 处理图片标签，将markdown的图片语法转换为HTML的img标签
html = handleImageTags(html);
// 配置Flying Saucer
ITextRenderer pdfRenderer = new ITextRenderer();
@ -610,6 +609,104 @@ public class DocumentTaskResultsServiceImpl implements IDocumentTaskResultsServi
}
}
/**
* 处理Markdown中的图片标签
* @param content Markdown内容
* @return 处理后的HTML内容
*/
private String handleImageTags(String content) {
// 使用非贪婪模式匹配alt文本，使用平衡组模式匹配路径
java.util.regex.Pattern pattern = java.util.regex.Pattern.compile("!\\[(.*?)\\]\\(([^()]*(?:\\([^()]*\\)[^()]*)*?)\\)");
java.util.regex.Matcher matcher = pattern.matcher(content);
StringBuffer sb = new StringBuffer();
while (matcher.find()) {
String alt = matcher.group(1);
String imagePath = matcher.group(2);
// 检查图片文件是否存在
File imageFile = new File(imagePath);
if (!imageFile.exists() && imagePath.contains("(")) {
// 如果文件不存在且路径中包含括号，尝试修正路径
String correctedPath = correctImagePath(imagePath);
imageFile = new File(correctedPath);
if (imageFile.exists()) {
imagePath = correctedPath;
}
}
if (imageFile.exists()) {
try {
// 将图片转换为Base64编码
byte[] imageBytes = Files.readAllBytes(imageFile.toPath());
String base64Image = java.util.Base64.getEncoder().encodeToString(imageBytes);
String imageType = getImageMimeType(imagePath);
// 创建data URL，使用固定的宽度来确保图片在页面内显示
String imgTag = String.format("<img src=\"data:%s;base64,%s\" alt=\"%s\" style=\"width: 600px; max-width: 100%%; height: auto; display: block; margin: 10px auto;\"/>",
imageType, base64Image, alt);
// 替换markdown图片语法为HTML img标签
matcher.appendReplacement(sb, imgTag.replace("$", "\\$"));
} catch (IOException e) {
log.error("处理图片失败: " + imagePath, e);
// 如果处理失败，保留原始文本
matcher.appendReplacement(sb, matcher.group(0).replace("$", "\\$"));
}
} else {
log.warn("图片文件不存在: " + imagePath);
// 如果图片不存在，保留原始文本
matcher.appendReplacement(sb, matcher.group(0).replace("$", "\\$"));
}
}
matcher.appendTail(sb);
return sb.toString();
}
/**
* 修正包含括号的图片路径
* @param originalPath 原始路径
* @return 修正后的路径
*/
private String correctImagePath(String originalPath) {
// 如果路径中包含多个括号，取最后一个右括号之前的内容
int lastRightBracket = originalPath.lastIndexOf(")");
if (lastRightBracket != -1) {
// 找到对应的左括号
int count = 1;
int leftBracket = lastRightBracket - 1;
while (leftBracket >= 0 && count > 0) {
if (originalPath.charAt(leftBracket) == ')') {
count++;
} else if (originalPath.charAt(leftBracket) == '(') {
count--;
}
leftBracket--;
}
if (count == 0) {
// 找到匹配的括号对，返回不包含最外层括号的路径
return originalPath.substring(0, leftBracket + 1) + originalPath.substring(lastRightBracket + 1);
}
}
return originalPath;
}
/**
* 根据文件扩展名获取MIME类型
* @param filePath 文件路径
* @return MIME类型
*/
private String getImageMimeType(String filePath) {
String extension = filePath.substring(filePath.lastIndexOf(".") + 1).toLowerCase();
return switch (extension) {
case "png" -> "image/png";
case "jpg", "jpeg" -> "image/jpeg";
case "gif" -> "image/gif";
case "bmp" -> "image/bmp";
default -> "image/jpeg";
};
}
// 添加文件到ZIP
private void addToZip(String filePath, String fileName, ZipOutputStream zos) throws IOException {
byte[] buffer = new byte[1024];

28
zaojiaManagement/zaojia-productManagement/src/main/java/org/dromara/productManagement/service/impl/DocumentTasksServiceImpl.java
View File

@ -1,11 +1,9 @@
package org.dromara.productManagement.service.impl;
import cn.dev33.satoken.stp.StpUtil;
import cn.hutool.core.io.FileUtil;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import lombok.Data;
import lombok.RequiredArgsConstructor;
import okhttp3.*;
import org.dromara.common.core.service.DictService;
import org.dromara.common.core.utils.MapstructUtils;
@ -308,28 +306,4 @@ public class DocumentTasksServiceImpl extends AbstractTaskProcessor<DocumentTask
return true;
}
// @Override
// protected Boolean processTaskItems(List<String> taskNames, BaseTaskBo bo, FileProcessResult fileResult, String taskType) {
// boolean flag = false;
// for (String taskName : taskNames) {
// DocumentTasks add = MapstructUtils.convert(bo, DocumentTasks.class);
// add.setTaskName(taskName);
// add.setDocumentName(fileResult.getFileName());
// add.setProgressStatus("PENDING");
// add.setTaskType(taskType);
//
// flag = baseMapper.insert(add) > 0;
//
// Long priority = calculatePriority(taskName);
// MyHttpUtils.sendTaskStartMessage(chatUrl + "/back/taskStart",
// add.getId(), taskName, fileResult.getFilePath(), priority);
//
// if (!flag) {
// throw new RuntimeException("新增文档任务失败");
// }
// }
// return flag;
//
// }
}

Write Preview
Loading…
Cancel
Save