106 lines
2.7 KiB
Java
106 lines
2.7 KiB
Java
package com.imyeyu.compress;
|
|
|
|
import com.imyeyu.io.IO;
|
|
import org.apache.commons.compress.archivers.sevenz.SevenZArchiveEntry;
|
|
import org.apache.commons.compress.archivers.sevenz.SevenZOutputFile;
|
|
|
|
import java.io.File;
|
|
import java.io.InputStream;
|
|
import java.io.OutputStream;
|
|
import java.nio.file.Files;
|
|
import java.nio.file.Path;
|
|
import java.util.List;
|
|
|
|
/**
|
|
* 7z 压缩器
|
|
*
|
|
* @author 夜雨
|
|
* @version 2024-06-30 19:40
|
|
*/
|
|
public class Z7Compressor extends Compressor<Z7Compressor> {
|
|
|
|
/**
|
|
* 创建压缩器
|
|
*
|
|
* @param fromPath 源路径
|
|
* @return 压缩器
|
|
*/
|
|
public static Z7Compressor from(String fromPath) {
|
|
return new Z7Compressor().of(fromPath);
|
|
}
|
|
|
|
/**
|
|
* 创建压缩器
|
|
*
|
|
* @param fromFile 源文件
|
|
* @return 压缩器
|
|
*/
|
|
public static Z7Compressor from(File fromFile) {
|
|
return new Z7Compressor().of(fromFile);
|
|
}
|
|
|
|
@Override
|
|
protected void toStream(String fromPath, OutputStream toStream) throws Exception {
|
|
Path tempFile = Files.createTempFile("timi-compress-", ".7z");
|
|
try {
|
|
of(fromPath).toFile(tempFile.toFile());
|
|
try (InputStream inputStream = Files.newInputStream(tempFile)) {
|
|
transfer(inputStream, toStream, false);
|
|
toStream.flush();
|
|
}
|
|
} finally {
|
|
Files.deleteIfExists(tempFile);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
protected void toFile(String fromPath, File toFile) throws Exception {
|
|
File fromFile = new File(fromPath);
|
|
List<File> files = IO.listFile(fromFile);
|
|
String basePath = files.getFirst().getParentFile().getAbsolutePath();
|
|
initByteProgress(IO.calcSize(fromFile));
|
|
try (SevenZOutputFile outputFile = new SevenZOutputFile(toFile)) {
|
|
for (File sourceFile : files) {
|
|
String name = sourceFile.getAbsolutePath().substring(basePath.length() + 1);
|
|
SevenZArchiveEntry entry = outputFile.createArchiveEntry(sourceFile, normalizeEntryName(name));
|
|
outputFile.putArchiveEntry(entry);
|
|
writeSevenZEntry(outputFile, sourceFile);
|
|
outputFile.closeArchiveEntry();
|
|
handleFile(sourceFile);
|
|
}
|
|
outputFile.finish();
|
|
finishProgress();
|
|
} catch (Exception exception) {
|
|
if (isInterrupt) {
|
|
IO.destroy(toFile);
|
|
}
|
|
throw exception;
|
|
} finally {
|
|
resetProgress();
|
|
}
|
|
if (isInterrupt) {
|
|
IO.destroy(toFile);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 写入 7z 条目内容
|
|
*
|
|
* @param outputFile 7z 输出文件
|
|
* @param sourceFile 源文件
|
|
* @throws Exception 写入失败
|
|
*/
|
|
private void writeSevenZEntry(SevenZOutputFile outputFile, File sourceFile) throws Exception {
|
|
byte[] buffer = new byte[8192];
|
|
try (InputStream inputStream = IO.getInputStream(sourceFile)) {
|
|
int length;
|
|
while ((length = inputStream.read(buffer)) != -1) {
|
|
awaitIfPaused();
|
|
ensureRunning();
|
|
outputFile.write(buffer, 0, length);
|
|
handleTransferred(length);
|
|
}
|
|
}
|
|
}
|
|
}
|