From 3fc51e7cbdabb7dcd1615c6efac52323f6ca8adb Mon Sep 17 00:00:00 2001 From: liusheng <337615773@qq.com> Date: 星期三, 03 七月 2024 18:28:34 +0800 Subject: [PATCH] 代码提交 --- ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java | 224 +++++++++++++++++++++++++++++++++++++++++++++---------- 1 files changed, 182 insertions(+), 42 deletions(-) diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java index cec5006..5268e61 100644 --- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java +++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CommonController.java @@ -1,9 +1,26 @@ package com.ruoyi.web.controller.common; +import java.io.*; import java.util.ArrayList; import java.util.List; +import java.util.regex.Matcher; +import java.util.regex.Pattern; +import javax.lang.model.util.Elements; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; +import javax.xml.parsers.DocumentBuilderFactory; +import javax.xml.parsers.ParserConfigurationException; + +import com.ruoyi.common.enums.PhotoEnum; +import com.ruoyi.common.enums.RadioEnum; +import com.ruoyi.common.enums.VadioEnum; +import org.apache.poi.hwpf.HWPFDocument; +import org.apache.poi.hwpf.converter.PicturesManager; +import org.apache.poi.hwpf.converter.WordToHtmlConverter; +import org.apache.poi.hwpf.usermodel.Picture; +import org.apache.poi.hwpf.usermodel.PictureType; +import org.jsoup.Jsoup; +import org.jsoup.nodes.Element; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Autowired; @@ -20,16 +37,17 @@ import com.ruoyi.common.utils.file.FileUploadUtils; import com.ruoyi.common.utils.file.FileUtils; import com.ruoyi.framework.config.ServerConfig; +import org.springframework.web.multipart.commons.CommonsMultipartFile; +import org.w3c.dom.Document; /** * 閫氱敤璇锋眰澶勭悊 - * + * * @author ruoyi */ @RestController @RequestMapping("/common") -public class CommonController -{ +public class CommonController { private static final Logger log = LoggerFactory.getLogger(CommonController.class); @Autowired @@ -39,17 +57,14 @@ /** * 閫氱敤涓嬭浇璇锋眰 - * + * * @param fileName 鏂囦欢鍚嶇О - * @param delete 鏄惁鍒犻櫎 + * @param delete 鏄惁鍒犻櫎 */ @GetMapping("/download") - public void fileDownload(String fileName, Boolean delete, HttpServletResponse response, HttpServletRequest request) - { - try - { - if (!FileUtils.checkAllowDownload(fileName)) - { + public void fileDownload(String fileName, Boolean delete, HttpServletResponse response, HttpServletRequest request) { + try { + if (!FileUtils.checkAllowDownload(fileName)) { throw new Exception(StringUtils.format("鏂囦欢鍚嶇О({})闈炴硶锛屼笉鍏佽涓嬭浇銆� ", fileName)); } String realFileName = System.currentTimeMillis() + fileName.substring(fileName.indexOf("_") + 1); @@ -58,14 +73,71 @@ response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE); FileUtils.setAttachmentResponseHeader(response, realFileName); FileUtils.writeBytes(filePath, response.getOutputStream()); - if (delete) - { + if (delete) { FileUtils.deleteFile(filePath); } - } - catch (Exception e) - { + } catch (Exception e) { log.error("涓嬭浇鏂囦欢澶辫触", e); + } + } + + /** + * 鍒嗙被涓婁紶璇锋眰 + */ + @PostMapping("/uploadSort") + public AjaxResult uploadFileSort(MultipartFile file) { + try { + // 涓婁紶鏂囦欢璺緞 + String filePath = RuoYiConfig.getUploadPath(); + String originalFilename = file.getOriginalFilename().split("\\.", 2)[1]; + if (VadioEnum.getDescByCode(originalFilename)) { + filePath = filePath + "/vadio"; + } else if (RadioEnum.getDescByCode(originalFilename)) { + filePath = filePath + "/radio"; + } else if (PhotoEnum.getDescByCode(originalFilename)) { + filePath = filePath + "/photo"; + } else { + filePath = filePath + "/file"; + } + // 涓婁紶骞惰繑鍥炴柊鏂囦欢鍚嶇О + String fileName = FileUploadUtils.uploadSort(filePath, file); + String url = serverConfig.getUrl() + fileName; + AjaxResult ajax = AjaxResult.success(); + ajax.put("url", url); + ajax.put("fileName", fileName); + ajax.put("newFileName", FileUtils.getName(fileName)); + ajax.put("originalFilename", file.getOriginalFilename()); + return ajax; + } catch (Exception e) { + return AjaxResult.error(e.getMessage()); + } + } + + /** + * 涓婁紶骞惰浆鎴恏tml璇锋眰 + */ + @PostMapping("/uploadShow") + public AjaxResult uploadFileShow(MultipartFile file) { + try { + // 涓婁紶鏂囦欢璺緞 + String filePath = RuoYiConfig.getUploadPath(); + String originalFilename = file.getOriginalFilename().split("\\.", 2)[1]; + filePath = filePath + "/show/" + file.getOriginalFilename().split("\\.", 2)[0]; + // 涓婁紶骞惰繑鍥炴柊鏂囦欢鍚嶇О + String fileName = FileUploadUtils.uploadSort(filePath, file); + //灏唚ord杞垚html + convertDocToHtml(filePath + "\\" + file.getOriginalFilename(), filePath + "\\" + file.getOriginalFilename().split("\\.", 2)[0] + ".html", filePath); + updatePicture(filePath + "\\" + file.getOriginalFilename().split("\\.", 2)[0] + ".html"); + + String url = serverConfig.getUrl() + fileName.replaceAll("\\.[^.]*$", ".html"); + AjaxResult ajax = AjaxResult.success(); + ajax.put("url", url); + ajax.put("fileName", fileName); + ajax.put("newFileName", FileUtils.getName(fileName)); + ajax.put("originalFilename", file.getOriginalFilename()); + return ajax; + } catch (Exception e) { + return AjaxResult.error(e.getMessage()); } } @@ -73,10 +145,8 @@ * 閫氱敤涓婁紶璇锋眰锛堝崟涓級 */ @PostMapping("/upload") - public AjaxResult uploadFile(MultipartFile file) throws Exception - { - try - { + public AjaxResult uploadFile(MultipartFile file) { + try { // 涓婁紶鏂囦欢璺緞 String filePath = RuoYiConfig.getUploadPath(); // 涓婁紶骞惰繑鍥炴柊鏂囦欢鍚嶇О @@ -88,9 +158,7 @@ ajax.put("newFileName", FileUtils.getName(fileName)); ajax.put("originalFilename", file.getOriginalFilename()); return ajax; - } - catch (Exception e) - { + } catch (Exception e) { return AjaxResult.error(e.getMessage()); } } @@ -99,18 +167,15 @@ * 閫氱敤涓婁紶璇锋眰锛堝涓級 */ @PostMapping("/uploads") - public AjaxResult uploadFiles(List<MultipartFile> files) throws Exception - { - try - { + public AjaxResult uploadFiles(List<MultipartFile> files) throws Exception { + try { // 涓婁紶鏂囦欢璺緞 String filePath = RuoYiConfig.getUploadPath(); List<String> urls = new ArrayList<String>(); List<String> fileNames = new ArrayList<String>(); List<String> newFileNames = new ArrayList<String>(); List<String> originalFilenames = new ArrayList<String>(); - for (MultipartFile file : files) - { + for (MultipartFile file : files) { // 涓婁紶骞惰繑鍥炴柊鏂囦欢鍚嶇О String fileName = FileUploadUtils.upload(filePath, file); String url = serverConfig.getUrl() + fileName; @@ -125,9 +190,7 @@ ajax.put("newFileNames", StringUtils.join(newFileNames, FILE_DELIMETER)); ajax.put("originalFilenames", StringUtils.join(originalFilenames, FILE_DELIMETER)); return ajax; - } - catch (Exception e) - { + } catch (Exception e) { return AjaxResult.error(e.getMessage()); } } @@ -136,13 +199,9 @@ * 鏈湴璧勬簮閫氱敤涓嬭浇 */ @GetMapping("/download/resource") - public void resourceDownload(String resource, HttpServletRequest request, HttpServletResponse response) - throws Exception - { - try - { - if (!FileUtils.checkAllowDownload(resource)) - { + public void resourceDownload(String resource, HttpServletRequest request, HttpServletResponse response) throws Exception { + try { + if (!FileUtils.checkAllowDownload(resource)) { throw new Exception(StringUtils.format("璧勬簮鏂囦欢({})闈炴硶锛屼笉鍏佽涓嬭浇銆� ", resource)); } // 鏈湴璧勬簮璺緞 @@ -154,10 +213,91 @@ response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE); FileUtils.setAttachmentResponseHeader(response, downloadName); FileUtils.writeBytes(downloadPath, response.getOutputStream()); - } - catch (Exception e) - { + } catch (Exception e) { log.error("涓嬭浇鏂囦欢澶辫触", e); } } + + + private void convertDocToHtml(String docFilePath, String outputHtmlFilePath, String imageDirPath) throws IOException, ParserConfigurationException { + // Load DOC into HWPFDocument + InputStream inputStream = new FileInputStream(docFilePath); + HWPFDocument document = new HWPFDocument(inputStream); + // Prepare HTML output + WordToHtmlConverter wordToHtmlConverter = new WordToHtmlConverter(DocumentBuilderFactory.newInstance().newDocumentBuilder().newDocument()); + // Set PicturesManager to handle image saving + wordToHtmlConverter.setPicturesManager(new PicturesManager() { + @Override + public String savePicture(byte[] content, PictureType pictureType, String suggestedName, float widthInches, float heightInches) { + File imageFile = new File(imageDirPath + "/" + suggestedName); + try (FileOutputStream fos = new FileOutputStream(imageFile)) { + fos.write(content); + } catch (IOException e) { + e.printStackTrace(); + } + return suggestedName; + } + }); + // Convert document to HTML + wordToHtmlConverter.processDocument(document); + // Save images + List<Picture> pics = document.getPicturesTable().getAllPictures(); + if (pics != null) { + for (Picture pic : pics) { + try (FileOutputStream fos = new FileOutputStream(imageDirPath + "/" + pic.suggestFullFileName())) { + fos.write(pic.getContent()); + } + } + } + // Convert to HTML + Document htmlDocument = wordToHtmlConverter.getDocument(); + ByteArrayOutputStream outStream = new ByteArrayOutputStream(); + org.apache.xml.serialize.OutputFormat format = new org.apache.xml.serialize.OutputFormat(htmlDocument); + org.apache.xml.serialize.XMLSerializer serializer = new org.apache.xml.serialize.XMLSerializer(outStream, format); + serializer.serialize(htmlDocument); + // Write HTML file + try (OutputStream outputStream = new FileOutputStream(outputHtmlFilePath)) { + outputStream.write(outStream.toByteArray()); + } + } + + private void updatePicture(String htmlFilePath) { + try { + // Load HTML file + File inputFile = new File(htmlFilePath); + org.jsoup.nodes.Document doc = Jsoup.parse(inputFile, "UTF-8"); + + // Select all elements with style attribute and modify their width and height to 90% + org.jsoup.select.Elements elementsWithStyle = doc.select("[style]"); + for (Element element : elementsWithStyle) { + String style = element.attr("style"); + String modifiedStyle = modifyStyle(style); + element.attr("style", modifiedStyle); + } + + // Save the modified HTML to output file + try (FileWriter writer = new FileWriter(htmlFilePath)) { + writer.write(doc.outerHtml()); + } + + System.out.println("Modification completed successfully."); + } catch (IOException e) { + e.printStackTrace(); + } + } + + private String modifyStyle(String style) { + // Define patterns for width and height + Pattern widthPattern = Pattern.compile("width\\s*:\\s*[^;]+"); + Pattern heightPattern = Pattern.compile("height\\s*:\\s*[^;]+"); + + // Replace width and height with 90% + Matcher widthMatcher = widthPattern.matcher(style); + style = widthMatcher.replaceAll("width: 90%"); + + Matcher heightMatcher = heightPattern.matcher(style); + style = heightMatcher.replaceAll("height: 90%"); + + return style; + } } -- Gitblit v1.9.3