shz.ZipHelp Maven / Gradle / Ivy
package shz;
import shz.msg.ServerFailureMsg;
import java.io.*;
import java.nio.charset.Charset;
import java.util.*;
import java.util.jar.*;
import java.util.zip.*;
public final class ZipHelp {
private ZipHelp() {
throw new IllegalStateException();
}
public static void updateZipFile(File file, Map packageFile, int dataSize) {
File tempFile;
try {
tempFile = File.createTempFile(file.getName(), null);
} catch (IOException e) {
throw PRException.of(e);
}
FileHelp.deleteFile(tempFile);
ServerFailureMsg.requireNon(!file.renameTo(tempFile), "文件不可重命名");
ZipInputStream zis = null;
ZipOutputStream zos = null;
ZipEntry entry;
try {
zis = IOHelp.getZis(tempFile);
zos = IOHelp.getZos(file);
while ((entry = zis.getNextEntry()) != null) {
if (!packageFile.containsKey(entry.getName())) {
zos.putNextEntry(new ZipEntry(entry.getName()));
IOHelp.read(zis, zos, dataSize, null, (i, o) -> {
try {
o.closeEntry();
} catch (IOException e) {
throw PRException.of(e);
}
});
}
}
} catch (IOException e) {
IOHelp.close(zos);
throw PRException.of(e);
} finally {
IOHelp.close(zis);
}
try {
for (Map.Entry kv : packageFile.entrySet()) {
BufferedInputStream bis = IOHelp.getBis(kv.getValue());
zos.putNextEntry(new ZipEntry(kv.getKey()));
IOHelp.read(bis, zos, dataSize, null, (i, o) -> {
try {
o.closeEntry();
} catch (IOException e) {
throw PRException.of(e);
} finally {
IOHelp.close(i);
}
});
}
} catch (IOException e) {
throw PRException.of(e);
} finally {
IOHelp.close(zos);
FileHelp.deleteFile(tempFile);
}
}
public static InputStream getIsFromZis(ZipInputStream zis, String path, int dataSize) {
if (zis == null || Validator.isBlank(path)) return null;
ZipEntry entry;
try {
while ((entry = zis.getNextEntry()) != null)
if (isEntry(entry, path)) return new ByteArrayInputStream(IOHelp.read(zis, dataSize, null));
} catch (IOException e) {
throw PRException.of(e);
}
return null;
}
private static boolean isEntry(ZipEntry entry, String path) {
String entryName = entry.getName();
entryName = entryName.charAt(0) == '/' ? entryName.substring(1) : entryName;
if (path.equals(entryName)) return true;
int index = entryName.lastIndexOf('/');
return index != -1 && path.equals(entryName.substring(index + 1));
}
public static InputStream getIsFromZipFile(ZipFile zip, String path, int dataSize) {
if (zip == null || Validator.isBlank(path)) return null;
Enumeration entries = zip.entries();
InputStream is;
try {
while (entries.hasMoreElements()) {
ZipEntry entry = entries.nextElement();
if (isEntry(entry, path)) {
is = zip.getInputStream(entry);
return new ByteArrayInputStream(IOHelp.read(is, dataSize, null));
}
}
} catch (IOException e) {
throw PRException.of(e);
}
return null;
}
public static String getFileFromZis(ZipInputStream zis, int dataSize, Set paths) {
ZipEntry entry;
try {
while ((entry = zis.getNextEntry()) != null && paths.size() > 0)
writeFiles(zis, entry, dataSize, paths);
} catch (IOException e) {
throw PRException.of(e);
} finally {
IOHelp.close(zis);
}
return temp_folder.getAbsolutePath();
}
private static final File temp_folder = new File(System.getProperty("user.home"), "from-zip-file-temp");
private static void writeFiles(InputStream is, ZipEntry entry, int dataSize, Set paths) {
if (temp_folder.mkdirs() || temp_folder.exists())
for (Iterator it = paths.iterator(); it.hasNext(); ) {
String next = it.next();
if (isEntry(entry, next)) {
IOHelp.read(is, IOHelp.getBos(new File(temp_folder, next)), dataSize, null, (i, o) -> IOHelp.close(o));
it.remove();
break;
}
}
}
public static void deleteTempFolder() {
FileHelp.deleteFile(temp_folder.getAbsolutePath());
}
public static String getFileFromZipFile(ZipFile zip, int dataSize, Set paths) {
Enumeration entries = zip.entries();
InputStream is = null;
try {
while (entries.hasMoreElements() && paths.size() > 0) {
ZipEntry entry = entries.nextElement();
is = zip.getInputStream(entry);
writeFiles(is, entry, dataSize, paths);
}
} catch (IOException e) {
throw PRException.of(e);
} finally {
IOHelp.close(is);
}
return temp_folder.getAbsolutePath();
}
public static void compress(Set srcs, String des, int dataSize) {
ZipOutputStream zos = IOHelp.getZos(des);
srcs.stream().map(File::new).filter(f -> f.exists() && f.canRead()).forEach(f -> compress0(zos, dataSize, "", f));
}
public static void compress0(ZipOutputStream zos, int dataSize, String path, File... files) {
Arrays.stream(files).forEach(f -> {
if (f.isFile()) {
BufferedInputStream bis = IOHelp.getBis(f);
try {
zos.putNextEntry(new ZipEntry(path + f.getName()));
} catch (IOException e) {
throw PRException.of(e);
}
IOHelp.read(bis, zos, dataSize, null, (i, o) -> {
try {
o.closeEntry();
} catch (IOException e) {
throw PRException.of(e);
} finally {
IOHelp.close(i);
}
});
} else {
File[] fs = f.listFiles();
if (Validator.isEmpty(fs)) try {
zos.putNextEntry(new ZipEntry(path + f.getName() + "/"));
} catch (IOException e) {
throw PRException.of(e);
}
else compress0(zos, dataSize, path + "/" + f.getName(), fs);
}
});
}
public static void compress(Set srcs, String des) {
compress(srcs, des, IOHelp.DEFAULT_DATA_SIZE);
}
public static void decompress(String src, String des, int dataSize) {
File desFile = FileHelp.checkCopyFile(des);
File file;
if ((file = FileHelp.findFile(src)) == null) return;
ZipFile zip;
try {
zip = new ZipFile(file);
} catch (IOException e) {
throw PRException.of(e);
}
Enumeration entries = zip.entries();
InputStream is = null;
try {
while (entries.hasMoreElements()) {
ZipEntry entry = entries.nextElement();
File f = new File(desFile, entry.getName());
if (entry.isDirectory()) {
if (!f.mkdirs() && !f.exists()) throw new IllegalArgumentException();
continue;
}
is = zip.getInputStream(entry);
IOHelp.read(is, IOHelp.getBos(f), dataSize, null, (i, o) -> IOHelp.close(o));
}
} catch (IOException e) {
throw PRException.of(e);
} finally {
IOHelp.close(is);
}
}
public static void decompress(String src, String des) {
decompress(src, des, IOHelp.DEFAULT_DATA_SIZE);
}
public static void updateJarFile(File file, Map packageFile, int dataSize) {
File tempFile;
try {
tempFile = File.createTempFile(file.getName(), null);
} catch (IOException e) {
throw PRException.of(e);
}
FileHelp.deleteFile(tempFile);
ServerFailureMsg.requireNon(!file.renameTo(tempFile), "文件不可重命名");
JarInputStream jis = null;
JarOutputStream jos = null;
JarEntry entry;
try {
jis = IOHelp.getJis(tempFile);
jos = IOHelp.getJos(file);
while ((entry = (JarEntry) jis.getNextEntry()) != null) {
if (!packageFile.containsKey(entry.getName())) {
jos.putNextEntry(new JarEntry(entry.getName()));
IOHelp.read(jis, jos, dataSize, null, (i, o) -> {
try {
o.closeEntry();
} catch (IOException e) {
throw PRException.of(e);
}
});
}
}
} catch (IOException e) {
IOHelp.close(jos);
throw PRException.of(e);
} finally {
IOHelp.close(jis);
}
try {
for (Map.Entry kv : packageFile.entrySet()) {
BufferedInputStream bis = IOHelp.getBis(kv.getValue());
jos.putNextEntry(new JarEntry(kv.getKey()));
IOHelp.read(bis, jos, dataSize, null, (i, o) -> {
try {
o.closeEntry();
} catch (IOException e) {
throw PRException.of(e);
} finally {
IOHelp.close(i);
}
});
}
} catch (IOException e) {
IOHelp.close(jos);
throw PRException.of(e);
} finally {
IOHelp.close(jos);
FileHelp.deleteFile(tempFile);
}
}
public static InputStream getIsFromJis(JarInputStream jis, String path, int dataSize) {
if (jis == null || Validator.isBlank(path)) return null;
JarEntry entry;
try {
while ((entry = (JarEntry) jis.getNextEntry()) != null)
if (isEntry(entry, path)) return new ByteArrayInputStream(IOHelp.read(jis, dataSize, null));
} catch (IOException e) {
throw PRException.of(e);
}
return null;
}
public static InputStream getIsFromJarFile(JarFile jar, String path, int dataSize) {
if (jar == null || Validator.isBlank(path)) return null;
Enumeration entries = jar.entries();
InputStream is;
try {
while (entries.hasMoreElements()) {
JarEntry entry = entries.nextElement();
if (isEntry(entry, path)) {
is = jar.getInputStream(entry);
return new ByteArrayInputStream(IOHelp.read(is, dataSize, null));
}
}
} catch (IOException e) {
throw PRException.of(e);
}
return null;
}
public static String getFileFromJis(JarInputStream jis, int dataSize, Set paths) {
JarEntry entry;
try {
while ((entry = (JarEntry) jis.getNextEntry()) != null && paths.size() > 0)
writeFiles(jis, entry, dataSize, paths);
} catch (IOException e) {
throw PRException.of(e);
} finally {
IOHelp.close(jis);
}
return temp_folder.getAbsolutePath();
}
public static String getFileFromJarFile(JarFile jar, int dataSize, Set paths) {
Enumeration entries = jar.entries();
InputStream is = null;
try {
while (entries.hasMoreElements() && paths.size() > 0) {
JarEntry entry = entries.nextElement();
is = jar.getInputStream(entry);
writeFiles(is, entry, dataSize, paths);
}
} catch (IOException e) {
throw PRException.of(e);
} finally {
IOHelp.close(is);
}
return temp_folder.getAbsolutePath();
}
public static void makeJar(File file, JarOutputStream jos) {
File[] files = file.listFiles();
if (Validator.nonEmpty(files)) try {
int idx = file.getAbsolutePath().replaceAll("\\\\", "/").lastIndexOf("/") + 1;
for (File f : files) makeJar0(f, jos, idx);
} catch (IOException e) {
throw PRException.of(e);
} finally {
IOHelp.close(jos);
}
}
private static void makeJar0(File file, JarOutputStream jos, int idx) throws IOException {
if (file.isDirectory()) {
File[] files = file.listFiles();
if (Validator.nonEmpty(files)) for (File f : files) makeJar0(f, jos, idx);
} else {
BufferedInputStream bis = IOHelp.getBis(file);
jos.putNextEntry(new JarEntry(file.getAbsolutePath().replaceAll("\\\\", "/").substring(idx)));
IOHelp.read(bis, jos, IOHelp.DEFAULT_DATA_SIZE, null, (i, o) -> {
try {
o.closeEntry();
} catch (IOException e) {
throw PRException.of(e);
} finally {
IOHelp.close(i);
}
});
}
}
public static void makeRunnableJar(JarInputStream jis, JarOutputStream jos) {
JarEntry entry;
try {
while ((entry = jis.getNextJarEntry()) != null) {
if (JarFile.MANIFEST_NAME.equals(entry.getName())) continue;
jos.putNextEntry(entry);
IOHelp.read(jis, jos, IOHelp.DEFAULT_DATA_SIZE, null, (i, o) -> {
try {
o.closeEntry();
} catch (IOException e) {
throw PRException.of(e);
}
});
}
} catch (IOException e) {
throw PRException.of(e);
} finally {
IOHelp.close(jos, jis);
}
}
public static void makeRunnableJar(JarInputStream jis, File newJar, Manifest manifest) {
makeRunnableJar(jis, IOHelp.getJos(newJar, false, manifest));
}
public static void makeRunnableJar(JarInputStream jis, File newJar, String mainClass, String version) {
Manifest manifest = jis.getManifest();
if (manifest == null) manifest = new Manifest();
Attributes a = manifest.getMainAttributes();
String oldMainClass = a.putValue("Main-Class", mainClass);
if (oldMainClass != null) {
IOHelp.close(jis);
return;
}
if (version == null) {
version = (String) a.get("Manifest-Version");
if (version == null) version = (String) a.get("Signature-Version");
}
a.putValue("Manifest-Version", version);
makeRunnableJar(jis, IOHelp.getJos(newJar, false, manifest));
}
public static String read(InputStream is, Charset charset) {
BufferedReader br;
try {
br = new BufferedReader(new InputStreamReader(new GZIPInputStream(new ByteArrayInputStream(IOHelp.read(is))), charset));
} catch (IOException e) {
throw PRException.of(e);
}
StringWriter sw = new StringWriter();
IOHelp.read(br, sw);
return sw.toString();
}
} © 2015 - 2025 Weber Informatics LLC | Privacy Policy