313 lines
11 KiB
C#
313 lines
11 KiB
C#
using ICSharpCode.SharpZipLib.Zip;
|
||
using System;
|
||
using System.Collections.Generic;
|
||
using System.IO;
|
||
using System.Linq;
|
||
using System.Text;
|
||
using System.Threading.Tasks;
|
||
|
||
namespace Api.Framework.Tools
|
||
{
|
||
public class Zip
|
||
{
|
||
|
||
/// <summary>
|
||
/// 根据给的文件参数，自动进行压缩或解压缩操作
|
||
/// </summary>
|
||
public static void Process(String[] files, String Password = null)
|
||
{
|
||
if (files.Length > 0)
|
||
{
|
||
if (files.Length == 1 && (files[0].ToLower().EndsWith(".zip") || files[0].ToLower().EndsWith(".rar")))
|
||
{
|
||
unzip(files[0], null, Password, null); // 解压缩
|
||
}
|
||
else
|
||
{
|
||
String zipPath = Tools.getPathNoExt(files[0]) + ".zip"; // 以待压缩的第一个文件命名生成的压缩文件
|
||
String BaseDir = Tools.getParent(files[0]); // 获取第一个文件的父路径信息
|
||
if (files.Length == 1) // 若载入的为单个目录，则已当前目录作为基础路径
|
||
{
|
||
String file = files[0];
|
||
if (Directory.Exists(file)) BaseDir = file + "\\";
|
||
}
|
||
|
||
String[] subFiles = Tools.getSubFiles(files); // 获取args对应的所有目录下的文件列表
|
||
zip(zipPath, BaseDir, subFiles, Password, null); // 对载入的文件进行压缩操作
|
||
}
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 压缩所有文件files，为压缩文件zipFile, 以相对于BaseDir的路径构建压缩文件子目录，ignoreNames指定要忽略的文件或目录
|
||
/// </summary>
|
||
public static bool zip(String zipPath, String BaseDir, String[] files, String Password = null, String[] ignoreNames = null)
|
||
{
|
||
if (files == null || files.Length == 0) return false;
|
||
if (zipPath == null || zipPath.Equals("")) zipPath = Tools.getPathNoExt(files[0]) + ".zip"; // 默认以第一个文件命名压缩文件
|
||
if (BaseDir == null || BaseDir.Equals("")) BaseDir = Tools.getParent(files[0]); // 默认以第一个文件的父目录作为基础路径
|
||
//Console.WriteLine("所有待压缩文件根目录：" + BaseDir);
|
||
|
||
try
|
||
{
|
||
Tools.mkdirs(Tools.getParent(zipPath)); // 创建目标路径
|
||
//Console.WriteLine("创建压缩文件：" + zipPath);
|
||
|
||
FileStream input = null;
|
||
ZipOutputStream zipStream = new ZipOutputStream(File.Create(zipPath));
|
||
if (Password != null && !Password.Equals("")) zipStream.Password = Password;
|
||
|
||
files = Tools.getSubFiles(files); // 获取子目录下所有文件信息
|
||
for (int i = 0; i < files.Length; i++)
|
||
{
|
||
if (ContainsIgnoreName(files[i], ignoreNames)) continue; // 跳过忽略的文件或目录
|
||
|
||
String entryName = Tools.relativePath(BaseDir, files[i]);
|
||
zipStream.PutNextEntry(new ZipEntry(entryName));
|
||
//Console.WriteLine("添加压缩文件：" + entryName);
|
||
|
||
if (File.Exists(files[i])) // 读取文件内容
|
||
{
|
||
input = File.OpenRead(files[i]);
|
||
Random rand = new Random();
|
||
byte[] buffer = new byte[10240];
|
||
int read = 0;
|
||
while ((read = input.Read(buffer, 0, 10240)) > 0)
|
||
{
|
||
zipStream.Write(buffer, 0, read);
|
||
}
|
||
input.Close();
|
||
}
|
||
}
|
||
zipStream.Close();
|
||
//Console.WriteLine("文件压缩完成！");
|
||
|
||
return true;
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
//Console.WriteLine(ex.ToString());
|
||
}
|
||
|
||
return false;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 解压文件 到指定的路径，可通过targeFileNames指定解压特定的文件
|
||
/// </summary>
|
||
public static bool unzip(String zipPath, String targetPath = null, String Password = null, String[] targeFileNames = null)
|
||
{
|
||
if (File.Exists(zipPath))
|
||
{
|
||
if (targetPath == null || targetPath.Equals("")) targetPath = Tools.getPathNoExt(zipPath);
|
||
//Console.WriteLine("解压文件：" + zipPath);
|
||
//Console.WriteLine("解压至目录：" + targetPath);
|
||
|
||
try
|
||
{
|
||
ZipInputStream zipStream = null;
|
||
FileStream bos = null;
|
||
|
||
zipStream = new ZipInputStream(File.OpenRead(zipPath));
|
||
if (Password != null && !Password.Equals("")) zipStream.Password = Password;
|
||
|
||
ZipEntry entry = null;
|
||
while ((entry = zipStream.GetNextEntry()) != null)
|
||
{
|
||
if (targeFileNames != null && targeFileNames.Length > 0) // 若指定了目标解压文件
|
||
{
|
||
if (!ContainsIgnoreName(entry.Name, targeFileNames)) continue; // 跳过非指定的文件
|
||
}
|
||
|
||
String target = targetPath + "\\" + entry.Name;
|
||
if (entry.IsDirectory) Tools.mkdirs(target); // 创建目标路径
|
||
if (entry.IsFile)
|
||
{
|
||
Tools.mkdirs(Tools.getParent(target));
|
||
|
||
bos = File.Create(target);
|
||
//Console.WriteLine("解压生成文件：" + target);
|
||
|
||
int read = 0;
|
||
byte[] buffer = new byte[10240];
|
||
while ((read = zipStream.Read(buffer, 0, 10240)) > 0)
|
||
{
|
||
bos.Write(buffer, 0, read);
|
||
}
|
||
bos.Flush();
|
||
bos.Close();
|
||
}
|
||
}
|
||
zipStream.CloseEntry();
|
||
|
||
//Console.WriteLine("解压完成！");
|
||
return true;
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
//Console.WriteLine(ex.ToString());
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 判断fileName中是否含有ignoreNames中的某一项
|
||
/// </summary>
|
||
private static bool ContainsIgnoreName(String fileName, String[] ignoreNames)
|
||
{
|
||
if (ignoreNames != null && ignoreNames.Length > 0)
|
||
{
|
||
foreach (string name in ignoreNames)
|
||
{
|
||
if (fileName.Contains(name)) return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
}
|
||
|
||
/// <summary>
|
||
/// 通用功能函数
|
||
/// </summary>
|
||
public class Tools
|
||
{
|
||
/// <summary>
|
||
/// 检测目录是否存在，若不存在则创建
|
||
/// </summary>
|
||
public static void mkdirs(string path)
|
||
{
|
||
if (!Directory.Exists(path))
|
||
{
|
||
Directory.CreateDirectory(path);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取去除拓展名的文件路径
|
||
/// </summary>
|
||
public static String getPathNoExt(String path)
|
||
{
|
||
if (File.Exists(path)) return Directory.GetParent(path).FullName + "\\" + Path.GetFileNameWithoutExtension(path);
|
||
else return Directory.GetParent(path).FullName + "\\" + Path.GetFileName(path);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取父目录的路径信息
|
||
/// </summary>
|
||
public static String getParent(String path)
|
||
{
|
||
return System.IO.Directory.GetParent(path).FullName + "\\";
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// 获取父目录的路径信息
|
||
/// </summary>
|
||
public static String getFileName(String path)
|
||
{
|
||
return System.IO.Path.GetFileName(path);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取filePath的相对于BaseDir的路径
|
||
/// </summary>
|
||
public static String relativePath(String BaseDir, String filePath)
|
||
{
|
||
String relativePath = "";
|
||
if (filePath.StartsWith(BaseDir)) relativePath = filePath.Substring(BaseDir.Length);
|
||
return relativePath;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取paths路径下所有文件信息
|
||
/// </summary>
|
||
public static String[] getSubFiles(String[] Paths)
|
||
{
|
||
List<String> list = new List<String>(); // paths路径下所有文件信息
|
||
|
||
foreach (String path in Paths)
|
||
{
|
||
List<String> subFiles = getSubFiles(path); // 获取路径path下所有文件列表信息
|
||
list = ListAdd(list, subFiles);
|
||
}
|
||
|
||
String[] A = List2Array(list); // 转化为数组形式
|
||
|
||
return A;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 合并list1和list2到新的list
|
||
/// </summary>
|
||
public static List<String> ListAdd(List<String> list1, List<String> list2)
|
||
{
|
||
List<String> list = new List<String>();
|
||
|
||
foreach (String path in list1) if (!list.Contains(path)) list.Add(path);
|
||
foreach (String path in list2) if (!list.Contains(path)) list.Add(path);
|
||
|
||
return list;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取file目录下所有文件列表
|
||
/// </summary>
|
||
public static List<String> getSubFiles(String file)
|
||
{
|
||
List<String> list = new List<String>();
|
||
|
||
if (File.Exists(file))
|
||
{
|
||
if (!list.Contains(file)) list.Add(file);
|
||
}
|
||
|
||
if (Directory.Exists(file))
|
||
{
|
||
// 获取目录下的文件信息
|
||
foreach (String iteam in Directory.GetFiles(file))
|
||
{
|
||
if (!list.Contains(iteam)) list.Add(iteam);
|
||
}
|
||
|
||
// 获取目录下的子目录信息
|
||
foreach (String iteam in Directory.GetDirectories(file))
|
||
{
|
||
List<String> L = getSubFiles(iteam); // 获取子目录下所有文件列表
|
||
foreach (String path in L)
|
||
{
|
||
if (!list.Contains(path)) list.Add(path);
|
||
}
|
||
}
|
||
|
||
// 记录当前目录
|
||
if (Directory.GetFiles(file).Length == 0 && Directory.GetDirectories(file).Length == 0)
|
||
{
|
||
if (!list.Contains(file)) list.Add(file + "\\");
|
||
}
|
||
}
|
||
|
||
return list;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 转化list为数组
|
||
/// </summary>
|
||
public static String[] List2Array(List<String> list)
|
||
{
|
||
int size = (list == null ? 0 : list.Count);
|
||
String[] A = new String[size];
|
||
|
||
int i = 0;
|
||
foreach (String S in list)
|
||
{
|
||
A[i++] = S;
|
||
}
|
||
|
||
return A;
|
||
}
|
||
|
||
}
|
||
}
|