1 files changed, 254 insertions, 0 deletions
diff --git a/mod/src/main/java/kr/syeyoung/dungeonsguide/rpc/RichPresenceManager.java b/mod/src/main/java/kr/syeyoung/dungeonsguide/rpc/RichPresenceManager.java
new file mode 100644
index 00000000..ef794b14
--- /dev/null
+++ b/mod/src/main/java/kr/syeyoung/dungeonsguide/rpc/RichPresenceManager.java
@@ -0,0 +1,254 @@
+/*
+ * Dungeons Guide - The most intelligent Hypixel Skyblock Dungeons Mod
+ * Copyright (C) 2021  cyoung06
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published
+ * by the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package kr.syeyoung.dungeonsguide.rpc;
+
+import com.sun.jna.Pointer;
+import com.sun.jna.ptr.IntByReference;
+import com.sun.jna.ptr.PointerByReference;
+import kr.syeyoung.dungeonsguide.DungeonsGuide;
+import kr.syeyoung.dungeonsguide.SkyblockStatus;
+import kr.syeyoung.dungeonsguide.chat.PartyContext;
+import kr.syeyoung.dungeonsguide.chat.PartyManager;
+import kr.syeyoung.dungeonsguide.dungeon.DungeonContext;
+import kr.syeyoung.dungeonsguide.events.DiscordUserJoinRequestEvent;
+import kr.syeyoung.dungeonsguide.events.DiscordUserUpdateEvent;
+import kr.syeyoung.dungeonsguide.features.FeatureRegistry;
+import kr.syeyoung.dungeonsguide.gamesdk.GameSDK;
+import kr.syeyoung.dungeonsguide.gamesdk.jna.NativeGameSDK;
+import kr.syeyoung.dungeonsguide.gamesdk.jna.datastruct.*;
+import kr.syeyoung.dungeonsguide.gamesdk.jna.enumuration.EDiscordActivityJoinRequestReply;
+import kr.syeyoung.dungeonsguide.gamesdk.jna.enumuration.EDiscordCreateFlags;
+import kr.syeyoung.dungeonsguide.gamesdk.jna.enumuration.EDiscordLogLevel;
+import kr.syeyoung.dungeonsguide.gamesdk.jna.enumuration.EDiscordResult;
+import kr.syeyoung.dungeonsguide.gamesdk.jna.interfacestruct.*;
+import kr.syeyoung.dungeonsguide.gamesdk.jna.typedef.*;
+import lombok.Getter;
+import net.minecraftforge.common.MinecraftForge;
+
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Optional;
+import java.util.Set;
+
+public class RichPresenceManager implements Runnable {
+    public static RichPresenceManager INSTANCE = new RichPresenceManager();
+    private Thread t = new Thread(this);
+
+    public RichPresenceManager() {
+        t.start();
+    }
+    @Getter
+    private NativeGameSDK nativeGameSDK;
+    @Getter
+    private IDiscordCore iDiscordCore;
+    private IDiscordActivityManager activityManager;
+    private IDiscordRelationshipManager relationshipManager;
+    private IDiscordActivityEvents.ByReference callbacks;
+    private IDiscordRelationshipEvents.ByReference relation_callbacks;
+
+    @Getter
+    private Map<Long, JDiscordRelation> relationMap = new HashMap<>();
+
+    private boolean ready = false;
+
+    @Getter
+    private int lastSetupCode = -99999;
+    public int setup() {
+        ready = false;
+        if (iDiscordCore != null) {
+            iDiscordCore.Destroy.destroy(iDiscordCore);
+            iDiscordCore = null;
+            activityManager = null; callbacks = null; relation_callbacks = null; relationMap.clear();
+        }
+
+        nativeGameSDK = GameSDK.getNativeGameSDK();
+        if (nativeGameSDK == null) return -9999;
+        DiscordCreateParams discordCreateParams = new DiscordCreateParams();
+        discordCreateParams.flags = new UInt64(EDiscordCreateFlags.DiscordCreateFlags_NoRequireDiscord.getValue());
+        discordCreateParams.client_id = new DiscordClientID(816298079732498473L);
+
+        callbacks = new IDiscordActivityEvents.ByReference();
+        callbacks.OnActivityInvite = (eventData, type, user, activity) -> {
+            try {
+                MinecraftForge.EVENT_BUS.post(new DiscordUserJoinRequestEvent(user, true));
+            } catch (Throwable t) {
+                t.printStackTrace();
+            }
+            System.out.println("Received Join Request from "+user.id.longValue()+" ("+GameSDK.readString(user.username)+")");
+        };
+        callbacks.OnActivityJoin = (eventData, secret) -> {
+            PartyManager.INSTANCE.joinWithToken(secret);
+            System.out.println("Trying to join with token "+secret);
+        };
+        callbacks.OnActivityJoinRequest = (eventData, user) -> {
+            try {
+                MinecraftForge.EVENT_BUS.post(new DiscordUserJoinRequestEvent(user, false));
+            } catch (Throwable t) {
+                t.printStackTrace();
+            }
+                System.out.println("Received Join Request from "+user.id.longValue()+" - "+GameSDK.readString(user.username));
+        };
+        callbacks.OnActivitySpectate = (eventData, secret) -> {
+        };
+        callbacks.write();
+
+        relation_callbacks = new IDiscordRelationshipEvents.ByReference();
+        relation_callbacks.OnRefresh = (p) -> {
+            try {
+                ready = true;
+                IDiscordRelationshipManager iDiscordRelationshipManager = iDiscordCore.GetRelationshipManager.getRelationshipManager(iDiscordCore);
+                iDiscordRelationshipManager.Filter.filter(iDiscordRelationshipManager, Pointer.NULL, (d, relation) -> true);
+                IntByReference intByReference = new IntByReference();
+                iDiscordRelationshipManager.Count.count(iDiscordRelationshipManager, intByReference);
+                int count = intByReference.getValue();
+                relationMap.clear();
+                for (int i = 0; i < count; i++) {
+                    DiscordRelationship discordRelationship = new DiscordRelationship();
+                    iDiscordRelationshipManager.GetAt.getAt(iDiscordRelationshipManager, new UInt32(i), discordRelationship);
+
+                    JDiscordRelation jDiscordRelation = JDiscordRelation.fromJNA(discordRelationship);
+                    relationMap.put(jDiscordRelation.getDiscordUser().getId(), jDiscordRelation);
+                }
+            } catch (Throwable e) {e.printStackTrace();}
+        };
+        relation_callbacks.OnRelationshipUpdate = (p, rel) -> {
+            try {
+                JDiscordRelation jDiscordRelation = JDiscordRelation.fromJNA(rel);
+                JDiscordRelation prev = relationMap.put(jDiscordRelation.getDiscordUser().getId(), jDiscordRelation);
+                MinecraftForge.EVENT_BUS.post(new DiscordUserUpdateEvent(prev, jDiscordRelation));
+            } catch (Throwable t) {
+                t.printStackTrace();
+            }
+        };
+        relation_callbacks.write();
+
+        discordCreateParams.activity_events = callbacks;
+        discordCreateParams.relationship_events = relation_callbacks; // 96
+
+        PointerByReference pointerByReference = new PointerByReference();
+        discordCreateParams.write();
+
+        EDiscordResult eDiscordResult = nativeGameSDK.DiscordCreate(new DiscordVersion(NativeGameSDK.DISCORD_VERSION), discordCreateParams, pointerByReference);
+        if (eDiscordResult != EDiscordResult.DiscordResult_Ok) return eDiscordResult.getValue();
+        if (pointerByReference.getValue() == Pointer.NULL) return -9998;
+        iDiscordCore = new IDiscordCore(pointerByReference.getValue());
+
+        iDiscordCore.SetLogHook.setLogHook(iDiscordCore, EDiscordLogLevel.DiscordLogLevel_Debug, Pointer.NULL, new IDiscordCore.LogHook() {
+            @Override
+            public void hook(Pointer hookData, EDiscordLogLevel level, String message) {
+                System.out.println(message+" - "+level+" - "+hookData);
+            }
+        });
+
+        activityManager = iDiscordCore.GetActivityManager.getActivityManager(iDiscordCore);
+        relationshipManager = iDiscordCore.GetRelationshipManager.getRelationshipManager(iDiscordCore);
+
+        return eDiscordResult.getValue();
+    }
+
+    private final SkyblockStatus skyblockStatus = DungeonsGuide.getDungeonsGuide().getSkyblockStatus();
+
+    public void respond(DiscordSnowflake userID, EDiscordActivityJoinRequestReply reply) {
+        if (activityManager == null) return;
+        activityManager.SendRequestReply.sendRequestReply(activityManager, userID, reply, Pointer.NULL, (callbackData, result) -> {
+            System.out.println("Discord Returned "+result+" For Replying "+reply+" To "+userID.longValue()+"L");
+        });
+    }
+
+    public void accept(DiscordSnowflake userID) {
+        if (activityManager == null) return;
+        activityManager.AcceptInvite.acceptInvite(activityManager, userID, Pointer.NULL, (callbackData, result) -> {
+            System.out.println("Discord Returned "+result+" For Accepting invite from "+userID.longValue()+"L");
+        });
+    }
+    public void updatePresence() {
+        if (!skyblockStatus.isOnHypixel() || !FeatureRegistry.DISCORD_RICHPRESENCE.isEnabled() || (!skyblockStatus.isOnSkyblock() && FeatureRegistry.DISCORD_RICHPRESENCE.<Boolean>getParameter("disablenotskyblock").getValue())) {
+            activityManager.ClearActivity.clearActivity(activityManager, Pointer.NULL, (callbackData, result) -> {
+            });
+        } else {
+            String name = skyblockStatus.getDungeonName() == null ? "" : skyblockStatus.getDungeonName();
+            if (!skyblockStatus.isOnSkyblock()) name ="Somewhere on Hypixel";
+            if (name.trim().equals("Your Island")) name = "Private Island";
+
+            DiscordActivity latestDiscordActivity = new DiscordActivity();
+            latestDiscordActivity.assets = new DiscordActivityAssets();
+            latestDiscordActivity.secrets = new DiscordActivitySecrets();
+            latestDiscordActivity.party = new DiscordActivityParty();
+            latestDiscordActivity.party.discordActivityParty = new DiscordPartySize();
+            latestDiscordActivity.timestamps = new DiscordActivityTimestamps();
+            GameSDK.writeString(latestDiscordActivity.assets.large_image, "mort");
+            GameSDK.writeString(latestDiscordActivity.assets.large_text, "mort");
+            GameSDK.writeString(latestDiscordActivity.state, name);
+
+                GameSDK.writeString(latestDiscordActivity.party.id, Optional.ofNullable( PartyManager.INSTANCE.getPartyContext()).map(PartyContext::getPartyID).orElse(""));
+                latestDiscordActivity.party.discordActivityParty.current_size = new Int32(Optional.ofNullable(PartyManager.INSTANCE.getPartyContext()).map(PartyContext::getPartyRawMembers).map(Set::size).orElse(1));
+                latestDiscordActivity.party.discordActivityParty.max_size = new Int32(PartyManager.INSTANCE.getMaxParty());
+
+            if (skyblockStatus.getContext() != null) {
+                DungeonContext dungeonContext = skyblockStatus.getContext();
+                long init = dungeonContext.getInit();
+                latestDiscordActivity.timestamps.start = new DiscordTimestamp(init);
+
+                if (dungeonContext.getBossfightProcessor() != null) {
+                    GameSDK.writeString(latestDiscordActivity.details, "Fighting "+dungeonContext.getBossfightProcessor().getBossName()+": "+dungeonContext.getBossfightProcessor().getCurrentPhase());
+                } else {
+                    GameSDK.writeString(latestDiscordActivity.details, "Clearing Rooms");
+                }
+            } else {
+                latestDiscordActivity.timestamps.start = new DiscordTimestamp(0);
+                GameSDK.writeString(latestDiscordActivity.details, "Dungeons Guide");
+            }
+            if (PartyManager.INSTANCE.getAskToJoinSecret() != null) {
+                GameSDK.writeString(latestDiscordActivity.secrets.join, PartyManager.INSTANCE.getAskToJoinSecret());
+            } else {
+                GameSDK.writeString(latestDiscordActivity.secrets.join, "");
+            }
+            activityManager.UpdateActivity.updateActivity(activityManager, latestDiscordActivity, Pointer.NULL, (callbackData, result) -> {
+            });
+        }
+    }
+
+    @Override
+    public void run() {
+        boolean setup = true;
+        int counter = 0;
+        while(!Thread.interrupted()) {
+            try {
+                if (iDiscordCore == null || setup) {
+                    long lastSetup = lastSetupCode;
+                    lastSetupCode = setup();
+                    if (lastSetup != lastSetupCode)
+                        System.out.println("Discord returned "+lastSetupCode+" for setup "+EDiscordResult.fromValue(lastSetupCode));
+                    setup = lastSetupCode != EDiscordResult.DiscordResult_Ok.getValue();
+                    counter = 0;
+                } else {
+                    EDiscordResult eDiscordResult = iDiscordCore.RunCallbacks.runCallbacks(iDiscordCore);
+                    if (eDiscordResult != EDiscordResult.DiscordResult_Ok) {
+                        setup = true;
+                    } else if (ready){
+                        if (counter == 0)
+                            updatePresence();
+                        if (++counter == 15) counter = 0;
+                    }
+                }
+                Thread.sleep(16L);
+            } catch (Exception e) {e.printStackTrace();}
+        }
+    }
+}