InstrumentationHelper.java

package me.earth.headlessmc.launcher.instrumentation;

import lombok.experimental.UtilityClass;
import lombok.val;
import me.earth.headlessmc.launcher.instrumentation.log4j.Patchers;
import me.earth.headlessmc.launcher.instrumentation.lwjgl.HmcLwjglTransformer;
import me.earth.headlessmc.launcher.instrumentation.modlauncher.BootstrapLauncherTransformer;
import me.earth.headlessmc.launcher.instrumentation.paulscode.PaulscodeTransformer;
import me.earth.headlessmc.launcher.launch.LaunchOptions;

import java.util.ArrayList;

@UtilityClass
public class InstrumentationHelper {
    public static final String RUNTIME_JAR = "headlessmc-runtime.jar";
    public static final String LWJGL_JAR = "headlessmc-lwjgl.jar";

    public static Instrumentation create(LaunchOptions options) {
        val transformers = new ArrayList<Transformer>(6);
        if (options.isLwjgl()) {
            transformers.add(new HmcLwjglTransformer());
            transformers.add(new ResourceExtractor(options.getFiles(), LWJGL_JAR));
        }

        if (options.isPaulscode()) {
            transformers.add(new PaulscodeTransformer());
        }

        if (options.isJndi()) {
            transformers.add(Patchers.JNDI);
        }

        if (options.isRuntime()) {
            transformers.add(new ResourceExtractor(options.getFiles(), RUNTIME_JAR));
        }

        if (options.isLookup()) {
            transformers.add(Patchers.LOOKUP);
        }

        if (options.isInMemory()) {
            transformers.add(new BootstrapLauncherTransformer());
        }

        return new Instrumentation(transformers, options.getFiles().getBase());
    }

}