Launcher/LaunchServer/src/main/java/ru/gravit/launchserver/binary/BuildContext.java

package ru.gravit.launchserver.binary;

import ru.gravit.utils.helper.IOHelper;

import java.io.IOException;
import java.io.InputStream;
import java.util.Set;
import java.util.jar.JarInputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

public class BuildContext {
    public final ZipOutputStream output;
    public final JAConfigurator config;

    public BuildContext(ZipOutputStream output, JAConfigurator config) {
        this.output = output;
        this.config = config;
    }
    public void pushFile(String filename,InputStream inputStream) throws IOException {
        ZipEntry zip = IOHelper.newZipEntry(filename);
        output.putNextEntry(zip);
        IOHelper.transfer(inputStream,output);
    }
    public void pushJarFile(JarInputStream input) throws IOException {
        ZipEntry e = input.getNextEntry();
        while (e != null) {
            output.putNextEntry(e);
            IOHelper.transfer(input,output);
            e = input.getNextEntry();
        }
    }
    public void pushJarFile(JarInputStream input, Set<String> blacklist) throws IOException {
        ZipEntry e = input.getNextEntry();
        while (e != null) {
            if(blacklist.contains(e.getName())){
                e = input.getNextEntry();
                continue;
            }
            output.putNextEntry(e);
            IOHelper.transfer(input,output);
            e = input.getNextEntry();
        }
    }
}