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());
        }
    }
}