package com.vci.web.util.file;

import com.vci.properties.VciFileTransProperties;
import com.vci.starter.web.exception.VciBaseException;
import com.vci.starter.web.util.LangBaseUtil;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;

import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.tools.zip.ZipEntry;
import org.apache.tools.zip.ZipOutputStream;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

/**
 * @author ludc
 * @date 2024/7/16 14:27
 */
@Component
public class VciZipUtil {
    @Autowired(
            required = false
    )
    private VciFileTransProperties transProperties;

    public VciZipUtil() {
    }

    private String getZipCharset() {
        return this.transProperties != null && !StringUtils.isBlank(this.transProperties.getZipCharset()) ? this.transProperties.getZipCharset() : "GBK";
    }

    public void folderToZipFile(String folder, String zipFileName) throws VciBaseException {
        this.folderToZipFile(folder, zipFileName, this.getZipCharset());
    }

    public void folderToZipFile(String folder, String zipFileName, String zipEncode) throws VciBaseException {
        File sourceFile = new File(folder);
        if (!sourceFile.exists()) {
            throw new VciBaseException("fileNotFound", new String[]{folder});
        } else {
            ZipOutputStream zos = null;

            try {
                zos = new ZipOutputStream(new FileOutputStream(zipFileName));
                zos.setEncoding(zipEncode);
                this.addFileToZip(zos, (new File(folder)).listFiles(), folder + File.separator);
            } catch (Throwable var14) {
                throw new VciBaseException(LangBaseUtil.getErrorMsg(var14), new String[0]);
            } finally {
                try {
                    if (zos != null) {
                        zos.close();
                    }
                } catch (Exception var13) {
                }

            }

        }
    }

    private void addFileToZip(ZipOutputStream zos, File[] files, String rootPath) throws VciBaseException {
        if (files != null && files.length > 0) {
            File[] var4 = files;
            int var5 = files.length;

            for(int var6 = 0; var6 < var5; ++var6) {
                File file = var4[var6];
                if (StringUtils.isBlank(rootPath)) {
                    try {
                        rootPath = file.getParent();
                        if (rootPath == null) {
                            rootPath = file.getPath();
                        }
                    } catch (Throwable var21) {
                        rootPath = file.getPath();
                    }
                }

                if (!file.exists()) {
                    throw new VciBaseException("fileNotFound", new String[]{file.getAbsolutePath()});
                }

                if (file.isDirectory()) {
                    this.addFileToZip(zos, file.listFiles(), rootPath);
                } else {
                    FileInputStream is = null;

                    try {
                        is = new FileInputStream(file);
                        String entryName = file.getAbsolutePath().replace(rootPath, "");
                        if (entryName.startsWith("\\")) {
                            entryName = entryName.substring(1);
                        }

                        ZipEntry zipEntry = new ZipEntry(entryName);
                        zos.putNextEntry(zipEntry);
                        byte[] b = new byte[100];
                        boolean var12 = false;

                        int length;
                        while((length = is.read(b)) != -1) {
                            zos.write(b, 0, length);
                        }
                    } catch (Exception var22) {
                        throw new VciBaseException("zipFileError", new String[]{file.getAbsolutePath()}, var22);
                    } finally {
                        try {
                            IOUtils.closeQuietly(is);
                        } catch (Exception var20) {
                        }

                    }
                }
            }
        }

    }

    public void addFileToZip(File file, String zipFile) throws VciBaseException {
        if (file != null && file.exists()) {
            if (StringUtils.isBlank(zipFile)) {
                throw new VciBaseException("zipFileNameNull", new String[0]);
            } else {
                this.addFileToZip(new File[]{file}, zipFile);
            }
        } else {
            throw new VciBaseException("fileNotFound", new String[]{file.getAbsolutePath()});
        }
    }

    public void addFileToZip(File[] files, String zipFile) throws VciBaseException {
        this.addFileToZip(files, zipFile, this.getZipCharset());
    }

    public void addFileToZip(File[] files, String zipFile, String zipEncode) throws VciBaseException {
        if (files != null && files.length != 0) {
            if (StringUtils.isBlank(zipFile)) {
                throw new VciBaseException("zipFileNameNull", new String[0]);
            } else {
                ZipOutputStream zos = null;

                try {
                    zos = new ZipOutputStream(new FileOutputStream(zipFile));
                    zos.setEncoding(zipEncode);
                    this.addFileToZip(zos, files, "");
                } catch (Throwable var13) {
                    throw new VciBaseException(LangBaseUtil.getErrorMsg(var13), new String[0], var13);
                } finally {
                    try {
                        if (zos != null) {
                            zos.close();
                        }
                    } catch (Exception var12) {
                    }

                }

            }
        } else {
            throw new VciBaseException("fileNull", new String[0]);
        }
    }
}