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

package ru.gravit.launchserver.binary;

import java.io.IOException;

import javassist.CannotCompileException;
import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtConstructor;
import javassist.CtMethod;
import javassist.NotFoundException;
import ru.gravit.launcher.LauncherConfig;

public class JAConfigurator implements AutoCloseable {
    public ClassPool pool;
    public CtClass ctClass;
    public CtConstructor ctConstructor;
    public CtMethod initModuleMethod;
    String classname;
    StringBuilder body;
    StringBuilder moduleBody;
    int autoincrement;

    public JAConfigurator(String configclass, JARLauncherBinary jarLauncherBinary) throws NotFoundException {
        pool = new ClassPool(false);
        pool.insertClassPath(jarLauncherBinary.cleanJar.toFile().getAbsolutePath());
        pool.appendSystemPath();
        classname = configclass;
        ctClass = pool.get(classname);
        ctConstructor = ctClass.getDeclaredConstructor(null);
        initModuleMethod = ctClass.getDeclaredMethod("initModules");
        body = new StringBuilder("{ isInitModules = false; ");
        moduleBody = new StringBuilder("{ isInitModules = true; ");
        autoincrement = 0;
    }

    public void addModuleClass(String fullName) {
        moduleBody.append("ru.gravit.launcher.modules.Module mod");
        moduleBody.append(autoincrement);
        moduleBody.append(" = new ");
        moduleBody.append(fullName);
        moduleBody.append("();");
        moduleBody.append("ru.gravit.launcher.Launcher.modulesManager.registerModule( mod");
        moduleBody.append(autoincrement);
        moduleBody.append(" , true );");
        autoincrement++;
    }

    @Override
    public void close() {
        ctClass.defrost();
    }

    public CtClass getCtClass() {
        return ctClass;
    }

    public byte[] getBytecode() throws IOException, CannotCompileException {
        return ctClass.toBytecode();
    }

    public void compile() throws CannotCompileException {
        body.append("}");
        moduleBody.append("}");
        ctConstructor.setBody(body.toString());
        initModuleMethod.insertAfter(moduleBody.toString());
        if (ctClass.isFrozen()) ctClass.defrost();
    }

    public String getZipEntryPath() {
        return classname.replace('.', '/').concat(".class");
    }

    public void setAddress(String address) {
        body.append("this.address = \"");
        body.append(address);
        body.append("\";");
    }

    public void setProjectName(String name) {
        body.append("this.projectname = \"");
        body.append(name);
        body.append("\";");
    }

    public void setSecretKey(String key) {
        body.append("this.secretKeyClient = \"");
        body.append(key);
        body.append("\";");
    }

    public void setPort(int port) {
        body.append("this.port = ");
        body.append(port);
        body.append(";");
    }

    public void setEnv(LauncherConfig.LauncherEnvironment env) {
        int i = 2;
        switch (env) {

            case DEV:
                i = 0;
                break;
            case DEBUG:
                i = 1;
                break;
            case STD:
                i = 2;
                break;
            case PROD:
                i = 3;
                break;
        }
        body.append("this.env = ");
        body.append(i);
        body.append(";");
    }

    public void setClientPort(int port) {
        body.append("this.clientPort = ");
        body.append(port);
        body.append(";");
    }

    public void setUsingWrapper(boolean b) {
        body.append("this.isUsingWrapper = ");
        body.append(b ? "true" : "false");
        body.append(";");
    }

    public void setDownloadJava(boolean b) {
        body.append("this.isDownloadJava = ");
        body.append(b ? "true" : "false");
        body.append(";");
    }

    public void setWarningMissArchJava(boolean b) {
        body.append("this.isWarningMissArchJava = ");
        body.append(b ? "true" : "false");
        body.append(";");
    }

    public ClassPool getPool() {
        return pool;
    }
}