268 lines
10 KiB
Java
268 lines
10 KiB
Java
package net.ME1312.SubServers.Bungee.Host.External;
|
|
|
|
import net.ME1312.SubServers.Bungee.Event.SubAddServerEvent;
|
|
import net.ME1312.SubServers.Bungee.Event.SubRemoveServerEvent;
|
|
import net.ME1312.SubServers.Bungee.Host.Executable;
|
|
import net.ME1312.SubServers.Bungee.Host.Host;
|
|
import net.ME1312.SubServers.Bungee.Host.SubCreator;
|
|
import net.ME1312.SubServers.Bungee.Host.SubServer;
|
|
import net.ME1312.SubServers.Bungee.Library.Config.YAMLSection;
|
|
import net.ME1312.SubServers.Bungee.Library.Exception.InvalidServerException;
|
|
import net.ME1312.SubServers.Bungee.Library.NamedContainer;
|
|
import net.ME1312.SubServers.Bungee.Library.Util;
|
|
import net.ME1312.SubServers.Bungee.Network.Client;
|
|
import net.ME1312.SubServers.Bungee.Network.ClientHandler;
|
|
import net.ME1312.SubServers.Bungee.Network.Packet.PacketExAddServer;
|
|
import net.ME1312.SubServers.Bungee.Network.Packet.PacketExDeleteServer;
|
|
import net.ME1312.SubServers.Bungee.Network.Packet.PacketExRemoveServer;
|
|
import net.ME1312.SubServers.Bungee.Network.Packet.PacketOutReset;
|
|
import net.ME1312.SubServers.Bungee.Network.PacketOut;
|
|
import net.ME1312.SubServers.Bungee.SubPlugin;
|
|
import org.json.JSONObject;
|
|
|
|
import java.net.InetAddress;
|
|
import java.util.*;
|
|
|
|
/**
|
|
* External Host Class
|
|
*/
|
|
public class ExternalHost extends Host implements ClientHandler {
|
|
private HashMap<String, SubServer> servers = new HashMap<String, SubServer>();
|
|
private String name;
|
|
private boolean enabled;
|
|
private InetAddress address;
|
|
private SubCreator creator;
|
|
private String directory;
|
|
private NamedContainer<Boolean, Client> client;
|
|
private LinkedList<PacketOut> queue;
|
|
private boolean clean;
|
|
protected SubPlugin plugin;
|
|
|
|
/**
|
|
* Creates an External Host
|
|
*
|
|
* @param plugin Plugin
|
|
* @param name Name
|
|
* @param enabled Enabled Status
|
|
* @param address Address
|
|
* @param directory Directory
|
|
* @param gitBash Git Bash Location
|
|
*/
|
|
public ExternalHost(SubPlugin plugin, String name, Boolean enabled, InetAddress address, String directory, String gitBash) {
|
|
super(plugin, name, enabled, address, directory, gitBash);
|
|
if (Util.isNull(plugin, name, enabled, address, directory, gitBash)) throw new NullPointerException();
|
|
this.plugin = plugin;
|
|
this.name = name;
|
|
this.enabled = enabled;
|
|
this.address = address;
|
|
this.creator = new ExternalSubCreator(this, gitBash);
|
|
this.directory = directory;
|
|
this.client = new NamedContainer<Boolean, Client>(false, null);
|
|
this.queue = new LinkedList<PacketOut>();
|
|
this.clean = false;
|
|
}
|
|
|
|
@Override
|
|
public Client getSubData() {
|
|
return client.get();
|
|
}
|
|
|
|
@Override
|
|
public void setSubData(Client client) {
|
|
this.client = new NamedContainer<Boolean, Client>(false, client);
|
|
}
|
|
|
|
protected void queue(PacketOut... packet) {
|
|
for (PacketOut p : packet) if (client.get() == null || client.name() == false) {
|
|
queue.add(p);
|
|
} else {
|
|
client.get().sendPacket(p);
|
|
}
|
|
}
|
|
private void requeue() {
|
|
if (!clean) {
|
|
client.get().sendPacket(new PacketOutReset("Prevent Desync"));
|
|
clean = true;
|
|
}
|
|
for (SubServer server : servers.values()) {
|
|
client.get().sendPacket(new PacketExAddServer(server.getName(), server.isEnabled(), server.isLogging(), server.getPath(), ((ExternalSubServer) server).exec, server.getStopCommand(), (server.isRunning())?((ExternalSubLogger) server.getLogger()).getExternalAddress():null));
|
|
}
|
|
while (queue.size() != 0) {
|
|
client.get().sendPacket(queue.get(0));
|
|
queue.remove(0);
|
|
}
|
|
client.rename(true);
|
|
}
|
|
|
|
@Override
|
|
public boolean isEnabled() {
|
|
return enabled && this.client != null;
|
|
}
|
|
|
|
@Override
|
|
public void setEnabled(boolean value) {
|
|
this.enabled = value;
|
|
}
|
|
|
|
@Override
|
|
public InetAddress getAddress() {
|
|
return address;
|
|
}
|
|
|
|
@Override
|
|
public String getPath() {
|
|
return directory;
|
|
}
|
|
|
|
@Override
|
|
public String getName() {
|
|
return name;
|
|
}
|
|
|
|
@Override
|
|
public int edit(UUID player, YAMLSection edit) {
|
|
return 0; // TODO
|
|
}
|
|
|
|
@Override
|
|
public SubCreator getCreator() {
|
|
return creator;
|
|
}
|
|
|
|
@Override
|
|
public Map<String, ? extends SubServer> getSubServers() {
|
|
return new TreeMap<String, SubServer>(servers);
|
|
}
|
|
|
|
@Override
|
|
public SubServer getSubServer(String name) {
|
|
if (Util.isNull(name)) throw new NullPointerException();
|
|
return getSubServers().get(name.toLowerCase());
|
|
}
|
|
|
|
@Override
|
|
public SubServer addSubServer(UUID player, String name, boolean enabled, int port, String motd, boolean log, String directory, Executable executable, String stopcmd, boolean start, boolean restart, boolean hidden, boolean restricted, boolean temporary) throws InvalidServerException {
|
|
if (plugin.api.getServers().keySet().contains(name.toLowerCase())) throw new InvalidServerException("A Server already exists with this name!");
|
|
SubServer server = new ExternalSubServer(this, name, enabled, port, motd, log, directory, executable, stopcmd, restart, hidden, restricted);
|
|
SubAddServerEvent event = new SubAddServerEvent(player, this, server);
|
|
plugin.getPluginManager().callEvent(event);
|
|
if (!event.isCancelled()) {
|
|
queue(new PacketExAddServer(name, enabled, log, directory, executable, stopcmd, (server.isRunning())?((ExternalSubLogger) server.getLogger()).getExternalAddress():null, json -> {
|
|
if (json.getInt("r") == 0) {
|
|
if (!((start || temporary) && !server.start()) && temporary) server.setTemporary(true);
|
|
servers.put(name.toLowerCase(), server);
|
|
}
|
|
}));
|
|
return server;
|
|
} else {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public boolean removeSubServer(UUID player, String name) throws InterruptedException {
|
|
if (Util.isNull(name)) throw new NullPointerException();
|
|
SubRemoveServerEvent event = new SubRemoveServerEvent(player, this, getSubServer(name));
|
|
plugin.getPluginManager().callEvent(event);
|
|
if (!event.isCancelled()) {
|
|
if (getSubServer(name).isRunning()) {
|
|
getSubServer(name).stop();
|
|
getSubServer(name).waitFor();
|
|
}
|
|
queue(new PacketExRemoveServer(name, json -> {
|
|
if (json.getInt("r") == 0) servers.remove(name.toLowerCase());
|
|
}));
|
|
return true;
|
|
} else return false;
|
|
}
|
|
|
|
@Override
|
|
public boolean forceRemoveSubServer(UUID player, String name) {
|
|
if (Util.isNull(name)) throw new NullPointerException();
|
|
SubRemoveServerEvent event = new SubRemoveServerEvent(player, this, getSubServer(name));
|
|
plugin.getPluginManager().callEvent(event);
|
|
if (getSubServer(name).isRunning()) {
|
|
getSubServer(name).terminate();
|
|
}
|
|
queue(new PacketExRemoveServer(name, json -> {
|
|
if (json.getInt("r") == 0) servers.remove(name.toLowerCase());
|
|
}));
|
|
return true;
|
|
}
|
|
|
|
@Override
|
|
public boolean deleteSubServer(UUID player, String name) throws InterruptedException {
|
|
if (Util.isNull(name)) throw new NullPointerException();
|
|
String server = servers.get(name.toLowerCase()).getName();
|
|
|
|
SubRemoveServerEvent event = new SubRemoveServerEvent(player, this, getSubServer(server));
|
|
plugin.getPluginManager().callEvent(event);
|
|
if (!event.isCancelled()) {
|
|
if (getSubServer(server).isRunning()) {
|
|
getSubServer(server).stop();
|
|
getSubServer(server).waitFor();
|
|
}
|
|
|
|
System.out.println("SubServers > Saving...");
|
|
JSONObject info = (plugin.config.get().getSection("Servers").getKeys().contains(server))?plugin.config.get().getSection("Servers").getSection(server).toJSON():new JSONObject();
|
|
info.put("Name", server);
|
|
info.put("Timestamp", Calendar.getInstance().getTime().getTime());
|
|
try {
|
|
if (plugin.config.get().getSection("Servers").getKeys().contains(server)) {
|
|
plugin.config.get().getSection("Servers").remove(server);
|
|
plugin.config.save();
|
|
}
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
}
|
|
|
|
System.out.println("SubServers > Removing Files...");
|
|
queue(new PacketExDeleteServer(server, info, json -> {
|
|
if (json.getInt("r") == 0) {
|
|
servers.remove(server.toLowerCase());
|
|
System.out.println("SubServers > Done!");
|
|
} else {
|
|
System.out.println("SubServers > Couldn't remove server from memory. See " + getName() + " console for more details");
|
|
}
|
|
}));
|
|
return true;
|
|
} else return false;
|
|
}
|
|
|
|
@Override
|
|
public boolean forceDeleteSubServer(UUID player, String name) throws InterruptedException {
|
|
if (Util.isNull(name)) throw new NullPointerException();
|
|
String server = servers.get(name.toLowerCase()).getName();
|
|
|
|
SubRemoveServerEvent event = new SubRemoveServerEvent(player, this, getSubServer(server));
|
|
plugin.getPluginManager().callEvent(event);
|
|
if (getSubServer(server).isRunning()) {
|
|
getSubServer(server).terminate();
|
|
}
|
|
|
|
System.out.println("SubServers > Saving...");
|
|
JSONObject info = (plugin.config.get().getSection("Servers").getKeys().contains(server))?plugin.config.get().getSection("Servers").getSection(server).toJSON():new JSONObject();
|
|
info.put("Name", server);
|
|
info.put("Timestamp", Calendar.getInstance().getTime().getTime());
|
|
try {
|
|
if (plugin.config.get().getSection("Servers").getKeys().contains(server)) {
|
|
plugin.config.get().getSection("Servers").remove(server);
|
|
plugin.config.save();
|
|
}
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
}
|
|
|
|
System.out.println("SubServers > Removing Files...");
|
|
queue(new PacketExDeleteServer(server, info, json -> {
|
|
if (json.getInt("r") == 0) {
|
|
servers.remove(server.toLowerCase());
|
|
System.out.println("SubServers > Done!");
|
|
} else {
|
|
System.out.println("SubServers > Couldn't remove server from memory. See " + getName() + " console for more details");
|
|
}
|
|
}));
|
|
return true;
|
|
}
|
|
}
|