java-insagent-pc/src/main/java/com/ydtech/utils/baidu/FileUtil.java

package com.ydtech.utils.baidu;

import com.ydtech.modules.ins.utils.CompressUtil;
import com.ydtech.utils.StringUtils;
import lombok.extern.slf4j.Slf4j;

import java.io.*;
import java.net.MalformedURLException;
import java.net.URI;
import java.net.URLEncoder;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * 文件读取工具类
 */
@Slf4j
public class FileUtil {

    /**
     * 读取文件内容，作为字符串返回
     */
    public static String readFileAsString(String filePath) throws IOException {
        File file = new File(filePath);
        if (!file.exists()) {
            throw new FileNotFoundException(filePath);
        }

        if (file.length() > 1024 * 1024 * 1024) {
            throw new IOException("File is too large");
        }

        StringBuilder sb = new StringBuilder((int) (file.length()));
        // 创建字节输入流
        FileInputStream fis = new FileInputStream(filePath);
        // 创建一个长度为10240的Buffer
        byte[] bbuf = new byte[10240];
        // 用于保存实际读取的字节数
        int hasRead = 0;
        while ((hasRead = fis.read(bbuf)) > 0) {
            sb.append(new String(bbuf, 0, hasRead));
        }
        fis.close();
        return sb.toString();
    }

    /**
     * 根据文件路径读取byte[] 数组
     */
    public static byte[] readFileByBytes(String filePath) throws IOException {
        File file = new File(filePath);
        if (!file.exists()) {
            throw new FileNotFoundException(filePath);
        } else {
            ByteArrayOutputStream bos = new ByteArrayOutputStream((int) file.length());
            BufferedInputStream in = null;

            try {
                in = new BufferedInputStream(new FileInputStream(file));
                short bufSize = 1024;
                byte[] buffer = new byte[bufSize];
                int len1;
                while (-1 != (len1 = in.read(buffer, 0, bufSize))) {
                    bos.write(buffer, 0, len1);
                }

                byte[] var7 = bos.toByteArray();
                return var7;
            } finally {
                try {
                    if (in != null) {
                        in.close();
                    }
                } catch (IOException var14) {
                    var14.printStackTrace();
                }

                bos.close();
            }
        }
    }


    /**
     * 网络url转文件
     *
     * @param url     保司返回的保单地址
     * @param dir     目录 杠结束  carInsPolicy/
     * @param path    文件真实目录 D:/upload/
     * @param showUrl 展示地址 http://sxzgkj.baoxianzhanggui.com/web-api/晋A12345/晋A12345-交强险保单.pdf
     * @author: lig
     * @date: 2023年09月21日 0021
     */
    public static String netUrlToFile(String url, String dir, String fileName, String path, String showUrl) {
//        String s = "http://eserviceinterface.ciitc.com.cn/ePolicyServices/insurancePolicy.do?downFiles=key&amp&pkid=a7370fea493945868b8ca847d3ef9d9e";
//        String path = "D:/upload/";
//        String dir = "carInsPolicy/";
        if (StringUtils.isEmpty(url)) return "";
        String showPath = String.format("%s%s%s", showUrl, dir, fileName);
        String uploadPath = path + dir;

        try {

            //存在
            File existFile = new File(uploadPath + fileName);
            if (existFile.exists()) {
                log.error("存在的文件：" + existFile.getPath());
                log.error("存在的文件的展示地址：" + showPath);
                return showPath;
            }

            File folder = new File(uploadPath);
            if (!folder.isDirectory()) {
                folder.mkdirs();
            }

            String encodeUrl = urlEncodeChinese(url);
            URI u = URI.create(encodeUrl);
            InputStream inputStream = u.toURL().openStream();
            File file = new File(uploadPath + fileName);
            copyInputStreamToFile(inputStream, file);

            log.error("文件路径：" + file.getPath());
            return showPath;

        } catch (MalformedURLException e) {
            throw new RuntimeException(e);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }

    }


    /**
     * 文件存在
     *
     * @param dir     目录 杠结束  carInsPolicy/
     * @param path    文件真实目录 D:/upload/
     * @param showUrl 展示地址 http://sxzgkj.baoxianzhanggui.com/web-api/晋A12345/晋A12345-交强险保单.pdf
     * @author: lig
     * @date: 2023年09月21日 0021
     */
    public static String fileExist(String dir, String fileName, String path, String showUrl) {
        String showPath = String.format("%s%s%s", showUrl, dir, fileName);
        String uploadPath = path + dir;


        //存在
        File existFile = new File(uploadPath + fileName);
        if (existFile.exists()) {
            log.error("存在的文件：" + existFile.getPath());
            log.error("存在的文件的展示地址：" + showPath);
            return showPath;
        }
        return "";


    }


    // InputStream -> File
    private static void copyInputStreamToFile(InputStream inputStream, File file)
            throws IOException {

        try (FileOutputStream outputStream = new FileOutputStream(file)) {

            int read;
            byte[] bytes = new byte[1024];

            while ((read = inputStream.read(bytes)) != -1) {
                outputStream.write(bytes, 0, read);
            }

            // commons-io
            //IOUtils.copy(inputStream, outputStream);

        }

    }

    public static String netUrlStoreToLocal(String url, String dir, String fileName, String path, String showUrl, String password,
                                            String licensePlate) {
        if (StringUtils.isEmpty(url)) return "";
        String showPath = String.format("%s%s%s", showUrl, dir, fileName);
        String uploadPath = path + dir;

        try {

            //存在
            File existFile = new File(uploadPath + fileName);
            if (existFile.exists()) {
                log.error("存在的文件：" + existFile.getPath());
                log.error("存在的文件的展示地址：" + showPath);
                return showPath;
            }

            File folder = new File(uploadPath);
            if (!folder.isDirectory()) {
                folder.mkdirs();
            }

            URI u = URI.create(url);
            InputStream inputStream = u.toURL().openStream();
            File file = new File(uploadPath + licensePlate + ".zip");
            copyInputStreamToFile(inputStream, file);

            CompressUtil.unZip(file.getPath(), uploadPath, password, fileName, licensePlate);
            file.delete();
            return showPath;

        } catch (MalformedURLException e) {
            throw new RuntimeException(e);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }

    }

    /**
     * 将 url 中的 中文转码
     *
     * @param url url链接
     * @return 转码后的url
     */
    public static String urlEncodeChinese(String url) {
        try {
            Matcher matcher = Pattern.compile("[\\u4e00-\\u9fa5]").matcher(url);
            String tmp;
            while (matcher.find()) {
                tmp = matcher.group();
                url = url.replaceAll(tmp, URLEncoder.encode(tmp, "UTF-8"));
            }
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        }
        return url;
    }


}