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/Impl/CmsTemplateServiceImpl.java |   31 ++++++++++++++++++++++++++-----
 1 files changed, 26 insertions(+), 5 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 2467205..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
@@ -20,6 +20,7 @@
 import com.xcong.farmer.cms.modules.system.mapper.CompanyMapper;
 import com.xcong.farmer.cms.modules.system.service.ICmsTemplateService;
 import com.xcong.farmer.cms.modules.system.util.LoginUserUtil;
+import lombok.extern.slf4j.Slf4j;
 import org.jsoup.Jsoup;
 import org.jsoup.nodes.Document;
 import org.jsoup.nodes.Element;
@@ -30,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;
@@ -42,6 +44,7 @@
  * @author wzy
  * @date 2022-07-04
  **/
+@Slf4j
 @Service
 public class CmsTemplateServiceImpl extends ServiceImpl<CmsTemplateMapper, CmsTemplateEntity> implements ICmsTemplateService {
 
@@ -178,7 +181,7 @@
     public Result dropdownList() {
         Long companyId = LoginUserUtil.getCompanyId();
         QueryWrapper<CmsTemplateEntity> objectQueryWrapper = new QueryWrapper<>();
-        objectQueryWrapper.eq("company_id",companyId);
+        objectQueryWrapper.eq("company_id", companyId);
         List<CmsTemplateEntity> cmsTemplateEntities = this.baseMapper.selectList(objectQueryWrapper);
         return Result.ok(cmsTemplateEntities);
     }
@@ -189,8 +192,10 @@
     @Override
     public Result viewTemplateInfo(Long id) {
         Result result = new Result();
+
+        CompanyEntity companyEntity = companyMapper.selectById(LoginUserUtil.getCompanyId());
         CmsTemplateEntity cmsTemplateEntity = this.baseMapper.selectById(id);
-        String templatePath = cmsProperties.getTemplatePath();
+        String templatePath = FileUtils.path(cmsProperties.getTemplatePath(), companyEntity.getCode());
         String pathName = FileUtils.path(templatePath, cmsTemplateEntity.getPath());
 
         byte[] bytes = new byte[0];
@@ -211,14 +216,17 @@
     public Result saveTemplateInfo(AdminSaveTemplateInfoDto adminSaveTemplateInfoDto) {
         Long companyId = LoginUserUtil.getCompanyId();
         Long id = adminSaveTemplateInfoDto.getId();
-        CmsTemplateEntity cmsTemplateEntity = this.baseMapper.selectByIdAndCompanyId(id,companyId);
+
+        CompanyEntity company = this.companyMapper.selectById(companyId);
+        CmsTemplateEntity cmsTemplateEntity = this.baseMapper.selectByIdAndCompanyId(id, companyId);
         String name = cmsTemplateEntity.getName();
         Integer type = cmsTemplateEntity.getType();
-        String templatePath = cmsProperties.getTemplatePath();
+        String templatePath = FileUtils.path(cmsProperties.getTemplatePath(), company.getCode());
         String path = cmsTemplateEntity.getPath();
 
-        this.baseMapper.delete(id,companyId);
+        this.baseMapper.delete(id, companyId);
         String pathNew = FileUtils.path(templatePath, path);
+        log.info("模板写入地址:{}", pathNew);
         File file = new File(pathNew);
         FileUtil.touch(file);
 
@@ -236,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