parent
7a0e7e5b3d
commit
c9cd4699a3
@ -1,232 +0,0 @@
|
|||||||
package com.ruoyi.gysl.utils;
|
|
||||||
|
|
||||||
import com.itextpdf.text.Document;
|
|
||||||
import com.itextpdf.text.DocumentException;
|
|
||||||
import com.itextpdf.text.Image;
|
|
||||||
import com.itextpdf.text.Paragraph;
|
|
||||||
import com.itextpdf.text.pdf.PdfWriter;
|
|
||||||
import com.itextpdf.tool.xml.XMLWorkerFontProvider;
|
|
||||||
import com.itextpdf.tool.xml.XMLWorkerHelper;
|
|
||||||
import com.ruoyi.common.utils.http.HttpUtils;
|
|
||||||
import org.jsoup.Jsoup;
|
|
||||||
import org.jsoup.nodes.Element;
|
|
||||||
import org.jsoup.select.Elements;
|
|
||||||
import org.slf4j.Logger;
|
|
||||||
import org.slf4j.LoggerFactory;
|
|
||||||
import org.springframework.mock.web.MockMultipartFile;
|
|
||||||
import org.springframework.web.multipart.MultipartFile;
|
|
||||||
|
|
||||||
import java.io.*;
|
|
||||||
import java.net.HttpURLConnection;
|
|
||||||
import java.net.URL;
|
|
||||||
import java.nio.charset.StandardCharsets;
|
|
||||||
import java.nio.file.Files;
|
|
||||||
import java.nio.file.Path;
|
|
||||||
import java.nio.file.Paths;
|
|
||||||
import java.text.SimpleDateFormat;
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.Base64;
|
|
||||||
import java.util.Date;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.regex.Matcher;
|
|
||||||
import java.util.regex.Pattern;
|
|
||||||
|
|
||||||
public class HtmlToPdfConverter {
|
|
||||||
private static final Logger log = LoggerFactory.getLogger(HttpUtils.class);
|
|
||||||
|
|
||||||
public static MultipartFile convertHtmlToPdf(String htmlArticleTitle, String htmlContent) throws IOException {
|
|
||||||
|
|
||||||
String ht = content2Html(htmlContent);
|
|
||||||
|
|
||||||
|
|
||||||
String path = System.getProperty("user.dir");
|
|
||||||
// 根据日期建文件夹
|
|
||||||
String format = new SimpleDateFormat("yyyy-MM-dd").format(new Date());
|
|
||||||
String[] split = format.split("-");
|
|
||||||
Path dirPath = Paths.get(path, "temporary", split[0], split[1], split[2]);
|
|
||||||
try {
|
|
||||||
// clearTemporaryDirectory(Paths.get(path, "temporary")); // 清理目录
|
|
||||||
Files.createDirectories(dirPath);
|
|
||||||
|
|
||||||
} catch (IOException e) {
|
|
||||||
throw new RuntimeException("创建文件夹失败: " + e.getMessage(), e);
|
|
||||||
}
|
|
||||||
File pdfFile = new File(dirPath.toFile(), "pdf-" + System.currentTimeMillis() + ".pdf");
|
|
||||||
// 创建PDF文档
|
|
||||||
Document document = new Document();
|
|
||||||
|
|
||||||
FileOutputStream fos = null;
|
|
||||||
InputStream is = null;
|
|
||||||
try {
|
|
||||||
fos = new FileOutputStream(pdfFile);
|
|
||||||
|
|
||||||
PdfWriter writer = PdfWriter.getInstance(document, fos);
|
|
||||||
XMLWorkerFontProvider fontProvider = new XMLWorkerFontProvider(XMLWorkerFontProvider.DONTLOOKFORFONTS);
|
|
||||||
fontProvider.register("C:/Windows/Fonts/simsun.ttc");
|
|
||||||
|
|
||||||
document.open();
|
|
||||||
// 添加标题
|
|
||||||
Paragraph paragraph = new Paragraph(htmlArticleTitle); // 设置你自定义的字体格式
|
|
||||||
document.add(paragraph);
|
|
||||||
String updatedHtmlContent = "";
|
|
||||||
if (htmlContent.contains("img")) {
|
|
||||||
updatedHtmlContent = replaceImgSrcWithBase64(ht);
|
|
||||||
|
|
||||||
if (updatedHtmlContent != null) {
|
|
||||||
is = new ByteArrayInputStream(updatedHtmlContent.getBytes(StandardCharsets.UTF_8));
|
|
||||||
XMLWorkerHelper.getInstance().parseXHtml(writer, document, is, StandardCharsets.UTF_8, fontProvider);
|
|
||||||
// 添加图片
|
|
||||||
Elements imgElements = Jsoup.parse(updatedHtmlContent).select("img");
|
|
||||||
for (Element img : imgElements) {
|
|
||||||
String imgSrc = img.attr("src");
|
|
||||||
if (imgSrc.startsWith("data:image/jpeg;base64,")) {
|
|
||||||
String base64Data = imgSrc.substring("data:image/jpeg;base64,".length());
|
|
||||||
byte[] imageBytes = Base64.getDecoder().decode(base64Data);
|
|
||||||
Image image = Image.getInstance(imageBytes);
|
|
||||||
document.add(image); // 添加图片到 PDF
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
} else {
|
|
||||||
is = new ByteArrayInputStream(ht.getBytes(StandardCharsets.UTF_8));
|
|
||||||
XMLWorkerHelper.getInstance().parseXHtml(writer, document, is, StandardCharsets.UTF_8, fontProvider);
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
} catch (DocumentException | IOException e) {
|
|
||||||
throw new RuntimeException("转换PDF失败: " + e.getMessage(), e);
|
|
||||||
} finally {
|
|
||||||
|
|
||||||
if (document.isOpen()) {
|
|
||||||
document.close(); // 确保Document关闭
|
|
||||||
}
|
|
||||||
if (fos != null) {
|
|
||||||
try {
|
|
||||||
fos.close(); // 关闭文件输出流
|
|
||||||
} catch (IOException e) {
|
|
||||||
e.printStackTrace();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (is != null) {
|
|
||||||
try {
|
|
||||||
is.close(); // 关闭输入流
|
|
||||||
} catch (IOException e) {
|
|
||||||
e.printStackTrace();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
MultipartFile multipartFile = getMultipartFile(pdfFile);
|
|
||||||
// 在此处删除创建的 PDF 文件
|
|
||||||
boolean deleted = pdfFile.delete();
|
|
||||||
if (!deleted) {
|
|
||||||
log.error("删除文件失败: " + pdfFile.getAbsolutePath());
|
|
||||||
|
|
||||||
}
|
|
||||||
return multipartFile;
|
|
||||||
}
|
|
||||||
|
|
||||||
private static MultipartFile getMultipartFile(File file) throws IOException {
|
|
||||||
try (FileInputStream fileInputStream = new FileInputStream(file)) {
|
|
||||||
return new MockMultipartFile(file.getName(), file.getName(), "application/pdf", fileInputStream);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public static String content2Html(String htmlContent) {
|
|
||||||
String COMPLETE_CONTENT = "<html><head></head><body style=\"font-family: SimSun;\">" + htmlContent + "</body></html>";
|
|
||||||
return COMPLETE_CONTENT.replace("<br>", "<br/>");
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
private static String replaceImgSrcWithBase64(String htmlContent) {
|
|
||||||
List<String> imgSrcs = new ArrayList<>();
|
|
||||||
String regex = "<img[^>]+src=\"([^\"]+)\"[^>]*>"; // 正确的正则表达式
|
|
||||||
Pattern pattern = Pattern.compile(regex);
|
|
||||||
Matcher matcher = pattern.matcher(htmlContent);
|
|
||||||
String baseUrl = "http://39.101.188.84:7071";
|
|
||||||
|
|
||||||
// 使用 StringBuilder 来构建新的 HTML
|
|
||||||
StringBuilder updatedHtml = new StringBuilder(htmlContent);
|
|
||||||
|
|
||||||
// 定义要替换的起始位置
|
|
||||||
int offset = 0;
|
|
||||||
|
|
||||||
while (matcher.find()) {
|
|
||||||
String imgSrc = matcher.group(1); // 捕获组返回 img src 的值
|
|
||||||
String imgUrl = baseUrl + imgSrc;
|
|
||||||
|
|
||||||
// 将 URL 转化为 Base64
|
|
||||||
try {
|
|
||||||
URL url = new URL(imgUrl);
|
|
||||||
HttpURLConnection connection = (HttpURLConnection) url.openConnection();
|
|
||||||
connection.setRequestProperty("User-Agent", "Mozilla/5.0"); // 添加 User-Agent
|
|
||||||
connection.setDoInput(true);
|
|
||||||
connection.connect();
|
|
||||||
InputStream inputStream = connection.getInputStream();
|
|
||||||
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
|
||||||
byte[] buffer = new byte[1024];
|
|
||||||
int bytesRead;
|
|
||||||
while ((bytesRead = inputStream.read(buffer)) != -1) {
|
|
||||||
outputStream.write(buffer, 0, bytesRead);
|
|
||||||
}
|
|
||||||
inputStream.close();
|
|
||||||
byte[] imageBytes = outputStream.toByteArray();
|
|
||||||
String base64 = Base64.getEncoder().encodeToString(imageBytes);
|
|
||||||
|
|
||||||
// 替换 img 标签的 src 属性为 Base64 数据
|
|
||||||
String base64ImgTag = "<img src=\"data:image/jpeg;base64," + base64 + "\" />";
|
|
||||||
int start = matcher.start() + offset; // 累计偏移量
|
|
||||||
int end = matcher.end();
|
|
||||||
|
|
||||||
// 替换原来的 img 标签
|
|
||||||
updatedHtml.replace(start, end, base64ImgTag);
|
|
||||||
//System.out.println(updatedHtml.toString());
|
|
||||||
// 更新偏移量
|
|
||||||
offset += base64ImgTag.length() - (end - start);
|
|
||||||
} catch (Exception e) {
|
|
||||||
e.printStackTrace();
|
|
||||||
return null; // 处理异常或返回特定的错误值
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return updatedHtml.toString();
|
|
||||||
}
|
|
||||||
|
|
||||||
public String convertImageToBase64(String imgUrl) {
|
|
||||||
try {
|
|
||||||
// String encodedImgUrl = URLEncoder.encode(imgUrl, StandardCharsets.UTF_8.toString());
|
|
||||||
|
|
||||||
URL url = new URL(imgUrl);
|
|
||||||
HttpURLConnection connection = (HttpURLConnection) url.openConnection();
|
|
||||||
connection.setRequestProperty("User-Agent", "Mozilla/5.0"); // 添加 User-Agent
|
|
||||||
connection.setDoInput(true);
|
|
||||||
connection.connect();
|
|
||||||
InputStream inputStream = connection.getInputStream();
|
|
||||||
|
|
||||||
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
|
||||||
byte[] buffer = new byte[1024];
|
|
||||||
int bytesRead;
|
|
||||||
while ((bytesRead = inputStream.read(buffer)) != -1) {
|
|
||||||
outputStream.write(buffer, 0, bytesRead);
|
|
||||||
}
|
|
||||||
inputStream.close();
|
|
||||||
|
|
||||||
byte[] imageBytes = outputStream.toByteArray();
|
|
||||||
String base64 = Base64.getEncoder().encodeToString(imageBytes);
|
|
||||||
return "data:image/jpeg;base64," + base64; // 根据实际情况修改 MIME 类型
|
|
||||||
} catch (Exception e) {
|
|
||||||
e.printStackTrace();
|
|
||||||
return null; // 处理异常或返回特定的错误值
|
|
||||||
}
|
|
||||||
}
|
|
||||||
private static void clearTemporaryDirectory(Path directory) {
|
|
||||||
File dir = directory.toFile();
|
|
||||||
if (dir.exists() && dir.isDirectory()) {
|
|
||||||
for (File file : dir.listFiles()) {
|
|
||||||
if (!file.isDirectory()) {
|
|
||||||
file.delete(); // 删除文件
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -1,6 +0,0 @@
|
|||||||
<?xml version="1.0" encoding="UTF-8"?>
|
|
||||||
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
|
|
||||||
<mapper namespace="com.ruoyi.docking.mapper.ProjectMapper">
|
|
||||||
|
|
||||||
</mapper>
|
|
||||||
|
|
||||||
Loading…
Reference in new issue