文件相关工具类Utils(WORD,PDF,PNG)

Mr_Huang_12023-12-21 6:38

导读，此工具类中包含以下功能：

1、文件后缀判断

2、图片后缀判断

3、word后缀判断

4、Excel后缀判断

5、PDF后缀判断

6、使用pdfbox将整个pdf转换成图片

7、pdf2word

8、word2PDF

9、向PDF指定页插入图片

10、PDF指定位置覆盖

11、删除文件夹及以下文件（可能出现删不全的情况）

12、zip文件返回

13、打压缩包

代码：

java 复制代码 
package cn.org.enst.report.office.utils.processdocument;

import cn.org.enst.common.exception.DbptException;
import cn.org.enst.common.utils.StringUtils;
import cn.org.enst.common.utils.file.FileUtils;
import com.baomidou.mybatisplus.core.toolkit.ArrayUtils;
import com.jacob.activeX.ActiveXComponent;
import com.jacob.com.ComThread;
import com.jacob.com.Dispatch;
import com.jacob.com.Variant;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.pdmodel.PDPage;
import org.apache.pdfbox.pdmodel.PDPageContentStream;
import org.apache.pdfbox.pdmodel.PDPageTree;
import org.apache.pdfbox.pdmodel.common.PDRectangle;
import org.apache.pdfbox.pdmodel.graphics.image.PDImageXObject;
import org.apache.pdfbox.rendering.PDFRenderer;
import org.apache.pdfbox.text.PDFTextStripper;

import javax.imageio.ImageIO;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.awt.image.BufferedImage;
import java.io.*;
import java.util.List;
import java.util.zip.CRC32;
import java.util.zip.CheckedOutputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

public class FileTypeUtils {

    /**
     * 文件后缀判断
     */
    public static boolean verificationFile(String originFileName) {
        // 后缀
        String extName = originFileName.substring(originFileName.lastIndexOf(".") + 1);
        // 判断是不是图片、文档、excel、pdf、压缩包等文件后缀
        if (extName.matches("(tif|gif|png|jpg|jpeg|bmp|doc|docx|pdf|xls|xlsx|rar|zip)")) {
            return true;
        }
        // "文件格式错误"
        return false;
    }

    /**
     * 图片后缀判断
     */
    public static boolean verificationImage(String originFileName) {
        // 后缀
        String extName = originFileName.substring(originFileName.lastIndexOf(".") + 1);
        // 判断是不是图片后缀
        if (extName.matches("(png|jpg|jpeg)")) {
            return true;
        }
        // "文件格式错误"
        return false;
    }

    /**
     * word后缀判断
     */
    public static boolean verificationWord(String originFileName) {
        // 后缀
        String extName = originFileName.substring(originFileName.lastIndexOf(".") + 1);
        // 判断是不是图片后缀
        if (extName.matches("(doc|docx)")) {
            return true;
        }
        // "文件格式错误"
        return false;
    }

    /**
     * Excel后缀判断
     */
    public static boolean verificationExcel(String originFileName) {
        // 后缀
        String extName = originFileName.substring(originFileName.lastIndexOf(".") + 1);
        // 判断是不是图片后缀
        if (extName.matches("(xls|xlsx)")) {
            return true;
        }
        // "文件格式错误"
        return false;
    }

    /**
     * PDF后缀判断
     */
    public static boolean verificationPDF(String originFileName) {
        // 后缀
        String extName = originFileName.substring(originFileName.lastIndexOf(".") + 1);
        // 判断是不是图片后缀
        if (extName.matches("(pdf)")) {
            return true;
        }
        // "文件格式错误"
        return false;
    }

    /**
     * 使用pdfbox将整个pdf转换成图片
     *
     * @param fileAddress 文件地址 如:C:\\Users\\user\\Desktop\\test
     * @param filename    PDF文件名不带后缀名
     * @param type        图片类型 png 和jpg
     * @param pathList    图片输出位置
     */
    public static void pdf2png(String fileAddress, String filename, String type, List<String> pathList) {
//        long startTime = System.currentTimeMillis();
        // 将文件地址和文件名拼接成路径 注意：线上环境不能使用\\拼接
        File file = new File(fileAddress + "\\" + filename + ".pdf");
        try {
            // 写入文件
            PDDocument doc = PDDocument.load(file);
            PDFRenderer renderer = new PDFRenderer(doc);
            int pageCount = doc.getNumberOfPages();
            for (int i = 0; i < pageCount; i++) {
                // dpi为144，越高越清晰，转换越慢
                BufferedImage image = renderer.renderImageWithDPI(i, 144); // Windows native DPI
                // 将图片写出到该路径下
                pathList.add(fileAddress + "\\" + filename + "_" + (i + 1) + "." + type);
                ImageIO.write(image, type, new File(fileAddress + "\\" + filename + "_" + (i + 1) + "." + type));
            }
            long endTime = System.currentTimeMillis();
//            System.out.println("共耗时：" + ((endTime - startTime) / 1000.0) + "秒");  //转化用时
        } catch (IOException e) {
            e.printStackTrace();
        }
    }


    /**
     * @param source 要转换的word文件
     * @param target 要转换成为的pdf文件
     */
    public static void pdf2word(String source, String target) {
        try {
            //将pdf加载到对象中去
            PDDocument doc = PDDocument.load(new File(source));
            //得到pdf的页数
            int pagenumber = doc.getNumberOfPages();
            //设置转换后的名字
//            pdfFile = pdfFile.substring(0, pdfFile.lastIndexOf("."));
//            String fileName = pdfFile + ".doc";
            File file = new File(target);
            if (!file.exists()) {
                file.createNewFile();
            }
            FileOutputStream fos = new FileOutputStream(target);
            //设置输出字符集为UTF-8 因此该word应该使用UTF-8格式打开 如果你出现乱码那么你可以自己修改一下这里的格式
            Writer writer = new OutputStreamWriter(fos, "UTF-8");
            PDFTextStripper stripper = new PDFTextStripper();
            stripper.setSortByPosition(true);// 排序
            stripper.setStartPage(1);// 设置转换的开始页
            stripper.setEndPage(pagenumber);// 设置转换的结束页
            stripper.writeText(doc, writer);
            writer.close();
            doc.close();
            System.out.println("pdf转换word成功！");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * \
     *
     * @param sourceUrl 要转换的word文件
     * @param targetUrl 要转换成为的pdf文件
     */
    public static void word2PDF(String sourceUrl, String targetUrl) {
        int wdFormatPDF = 17;// PDF 格式
        ActiveXComponent app = null;
        Dispatch doc = null;
        try {
            app = new ActiveXComponent("Word.Application");
            app.setProperty("Visible", new Variant(false));
            Dispatch docs = app.getProperty("Documents").toDispatch();

            //转换前的文件路径
            String startFile = sourceUrl;
            //转换后的文件路劲
            String overFile = targetUrl;

            doc = Dispatch.call(docs, "Open", startFile).toDispatch();
            File tofile = new File(overFile);
            if (tofile.exists()) {
                tofile.delete();
            }
            Dispatch.call(doc, "SaveAs", overFile, wdFormatPDF);
        } catch (Exception e) {
            System.out.println(e.getMessage());
        } finally {
            Dispatch.call(doc, "Close", false);
            if (app != null)
                app.invoke("Quit", new Variant[]{});
        }
        //结束后关闭进程
        ComThread.Release();
    }

    /**
     * @param tempContentPathString pdf文件路径
     * @param tempSignPathList      jpg图片路径集合
     * @param i                     图片放在pdf第几页(下标从0开始)
     * @param pdfUrl                pdf输出位置
     */
    public static void PDFMergeJPGIndex(String tempContentPathString, List<String> tempSignPathList, int i, String pdfUrl) throws Exception {
        new File(pdfUrl.substring(0,pdfUrl.lastIndexOf(File.separator))).mkdirs();
        //"D:\\temp\\电力行标-报告.pdf"
        //打开PDF文件
        PDDocument doc = PDDocument.load(new FileInputStream(tempContentPathString));
        //获取所有页
        PDPageTree pages = doc.getDocumentCatalog().getPages();
        //获取第一页
        PDPage pdPage0 = pages.get(0);
        //获取第一页纸张信息
        PDRectangle mediaBox = pdPage0.getMediaBox();
        float lowerLeftX = mediaBox.getLowerLeftX();
        float lowerLeftY = mediaBox.getLowerLeftY();
        float width = mediaBox.getWidth();
        float height = mediaBox.getHeight();
        //有几张照片循环几次
        for (String s : tempSignPathList) {
            // 根据第一个纸张信息创建一张空白页
            PDPage pageOne = new PDPage(mediaBox);
            //获取照片
            PDImageXObject pdImage = PDImageXObject.createFromFile(s, doc);
            // 获取此页内容流
            PDPageContentStream contentStream = new PDPageContentStream(doc, pageOne, PDPageContentStream.AppendMode.APPEND, true);
            //照片写入
            contentStream.drawImage(pdImage, lowerLeftX, lowerLeftY, width, height);
            // 关闭内容流
            contentStream.close();
            pages.insertBefore(pageOne, pages.get(i + tempSignPathList.indexOf(s)));
        }
        doc.save(new File(pdfUrl));
        doc.close();
    }

    /**
     * pdf遮盖（不通用）
     * pdfUrl pdf路径
     * type 文档类型（区分不同文档类型，区分覆盖位置）
     */
    public static void PDFFill(String pdfUrl, int type) throws Exception {
        //初始化数据
        float x = 0; // 替换为您需要遮盖的区域的X轴坐标
        float y = 0; // 替换为您需要遮盖的区域的Y轴坐标
        float w = 0; // 替换为遮盖区域的宽度
        float h = 0; // 替换为遮盖区域的高度
        //打开PDF文件
        PDDocument doc = PDDocument.load(new FileInputStream(pdfUrl));
        PDPageTree pages = doc.getDocumentCatalog().getPages();
        PDPage pdPage0 = pages.get(0);
        PDRectangle mediaBox = pdPage0.getMediaBox();
        // 创建内容流，将第一页的内容绘制到新页上（如果需要删除/遮盖某个区域，请在此内容流中进行相应操作）
        PDPageContentStream contentStream = new PDPageContentStream(doc, pdPage0, PDPageContentStream.AppendMode.APPEND, true);
        // 设置填充颜色（这里设置为白色，您可以根据需要更改颜色）
        contentStream.setNonStrokingColor(255, 255, 255); // 白色
        if (0 == type) {
            //方案、测评表
            x = 0f; // 替换为您需要遮盖的区域的X轴坐标
            y = 200f; // 替换为您需要遮盖的区域的Y轴坐标
            w = 500f; // 替换为遮盖区域的宽度
            h = 300f; // 替换为遮盖区域的高度
        }else if(1 == type){
            //调查表
            x = 0f; // 替换为您需要遮盖的区域的X轴坐标
            y = 145f; // 替换为您需要遮盖的区域的Y轴坐标
            w = 600f; // 替换为遮盖区域的宽度
            h = 150f; // 替换为遮盖区域的高度
        }
        // 绘制矩形遮盖指定区域
        contentStream.addRect(x, y, w, h);
        contentStream.fill();
        // 结束内容流
        contentStream.close();
        // 保存修改后的文档
        doc.save(pdfUrl);
        // 关闭文档
        doc.close();
    }

    /**
     * 删除文件夹及以下的文件
     */
    public static void deleteFolder(File folder) {
        File[] files = folder.listFiles();
        if (files != null) {
            for (File file : files) {
                if (file.isDirectory()) {
                    deleteFolder(file);
                } else {
                    file.delete();
                }
            }
        }
        folder.delete();
    }

    /**
     * 文件返回
     */
    public static void downloadZip(HttpServletRequest request, HttpServletResponse response, String zipPath) {
        //截取文件名
        String fileName = zipPath.substring(zipPath.lastIndexOf(File.separator) + 1);
        try {
            response.setCharacterEncoding("utf-8");
            // 导出使用"application/octet-stream"更标准
            response.setContentType("application/octet-stream");
            response.setHeader("Content-Disposition",
                    "attachment;filename=" + FileUtils.setFileDownloadHeader(request, fileName));
            FileUtils.writeBytes(zipPath, response.getOutputStream());
        } catch (IOException e) {
            e.printStackTrace();
            throw new DbptException(StringUtils.format("导出文件 ({}) ！", fileName, e));
        }
    }

    /**
     * 打压缩包
     */
    public static void compress(String fromPath, String toPath) throws Exception {
        File fromFile = new File(fromPath);
        File toFile = new File(toPath);
        if (!fromFile.exists()) {
            throw new DbptException(fromPath + "不存在！");
        }
        try (FileOutputStream outputStream = new FileOutputStream(toFile); CheckedOutputStream checkedOutputStream = new CheckedOutputStream(outputStream, new CRC32()); ZipOutputStream zipOutputStream = new ZipOutputStream(checkedOutputStream)) {
            //区分重载方法，无实际意义
            String baseDir = "";
            //进行一次循环为了去除最外层文件夹
            File[] files = fromFile.listFiles();
            for (File file : files) {
                compress(file, zipOutputStream, baseDir);
            }
        }
    }

    private static void compress(File file, ZipOutputStream zipOut, String baseDir) throws IOException {
        if (file.isDirectory()) {
            compressDirectory(file, zipOut, baseDir);
        } else {
            compressFile(file, zipOut, baseDir);
        }
    }

    private static void compressFile(File file, ZipOutputStream zipOut, String baseDir) throws IOException {
        if (!file.exists()) {
            return;
        }
        try (BufferedInputStream bis = new BufferedInputStream(new FileInputStream(file))) {
            ZipEntry entry = new ZipEntry(baseDir + file.getName());
            zipOut.putNextEntry(entry);
            int count;
            byte[] data = new byte[1024];
            while ((count = bis.read(data, 0, 1024)) != -1) {
                zipOut.write(data, 0, count);
            }
        }
    }

    private static void compressDirectory(File dir, ZipOutputStream zipOut, String baseDir) throws IOException {
        File[] files = dir.listFiles();
        if (files != null && ArrayUtils.isNotEmpty(files)) {
            for (File file : files) {
                compress(file, zipOut, baseDir + dir.getName() + File.separator);
            }
        }
    }
}
相关推荐
考虑考虑4 分钟前
点阵图更改背景文字
java·后端·java ee
ZHE|张恒12 分钟前
Spring Boot 3 + Flyway 全流程教程
java·spring boot·后端
TDengine （老段）36 分钟前
TDengine 数学函数 CRC32 用户手册
java·大数据·数据库·sql·时序数据库·tdengine·1024程序员节
心随雨下1 小时前
Tomcat日志配置与优化指南
java·服务器·tomcat
Kapaseker1 小时前
Java 25 中值得关注的新特性
java
wljt1 小时前
Linux 常用命令速查手册（Java开发版）
java·linux·python
撩得Android一次心动1 小时前
Android 四大组件——BroadcastReceiver(广播)
android·java·android 四大组件
canonical_entropy1 小时前
Nop平台到底有什么独特之处，它能用在什么场景？
java·后端·领域驱动设计
chilavert3181 小时前
技术演进中的开发沉思-174 java-EJB：分布式通信
java·分布式
不是株2 小时前
JavaWeb(后端进阶)
java·开发语言·后端
热门推荐
01GitHub 镜像站点02综合整理：pdf预览显示：你尝试预览的文件可能对你的计算机有害。如果你信任此文件以及其来源，请打开此文件以看其内容，如何解决以正常预览文件03UV安装并设置国内源04Linux下V2Ray安装配置指南05npm使用国内淘宝镜像的方法06安娜的档案(Anna’s Archive) 镜像网站/国内最新可访问入口（持续更新）07BongoCat - 跨平台键盘猫动画工具08《大数据技术原理与应用》实验报告三 熟悉HBase常用操作09NVIDIA显卡驱动、CUDA、cuDNN 和 TensorRT 版本匹配指南10GitLab 零基础入门指南：从安装到项目管理全流程