/**
* Copyright 2002-2010 the original author is huanghe.
*/
package com.ucap.web.cm.webapp.util;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.apache.tools.zip.ZipEntry;
import org.apache.tools.zip.ZipFile;
import org.apache.tools.zip.ZipOutputStream;
import com.ucap.template.Constants;
import com.ucap.utils.UUIDGenerator;
import com.ucap.utils.formatString.FormatString;
import com.ucap.utils.formatString.Validator;
/**
* 压缩和解压缩工具类
*/
@SuppressWarnings("unchecked")
public class ZipUtil {
private static int bufSize = 4096;
private static byte[] buf = new byte[bufSize];
private static String OS_TYPE;
static {
if (System.getProperty("os.name").equals("Linux")) {
OS_TYPE = "linux";
} else if (System.getProperty("os.name").indexOf("Windows") != -1) {
OS_TYPE = "windows";
}
}
public ZipUtil() {
}
/**
* 压缩文件夹内的文件
*
* @param zipDirectory
* 需要压缩的文件夹名
* @return File 压缩文件对象
*/
public static File doZip(String zipDirectory) {
ZipOutputStream zipOut;
File zipDir = new File(zipDirectory);
String zipFileName = zipDir.getName() + ".zip";// 压缩后生成的zip文件名
if (System.getProperty("os.name").startsWith("Windows")) {
if (!zipDirectory.endsWith("\\"))
zipDirectory = zipDirectory + "\\";
} else {
if (!zipDirectory.endsWith("/"))
zipDirectory = zipDirectory + "/";
}
//判断压缩文件是否已经存在，如果存在则删除
File preZip = new File(zipDirectory + "/" + zipFileName);
if (preZip.exists()) {
try {
FileUtils.forceDelete(preZip);
} catch (IOException e) {
e.printStackTrace();
}
}
//创建临时目录
File tempFolder = createTempFolder();
String tempPath = tempFolder.getAbsolutePath();
File zipFile = new File(tempPath + "/" + zipFileName);
if (!zipFile.getParentFile().exists())
zipFile.getParentFile().mkdirs();
if (zipFile.exists() && zipFile.canWrite())
zipFile.delete();// 如果文件存在就删除原来的文件
try {
zipOut = new ZipOutputStream(new BufferedOutputStream(new FileOutputStream(zipFile)));
handleDir(zipOut, zipDir, "");
zipOut.close();
FileUtils.copyFileToDirectory(zipFile, zipDir);
} catch (IOException ioe) {
ioe.printStackTrace();
} finally {
//删除临时文件夹
if (tempFolder.exists()) {
try {
FileUtils.deleteDirectory(tempFolder);
} catch (IOException e) {
e.printStackTrace();
}
}
}
File zip = new File(zipDir + "/" + zipFileName);
return zip;
}
/**
* 由doZip调用,递归完成目录文件读取
*
*/
private static void handleDir(ZipOutputStream out, File f, String base) throws IOException {
if (f.isDirectory()) {
File[] fl = f.listFiles();
if (System.getProperty("os.name").startsWith("Windows")) {
base = base.length() == 0 ? "" : base + "\\";
//out.putNextEntry(new org.apache.tools.zip.ZipEntry(base));
} else {
base = base.length() == 0 ? "" : base + "/";
//out.putNextEntry(new org.apache.tools.zip.ZipEntry(base));
}
for (int i = 0; i < fl.length; i++) {
handleDir(out, fl[i], base + fl[i].getName());
}
} else {
out.putNextEntry(new org.apache.tools.zip.ZipEntry(base));
FileInputStream in = new FileInputStream(f);
byte b[] = new byte[512];
int len = 0;
while ((len = in.read(b)) != -1) {
out.write(b, 0, len);
}
out.closeEntry();
in.close();
}
}
/**
* 解压指定zip文件
*
* @param unZipfileName
* 需要解压的zip文件
* @param destPath
* 目录文件夹，如果目标文件夹为null ，则解压到当前目录下
* @param isDeleteSrc
* 是否删除原压缩文件
* @throws Exception
*/
public static List<String> unZip(File zipfileName, String destPath, boolean isDeleteSrc)
throws Exception {
List<String> ret = new ArrayList<String>();
if (zipfileName == null)
return ret;
if (destPath == null)
destPath = zipfileName.getAbsolutePath().substring(0,
zipfileName.getAbsolutePath().lastIndexOf("\\"))
+ "\\";
FileOutputStream fileOut;
File file;
InputStream inputStream;
ZipFile zipFile;
int readedBytes;
File tempFolder = createTempFolder();
String tempPath = tempFolder.getAbsolutePath();
try {
if (System.getProperty("os.name").equals("Linux"))
zipFile = new org.apache.tools.zip.ZipFile(zipfileName,"GBK");
else
zipFile = new org.apache.tools.zip.ZipFile(zipfileName);
for (Enumeration entries = zipFile.getEntries(); entries.hasMoreElements();) {
ZipEntry entry = (ZipEntry) entries.nextElement();
if (System.getProperty("os.name").equals("Linux"))
entry.setUnixMode(644);//解决linux乱码
file = new File(tempPath + "/" + entry.getName());
if (entry.isDirectory()) {
if (!file.exists())
FileUtils.forceMkdir(file);
} else {
// 如果指定文件的目录不存在,则创建之.
File parent = file.getParentFile();
if (!parent.exists()) {
FileUtils.forceMkdir(parent);
}
ret.add(entry.getName());
inputStream = zipFile.getInputStream(entry);
if (isRequiredSuffix(file.getAbsolutePath(), Constants.REQUIRED_ENCODE_SUFFIXS)) {
String content = IOUtils.toString(inputStream, "UTF-8");
FileUtils.writeStringToFile(file, content, "UTF-8");
} else {
fileOut = new FileOutputStream(file);
while ((readedBytes = inputStream.read(buf)) > 0) {
fileOut.write(buf, 0, readedBytes);
}
fileOut.close();
inputStream.close();
}
}
}
zipFile.close();
File destFolder = new File(destPath);
if (!destFolder.exists()) {
destFolder.mkdir();
}
FileUtils.copyDirectory(tempFolder, destFolder);
} catch (IOException ioe) {
ioe.printStackTrace();
} finally {
//删除临时文件夹
if (tempFolder.exists()) {
try {
FileUtils.deleteDirectory(tempFolder);
} catch (IOException e) {
e.printStackTrace();
}
}
//删除上传的压缩文件
if (isDeleteSrc)
zipfileName.delete();
}
return ret;
}
private static File createTempFolder() {
File tempFolder = null;
String tempPath = "";
try{
String tempFileName = UUIDGenerator.generate();
if (OS_TYPE.equals("window"))
tempPath = "C:/" + tempFileName;
else
tempPath = "/tmp/" + tempFileName;
tempFolder = new File(tempPath);
if (!tempFolder.exists()) {
tempFolder.mkdir();
}
}catch (Exception e) {
System.out.println("CreateTempFolder："+tempPath +" Exception:" + e.getMessage());
}
return tempFolder;
}
// 设置缓冲区大小
public void setBufSize(int bufSize) {
this.bufSize = bufSize;
}
// 测试AntZip类
public static void main(String[] args) throws Exception {
ZipUtil m_zip = new ZipUtil();
String filepath = "C:\\template\\template_upload/site/";
try {
m_zip.doZip(filepath);
} catch (Exception ex) {
ex.printStackTrace();
}
}
/**
* 判断文件的后缀名是否包含在是否以suffixs中
* @param fileName
* @param suffixs
* @return
*/
public static boolean isRequiredSuffix(String fileName, String... suffixs) {
if (Validator.isEmpty(fileName)) {
return false;
}
if (suffixs == null || suffixs.length < 1) {
return false;
}
for (String str : suffixs) {
if (fileName.indexOf("." + str) == fileName.length() - ("." + str).length()) {
return true;
}
}
return false;
}
/**
* 判断解压的文件是否包含汉字。
*
* @param zipfileName 要解压的文件
* @return 返回判断结果，true 含有 ;false 不含有
*/
public static boolean isHaveChinese(File zipfileName) {
ZipFile zipFile = null;
try {
zipFile = new ZipFile(zipfileName);
ZipEntry zipEntry = null;
Enumeration e = zipFile.getEntries();
while (e.hasMoreElements()) {
zipEntry = (ZipEntry) e.nextElement();
if (FormatString.IsHaveChinese(zipEntry.getName())) {
return true;
}
}
return false;
} catch (IOException e1) {
e1.printStackTrace();
} finally {
try {
zipFile.close();
} catch (IOException e) {
e.printStackTrace();
}
}
return false;
}
}