From c305b09fbd20849a39631ddd43980cb09e1805e9 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Mon, 04 Jul 2022 17:10:45 +0800
Subject: [PATCH] fix seeColumnInfo

---
 src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CmsTemplateServiceImpl.java |   33 ++++++++++++++++++++++++++++++++-
 1 files changed, 32 insertions(+), 1 deletions(-)

diff --git a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CmsTemplateServiceImpl.java b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CmsTemplateServiceImpl.java
index f1795c0..f1e072f 100644
--- a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CmsTemplateServiceImpl.java
+++ b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CmsTemplateServiceImpl.java
@@ -4,6 +4,9 @@
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.xcong.farmer.cms.common.exception.GlobalException;
+import com.xcong.farmer.cms.common.utils.FileUtils;
+import com.xcong.farmer.cms.configurations.GlobalExceptionHandler;
 import com.xcong.farmer.cms.core.template.TemplateConfiguration;
 import com.xcong.farmer.cms.modules.system.dto.TemplateListDto;
 import com.xcong.farmer.cms.modules.system.entity.CmsTemplateEntity;
@@ -13,6 +16,12 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.web.multipart.MultipartFile;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
 
 /**
  * @author wzy
@@ -24,10 +33,32 @@
     @Autowired
     private TemplateConfiguration cfg;
 
+    private List<String> fileSuffix = Arrays.asList(".zip", ".html");
+
     @Override
-    public void updateTemplate(MultipartFile file) {
+    public void updateTemplate(MultipartFile upload) {
         String templatePath = cfg.templatePath;
         String staticPath = cfg.staticPath;
+
+        String filename = upload.getOriginalFilename();
+        String suffix = filename.substring(filename.lastIndexOf("."));
+
+        if (!fileSuffix.contains(suffix)) {
+            throw new GlobalException("请上传正确格式文件");
+        }
+
+        try {
+            if (".zip".equals(suffix)) {
+                String path = FileUtils.path(templatePath, filename);
+                System.out.println(path);
+                File file = new File(path);
+                upload.transferTo(file);
+                FileUtils.zipUpload(file, templatePath, staticPath);
+            }
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+        System.out.println(suffix);
     }
 
     @Override

--
Gitblit v1.9.1