ZhouXY108/hutool
1
0
Fork 0
mirror of https://gitee.com/chinabugotech/hutool.git synced 2025-05-09 23:51:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

!1017 添加zip替换工具类ZipReplacer

Browse Source 
Merge pull request !1017 from 魏凡/v6-dev
This commit is contained in:
Looly 2023-06-27 02:32:44 +00:00 committed by Gitee
commit 17cf97d2ba
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 276 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

276
hutool-core/src/main/java/org/dromara/hutool/core/compress/ZipReplacer.java Normal file
View File

@ -0,0 +1,276 @@
package org.dromara.hutool.core.compress;
import org.dromara.hutool.core.io.IORuntimeException;
import org.dromara.hutool.core.io.file.FileUtil;
import org.dromara.hutool.core.text.StrUtil;
import org.dromara.hutool.core.util.CharsetUtil;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.Charset;
import java.util.Arrays;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.List;
import java.util.regex.Matcher;
import java.util.stream.Collectors;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipOutputStream;
public class ZipReplacer {
/**
* 不解压zip，替换文件内容，也可用于转换zip编码格式
*
* @param srcFile 源文件
* @param tarFile 输出文件
* @param innerFiles 替换文件在zip中的相对路径
* @param replaceFiles 替换文件到系统中到存储路径
* @param charset 读取编码格式
* @param charsetOut 输出编码格式
*/
public static void replace(File srcFile, File tarFile, String[] innerFiles, List<File> replaceFiles, Charset charset, Charset charsetOut) {
// 记录zip中是否存在相同路径的文件， 是 更新文件 否，添加文件
boolean[] updates = new boolean[replaceFiles.size()];
try (ZipFile zipFile = new ZipFile(srcFile, charset);
FileOutputStream fos = new FileOutputStream(tarFile);
ZipOutputStream zos = new ZipOutputStream(fos);
ZipWriter zipWriter = new ZipWriter(zos, charsetOut)) {
HashMap<String, InputStream> data = new HashMap<>();
for (Enumeration<? extends ZipEntry> entries = zipFile.entries(); entries.hasMoreElements(); ) {
ZipEntry zipEntryIn = entries.nextElement();
String zipEntryInName = zipEntryIn.getName();
// false 未被替换
boolean update = false;
for (int i = 0; i < innerFiles.length; i++) {
update = samePath(zipEntryInName, innerFiles[i]);
// 存在同路径文件，替换
if (update) {
updates[i] = true;
data.put(zipEntryInName, FileUtil.getInputStream(replaceFiles.get(i)));
break;
}
}
// 不存在同路径文件，直接添加原zipEntry
if (!update) {
data.put(zipEntryInName, ZipUtil.getStream(zipFile, zipEntryIn));
}
}
// 确认replaceFiles是否替换原zip中文件，没有替换直接添加
for (int i = 0; i < updates.length; i++) {
if (!updates[i]) {
// 原zip中不存在同路径文件，添加到制定目录
data.put(innerFiles[i], FileUtil.getInputStream(replaceFiles.get(i)));
}
}
for (String key : data.keySet()) {
zipWriter.add(key, data.get(key));
}
} catch (IOException e) {
throw new IORuntimeException(e);
}
}
public static void replace(File srcFile, File tarFile, String[] innerFiles, File[] replaceFiles, Charset charset, Charset charsetOut) {
replace(srcFile, tarFile, innerFiles, Arrays.asList(replaceFiles), charset, charsetOut);
}
public static void replace(File srcFile, File tarFile, String[] innerFiles, String[] replaceFiles, Charset charset, Charset charsetOut) {
List<File> files = Arrays.stream(replaceFiles).map(FileUtil::file).collect(Collectors.toList());
replace(srcFile, tarFile, innerFiles, files, charset, charsetOut);
}
/**
* 不解压zip，替换文件内容，也可用于转换zip编码格式
*
* @param srcFile 源文件
* @param tarFile 输出文件
* @param innerFiles 替换文件在zip中的相对路径
* @param replaceFiles 替换文件到系统中到存储路径
* @param charset 读取输出编码格式
*/
public static void replace(File srcFile, File tarFile, String[] innerFiles, String[] replaceFiles, Charset charset) {
replace(srcFile, tarFile, innerFiles, replaceFiles, charset, charset);
}
public static void replace(File srcFile, File tarFile, String[] innerFiles, File[] replaceFiles, Charset charset) {
replace(srcFile, tarFile, innerFiles, replaceFiles, charset, charset);
}
public static void replace(File srcFile, File tarFile, String[] innerFiles, List<File> replaceFiles, Charset charset) {
replace(srcFile, tarFile, innerFiles, replaceFiles, charset, charset);
}
/**
* 不解压zip，替换文件内容，也可用于转换zip编码格式
*
* @param srcFile 源文件
* @param tarFile 输出文件
* @param innerFiles 要替换文件在zip中的相对路径
* @param replaceFiles 替换文件到系统中到存储路径
*/
public static void replace(File srcFile, File tarFile, String[] innerFiles, String[] replaceFiles) {
replace(srcFile, tarFile, innerFiles, replaceFiles, CharsetUtil.defaultCharset());
}
public static void replace(File srcFile, File tarFile, String[] innerFiles, File[] replaceFiles) {
replace(srcFile, tarFile, innerFiles, replaceFiles, CharsetUtil.defaultCharset());
}
public static void replace(File srcFile, File tarFile, String[] innerFiles, List<File> replaceFiles) {
replace(srcFile, tarFile, innerFiles, replaceFiles, CharsetUtil.defaultCharset());
}
/**
* 不解压zip，替换文件内容，也可用于转换zip编码格式
*
* @param srcFile 源文件，也是输出路径
* @param innerFiles 替换文件在zip中的相对路径
* @param replaceFiles 替换文件到系统中到存储路径
* @param charset 读取输出编码格式
*/
public static void replace(File srcFile, String[] innerFiles, String[] replaceFiles, Charset charset) {
replace(srcFile, srcFile, innerFiles, replaceFiles, charset);
}
public static void replace(File srcFile, String[] innerFiles, File[] replaceFiles, Charset charset) {
replace(srcFile, srcFile, innerFiles, replaceFiles, charset);
}
public static void replace(File srcFile, String[] innerFiles, List<File> replaceFiles, Charset charset) {
replace(srcFile, srcFile, innerFiles, replaceFiles, charset);
}
/**
* 不解压zip，替换文件内容，也可用于转换zip编码格式
*
* @param srcFile 源文件
* @param innerFiles 要替换文件在zip中的相对路径
* @param replaceFiles 要替换的文件绝对路径
*/
public static void replace(File srcFile, String[] innerFiles, String[] replaceFiles) {
replace(srcFile, srcFile, innerFiles, replaceFiles, CharsetUtil.defaultCharset());
}
public static void replace(File srcFile, String[] innerFiles, File[] replaceFiles) {
replace(srcFile, srcFile, innerFiles, replaceFiles, CharsetUtil.defaultCharset());
}
public static void replace(File srcFile, String[] innerFiles, List<File> replaceFiles) {
replace(srcFile, srcFile, innerFiles, replaceFiles, CharsetUtil.defaultCharset());
}
/**
* 不解压zip，替换文件内容，也可用于转换zip编码格式
*
* @param srcFilePath 源文件路径
* @param tarFilePath 输出文件路径
* @param innerFiles 要替换文件在zip中的相对路径
* @param replaceFiles 要替换的文件绝对路径
* @param charset 读取编码格式
* @param charsetOut 输出编码格式
*/
public static void replace(String srcFilePath, String tarFilePath, String[] innerFiles, String[] replaceFiles, Charset charset, Charset charsetOut) {
replace(FileUtil.file(srcFilePath), FileUtil.file(tarFilePath), innerFiles, replaceFiles, charset, charsetOut);
}
public static void replace(String srcFilePath, String tarFilePath, String[] innerFiles, File[] replaceFiles, Charset charset, Charset charsetOut) {
replace(FileUtil.file(srcFilePath), FileUtil.file(tarFilePath), innerFiles, replaceFiles, charset, charsetOut);
}
public static void replace(String srcFilePath, String tarFilePath, String[] innerFiles, List<File> replaceFiles, Charset charset, Charset charsetOut) {
replace(FileUtil.file(srcFilePath), FileUtil.file(tarFilePath), innerFiles, replaceFiles, charset, charsetOut);
}
/**
* 不解压zip，替换文件内容，也可用于转换zip编码格式
*
* @param srcFilePath zip源文件路径
* @param tarFilePath zip输出文件路径
* @param innerFiles 要替换文件在zip中的相对路径
* @param replaceFiles 要替换的文件绝对路径
* @param charset 读取输出编码格式
*/
public static void replace(String srcFilePath, String tarFilePath, String[] innerFiles, String[] replaceFiles, Charset charset) {
replace(FileUtil.file(srcFilePath), FileUtil.file(tarFilePath), innerFiles, replaceFiles, charset, charset);
}
public static void replace(String srcFilePath, String tarFilePath, String[] innerFiles, File[] replaceFiles, Charset charset) {
replace(FileUtil.file(srcFilePath), FileUtil.file(tarFilePath), innerFiles, replaceFiles, charset, charset);
}
public static void replace(String srcFilePath, String tarFilePath, String[] innerFiles, List<File> replaceFiles, Charset charset) {
replace(FileUtil.file(srcFilePath), FileUtil.file(tarFilePath), innerFiles, replaceFiles, charset, charset);
}
/**
* 不解压zip，替换文件内容，也可用于转换zip编码格式
*
* @param srcFilePath 源文件路径
* @param tarFilePath 输出文件路径
* @param innerFiles 替换文件在zip中的相对路径
* @param replaceFiles 替换文件到系统中到存储路径
*/
public static void replace(String srcFilePath, String tarFilePath, String[] innerFiles, String[] replaceFiles) {
replace(FileUtil.file(srcFilePath), FileUtil.file(tarFilePath), innerFiles, replaceFiles, CharsetUtil.defaultCharset());
}
public static void replace(String srcFilePath, String tarFilePath, String[] innerFiles, File[] replaceFiles) {
replace(FileUtil.file(srcFilePath), FileUtil.file(tarFilePath), innerFiles, replaceFiles, CharsetUtil.defaultCharset());
}
public static void replace(String srcFilePath, String tarFilePath, String[] innerFiles, List<File> replaceFiles) {
replace(FileUtil.file(srcFilePath), FileUtil.file(tarFilePath), innerFiles, replaceFiles, CharsetUtil.defaultCharset());
}
/**
* 不解压zip，替换文件内容，也可用于转换zip编码格式
*
* @param srcFilePath 源文件路径，也是输出路径
* @param innerFiles 要替换文件在zip中的相对路径
* @param replaceFiles 要替换的文件绝对路径
*/
public static void replace(String srcFilePath, String[] innerFiles, String[] replaceFiles) {
replace(FileUtil.file(srcFilePath), FileUtil.file(srcFilePath), innerFiles, replaceFiles, CharsetUtil.defaultCharset());
}
public static void replace(String srcFilePath, String[] innerFiles, File[] replaceFiles) {
replace(FileUtil.file(srcFilePath), FileUtil.file(srcFilePath), innerFiles, replaceFiles, CharsetUtil.defaultCharset());
}
public static void replace(String srcFilePath, String[] innerFiles, List<File> replaceFiles) {
replace(FileUtil.file(srcFilePath), FileUtil.file(srcFilePath), innerFiles, replaceFiles, CharsetUtil.defaultCharset());
}
/**
* 判断路径是否相等
*
* @param entryPath 路径A
* @param targetPath 路径B
* @param ignoreCase 是否忽略大小写
* @return ture 路径相等
*/
public static boolean samePath(String entryPath, String targetPath, boolean ignoreCase) {
entryPath = entryPath.replaceAll("[/\\\\]", Matcher.quoteReplacement(File.separator));
targetPath = targetPath.replaceAll("[/\\\\]", Matcher.quoteReplacement(File.separator));
if (entryPath.startsWith(File.separator)) {
entryPath = entryPath.substring(1);
}
if (targetPath.startsWith(File.separator)) {
targetPath = targetPath.substring(1);
}
if (ignoreCase) {
return StrUtil.equalsIgnoreCase(entryPath, targetPath);
} else {
return StrUtil.equals(entryPath, targetPath);
}
}
public static boolean samePath(String entryPath, String targetPath) {
return samePath(entryPath, targetPath, true);
}
}