141 lines
4.9 KiB
Java
141 lines
4.9 KiB
Java
package me.mctp.socket;
|
|
|
|
import me.mctp.Main;
|
|
import me.mctp.managers.PinManager;
|
|
import org.bukkit.Bukkit;
|
|
import org.java_websocket.client.WebSocketClient;
|
|
import org.java_websocket.handshake.ServerHandshake;
|
|
import org.json.simple.JSONObject;
|
|
|
|
import java.net.URI;
|
|
import java.net.URISyntaxException;
|
|
import java.util.UUID;
|
|
|
|
public class Client {
|
|
private String url;
|
|
private int taskID = 0;
|
|
private int controlID = 0;
|
|
private WebSocketClient wsc;
|
|
private boolean connected = false;
|
|
|
|
public Client(String url) {
|
|
this.url = url;
|
|
}
|
|
|
|
public void connect() {
|
|
if (!this.connected) {
|
|
this.connected = true;
|
|
this.controlID = Bukkit.getScheduler().runTaskTimer(Main.getPlugin(), () -> {
|
|
if (!this.connected) {
|
|
Bukkit.getScheduler().cancelTask(this.controlID);
|
|
this.controlID = 0;
|
|
} else if (this.wsc == null || !this.wsc.isOpen()) {
|
|
if (this.wsc != null) {
|
|
this.wsc.closeConnection(404, "Disconnected from socket");
|
|
this.wsc = null;
|
|
}
|
|
|
|
this.connect();
|
|
}
|
|
}, 600L, 600L).getTaskId();
|
|
}
|
|
|
|
if (this.url != null && this.wsc == null) {
|
|
try {
|
|
URI uri = new URI(this.url + "?type=SERVER&");
|
|
|
|
this.wsc = new WebSocketClient(uri) {
|
|
@Override
|
|
public void onOpen(ServerHandshake serverHandshake) {}
|
|
|
|
@Override
|
|
public void onMessage(String s) {
|
|
JSONObject json = JSONUtil.parse(s);
|
|
if (json == null) return;
|
|
|
|
String str = JSONUtil.getValue(json, "task");
|
|
if (str == null || str.isEmpty()) return;
|
|
|
|
if (str.equals("VERIFY")) {
|
|
String uuid = JSONUtil.getValue(json, "uuid");
|
|
if (uuid == null || uuid.isEmpty()) return;
|
|
|
|
String pin = JSONUtil.getValue(json, "pin");
|
|
if (pin == null || pin.isEmpty()) return;
|
|
|
|
UUID pUUID = JSONUtil.formatFromInput(uuid);
|
|
boolean verified = false;
|
|
if (Bukkit.getPlayer(pUUID) != null) {
|
|
verified = PinManager.checkPin(pUUID, pin);
|
|
}
|
|
|
|
JSONObject reply = new JSONObject();
|
|
reply.put("task", "AUTHENTICATION");
|
|
reply.put("verified", verified);
|
|
reply.put("uuid", uuid);
|
|
this.send(reply.toJSONString());
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void onClose(int i, String s, boolean b) {
|
|
Client.this.wsc = null;
|
|
Bukkit.getScheduler().cancelTask(Client.this.taskID);
|
|
Client.this.taskID = 0;
|
|
}
|
|
|
|
@Override
|
|
public void onError(Exception e) {
|
|
e.printStackTrace();
|
|
}
|
|
};
|
|
|
|
this.wsc.connect();
|
|
|
|
if (this.taskID == 0) {
|
|
this.taskID = Bukkit.getScheduler().runTaskTimerAsynchronously(Main.getPlugin(), () -> {
|
|
if (Client.this.wsc != null && Client.this.wsc.isOpen()) {
|
|
Client.this.wsc.send("__PING__");
|
|
} else {
|
|
if (Client.this.wsc != null) {
|
|
Client.this.wsc.closeConnection(404, "Disconnected from socket");
|
|
Client.this.wsc = null;
|
|
}
|
|
|
|
Bukkit.getScheduler().cancelTask(Client.this.taskID);
|
|
Client.this.taskID = 0;
|
|
}
|
|
}, 200L, 200L).getTaskId();
|
|
}
|
|
} catch (URISyntaxException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
}
|
|
|
|
public void disconnect() {
|
|
if (this.wsc != null) {
|
|
this.wsc.closeConnection(404, "Disconnected from socket");
|
|
}
|
|
|
|
this.wsc = null;
|
|
|
|
if (this.taskID != 0) {
|
|
Bukkit.getScheduler().cancelTask(this.taskID);
|
|
this.taskID = 0;
|
|
}
|
|
|
|
this.connected = false;
|
|
|
|
if (this.controlID != 0) {
|
|
Bukkit.getScheduler().cancelTask(this.controlID);
|
|
this.controlID = 0;
|
|
}
|
|
}
|
|
|
|
public void sendData(JSONObject object) {
|
|
if (this.wsc != null && this.wsc.isOpen() && object != null && object.toJSONString() != null) {
|
|
this.wsc.send(object.toJSONString());
|
|
}
|
|
}
|
|
}
|