| New file |
| | |
|---|
| | | package com.ruoyi.web.controller.tool; |
|---|
| | | |
|---|
| | | import com.documents4j.api.DocumentType; |
|---|
| | | import com.documents4j.api.IConverter; |
|---|
| | | import com.documents4j.job.LocalConverter; |
|---|
| | | import lombok.extern.slf4j.Slf4j; |
|---|
| | | import org.apache.poi.xwpf.usermodel.*; |
|---|
| | | import org.springframework.beans.factory.annotation.Autowired; |
|---|
| | | import org.springframework.mock.web.MockMultipartFile; |
|---|
| | | import org.springframework.stereotype.Component; |
|---|
| | | import org.springframework.web.multipart.MultipartFile; |
|---|
| | | |
|---|
| | | import java.io.*; |
|---|
| | | import java.net.URL; |
|---|
| | | import java.nio.charset.StandardCharsets; |
|---|
| | | import java.nio.file.Files; |
|---|
| | | import java.util.ArrayList; |
|---|
| | | import java.util.Arrays; |
|---|
| | | import java.util.List; |
|---|
| | | import java.util.Map; |
|---|
| | | import java.util.concurrent.TimeUnit; |
|---|
| | | |
|---|
| | | @Slf4j |
|---|
| | | @Component |
|---|
| | | public class PdfUtils { |
|---|
| | | @Autowired |
|---|
| | | private TencentCosUtil tencentCosUtil; |
|---|
| | | /** |
|---|
| | | * word 转 pdf |
|---|
| | | * |
|---|
| | | */ |
|---|
| | | // public String wordToPdf(String url,String filePath, String fileName) { |
|---|
| | | // try { |
|---|
| | | // DocumentType documentType = DocumentType.DOC; |
|---|
| | | // if(url.contains(".docx")){ |
|---|
| | | // documentType = DocumentType.DOCX; |
|---|
| | | // } |
|---|
| | | // if(url.contains(".doc")){ |
|---|
| | | // documentType = DocumentType.DOC; |
|---|
| | | // } |
|---|
| | | // if(url.contains(".xlsx")){ |
|---|
| | | // documentType = DocumentType.XLSX; |
|---|
| | | // }else { |
|---|
| | | // if(url.contains(".xls")){ |
|---|
| | | // documentType = DocumentType.XLS; |
|---|
| | | // } |
|---|
| | | // } |
|---|
| | | // InputStream inputStream = new URL(url).openStream(); |
|---|
| | | // ByteArrayOutputStream stream = new ByteArrayOutputStream(); |
|---|
| | | // IConverter converter = LocalConverter.builder().build(); |
|---|
| | | // converter.convert(inputStream) |
|---|
| | | // .as(documentType) |
|---|
| | | // .to(stream) |
|---|
| | | // .as(DocumentType.PDF).execute(); |
|---|
| | | // |
|---|
| | | // //上传图片 |
|---|
| | | // byte2File(stream.toByteArray(),filePath + "/pdf",fileName.substring(0,fileName.lastIndexOf(".")) + ".pdf"); |
|---|
| | | // MultipartFile multipartFile = convertToMultipartFile(stream,fileName.substring(0,fileName.lastIndexOf(".")) ); |
|---|
| | | // String s = tencentCosUtil.upLoadFile(multipartFile,"/wordToPdf"); |
|---|
| | | // |
|---|
| | | // stream.close(); |
|---|
| | | // inputStream.close(); |
|---|
| | | // return s; |
|---|
| | | // } catch (Exception e) { |
|---|
| | | // e.printStackTrace(); |
|---|
| | | // } |
|---|
| | | // return null; |
|---|
| | | // } |
|---|
| | | |
|---|
| | | public String wordToPdf(String filePath, String fileName) { |
|---|
| | | try { |
|---|
| | | // 确保路径正确性 |
|---|
| | | String inputFile = new File(filePath, fileName).getAbsolutePath(); |
|---|
| | | String outputDir = new File(filePath, "pdf").getAbsolutePath(); |
|---|
| | | |
|---|
| | | // 创建输出目录 |
|---|
| | | new File(outputDir).mkdirs(); |
|---|
| | | |
|---|
| | | // 使用完整的转换参数 |
|---|
| | | List<String> command = Arrays.asList( |
|---|
| | | "/usr/bin/libreoffice", // 使用完整路径 |
|---|
| | | "--headless", |
|---|
| | | "--norestore", |
|---|
| | | "--convert-to", |
|---|
| | | "pdf:writer_pdf_Export:PDFExport{" + |
|---|
| | | "EmbedStandardFonts=1;" + |
|---|
| | | "EmbedFonts=1;" + |
|---|
| | | "EmbedOnlyUsedFonts=0;" + |
|---|
| | | "UseTaggedPDF=1" + |
|---|
| | | "}", |
|---|
| | | "--outdir", |
|---|
| | | outputDir, |
|---|
| | | inputFile |
|---|
| | | ); |
|---|
| | | |
|---|
| | | // 创建进程构建器 |
|---|
| | | ProcessBuilder pb = new ProcessBuilder(command); |
|---|
| | | |
|---|
| | | // 设置环境变量 |
|---|
| | | Map<String, String> env = pb.environment(); |
|---|
| | | env.put("LC_ALL", "zh_CN.UTF-8"); |
|---|
| | | env.put("LANG", "zh_CN.UTF-8"); |
|---|
| | | env.put("LANGUAGE", "zh_CN.UTF-8"); |
|---|
| | | |
|---|
| | | // 重定向错误流到标准输出 |
|---|
| | | pb.redirectErrorStream(true); |
|---|
| | | |
|---|
| | | // 启动进程 |
|---|
| | | Process process = pb.start(); |
|---|
| | | |
|---|
| | | // 读取输出 |
|---|
| | | StringBuilder output = new StringBuilder(); |
|---|
| | | try (BufferedReader reader = new BufferedReader( |
|---|
| | | new InputStreamReader(process.getInputStream(), StandardCharsets.UTF_8))) { |
|---|
| | | String line; |
|---|
| | | while ((line = reader.readLine()) != null) { |
|---|
| | | output.append(line).append("\n"); |
|---|
| | | System.out.println(line); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 等待进程完成，设置超时 |
|---|
| | | if (!process.waitFor(120, TimeUnit.SECONDS)) { |
|---|
| | | process.destroyForcibly(); |
|---|
| | | throw new RuntimeException("转换超时"); |
|---|
| | | } |
|---|
| | | |
|---|
| | | int exitCode = process.exitValue(); |
|---|
| | | if (exitCode != 0) { |
|---|
| | | throw new RuntimeException("转换失败，退出码：" + exitCode + "\n输出：" + output); |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 检查生成的PDF文件 |
|---|
| | | String pdfFileName = fileName.substring(0, fileName.lastIndexOf(".")) + ".pdf"; |
|---|
| | | File pdfFile = new File(outputDir, pdfFileName); |
|---|
| | | |
|---|
| | | if (!pdfFile.exists() || pdfFile.length() == 0) { |
|---|
| | | throw new RuntimeException("PDF文件未生成或为空"); |
|---|
| | | } |
|---|
| | | String absolutePath = pdfFile.getAbsolutePath(); |
|---|
| | | |
|---|
| | | MultipartFile multipartFile = convertFileToMultipartFile(pdfFile); |
|---|
| | | String s = tencentCosUtil.upLoadFile(multipartFile,"/wordToPdf"); |
|---|
| | | return s; |
|---|
| | | |
|---|
| | | } catch (Exception e) { |
|---|
| | | throw new RuntimeException("PDF转换失败: " + e.getMessage(), e); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | |
|---|
| | | public MultipartFile convertFileToMultipartFile(File file) throws IOException { |
|---|
| | | // 读取文件内容到字节数组 |
|---|
| | | byte[] fileContent = Files.readAllBytes(file.toPath()); |
|---|
| | | |
|---|
| | | // 创建 MultipartFile 对象 |
|---|
| | | MultipartFile multipartFile = new MockMultipartFile( |
|---|
| | | file.getName(), // 文件名 |
|---|
| | | file.getName(), // 原始文件名 |
|---|
| | | "application/pdf", // 内容类型，根据实际情况调整 |
|---|
| | | fileContent // 文件内容 |
|---|
| | | ); |
|---|
| | | |
|---|
| | | return multipartFile; |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 在使用前检查和配置环境 |
|---|
| | | public static void setupEnvironment() { |
|---|
| | | try { |
|---|
| | | // 1. 检查LibreOffice安装 |
|---|
| | | checkLibreOffice(); |
|---|
| | | |
|---|
| | | // 2. 检查和安装字体 |
|---|
| | | installFonts(); |
|---|
| | | |
|---|
| | | // 3. 配置字体 |
|---|
| | | configureFonts(); |
|---|
| | | |
|---|
| | | // 4. 验证环境变量 |
|---|
| | | checkEnvironment(); |
|---|
| | | |
|---|
| | | } catch (Exception e) { |
|---|
| | | throw new RuntimeException("环境设置失败: " + e.getMessage(), e); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | |
|---|
| | | private static void checkLibreOffice() throws IOException, InterruptedException { |
|---|
| | | Process process = Runtime.getRuntime().exec("which libreoffice"); |
|---|
| | | if (process.waitFor() != 0) { |
|---|
| | | throw new RuntimeException("LibreOffice未安装"); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | |
|---|
| | | private static void installFonts() throws IOException, InterruptedException { |
|---|
| | | // 创建字体安装脚本 |
|---|
| | | String scriptContent = |
|---|
| | | "#!/bin/bash\n" + |
|---|
| | | "apt-get update\n" + |
|---|
| | | "apt-get install -y fonts-wqy-zenhei fonts-wqy-microhei fonts-arphic-ukai fonts-arphic-uming\n" + |
|---|
| | | "fc-cache -fv\n"; |
|---|
| | | |
|---|
| | | File script = new File("/tmp/install_fonts.sh"); |
|---|
| | | Files.write(script.toPath(), scriptContent.getBytes()); |
|---|
| | | script.setExecutable(true); |
|---|
| | | |
|---|
| | | // 执行脚本 |
|---|
| | | Process process = Runtime.getRuntime().exec("sudo /tmp/install_fonts.sh"); |
|---|
| | | process.waitFor(); |
|---|
| | | |
|---|
| | | // 清理脚本 |
|---|
| | | script.delete(); |
|---|
| | | } |
|---|
| | | |
|---|
| | | private static void configureFonts() throws IOException { |
|---|
| | | // 创建字体配置文件 |
|---|
| | | String fontConfig = |
|---|
| | | "<?xml version=\"1.0\"?>\n" + |
|---|
| | | "<!DOCTYPE fontconfig SYSTEM \"fonts.dtd\">\n" + |
|---|
| | | "<fontconfig>\n" + |
|---|
| | | " <match target=\"pattern\">\n" + |
|---|
| | | " <test name=\"family\"><string>serif</string></test>\n" + |
|---|
| | | " <edit name=\"family\" mode=\"prepend\">\n" + |
|---|
| | | " <string>WenQuanYi Zen Hei</string>\n" + |
|---|
| | | " </edit>\n" + |
|---|
| | | " </match>\n" + |
|---|
| | | " <match target=\"pattern\">\n" + |
|---|
| | | " <test name=\"family\"><string>sans-serif</string></test>\n" + |
|---|
| | | " <edit name=\"family\" mode=\"prepend\">\n" + |
|---|
| | | " <string>WenQuanYi Zen Hei</string>\n" + |
|---|
| | | " </edit>\n" + |
|---|
| | | " </match>\n" + |
|---|
| | | "</fontconfig>"; |
|---|
| | | |
|---|
| | | // 写入配置文件 |
|---|
| | | File configFile = new File(System.getProperty("user.home") + "/.fonts.conf"); |
|---|
| | | Files.write(configFile.toPath(), fontConfig.getBytes()); |
|---|
| | | } |
|---|
| | | |
|---|
| | | private static void checkEnvironment() { |
|---|
| | | // 检查环境变量 |
|---|
| | | String[] requiredVars = {"LANG", "LC_ALL", "LANGUAGE"}; |
|---|
| | | for (String var : requiredVars) { |
|---|
| | | String value = System.getenv(var); |
|---|
| | | if (value == null || !value.contains("zh_CN")) { |
|---|
| | | System.err.println("警告: " + var + " 环境变量未正确设置"); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
| | | |
|---|
| | | public static MultipartFile convertToMultipartFile(ByteArrayOutputStream baos, String fileName) throws IOException { |
|---|
| | | // 创建一个临时文件 |
|---|
| | | File tempFile = File.createTempFile(fileName, ".pdf"); |
|---|
| | | |
|---|
| | | // 将ByteArrayOutputStream中的数据写入临时文件 |
|---|
| | | try (FileOutputStream fos = new FileOutputStream(tempFile)) { |
|---|
| | | baos.writeTo(fos); |
|---|
| | | } catch (IOException e) { |
|---|
| | | e.printStackTrace(); |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 创建一个MultipartFile对象 |
|---|
| | | return new MockMultipartFile( |
|---|
| | | fileName + ".pdf", // 参数名称 |
|---|
| | | fileName + ".pdf", // 文件名 |
|---|
| | | "application/pdf", // 内容类型 |
|---|
| | | Files.readAllBytes(tempFile.toPath()) // 文件内容 |
|---|
| | | ); |
|---|
| | | } |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * file转byte |
|---|
| | | */ |
|---|
| | | public static byte[] file2byte(File file){ |
|---|
| | | byte[] buffer = null; |
|---|
| | | try{ |
|---|
| | | FileInputStream fis = new FileInputStream(file); |
|---|
| | | ByteArrayOutputStream bos = new ByteArrayOutputStream(); |
|---|
| | | byte[] b = new byte[1024]; |
|---|
| | | int n; |
|---|
| | | while ((n = fis.read(b)) != -1) |
|---|
| | | { |
|---|
| | | bos.write(b, 0, n); |
|---|
| | | } |
|---|
| | | fis.close(); |
|---|
| | | bos.close(); |
|---|
| | | buffer = bos.toByteArray(); |
|---|
| | | }catch (FileNotFoundException e){ |
|---|
| | | e.printStackTrace(); |
|---|
| | | } |
|---|
| | | catch (IOException e){ |
|---|
| | | e.printStackTrace(); |
|---|
| | | } |
|---|
| | | return buffer; |
|---|
| | | } |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * byte 转file |
|---|
| | | */ |
|---|
| | | public static File byte2File(byte[] buf, String filePath, String fileName){ |
|---|
| | | BufferedOutputStream bos = null; |
|---|
| | | FileOutputStream fos = null; |
|---|
| | | OutputStreamWriter osw = null; |
|---|
| | | File file = null; |
|---|
| | | try{ |
|---|
| | | File dir = new File(filePath+"/"); |
|---|
| | | if (!dir.exists()){ |
|---|
| | | dir.mkdirs(); |
|---|
| | | } |
|---|
| | | file = new File(filePath +File.separator + fileName); |
|---|
| | | fos = new FileOutputStream(file); |
|---|
| | | bos = new BufferedOutputStream(fos); |
|---|
| | | // osw = new OutputStreamWriter(fos, StandardCharsets.UTF_8); |
|---|
| | | bos.write(buf); |
|---|
| | | }catch (Exception e){ |
|---|
| | | e.printStackTrace(); |
|---|
| | | } |
|---|
| | | finally{ |
|---|
| | | if (bos != null){ |
|---|
| | | try{ |
|---|
| | | bos.close(); |
|---|
| | | }catch (IOException e){ |
|---|
| | | e.printStackTrace(); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | if (fos != null){ |
|---|
| | | try{ |
|---|
| | | fos.close(); |
|---|
| | | }catch (IOException e){ |
|---|
| | | e.printStackTrace(); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
| | | return file; |
|---|
| | | } |
|---|
| | | /** |
|---|
| | | * multipartFile转File |
|---|
| | | **/ |
|---|
| | | public static File multipartFile2File(MultipartFile multipartFile){ |
|---|
| | | File file = null; |
|---|
| | | if (multipartFile != null){ |
|---|
| | | try { |
|---|
| | | file= File.createTempFile("tmp", null); |
|---|
| | | multipartFile.transferTo(file); |
|---|
| | | System.gc(); |
|---|
| | | file.deleteOnExit(); |
|---|
| | | }catch (Exception e){ |
|---|
| | | e.printStackTrace(); |
|---|
| | | log.warn("multipartFile转File发生异常："+e); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | return file; |
|---|
| | | } |
|---|
| | | |
|---|
| | | |
|---|
| | | public static void main(String[] args) { |
|---|
| | | // String url = "file:///E:\\qiyeweixin\\WXWork\\1688855207501340\\Cache\\File\\2024-09\\专业技术工作总结.docx"; |
|---|
| | | // String filePath = "E:\\qiyeweixin\\WXWork\\1688855207501340\\Cache\\File\\2024-09"; |
|---|
| | | // String fileName = "专业技术工作总结.docx"; |
|---|
| | | // String s = wordToPdf(url, filePath, fileName); |
|---|
| | | // System.err.println(s); |
|---|
| | | |
|---|
| | | // String url = "file:///F:\\测试动态列表Word.doc"; |
|---|
| | | //// String filePath = "E:\\qiyeweixin\\WXWork\\1688855207501340\\Cache\\File\\2024-09"; |
|---|
| | | //// String fileName = "专业技术工作总结.docx";4 |
|---|
| | | // String filePath = "F:\\"; |
|---|
| | | // |
|---|
| | | // String s = wordToPdf(url, filePath, "测试动态列表Word.doc"); |
|---|
| | | // System.err.println(s); |
|---|
| | | |
|---|
| | | |
|---|
| | | |
|---|
| | | // TODO Auto-generated method stub |
|---|
| | | |
|---|
| | | |
|---|
| | | // HashMap<String, Object> paramMap = new HashMap<>(); |
|---|
| | | // paramMap.put("CorpID", "SCZT006959"); |
|---|
| | | // paramMap.put("Pwd", "123456"); |
|---|
| | | // paramMap.put("Mobile", "19522115070"); |
|---|
| | | // paramMap.put("Cell", ""); |
|---|
| | | // paramMap.put("SendTime", ""); |
|---|
| | | // paramMap.put("Content", java.net.URLEncoder.encode("你好，这是测试短信发送。【职评网】")); |
|---|
| | | // String result3 = HttpUtil.post("http://sdk2.028lk.com/sdk2/BatchSend2.aspx", paramMap); |
|---|
| | | // if (result3 == null) { |
|---|
| | | // result3 = ""; |
|---|
| | | // } |
|---|
| | | // result3 = StrUtil.trim(result3); |
|---|
| | | // log.info("返回结果:" + result3); |
|---|
| | | // |
|---|
| | | // if (result3.matches("^\\d+$") && !result3.equals("0")) { |
|---|
| | | // return; |
|---|
| | | // } |
|---|
| | | |
|---|
| | | } |
|---|
| | | |
|---|
| | | public String test(String fileName){ |
|---|
| | | // String url = "file:///usr/local/project/file/"+fileName; |
|---|
| | | // String filePath = "E:\\qiyeweixin\\WXWork\\1688855207501340\\Cache\\File\\2024-09"; |
|---|
| | | // String fileName = "专业技术工作总结.docx";4 |
|---|
| | | String filePath = "/usr/local/project/file/"; |
|---|
| | | |
|---|
| | | String s = wordToPdf(filePath, fileName); |
|---|
| | | System.err.println(s); |
|---|
| | | |
|---|
| | | return s; |
|---|
| | | } |
|---|
| | | |
|---|
| | | } |
|---|
