package com.javarush.task.task31.task3105;
import java.io.*;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;
import java.util.zip.ZipOutputStream;
/*
Добавление файла в архив
*/
public class Solution {
public static void main(String[] args) throws IOException {
Map<ZipEntry, byte[]> dataFromZip = new HashMap<>();
byte[] buffer = new byte[1024];
try (ZipInputStream zipInputStream = new ZipInputStream(new FileInputStream(args[1]))) {
ZipEntry entry;
while ((entry = zipInputStream.getNextEntry()) != null) {
ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
int count;
while ((count = zipInputStream.read(buffer)) != -1) {
byteArrayOutputStream.write(buffer, 0, count);
}
dataFromZip.put(entry, byteArrayOutputStream.toByteArray());
}
}
FileOutputStream fileOutputStream = new FileOutputStream(args[1]);
ZipOutputStream zipOutputStream = new ZipOutputStream(fileOutputStream);
Path path = Paths.get(args[0]);
List<ZipEntry> zipEntries = new ArrayList<>(dataFromZip.keySet());
boolean isContains = false;
for (ZipEntry zipEntry : zipEntries) {
if (zipEntry.getName().equals(path.getFileName().toString())) {
isContains = true;
}
}
if (isContains) {
try {
zipOutputStream.putNextEntry(new ZipEntry(path.getFileName().toString()));
Files.copy(path, zipOutputStream);
zipOutputStream.closeEntry();
} catch (IOException e) {
e.printStackTrace();
}
} else {
try {
zipOutputStream.putNextEntry(new ZipEntry("new/"));
zipOutputStream.putNextEntry(new ZipEntry("new/" + path.getFileName().toString()));
Files.copy(path, zipOutputStream);
zipOutputStream.closeEntry();
} catch (IOException e) {
e.printStackTrace();
}
}
dataFromZip.forEach((key, value) -> {
if (!key.getName().equals(path.getFileName().toString())) {
try {
zipOutputStream.putNextEntry(key);
zipOutputStream.write(value);
zipOutputStream.closeEntry();
} catch (IOException e) {
e.printStackTrace();
}
}
});
zipOutputStream.close();
}
}