From 1581fd277f93678e8c3d8f8419ebd11e9f662759 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=AB=98=E9=9B=84?= <admin@cxcp.com>
Date: Fri, 12 May 2023 01:50:56 +0000
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96CAD=20=E9=87=8A=E6=94=BE?=
 =?UTF-8?q?=E8=B5=84=E6=BA=90=E6=96=B9=E6=B3=95=20=E4=BC=98=E5=8C=96CAD=20?=
 =?UTF-8?q?=E9=87=8A=E6=94=BE=E8=B5=84=E6=BA=90=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: 高雄 <admin@cxcp.com>
---
 .../cn/keking/service/FileHandlerService.java | 39 ++++++++++++-------
 1 file changed, 25 insertions(+), 14 deletions(-)

diff --git a/server/src/main/java/cn/keking/service/FileHandlerService.java b/server/src/main/java/cn/keking/service/FileHandlerService.java
index 393c2389..a7080cde 100644
--- a/server/src/main/java/cn/keking/service/FileHandlerService.java
+++ b/server/src/main/java/cn/keking/service/FileHandlerService.java
@@ -15,12 +15,13 @@ import com.aspose.cad.Image;
 import com.aspose.cad.LoadOptions;
 import com.aspose.cad.imageoptions.CadRasterizationOptions;
 import com.aspose.cad.imageoptions.PdfOptions;
-import com.itextpdf.text.exceptions.BadPasswordException;
 import com.itextpdf.text.pdf.PdfReader;
+import org.apache.commons.lang3.exception.ExceptionUtils;
 import org.apache.pdfbox.pdmodel.PDDocument;
 import org.apache.pdfbox.rendering.ImageType;
 import org.apache.pdfbox.rendering.PDFRenderer;
 import org.apache.pdfbox.tools.imageio.ImageIOUtil;
+import org.apache.poi.EncryptedDocumentException;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Value;
@@ -50,6 +51,7 @@ public class FileHandlerService {
     private final String fileDir = ConfigConstants.getFileDir();
     private final static String pdf2jpg_image_format = ".jpg";
     private final CacheService cacheService;
+    private static final String pdf_password_msg = "password";
 
     @Value("${server.tomcat.uri-encoding:UTF-8}")
     private String uriEncoding;
@@ -223,6 +225,7 @@ public class FileHandlerService {
     public List<String> pdf2jpg(String pdfFilePath, String pdfName, FileAttribute fileAttribute) throws Exception {
         boolean forceUpdatedCache = fileAttribute.forceUpdatedCache();
         String filePassword = fileAttribute.getFilePassword();
+        String pdfPassword = null;
         PDDocument doc = null;
         PdfReader pdfReader = null;
         if (!forceUpdatedCache) {
@@ -256,18 +259,25 @@ public class FileHandlerService {
                 imageUrls.add(imageUrl);
             }
             try {
-                pdfReader =  new PdfReader(pdfFilePath);   //判断pdf文件是否加密 缓存不加密文件
-                this.addPdf2jpgCache(pdfFilePath, pageCount);
-            } catch (BadPasswordException e) {
-
-            } catch (Exception e) {
-                e.printStackTrace();
+                pdfReader =  new PdfReader(pdfFilePath);   //读取PDF文件
+            } catch (Exception e) {  //获取异常方法 判断是否有加密字符串
+                Throwable[] throwableArray = ExceptionUtils.getThrowables(e);
+                for (Throwable throwable : throwableArray) {
+                    if (throwable instanceof IOException || throwable instanceof EncryptedDocumentException) {
+                        if (e.getMessage().toLowerCase().contains(pdf_password_msg)) {
+                            pdfPassword = pdf_password_msg;
+                        }
+                    }
+                }
+                logger.error("Convert pdf exception, pdfFilePath：{}", pdfFilePath, e);
             }finally {
                 if (pdfReader != null) {   //关闭
                     pdfReader.close();
                 }
             }
-
+            if(!pdfPassword.equals(pdf_password_msg)){  //判断是否加密文件 加密文件不缓存
+                this.addPdf2jpgCache(pdfFilePath, pageCount);
+            }
         } catch (IOException e) {
             logger.error("Convert pdf to jpg exception, pdfFilePath：{}", pdfFilePath, e);
             throw new Exception(e);
@@ -286,7 +296,7 @@ public class FileHandlerService {
      * @param outputFilePath pdf输出文件路径
      * @return 转换是否成功
      */
-    public boolean cadToPdf(String inputFilePath, String outputFilePath) {
+    public String cadToPdf(String inputFilePath, String outputFilePath)  throws Exception  {
         File outputFile = new File(outputFilePath);
         LoadOptions opts = new LoadOptions();
         opts.setSpecifiedEncoding(CodePages.SimpChinese);
@@ -300,21 +310,22 @@ public class FileHandlerService {
         cadRasterizationOptions.setDrawType(1);
         PdfOptions pdfOptions = new PdfOptions();
         pdfOptions.setVectorRasterizationOptions(cadRasterizationOptions);
-        OutputStream stream;
+        OutputStream stream = null;
         try {
             stream = new FileOutputStream(outputFile);
             cadImage.save(stream, pdfOptions);
-            stream.close();
-            cadImage.close();
-            return true;
         } catch (IOException e) {
             logger.error("PDFFileNotFoundException，inputFilePath：{}", inputFilePath, e);
+            return "null";
         } finally {
+            if (stream != null) {   //关闭
+                stream.close();
+            }
             if (cadImage != null) {   //关闭
                 cadImage.close();
             }
         }
-        return false;
+        return "true";
     }
     /**
      *