From d39169d41038ed62d1f6a9b25513b247a1a8c02f Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Tue, 30 Aug 2022 16:48:57 +0800
Subject: [PATCH] fix

---
 src/main/java/com/xcong/farmer/cms/modules/system/service/ICmsTemplateService.java         |    2 ++
 src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CmsTemplateServiceImpl.java |   14 ++++++++++++++
 src/main/java/com/xcong/farmer/cms/modules/system/controller/CmsTemplateController.java    |    5 +++++
 3 files changed, 21 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/xcong/farmer/cms/modules/system/controller/CmsTemplateController.java b/src/main/java/com/xcong/farmer/cms/modules/system/controller/CmsTemplateController.java
index 2a655e7..5c809a7 100644
--- a/src/main/java/com/xcong/farmer/cms/modules/system/controller/CmsTemplateController.java
+++ b/src/main/java/com/xcong/farmer/cms/modules/system/controller/CmsTemplateController.java
@@ -64,4 +64,9 @@
         return cmsTemplateService.saveTemplateInfo(adminSaveTemplateInfoDto);
     }
 
+    @ApiOperation(value = "下载模板", notes = "下载模板")
+    @PostMapping(value = "/downloadTemplate")
+    public Result downloadTemplate() {
+        return cmsTemplateService.downloadTemplate();
+    }
 }
diff --git a/src/main/java/com/xcong/farmer/cms/modules/system/service/ICmsTemplateService.java b/src/main/java/com/xcong/farmer/cms/modules/system/service/ICmsTemplateService.java
index 21b0018..9c893cd 100644
--- a/src/main/java/com/xcong/farmer/cms/modules/system/service/ICmsTemplateService.java
+++ b/src/main/java/com/xcong/farmer/cms/modules/system/service/ICmsTemplateService.java
@@ -22,4 +22,6 @@
     Result viewTemplateInfo(Long id);
 
     Result saveTemplateInfo(AdminSaveTemplateInfoDto adminSaveTemplateInfoDto);
+
+    Result downloadTemplate();
 }
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 b0022d6..f08dbff 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
@@ -31,6 +31,7 @@
 import org.springframework.web.multipart.MultipartFile;
 
 import java.io.File;
+import java.io.FileNotFoundException;
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.nio.charset.StandardCharsets;
@@ -243,4 +244,17 @@
         return Result.ok("保存成功");
     }
 
+    @Override
+    public Result downloadTemplate() {
+        Long companyId = LoginUserUtil.getCompanyId();
+        CompanyEntity company = this.companyMapper.selectById(companyId);
+
+        String templatePath = FileUtils.path(cmsProperties.getTemplatePath(), company.getCode());
+        String staticPath = FileUtils.path(cmsProperties.getStaticPath(), company.getCode());
+
+        String fileName = "template.zip";
+        ZipUtil.zip(templatePath, staticPath + "/template.zip", true);
+
+        return Result.ok("success", cmsProperties.getStaticUrl() + fileName);
+    }
 }

--
Gitblit v1.9.1