From 60ed3f3460163b8cd6669506582285faadd51675 Mon Sep 17 00:00:00 2001
From: yjz <3131811435@qq.com>
Date: Tue, 2 Apr 2024 13:26:26 +0800
Subject: [PATCH] =?UTF-8?q?feat:=E6=8B=86=E5=88=86=E5=AF=BC=E5=85=A5?=
 =?UTF-8?q?=E3=80=81=E4=BC=81=E6=A0=87=E7=A8=BD=E6=9F=A5=E8=AE=A1=E5=88=92?=
 =?UTF-8?q?=E5=AF=BC=E5=85=A5=E8=A7=A3=E5=AF=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../impl/ProcessEsInspectPlanServiceImpl.java |  6 ++-
 .../impl/FileSplitItemsEOServiceImpl.java     | 41 ++++++++++++++++++-
 2 files changed, 43 insertions(+), 4 deletions(-)

diff --git a/laws-modules/src/main/java/com/jero/modules/activiti/process/esinspectplan/service/impl/ProcessEsInspectPlanServiceImpl.java b/laws-modules/src/main/java/com/jero/modules/activiti/process/esinspectplan/service/impl/ProcessEsInspectPlanServiceImpl.java
index f2384130..372afbf8 100644
--- a/laws-modules/src/main/java/com/jero/modules/activiti/process/esinspectplan/service/impl/ProcessEsInspectPlanServiceImpl.java
+++ b/laws-modules/src/main/java/com/jero/modules/activiti/process/esinspectplan/service/impl/ProcessEsInspectPlanServiceImpl.java
@@ -11,6 +11,7 @@ import com.jero.common.exception.JeroBootException;
 import com.jero.common.system.vo.LoginUser;
 import com.jero.common.util.BeanCopyUtils;
 import com.jero.common.util.ImportExcelUtil;
+import com.jero.common.util.IntekeyUtils;
 import com.jero.modules.activiti.process.esinspectplan.common.EsInspectPlanCommon;
 import com.jero.modules.activiti.process.esinspectplan.entity.ProcessEsInspectPlan;
 import com.jero.modules.activiti.process.esinspectplan.entity.ProcessEsInspectPlanOpinion;
@@ -215,7 +216,8 @@ public class ProcessEsInspectPlanServiceImpl extends ServiceImpl<ProcessEsInspec
 		Map<String, MultipartFile> fileMap = multipartRequest.getFileMap();
 		for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) {
 			// 获取上传文件对象
-			MultipartFile file = entity.getValue();
+			// 解密excel
+			MultipartFile file = IntekeyUtils.autoDecryptFile(entity.getValue());
 			ImportParams params = new ImportParams();
 			params.setNeedSave(true);
 			params.setHeadRows(1);
@@ -286,7 +288,7 @@ public class ProcessEsInspectPlanServiceImpl extends ServiceImpl<ProcessEsInspec
 		Map<String, MultipartFile> fileMap = multipartRequest.getFileMap();
 		for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) {
 			// 获取上传文件对象
-			MultipartFile file = entity.getValue();
+			MultipartFile file = IntekeyUtils.autoDecryptFile(entity.getValue());
 			ImportParams params = new ImportParams();
 			params.setNeedSave(true);
 			params.setHeadRows(1);
diff --git a/laws-modules/src/main/java/com/jero/modules/split/service/impl/FileSplitItemsEOServiceImpl.java b/laws-modules/src/main/java/com/jero/modules/split/service/impl/FileSplitItemsEOServiceImpl.java
index e6ea63b6..6c7bc91c 100644
--- a/laws-modules/src/main/java/com/jero/modules/split/service/impl/FileSplitItemsEOServiceImpl.java
+++ b/laws-modules/src/main/java/com/jero/modules/split/service/impl/FileSplitItemsEOServiceImpl.java
@@ -155,6 +155,32 @@ public class FileSplitItemsEOServiceImpl extends ServiceImpl<FileSplitItemsEOMap
     @Value(value = "${jero.path.uploadCos}")
     private String uploadCospath;
 
+    public static boolean ENCRYPT_ENABLE;
+    public static String DECRYPT_URL;
+    public static String DECRYPT_APP_CODE;
+    public static String DECRYPT_SECRET_KEY;
+
+    @Value("${download.enable}")
+    public void setEnable(boolean enable) {
+        ENCRYPT_ENABLE = enable;
+    }
+
+    @Value("${download.decrypt.url}")
+    public void setDecryptUrl(String decryptUrl) {
+        DECRYPT_URL = decryptUrl;
+    }
+
+    @Value("${download.decrypt.app_code}")
+    public void setDecryptAppCode(String decryptAppCode) {
+        DECRYPT_APP_CODE = decryptAppCode;
+    }
+
+    @Value("${download.decrypt.secret_key}")
+    public void setDecryptSecretKey(String decryptSecretKey) {
+        DECRYPT_SECRET_KEY = decryptSecretKey;
+    }
+
+
     @Override
     public int addItemContent(Map<String, Object> parameter){
         List<SarFileSplitItemsValEO>  itemValEOList = (List<SarFileSplitItemsValEO>) parameter.get("itemValEOList");
@@ -1596,9 +1622,20 @@ public class FileSplitItemsEOServiceImpl extends ServiceImpl<FileSplitItemsEOMap
                 //excel读取到的数据
                 List<Map<String, Object>> datas = result;
                 if(datas!=null &&!datas.isEmpty()){
-                    try {
+                    // 将excel文件解密
+                    InputStream inputStream = null;
+                    if (ENCRYPT_ENABLE) {
+                        inputStream = IntekeyUtils.decryptFile(DECRYPT_URL, DECRYPT_APP_CODE, DECRYPT_SECRET_KEY, null,
+                                new MockMultipartFile(
+                                        excelfile.getName(),
+                                        excelfile.getName(),
+                                        (String) null,
+                                        Files.newInputStream(excelfile.toPath())));
+                    } else {
+                        inputStream = Files.newInputStream(excelfile.toPath());
+                    }
+                    try (XSSFWorkbook workbook=new XSSFWorkbook(inputStream)){
                         //获取全部sheet页中表格
-                        XSSFWorkbook workbook=new XSSFWorkbook(new FileInputStream(excelfile));
                         XSSFSheet sheet=null;
                         List<SplitTableInfo> getSheetTableList = new ArrayList();
                         for (int i = 1; i < workbook.getNumberOfSheets(); i++) {//获取每个Sheet表