From 05f3eec8ba39fb05eac09d8d3f2930cd77dd41f7 Mon Sep 17 00:00:00 2001 From: xiaoyong931011 <15274802129@163.com> Date: Tue, 12 Jul 2022 11:39:30 +0800 Subject: [PATCH] 20220606 --- src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CmsTemplateServiceImpl.java | 69 ++++++++++++++++++++++++---------- 1 files changed, 48 insertions(+), 21 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 e21b47d..5c20554 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 @@ -1,7 +1,7 @@ package com.xcong.farmer.cms.modules.system.service.Impl; import cn.hutool.core.io.FileUtil; -import cn.hutool.core.util.IdUtil; +import cn.hutool.core.io.file.FileWriter; import cn.hutool.core.util.StrUtil; import cn.hutool.core.util.ZipUtil; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; @@ -11,12 +11,10 @@ import com.xcong.farmer.cms.common.exception.GlobalException; import com.xcong.farmer.cms.common.response.Result; import com.xcong.farmer.cms.common.utils.FileUtils; -import com.xcong.farmer.cms.configurations.GlobalExceptionHandler; import com.xcong.farmer.cms.configurations.properties.CmsProperties; -import com.xcong.farmer.cms.core.template.TemplateConfiguration; +import com.xcong.farmer.cms.modules.system.dto.AdminSaveTemplateInfoDto; import com.xcong.farmer.cms.modules.system.dto.TemplateListDto; import com.xcong.farmer.cms.modules.system.entity.CmsTemplateEntity; -import com.xcong.farmer.cms.modules.system.entity.UserEntity; import com.xcong.farmer.cms.modules.system.mapper.CmsTemplateMapper; import com.xcong.farmer.cms.modules.system.service.ICmsTemplateService; import com.xcong.farmer.cms.modules.system.util.LoginUserUtil; @@ -24,16 +22,14 @@ import org.jsoup.nodes.Document; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; import org.springframework.web.multipart.MultipartFile; import java.io.File; -import java.io.FileInputStream; import java.io.IOException; -import java.io.InputStream; import java.nio.charset.StandardCharsets; import java.nio.file.Files; import java.nio.file.Paths; -import java.util.ArrayList; import java.util.Arrays; import java.util.List; @@ -45,14 +41,15 @@ public class CmsTemplateServiceImpl extends ServiceImpl<CmsTemplateMapper, CmsTemplateEntity> implements ICmsTemplateService { @Autowired - private TemplateConfiguration cfg; + private CmsProperties properties; private List<String> fileSuffix = Arrays.asList(".zip", ".html"); @Override public void updateTemplate(MultipartFile upload) { - String templatePath = cfg.templatePath; - String staticPath = cfg.staticPath; + String templatePath = properties.getTemplatePath(); + String staticPath = properties.getStaticPath(); + Long companyId = LoginUserUtil.getCompanyId(); String filename = upload.getOriginalFilename(); String suffix = filename.substring(filename.lastIndexOf(".")); @@ -78,6 +75,7 @@ for (File templateFile : files) { if (!templateFile.isFile()) { + FileUtil.move(templateFile, new File(staticPath), true); continue; } @@ -86,14 +84,14 @@ } String name = templateFile.getName(); - if (!name.endsWith(".list.html") && !name.endsWith(".article.html") && !name.endsWith(".index.html")) { + if (!name.endsWith(".list.html") && !name.endsWith(".article.html") && !name.endsWith("index.html")) { continue; } Document parse = Jsoup.parse(templateFile, null); String attr = parse.head().attr("name"); CmsTemplateEntity cmsTemplate = new CmsTemplateEntity(); - cmsTemplate.setCompanyId(10L); + cmsTemplate.setCompanyId(companyId); if (name.endsWith(".list.html")) { cmsTemplate.setType(2); } else if (name.endsWith(".article.html")) { @@ -102,7 +100,7 @@ cmsTemplate.setType(1); } cmsTemplate.setName(StrUtil.isNotBlank(attr) ? attr : templateFile.getName()); - cmsTemplate.setPath(IdUtil.simpleUUID()+templateFile.getName()); + cmsTemplate.setPath(templateFile.getName()); this.baseMapper.insert(cmsTemplate); } @@ -115,7 +113,7 @@ Document parse = Jsoup.parse(file, null); String attr = parse.head().attr("name"); CmsTemplateEntity cmsTemplate = new CmsTemplateEntity(); - cmsTemplate.setCompanyId(LoginUserUtil.getCompanyId()); + cmsTemplate.setCompanyId(companyId); if (file.getName().endsWith(".list.html")) { cmsTemplate.setType(2); } else if (file.getName().endsWith(".article.html")) { @@ -124,7 +122,7 @@ cmsTemplate.setType(1); } cmsTemplate.setName(StrUtil.isNotBlank(attr) ? attr : file.getName()); - cmsTemplate.setPath(IdUtil.simpleUUID()+file.getName()); + cmsTemplate.setPath(file.getName()); this.baseMapper.insert(cmsTemplate); } @@ -163,23 +161,52 @@ @Override public Result viewTemplateInfo(Long id) { - Long companyId = LoginUserUtil.getCompanyId(); + Result result = new Result(); CmsTemplateEntity cmsTemplateEntity = this.baseMapper.selectById(id); - String baseUrl = cmsProperties.getBaseUrl(); String templatePath = cmsProperties.getTemplatePath(); - String htmlUrl = baseUrl + templatePath; - String pathName = htmlUrl + "\\" + cmsTemplateEntity.getPath(); + String pathName = FileUtils.path(templatePath, cmsTemplateEntity.getPath()); byte[] bytes = new byte[0]; try { bytes = Files.readAllBytes(Paths.get(pathName)); } catch (IOException e) { e.printStackTrace(); - return Result.ok("未找到模板"); + return result.fail("未找到模板"); } String content = new String(bytes, StandardCharsets.UTF_8); - return Result.ok(content ); + result.setData(content); + return result; + } + + @Override + @Transactional + public Result saveTemplateInfo(AdminSaveTemplateInfoDto adminSaveTemplateInfoDto) { + Long companyId = LoginUserUtil.getCompanyId(); + Long id = adminSaveTemplateInfoDto.getId(); + CmsTemplateEntity cmsTemplateEntity = this.baseMapper.selectByIdAndCompanyId(id,companyId); + String name = cmsTemplateEntity.getName(); + Integer type = cmsTemplateEntity.getType(); + String templatePath = cmsProperties.getTemplatePath(); + String path = cmsTemplateEntity.getPath(); + + this.baseMapper.delete(id,companyId); + String pathNew = FileUtils.path(templatePath, path); + File file = new File(pathNew); + FileUtil.touch(file); + + //文件写入,直接覆盖 + FileWriter writer = new FileWriter(file); + writer.write(adminSaveTemplateInfoDto.getTemplateInfo(), false); + + CmsTemplateEntity cmsTemplate = new CmsTemplateEntity(); + cmsTemplate.setCompanyId(companyId); + cmsTemplate.setType(type); + cmsTemplate.setName(name); + cmsTemplate.setPath(path); + this.baseMapper.insert(cmsTemplate); + + return Result.ok("保存成功"); } } -- Gitblit v1.9.1