package com.vci.ubcs.file.util;


import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.vci.ubcs.file.constant.VciFileLangCodeConstant;
import com.vci.ubcs.file.properties.VciFileTransProperties;
import com.vci.ubcs.starter.exception.VciBaseException;
import com.vci.ubcs.starter.web.util.LangBaseUtil;
import org.apache.commons.io.IOUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import org.apache.tools.zip.ZipEntry;
import org.apache.tools.zip.ZipOutputStream;

/**
 * zip相关的工具类
 * @author weidy
 * @date 2020/3/12
 */
@Component
public class VciZipUtil {

	/**
	 * 文件传输配置
	 */
	@Autowired(required = false)
	private VciFileTransProperties transProperties;

	/**
	 * 获取zip的字符集
	 * @return 默认为GBK，否则需要配置
	 */
	private String getZipCharset(){
		return (transProperties == null || StringUtils.isBlank(transProperties.getZipCharset()))?"GBK":transProperties.getZipCharset();
	}

	/**
	 * 将文件夹压缩成压缩包
	 * @param folder 文件夹路径
	 * @param zipFileName zip文件名
	 * @throws VciBaseException 参数为空或者添加出错的会抛出异常
	 */
	public void folderToZipFile(String folder, String zipFileName) throws VciBaseException {
		folderToZipFile(folder,zipFileName,getZipCharset());
	}

	/**
	 * 将文件夹压缩成压缩包
	 * @param folder 文件夹路径
	 * @param zipFileName zip文件名
	 * @param zipEncode zip的字符集，windows应该为GBK
	 * @throws VciBaseException 参数为空或者添加出错的会抛出异常
	 */
	public void folderToZipFile(String folder, String zipFileName,String zipEncode) throws VciBaseException{
		File sourceFile = new File(folder);
		if(!sourceFile.exists()){
			throw new VciBaseException(VciFileLangCodeConstant.FILE_NOT_FOUND,new String[]{folder});
		}
		ZipOutputStream zos = null;
		try {
			zos = new ZipOutputStream(new FileOutputStream(zipFileName));
			zos.setEncoding(zipEncode);
			//默认是UTF-8格式的
			addFileToZip(zos, new File(folder).listFiles(),folder + File.separator);
		}catch (Throwable e){
			throw new VciBaseException(LangBaseUtil.getErrorMsg(e),new String[]{});
		}finally {
			try{
				if(zos!=null){
					zos.close();
				}
			}catch (Exception e){

			}
		}
	}

	/**
	 * 将文件夹压缩成压缩包
	 * @param zos zip输出流
	 * @param files 要压缩的文件
	 * @param rootPath 压缩的文件中的顶层目录
	 * @throws VciBaseException 文件不存在，或者内容出现错误会抛出异常
	 */
	private void addFileToZip(ZipOutputStream zos,File[] files,String rootPath) throws VciBaseException{
		if(files!=null && files.length>0) {
			for (File file : files) {
				if(StringUtils.isBlank(rootPath)){
					try {
						rootPath = file.getParent();
						if (rootPath == null) {
							rootPath = file.getPath();
						}
					}catch (Throwable e){
						rootPath = file.getPath();
					}
				}
				if (!file.exists()) {
					throw new VciBaseException(VciFileLangCodeConstant.FILE_NOT_FOUND,new String[]{file.getAbsolutePath()});
				}
				if (file.isDirectory()) {
					addFileToZip(zos, file.listFiles(),rootPath);
				}else{
					InputStream 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];
						int length = 0;
						while ((length = is.read(b)) != -1) {
							zos.write(b, 0, length);
						}
					} catch (Exception e) {
						throw new VciBaseException(VciFileLangCodeConstant.ZIP_FILE_ERROR,new String[]{file.getAbsolutePath()},e);
					} finally {
						try {
							IOUtils.closeQuietly(is);
						} catch (Exception e) {

						}
					}
				}
			}
		}
	}

	/**
	 * 将单个文件添加到zip中
	 * @param file 文件
	 * @param zipFile zip文件
	 * @throws VciBaseException 参数为空或者添加出错的会抛出异常
	 */
	public void addFileToZip(File file, String zipFile) throws VciBaseException{
		if(file == null || !file.exists() ){
			throw new VciBaseException(VciFileLangCodeConstant.FILE_NOT_FOUND ,new String[]{file.getAbsolutePath()});
		}
		if(StringUtils.isBlank(zipFile)){
			throw new VciBaseException(VciFileLangCodeConstant.ZIP_FILE_NAME_NULL,new String[0]);
		}
		addFileToZip(new File[]{file},zipFile);
	}


	/**
	 * 将多个个文件添加到zip中
	 * @param files 文件
	 * @param zipFile zip文件
	 * @throws VciBaseException 参数为空或者添加出错的会抛出异常
	 */
	public void addFileToZip(File[] files, String zipFile) throws VciBaseException{
		addFileToZip(files, zipFile,getZipCharset());
	}

	/**
	 * 将多个个文件添加到zip中
	 * @param files 文件
	 * @param zipFile zip文件
	 * @param zipEncode zip的字符集
	 * @throws VciBaseException 参数为空或者添加出错的会抛出异常
	 */
	public void addFileToZip(File[] files, String zipFile,String zipEncode) throws VciBaseException{
		if(files == null || files.length == 0){
			throw new VciBaseException(VciFileLangCodeConstant.FILES_NULL,new String[0]);
		}
		if(StringUtils.isBlank(zipFile)){
			throw new VciBaseException(VciFileLangCodeConstant.ZIP_FILE_NAME_NULL,new String[0]);
		}
		ZipOutputStream zos = null;
		try {
			zos = new ZipOutputStream(new FileOutputStream(zipFile));
			zos.setEncoding(zipEncode);
			addFileToZip(zos, files,"");
		}catch (Throwable e){
			throw new VciBaseException(LangBaseUtil.getErrorMsg(e),new String[]{},e);
		}finally {
			try{
				if(zos!=null){
					zos.close();
				}
			}catch (Exception e){

			}
		}
	}
}