ssl better

1 files changed, 219 insertions, 0 deletions

diff --git a/src/main/java/kr/syeyoung/dungeonsguide/party/PartyManager.java b/src/main/java/kr/syeyoung/dungeonsguide/party/PartyManager.java
new file mode 100644
index 00000000..08492ae8
--- /dev/null
+++ b/src/main/java/kr/syeyoung/dungeonsguide/party/PartyManager.java
@@ -0,0 +1,219 @@
+package kr.syeyoung.dungeonsguide.party;
+
+import kr.syeyoung.dungeonsguide.RichPresenceManager;
+import kr.syeyoung.dungeonsguide.commands.CommandReparty;
+import kr.syeyoung.dungeonsguide.e;
+import kr.syeyoung.dungeonsguide.events.SkyblockJoinedEvent;
+import kr.syeyoung.dungeonsguide.events.StompConnectedEvent;
+import kr.syeyoung.dungeonsguide.stomp.StompInterface;
+import kr.syeyoung.dungeonsguide.stomp.StompMessageHandler;
+import kr.syeyoung.dungeonsguide.stomp.StompPayload;
+import kr.syeyoung.dungeonsguide.stomp.StompSubscription;
+import kr.syeyoung.dungeonsguide.utils.TextUtils;
+import lombok.Getter;
+import net.minecraft.client.Minecraft;
+import net.minecraftforge.client.event.ClientChatReceivedEvent;
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
+import org.json.JSONArray;
+import org.json.JSONObject;
+import scala.util.parsing.json.JSON;
+
+import java.security.SecureRandom;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Map;
+import java.util.Set;
+
+public class PartyManager implements StompMessageHandler {
+    public static final PartyManager INSTANCE = new PartyManager();
+
+    @Getter
+    private String partyID = null;
+    @Getter
+    private String askToJoinSecret = null;
+
+    private SecureRandom random = new SecureRandom();
+    private static final String validChars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-_";
+
+    @Getter
+    private boolean allowAskToJoin = false;
+    @Getter
+    private boolean canInvite = false;
+
+    public void toggleAllowAskToJoin() {
+        if (canInvite) allowAskToJoin = !allowAskToJoin;
+        if (allowAskToJoin) {
+            generateNewAskToJoinSecret();
+        }
+    }
+
+    public void setPartyID(String partyID) {
+        if (this.partyID != null && partyID == null) {
+            JSONObject object = new JSONObject();
+            object.put("members", new JSONArray());
+            StompInterface stompInterface = e.getDungeonsGuide().getStompConnection();
+            stompInterface.send(new StompPayload().payload(object.toString()).header("destination", "/app/party.join"));
+        }
+        this.partyID = partyID;
+        this.askToJoinSecret = null;
+
+        if (partyID != null && !partyID.equals(this.partyID)) {
+            Minecraft.getMinecraft().thePlayer.sendChatMessage("/p invite -");
+        } else {
+            canInvite = true;
+            allowAskToJoin = false;
+        }
+
+        if (allowAskToJoin) {
+            generateNewAskToJoinSecret();
+        } else {
+            RichPresenceManager.INSTANCE.updatePresence();
+        }
+    }
+
+    public void generateNewAskToJoinSecret() {
+        if (partyID == null) {
+            JSONObject object = new JSONObject();
+            object.put("members", new JSONArray());
+            StompInterface stompInterface = e.getDungeonsGuide().getStompConnection();
+            stompInterface.send(new StompPayload().payload(object.toString()).header("destination", "/app/party.join"));
+        }
+
+        StringBuilder secretBuilder = new StringBuilder();
+        for (int i = 0; i < 20; i++) {
+            secretBuilder.append(validChars.charAt(random.nextInt(validChars.length())));
+        }
+        this.askToJoinSecret = secretBuilder.toString();
+
+        StompInterface stompInterface = e.getDungeonsGuide().getStompConnection();
+        stompInterface.send(new StompPayload().payload(new JSONObject().put("secret", askToJoinSecret).toString()).header("destination", "/app/party.setjoinsecret"));
+        RichPresenceManager.INSTANCE.updatePresence();
+    }
+
+    private boolean partyJoin = false;
+    private Set<String> members = new HashSet<>();
+    private Map<String, Long> recentlyJoined = new HashMap<>();
+    @SubscribeEvent
+    public void onMessage(ClientChatReceivedEvent chatReceivedEvent) {
+        if (chatReceivedEvent.type == 2) return;
+
+        String str = chatReceivedEvent.message.getFormattedText();
+
+        if (str.startsWith("§eYou have joined ")) {
+            setPartyID(null);
+            members.clear();
+            String strs[] = TextUtils.stripColor(str).split(" ");
+            for (String s : strs) {
+                if (s.endsWith("'s")) {
+                    members.add(s.substring(0, s.indexOf("'s")));
+                    partyJoin = true;
+                    break;
+                }
+            }
+        } else if (str.startsWith("§eYou'll be partying with: ")) {
+            String[] players = TextUtils.stripColor(str.substring(27)).split(" ");
+            for (String player : players) {
+                if (player.startsWith("[")) continue;
+                members.add(player);
+            }
+        } else if (str.equals("§9§m-----------------------------§r")) {
+            if (partyJoin) {
+                partyJoin = false;
+                // REQ PARTY JOIN
+
+                JSONArray jsonArray = new JSONArray();
+                for (String member : members) {
+                    jsonArray.put(member);
+                }
+                JSONObject object = new JSONObject();
+                object.put("members", jsonArray);
+                StompInterface stompInterface = e.getDungeonsGuide().getStompConnection();
+                stompInterface.send(new StompPayload().payload(object.toString()).header("destination", "/app/party.join"));
+            }
+        } else if (str.endsWith("§ejoined the party.§r")) {
+            String asd = null;
+            for (String s : TextUtils.stripColor(str).split(" ")) {
+                if (s.startsWith("[")) continue;
+                asd = s;
+            }
+            if (asd != null)
+                recentlyJoined.put(asd, System.currentTimeMillis());
+        } else if ((str.equals("§eYou left the party.§r")
+        || str.equals("§cThe party was disbanded because all invites expired and the party was empty§r"))
+        ){
+            setPartyID(null);
+        } else if (str.startsWith("§6Party Members ")) {
+            partyJoin = true;
+            members.clear();
+        } else if (str.startsWith("§cYou are not currently in a party.§r")) {
+            members.clear();
+            setPartyID(null);
+        } else if (str.startsWith("§eParty ") && str.contains(":")) {
+            String playerNames = TextUtils.stripColor(str.split(":")[1]);
+            for (String s : playerNames.split(" ")) {
+                if (s.isEmpty()) continue;
+                if (s.equals("●")) continue;
+                if (s.startsWith("[")) continue;
+                members.add(s);
+            }
+        } else if (str.equals("§cYou are not allowed to invite players.§r")) {
+            canInvite = false;
+            allowAskToJoin = false;
+            askToJoinSecret = "";
+            RichPresenceManager.INSTANCE.updatePresence();
+        } else if (str.equals("§cCouldn't find a player with that name!§r")) {
+            canInvite = true;
+        } else if (str.endsWith("§aenabled All Invite§r")) {
+            canInvite = true;
+        } else if (str.endsWith("§cdisabled All Invite§r")) {
+            canInvite = false;
+            allowAskToJoin = false;
+            askToJoinSecret = "";
+            RichPresenceManager.INSTANCE.updatePresence();
+            Minecraft.getMinecraft().thePlayer.sendChatMessage("/p invite -");
+        }
+    }
+
+    @SubscribeEvent
+    public void onSBJoin(SkyblockJoinedEvent skyblockJoinedEvent) {
+        Minecraft.getMinecraft().thePlayer.sendChatMessage("/pl");
+    }
+
+    @Override
+    public void handle(StompInterface stompInterface, StompPayload stompPayload) {
+        JSONObject object = new JSONObject(stompPayload.payload());
+        if ("/queue/party.check".equals(stompPayload.headers().get("destination"))) {
+            String playerName = object.getString("player");
+            String token = object.getString("token");
+            Long loong = recentlyJoined.get(playerName);
+            if (loong == null) {
+                stompInterface.send(new StompPayload().payload(new JSONObject().put("status", "failure").put("token", token).toString()).header("destination", "/app/party.check.resp"));
+            } else if (loong > System.currentTimeMillis() - 2000){
+                stompInterface.send(new StompPayload().payload(new JSONObject().put("status", "success").put("token", token).toString()).header("destination", "/app/party.check.resp"));
+            }
+        } else if ("/queue/party.join".equals(stompPayload.headers().get("destination"))) {
+            String playerName = object.getString("player");
+            String secret = object.getString("secret");
+            if (secret.equals(askToJoinSecret) && partyID != null) {
+                Minecraft.getMinecraft().thePlayer.sendChatMessage("/p invite "+playerName);
+            }
+        } else {
+            String str = object.getString("status");
+            if ("success".equals(str)) {
+                setPartyID(object.getString("partyId"));
+            } else {
+                setPartyID(null);
+            }
+        }
+    }
+
+    @SubscribeEvent
+    public void stompConnect(StompConnectedEvent stompConnectedEvent) {
+        stompConnectedEvent.getStompInterface().subscribe(StompSubscription.builder()
+        .stompMessageHandler(this).ackMode(StompSubscription.AckMode.AUTO).destination("/user/queue/party.resp").build());
+        stompConnectedEvent.getStompInterface().subscribe(StompSubscription.builder()
+                .stompMessageHandler(this).ackMode(StompSubscription.AckMode.AUTO).destination("/user/queue/party.check").build());
+        stompConnectedEvent.getStompInterface().subscribe(StompSubscription.builder()
+                .stompMessageHandler(this).ackMode(StompSubscription.AckMode.AUTO).destination("/user/queue/party.join").build());
+    }
+}