package net.ME1312.SubServers.Host.Executable; import net.ME1312.Galaxi.Engine.GalaxiEngine; import net.ME1312.Galaxi.Library.Config.YAMLSection; import net.ME1312.Galaxi.Library.Container.NamedContainer; import net.ME1312.Galaxi.Library.Map.ObjectMap; import net.ME1312.Galaxi.Library.Container.Container; import net.ME1312.Galaxi.Library.Map.ObjectMapValue; import net.ME1312.Galaxi.Library.UniversalFile; import net.ME1312.Galaxi.Library.Util; import net.ME1312.Galaxi.Library.Version.Version; import net.ME1312.SubData.Client.SubDataClient; import net.ME1312.SubServers.Host.ExHost; import net.ME1312.SubServers.Host.Library.Exception.InvalidServerException; import net.ME1312.SubServers.Host.Library.Exception.InvalidTemplateException; import net.ME1312.SubServers.Host.Library.Exception.SubCreatorException; import net.ME1312.SubServers.Host.Library.ReplacementScanner; import net.ME1312.SubServers.Host.Network.API.SubCreator.ServerType; import net.ME1312.SubServers.Host.Network.Packet.PacketExCreateServer; import net.ME1312.SubServers.Host.Network.Packet.PacketOutExLogMessage; import net.ME1312.SubServers.Host.SubAPI; import org.json.JSONObject; import java.io.*; import java.net.URL; import java.nio.charset.Charset; import java.nio.file.Files; import java.nio.file.LinkOption; import java.nio.file.StandardCopyOption; import java.security.MessageDigest; import java.util.*; /** * Internal SubCreator Class */ public class SubCreatorImpl { private ExHost host; private TreeMap thread; public static class ServerTemplate extends net.ME1312.SubServers.Host.Network.API.SubCreator.ServerTemplate { private String name; private String nick = null; private boolean enabled; private String icon; private File directory; private ServerType type; private ObjectMap build; private ObjectMap options; /** * Create a SubCreator Template * * @param name Template Name * @param directory Template Directory * @param build Build Options * @param options Configuration Options */ public ServerTemplate(String name, boolean enabled, String icon, File directory, ObjectMap build, ObjectMap options) { super(toRaw(name, enabled, icon, directory, build, options)); if (name.contains(" ")) throw new InvalidTemplateException("Template names cannot have spaces: " + name); this.name = name; this.enabled = enabled; this.icon = icon; this.directory = directory; this.type = (build.contains("Server-Type"))?ServerType.valueOf(build.getRawString("Server-Type").toUpperCase()):ServerType.CUSTOM; this.build = build; this.options = options; } /** * Get the Name of this Template * * @return Template Name */ public String getName() { return name; } /** * Get the Display Name of this Template * * @return Display Name */ public String getDisplayName() { return (nick == null)?getName():nick; } /** * Sets the Display Name for this Template * * @param value Value (or null to reset) */ public void setDisplayName(String value) { if (value == null || value.length() == 0 || getName().equals(value)) { this.nick = null; } else { this.nick = value; } } /** * Get the Enabled Status of this Template * * @return Enabled Status */ public boolean isEnabled() { return enabled; } /** * Set the Enabled Status of this Template * * @param value Value */ public void setEnabled(boolean value) { enabled = value; } /** * Get the Item Icon for this Template * * @return Item Icon Name/ID */ public String getIcon() { return icon; } /** * Set the Item Icon for this Template * * @param value Value */ public void setIcon(String value) { icon = value; } /** * Get the Directory for this Template * * @return Directory */ public File getDirectory() { return directory; } /** * Get the Type of this Template * * @return Template Type */ public ServerType getType() { return type; } /** * Get whether this Template requires the Version argument * * @return Version Requirement */ public boolean requiresVersion() { return getBuildOptions().getBoolean("Require-Version", false); } /** * Get whether this Template can be used to update it's servers * * @return Updatable Status */ public boolean canUpdate() { return getBuildOptions().getBoolean("Can-Update", false); } /** * Get the Build Options for this Template * * @return Build Options */ public ObjectMap getBuildOptions() { return build; } /** * Get the Configuration Options for this Template * * @return Configuration Options */ public ObjectMap getConfigOptions() { return options; } private static ObjectMap toRaw(String name, boolean enabled, String icon, File directory, ObjectMap build, ObjectMap options) { if (Util.isNull(name, enabled, directory, build, options)) throw new NullPointerException(); ObjectMap tinfo = new ObjectMap(); tinfo.set("enabled", enabled); tinfo.set("name", name); tinfo.set("display", name); tinfo.set("icon", icon); tinfo.set("type", (build.contains("Server-Type"))?ServerType.valueOf(build.getRawString("Server-Type").toUpperCase()):ServerType.CUSTOM); return tinfo; } } private class CreatorTask extends Thread { private final SubServerImpl update; private final UUID player; private final String name; private final ServerTemplate template; private final Version version; private final int port; private final UUID address; private final UUID tracker; private final SubLoggerImpl log; private final HashMap replacements; private Process process; private CreatorTask(UUID player, String name, ServerTemplate template, Version version, int port, UUID address, UUID tracker) { super(SubAPI.getInstance().getAppInfo().getName() + "::SubCreator_Process_Handler(" + name + ')'); this.update = host.servers.getOrDefault(name.toLowerCase(), null); this.player = player; this.name = name; this.template = template; this.version = version; this.port = port; this.log = new SubLoggerImpl(null, this, name + File.separator + ((update == null)?"Creator":"Updater"), address, new Container(true), null); this.replacements = new HashMap(); this.address = address; this.tracker = tracker; } private ObjectMap build(File dir, ServerTemplate template, List history) throws SubCreatorException { ObjectMap server = new ObjectMap(); Version version = this.version; HashMap var = new HashMap(); boolean error = false; if (history.contains(template)) throw new IllegalStateException("Template Import loop detected"); history.add(template); for (String other : template.getBuildOptions().getStringList("Import", new ArrayList())) { if (host.templates.keySet().contains(other.toLowerCase())) { if (host.templates.get(other.toLowerCase()).isEnabled()) { if (version != null || !host.templates.get(other.toLowerCase()).requiresVersion()) { if (update == null || host.templates.get(other.toLowerCase()).canUpdate()) { ObjectMap config = build(dir, host.templates.get(other.toLowerCase()), history); if (config == null) { throw new SubCreatorException(); } else { server.setAll(config); } } else { log.logger.warn.println("Skipping template that cannot be run in update mode: " + other); ((SubDataClient) SubAPI.getInstance().getSubDataNetwork()[0]).sendPacket(new PacketOutExLogMessage(address, "Skipping template that cannot be run in update mode: " + other)); } } else { log.logger.warn.println("Skipping template that requires extra versioning: " + other); ((SubDataClient) SubAPI.getInstance().getSubDataNetwork()[0]).sendPacket(new PacketOutExLogMessage(address, "Skipping template that requires extra versioning: " + other)); } } else { log.logger.warn.println("Skipping disabled template: " + other); ((SubDataClient) SubAPI.getInstance().getSubDataNetwork()[0]).sendPacket(new PacketOutExLogMessage(address, "Skipping disabled template: " + other)); } } else { log.logger.warn.println("Skipping missing template: " + other); ((SubDataClient) SubAPI.getInstance().getSubDataNetwork()[0]).sendPacket(new PacketOutExLogMessage(address, "Skipping missing template: " + other)); } } server.setAll(template.getConfigOptions()); try { log.logger.info.println("Loading Template: " + template.getDisplayName()); ((SubDataClient) SubAPI.getInstance().getSubDataNetwork()[0]).sendPacket(new PacketOutExLogMessage(address, "Loading Template: " + template.getDisplayName())); if (template.getBuildOptions().getBoolean("Update-Files", false)) updateDirectory(template.getDirectory(), dir); else Util.copyDirectory(template.getDirectory(), dir); for (ObjectMapValue replacement : template.getBuildOptions().getMap("Replacements").getValues()) if (!replacement.isNull()) { replacements.put(replacement.getHandle().toLowerCase().replace('-', '_').replace(' ', '_'), replacement.asRawString()); } var.putAll(replacements); var.put("java", System.getProperty("java.home") + File.separator + "bin" + File.separator + "java"); var.put("mode", (update == null)?"CREATE":"UPDATE"); if (player != null) var.put("player", player.toString().toUpperCase()); else var.remove("player"); var.put("name", name); var.put("host", SubAPI.getInstance().getName()); var.put("template", template.getName()); var.put("type", template.getType().toString().toUpperCase()); if (version != null) var.put("version", version.toString()); else var.remove("version"); var.put("address", getAddress()); var.put("port", Integer.toString(port)); switch (template.getType()) { case SPONGE: case FORGE: if (version != null) { log.logger.info.println("Searching Versions..."); ((SubDataClient) SubAPI.getInstance().getSubDataNetwork()[0]).sendPacket(new PacketOutExLogMessage(address, "Searching Versions...")); YAMLSection spversionmanifest = new YAMLSection(new JSONObject("{\"versions\":" + Util.readAll(new BufferedReader(new InputStreamReader(new URL("https://dl-api.spongepowered.org/v1/org.spongepowered/sponge" + ((template.getType() == ServerType.FORGE)?"forge":"vanilla") + "/downloads?type=stable&minecraft=" + version).openStream(), Charset.forName("UTF-8")))) + '}')); ObjectMap spprofile = null; Version spversion = null; for (ObjectMap profile : spversionmanifest.getMapList("versions")) { if (profile.getMap("dependencies").getRawString("minecraft").equalsIgnoreCase(version.toString()) && (spversion == null || new Version(profile.getRawString("version")).compareTo(spversion) >= 0)) { spprofile = profile; spversion = new Version(profile.getRawString("version")); } } if (spversion == null) throw new InvalidServerException("Cannot find Sponge version for Minecraft " + version.toString()); log.logger.info.println("Found \"sponge" + ((template.getType() == ServerType.FORGE)?"forge":"vanilla") + "-" + spversion.toString() + '"'); ((SubDataClient) SubAPI.getInstance().getSubDataNetwork()[0]).sendPacket(new PacketOutExLogMessage(address, "Found \"sponge" + ((template.getType() == ServerType.FORGE)?"forge":"vanilla") + "-" + spversion.toString() + '"')); if (template.getType() == ServerType.FORGE) { Version mcfversion = new Version(((spprofile.getMap("dependencies").getRawString("forge").contains("-"))?"":spprofile.getMap("dependencies").getRawString("minecraft") + '-') + spprofile.getMap("dependencies").getRawString("forge")); log.logger.info.println("Found \"forge-" + mcfversion.toString() + '"'); ((SubDataClient) SubAPI.getInstance().getSubDataNetwork()[0]).sendPacket(new PacketOutExLogMessage(address, "Found \"forge-" + mcfversion.toString() + '"')); var.put("mcf_version", mcfversion.toString()); } var.put("sp_version", spversion.toString()); } break; } } catch (Exception e) { log.logger.error.println(e); } if (template.getBuildOptions().contains("Executable")) { File cache; if (template.getBuildOptions().getBoolean("Use-Cache", true)) { cache = new UniversalFile(GalaxiEngine.getInstance().getRuntimeDirectory(), "Cache:Templates:" + template.getName()); cache.mkdirs(); String c = cache.toString(); if (System.getProperty("os.name").toLowerCase().startsWith("windows") && (template.getBuildOptions().getRawString("Executable").toLowerCase().startsWith("bash ") || template.getBuildOptions().getRawString("Executable").toLowerCase().startsWith("sh "))) c = c.replace(File.separatorChar, '/'); var.put("cache", c); } else { cache = null; } try { log.logger.info.println("Launching Build Script..."); ((SubDataClient) SubAPI.getInstance().getSubDataNetwork()[0]).sendPacket(new PacketOutExLogMessage(address, "Launching Build Script...")); ProcessBuilder pb = new ProcessBuilder().command(Executable.parse(host.host.getRawString("Git-Bash"), template.getBuildOptions().getRawString("Executable"))).directory(dir); pb.environment().putAll(var); process = pb.start(); log.file = new File(dir, "SubCreator-" + template.getName() + "-" + ((version != null)?"-"+version.toString():"") + ".log"); log.process = process; log.start(); process.waitFor(); Thread.sleep(500); if (process.exitValue() != 0) error = true; } catch (InterruptedException e) { error = true; } catch (Exception e) { error = true; log.logger.error.println(e); } if (cache != null) { if (cache.isDirectory() && cache.listFiles().length == 0) cache.delete(); cache = new UniversalFile(GalaxiEngine.getInstance().getRuntimeDirectory(), "Cache:Templates"); if (cache.isDirectory() && cache.listFiles().length == 0) cache.delete(); cache = new UniversalFile(GalaxiEngine.getInstance().getRuntimeDirectory(), "Cache"); if (cache.isDirectory() && cache.listFiles().length == 0) cache.delete(); } } new UniversalFile(dir, "template.yml").delete(); if (error) throw new SubCreatorException(); return server; } @SuppressWarnings("unchecked") public void run() { Runnable declaration = () -> { replacements.put("player", (player == null)?"":player.toString()); replacements.put("name", name); replacements.put("host", SubAPI.getInstance().getName()); replacements.put("template", template.getName()); replacements.put("type", template.getType().toString()); replacements.put("version", (version != null)?version.toString():""); replacements.put("address", getAddress()); replacements.put("port", Integer.toString(port)); }; declaration.run(); File dir = (update != null)?new File(update.getFullPath()):new File(host.host.getRawString("Directory"), (template.getConfigOptions().contains("Directory"))?new ReplacementScanner(replacements).replace(template.getConfigOptions().getRawString("Directory")).toString():name); dir.mkdirs(); ObjectMap config; try { config = build(dir, template, new LinkedList<>()); } catch (SubCreatorException e) { config = null; } catch (Exception e) { config = null; log.logger.error.println(e); } declaration.run(); ReplacementScanner replacements = new ReplacementScanner(this.replacements); if (config != null) { try { if (template.getBuildOptions().getBoolean("Install-Client", true)) generateClient(dir, template.getType(), name); LinkedList masks = new LinkedList<>(); masks.add("/server.properties"); masks.addAll(template.getBuildOptions().getRawStringList("Replace", Collections.emptyList())); replacements.replace(dir, masks.toArray(new String[0])); } catch (Exception e) { config = null; e.printStackTrace(); } } if (config != null) { ((SubDataClient) SubAPI.getInstance().getSubDataNetwork()[0]).sendPacket(new PacketExCreateServer(0, null, (Map) replacements.replace(config.get()), tracker)); } else { log.logger.info.println("Couldn't build the server jar. Check the SubCreator logs for more detail."); ((SubDataClient) SubAPI.getInstance().getSubDataNetwork()[0]).sendPacket(new PacketExCreateServer(-1, "Couldn't build the server jar. Check the SubCreator logs for more detail.", tracker)); } SubCreatorImpl.this.thread.remove(name.toLowerCase()); } } /** * Creates a SubCreator Instance * * @param host SubServers.Host */ public SubCreatorImpl(ExHost host) { if (Util.isNull(host)) throw new NullPointerException(); this.host = host; this.thread = new TreeMap<>(); } public boolean create(UUID player, String name, ServerTemplate template, Version version, int port, UUID address, UUID tracker) { if (Util.isNull(name, template, port, address)) throw new NullPointerException(); CreatorTask task = new CreatorTask(player, name, template, version, port, address, tracker); this.thread.put(name.toLowerCase(), task); task.start(); return true; } public void terminate() { HashMap temp = new HashMap(); temp.putAll(thread); for (String i : temp.keySet()) { terminate(i); } } public void terminate(String name) { if (this.thread.keySet().contains(name.toLowerCase())) { if (this.thread.get(name.toLowerCase()).process != null && this.thread.get(name.toLowerCase()).process.isAlive()) { Executable.terminate(this.thread.get(name.toLowerCase()).process); } else if (this.thread.get(name.toLowerCase()).isAlive()) { this.thread.get(name.toLowerCase()).interrupt(); this.thread.remove(name.toLowerCase()); } } } public void waitFor() throws InterruptedException { HashMap temp = new HashMap(); temp.putAll(thread); for (String i : temp.keySet()) { waitFor(i); } } public void waitFor(String name) throws InterruptedException { while (this.thread.keySet().contains(name.toLowerCase()) && this.thread.get(name.toLowerCase()).isAlive()) { Thread.sleep(250); } } public List getLoggers() { List loggers = new ArrayList(); HashMap temp = new HashMap(); temp.putAll(thread); for (String i : temp.keySet()) { loggers.add(getLogger(i)); } return loggers; } public SubLoggerImpl getLogger(String name) { return this.thread.get(name).log; } private static NamedContainer address = null; private String getAddress() { if (address == null || host.config.get() != address.name()) { address = new NamedContainer<>(host.config.get(), host.config.get().getMap("Settings").getRawString("Server-Bind")); } return address.get(); } private static NamedContainer> subdata = null; private Map getSubData() { if (subdata == null || host.config.get() != subdata.name()) { Map map = new HashMap(); map.put("Address", host.config.get().getMap("Settings").getMap("SubData").getRawString("Address")); if (host.config.get().getMap("Settings").getMap("SubData").getRawString("Password", "").length() > 0) map.put("Password", host.config.get().getMap("Settings").getMap("SubData").getRawString("Password")); subdata = new NamedContainer<>(host.config.get(), map); } return subdata.get(); } private void generateClient(File dir, ServerType type, String name) throws IOException { boolean installed = false; if (type == ServerType.SPIGOT) { installed = true; if (!new UniversalFile(dir, "plugins").exists()) new UniversalFile(dir, "plugins").mkdirs(); if (!new UniversalFile(dir, "plugins:SubServers.Client.jar").exists()) Util.copyFromJar(ExHost.class.getClassLoader(), "net/ME1312/SubServers/Host/Library/Files/client.jar", new UniversalFile(dir, "plugins:SubServers.Client.jar").getPath()); } else if (type == ServerType.FORGE || type == ServerType.SPONGE) { installed = true; if (!new UniversalFile(dir, "mods").exists()) new UniversalFile(dir, "mods").mkdirs(); if (!new UniversalFile(dir, "mods:SubServers.Client.jar").exists()) Util.copyFromJar(ExHost.class.getClassLoader(), "net/ME1312/SubServers/Host/Library/Files/client.jar", new UniversalFile(dir, "mods:SubServers.Client.jar").getPath()); } if (installed) { YAMLSection config = new YAMLSection(); FileWriter writer = new FileWriter(new UniversalFile(dir, "subdata.json"), false); config.set("Name", name); config.setAll(getSubData()); writer.write(config.toJSON().toString()); writer.close(); if (!new UniversalFile(dir, "subdata.rsa.key").exists() && new UniversalFile("subdata.rsa.key").exists()) { Files.copy(new UniversalFile("subdata.rsa.key").toPath(), new UniversalFile(dir, "subdata.rsa.key").toPath()); } } } private void updateDirectory(File from, File to) { if (from.isDirectory() && !Files.isSymbolicLink(from.toPath())) { if (!to.exists()) { to.mkdirs(); } String files[] = from.list(); for (String file : files) { File srcFile = new File(from, file); File destFile = new File(to, file); updateDirectory(srcFile, destFile); } } else { try { if (!to.exists() || from.length() != to.length() || !Arrays.equals(generateSHA256(to), generateSHA256(from))) { if (to.exists()) { if (to.isDirectory()) Util.deleteDirectory(to); else to.delete(); } Files.copy(from.toPath(), to.toPath(), LinkOption.NOFOLLOW_LINKS, StandardCopyOption.REPLACE_EXISTING); } } catch (Exception e) { e.printStackTrace(); } } } private byte[] generateSHA256(File file) throws Exception { MessageDigest md = MessageDigest.getInstance("SHA-256"); FileInputStream fis = new FileInputStream(file); byte[] dataBytes = new byte[1024]; int nread; while ((nread = fis.read(dataBytes)) != -1) { md.update(dataBytes, 0, nread); } fis.close(); return md.digest(); } }