Launcher/ServerWrapper/src/main/java/ru/gravit/launcher/server/ServerWrapper.java

package ru.gravit.launcher.server;


import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.lang.invoke.MethodHandle;
import java.lang.invoke.MethodHandles;
import java.lang.invoke.MethodType;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.HashMap;

import ru.gravit.launcher.Launcher;
import ru.gravit.launcher.LauncherConfig;
import ru.gravit.launcher.serialize.config.ConfigObject;
import ru.gravit.launcher.serialize.config.TextConfigReader;
import ru.gravit.launcher.serialize.config.TextConfigWriter;
import ru.gravit.launcher.serialize.config.entry.BlockConfigEntry;
import ru.gravit.launcher.serialize.config.entry.BooleanConfigEntry;
import ru.gravit.launcher.serialize.config.entry.IntegerConfigEntry;
import ru.gravit.launcher.serialize.config.entry.StringConfigEntry;
import ru.gravit.utils.helper.IOHelper;
import ru.gravit.utils.helper.LogHelper;
import ru.gravit.launcher.profiles.ClientProfile;
import ru.gravit.launcher.request.update.ProfilesRequest;
import ru.gravit.launcher.serialize.HInput;
import ru.gravit.launcher.serialize.signed.SignedObjectHolder;
import ru.gravit.utils.helper.SecurityHelper;
import sun.security.rsa.RSAPublicKeyImpl;

public class ServerWrapper {
    public static ModulesManager modulesManager;
    public static Path configFile;
    public static Config config;
    public static void main(String[] args) throws Throwable {
        ServerWrapper wrapper = new ServerWrapper();
        modulesManager = new ModulesManager(wrapper);
        modulesManager.autoload(Paths.get("modules"));
        Launcher.modulesManager = modulesManager;
        configFile = Paths.get("ServerWrapper.cfg");
        modulesManager.preInitModules();
        generateConfigIfNotExists();
        try (BufferedReader reader = IOHelper.newReader(configFile)) {
            config = new Config(TextConfigReader.read(reader, true));
        }
        LauncherConfig cfg = new LauncherConfig(config.address, config.port, SecurityHelper.toPublicRSAKey(IOHelper.read(Paths.get("public.key"))),new HashMap<>(),config.projectname);
        ProfilesRequest.Result result = new ProfilesRequest(cfg).request();
        for (SignedObjectHolder<ClientProfile> p : result.profiles) {
            LogHelper.debug("Get profile: %s", p.object.getTitle());
            if (p.object.getTitle().equals(config.title)) {
                wrapper.profile = p.object;
                Launcher.profile = p.object;
                LogHelper.debug("Found profile: %s", Launcher.profile.getTitle());
                break;
            }
        }
        modulesManager.initModules();
        String classname = config.mainclass.isEmpty() ? args[0] : config.mainclass;
        Class<?> mainClass;
        if(config.customClassLoader) {
            Class<ClassLoader> classloader_class = (Class<ClassLoader>) Class.forName(config.classloader);
            ClassLoader loader = classloader_class.getConstructor(ClassLoader.class).newInstance(ClassLoader.getSystemClassLoader());
            Thread.currentThread().setContextClassLoader(loader);
            mainClass = Class.forName(classname,false,loader);
        }
        else mainClass = Class.forName(classname);
        MethodHandle mainMethod = MethodHandles.publicLookup().findStatic(mainClass, "main", MethodType.methodType(void.class, String[].class));
        String[] real_args = new String[args.length - 1];
        System.arraycopy(args, 1, real_args, 0, args.length - 1);
        modulesManager.postInitModules();
        mainMethod.invoke(real_args);
    }
    private static void generateConfigIfNotExists() throws IOException {
        if (IOHelper.isFile(configFile))
            return;

        // Create new config
        LogHelper.info("Creating LaunchWrapper config");
        Config newConfig;
        try (BufferedReader reader = IOHelper.newReader(IOHelper.getResourceURL("ru/gravit/launcher/server/config.cfg"))) {
            newConfig = new Config(TextConfigReader.read(reader, false));
        }

        LogHelper.warning("Title is not set. Please show ServerWrapper.cfg");

        // Write LaunchServer config
        LogHelper.info("Writing LaunchWrapper config file");
        try (BufferedWriter writer = IOHelper.newWriter(configFile)) {
            TextConfigWriter.write(newConfig.block, writer, true);
        }
    }
    public static final class Config extends ConfigObject {
        public String title;
        public String projectname;
        public String address;
        public int port;
        public boolean customClassLoader;
        public String classloader;
        public String mainclass;
        protected Config(BlockConfigEntry block) {
            super(block);
            title = block.getEntryValue("title",StringConfigEntry.class);
            address = block.getEntryValue("address",StringConfigEntry.class);
            projectname = block.getEntryValue("projectName",StringConfigEntry.class);
            port = block.getEntryValue("port", IntegerConfigEntry.class);
            customClassLoader = block.getEntryValue("customClassLoader", BooleanConfigEntry.class);
            if(customClassLoader)
                classloader = block.getEntryValue("classloader",StringConfigEntry.class);
            mainclass = block.getEntryValue("MainClass",StringConfigEntry.class);
        }
    }

    public ClientProfile profile;
}