ResourcePackConverter.java

package i18nupdatemod.core;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import i18nupdatemod.util.FileUtil;
import i18nupdatemod.util.Log;
import org.apache.commons.io.IOUtils;

import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.Enumeration;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipOutputStream;

public class ResourcePackConverter {
    private static final Gson GSON = new GsonBuilder().setPrettyPrinting().create();
    private final List<Path> sourcePath;
    private final Path filePath;
    private final Path tmpFilePath;

    public ResourcePackConverter(List<ResourcePack> resourcePack, String filename) {
        this.sourcePath = resourcePack.stream().map(ResourcePack::getTmpFilePath).collect(Collectors.toList());
        this.filePath = FileUtil.getResourcePackPath(filename);
        this.tmpFilePath = FileUtil.getTemporaryPath(filename);
    }

    public void convert(int packFormat, String description) throws Exception {
        Set<String> fileList = new HashSet<>();
        try (ZipOutputStream zos = new ZipOutputStream(
                Files.newOutputStream(tmpFilePath),
                StandardCharsets.UTF_8)) {
//            zos.setMethod(ZipOutputStream.STORED);
            for (Path p : sourcePath) {
                Log.info("Converting: " + p);
                LoadDetailUI.appendLog("正在转换 " + p);
                try (ZipFile zf = new ZipFile(p.toFile(), StandardCharsets.UTF_8)) {
                    for (Enumeration<? extends ZipEntry> e = zf.entries(); e.hasMoreElements(); ) {
                        ZipEntry ze = e.nextElement();
                        String name = ze.getName();
                        // Don't put same file
                        if (fileList.contains(name)) {
//                            Log.debug(name + ": DUPLICATE");
                            continue;
                        }
                        fileList.add(name);
//                        Log.debug(name);

                        // Put file into new zip
                        zos.putNextEntry(new ZipEntry(name));
                        InputStream is = zf.getInputStream(ze);
                        if (name.equalsIgnoreCase("pack.mcmeta")) {
                            //Convert pack.mcmeta
                            zos.write(convertPackMeta(is, packFormat, description));
                        } else {
                            //Copy other file
                            IOUtils.copy(is, zos);
                        }
                        zos.closeEntry();
                    }
                }
            }
            zos.close();
            Log.info("Converted: %s -> %s", sourcePath, tmpFilePath);
            FileUtil.syncTmpFile(tmpFilePath, filePath, true);
        } catch (Exception e) {
            throw new Exception(String.format("Error converting %s to %s: %s", sourcePath, tmpFilePath, e));
        }
    }

    private byte[] convertPackMeta(InputStream is, int packFormat, String description) {
        PackMeta meta = GSON.fromJson(new InputStreamReader(is, StandardCharsets.UTF_8), PackMeta.class);
        meta.pack.pack_format = packFormat;
        meta.pack.description = description;
        return GSON.toJson(meta).getBytes(StandardCharsets.UTF_8);
    }

    private static class PackMeta {
        Pack pack;

        private static class Pack {
            int pack_format;
            String description;
        }
    }
}