- cleanup pain

Signed-off-by: syeyoung <cyoung06@naver.com>

6 files changed, 155 insertions, 17 deletions

diff --git a/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/DungeonsGuide.java b/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/DungeonsGuide.java
index f45164e9..f3c600fc 100755
--- a/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/DungeonsGuide.java
+++ b/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/DungeonsGuide.java
@@ -28,6 +28,8 @@ import kr.syeyoung.dungeonsguide.mod.commands.CommandDungeonsGuide;
 import kr.syeyoung.dungeonsguide.mod.commands.CommandReparty;
 import kr.syeyoung.dungeonsguide.mod.config.Config;
 import kr.syeyoung.dungeonsguide.mod.cosmetics.CosmeticsManager;
+import kr.syeyoung.dungeonsguide.mod.cosmetics.CustomNetworkPlayerInfo;
+import kr.syeyoung.dungeonsguide.mod.discord.gamesdk.GameSDK;
 import kr.syeyoung.dungeonsguide.mod.discord.rpc.RichPresenceManager;
 import kr.syeyoung.dungeonsguide.mod.dungeon.DungeonFacade;
 import kr.syeyoung.dungeonsguide.mod.events.listener.DungeonListener;
@@ -44,11 +46,21 @@ import kr.syeyoung.dungeonsguide.mod.utils.cursor.GLCursors;
 import kr.syeyoung.dungeonsguide.mod.wsresource.StaticResourceCache;
 import lombok.Getter;
 import net.minecraft.client.Minecraft;
+import net.minecraft.client.entity.AbstractClientPlayer;
+import net.minecraft.client.entity.EntityPlayerSP;
 import net.minecraft.client.gui.*;
+import net.minecraft.client.network.NetHandlerPlayClient;
+import net.minecraft.client.network.NetworkPlayerInfo;
+import net.minecraft.client.renderer.entity.RenderManager;
 import net.minecraft.client.resources.IResourceManager;
 import net.minecraft.client.resources.IResourcePack;
+import net.minecraft.command.CommandHandler;
 import net.minecraft.command.ICommand;
+import net.minecraft.entity.player.EntityPlayer;
+import net.minecraft.entity.player.EntityPlayerMP;
 import net.minecraft.launchwrapper.LaunchClassLoader;
+import net.minecraft.network.play.server.S38PacketPlayerListItem;
+import net.minecraft.world.World;
 import net.minecraftforge.client.ClientCommandHandler;
 import net.minecraftforge.client.event.GuiOpenEvent;
 import net.minecraftforge.common.MinecraftForge;
@@ -60,9 +72,7 @@ import org.apache.logging.log4j.Logger;
 
 import java.io.File;
 import java.io.IOException;
-import java.util.ArrayList;
-import java.util.List;
-import java.util.Set;
+import java.util.*;
 import java.util.concurrent.ThreadFactory;
 import java.util.concurrent.atomic.AtomicInteger;
 
@@ -139,10 +149,8 @@ public class DungeonsGuide implements DGInterface {
     }
 
 
+    private PacketListener packetListener;
     public void init(File f) {
-        ClassLoader orignalLoader = Thread.currentThread().getContextClassLoader();
-
-        Thread.currentThread().setContextClassLoader(getClass().getClassLoader());
         ProgressManager.ProgressBar progressbar = ProgressManager.push("DungeonsGuide", 4);
 
         progressbar.step("Creating Configuration");
@@ -210,7 +218,7 @@ public class DungeonsGuide implements DGInterface {
         registerEventsForge(commandReparty = new CommandReparty());
 
         registerEventsForge(new FeatureListener());
-        registerEventsForge(new PacketListener());
+        registerEventsForge(packetListener = new PacketListener());
         registerEventsForge(new Keybinds());
 
         registerEventsForge(PartyManager.INSTANCE);
@@ -232,7 +240,7 @@ public class DungeonsGuide implements DGInterface {
         }
 
         if (FeatureRegistry.ETC_REPARTY.isEnabled()) {
-            ClientCommandHandler.instance.registerCommand(commandReparty);
+            registerCommands(commandReparty);
         }
 
         if (FeatureRegistry.DISCORD_DONOTUSE.isEnabled()) {
@@ -244,7 +252,29 @@ public class DungeonsGuide implements DGInterface {
 
         ProgressManager.pop(progressbar);
 
-        Thread.currentThread().setContextClassLoader(orignalLoader);
+    }
+
+    // hotswap fails in dev env due to intellij auto log collection or smth. it holds ref to stacktrace.
+
+    private void transform(AbstractClientPlayer abstractClientPlayer) {
+        NetworkPlayerInfo uuidNetworkPlayerInfoEntry = ReflectionHelper.getPrivateValue(AbstractClientPlayer.class,
+                abstractClientPlayer,
+                "playerInfo", "field_175157_a", "a"
+        );
+        if (uuidNetworkPlayerInfoEntry instanceof CustomNetworkPlayerInfo) {
+            S38PacketPlayerListItem s38PacketPlayerListItem = new S38PacketPlayerListItem();
+            NetworkPlayerInfo newInfo = new NetworkPlayerInfo(s38PacketPlayerListItem.new AddPlayerData(
+                    uuidNetworkPlayerInfoEntry.getGameProfile(),
+                    uuidNetworkPlayerInfoEntry.getResponseTime(),
+                    uuidNetworkPlayerInfoEntry.getGameType(),
+                    ((CustomNetworkPlayerInfo)uuidNetworkPlayerInfoEntry).getOriginalDisplayName()
+            ));
+            ReflectionHelper.setPrivateValue(AbstractClientPlayer.class,
+                    abstractClientPlayer,
+                    newInfo,
+                    "playerInfo", "field_175157_a", "a"
+            );
+        }
     }
 
     @Override
@@ -255,7 +285,7 @@ public class DungeonsGuide implements DGInterface {
         for (Object registeredListener : registeredListeners) {
             MinecraftForge.EVENT_BUS.unregister(registeredListener);
         }
-        Set<ICommand> commands = ReflectionHelper.getPrivateValue(ClientCommandHandler.class, ClientCommandHandler.instance, "commandSet");
+        Set<ICommand> commands = ReflectionHelper.getPrivateValue(CommandHandler.class, ClientCommandHandler.instance, "commandSet");
 
         for (ICommand registeredCommand : registeredCommands) {
             ClientCommandHandler.instance.getCommands().remove(registeredCommand.getCommandName());
@@ -264,13 +294,72 @@ public class DungeonsGuide implements DGInterface {
             }
             commands.remove(registeredCommand);
         }
+
+        if (packetListener != null) packetListener.cleanup();
+
+        try {
+            EntityPlayerSP ep = (EntityPlayerSP) Minecraft.getMinecraft().getRenderManager().livingPlayer;
+            transform(ep);
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+
+        try {
+            if (Minecraft.getMinecraft().pointedEntity instanceof AbstractClientPlayer) {
+                AbstractClientPlayer ep = (AbstractClientPlayer) Minecraft.getMinecraft().pointedEntity;
+                transform(ep);
+            }
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        NetHandlerPlayClient netHandlerPlayClient = Minecraft.getMinecraft().getNetHandler();
+        if (netHandlerPlayClient == null)
+            netHandlerPlayClient = ((EntityPlayerSP) Minecraft.getMinecraft().getRenderManager().livingPlayer).sendQueue;
+
+        if (netHandlerPlayClient != null) {
+            Map<UUID, NetworkPlayerInfo> playerInfoMap = ReflectionHelper.getPrivateValue(NetHandlerPlayClient.class,
+                    netHandlerPlayClient, "playerInfoMap", "field_147310_i", "i");
+            for (Map.Entry<UUID, NetworkPlayerInfo> uuidNetworkPlayerInfoEntry : playerInfoMap.entrySet()) {
+                if (uuidNetworkPlayerInfoEntry.getValue() instanceof CustomNetworkPlayerInfo) {
+                    S38PacketPlayerListItem s38PacketPlayerListItem = new S38PacketPlayerListItem();
+                    NetworkPlayerInfo newInfo =  new NetworkPlayerInfo(s38PacketPlayerListItem.new AddPlayerData(
+                            uuidNetworkPlayerInfoEntry.getValue().getGameProfile(),
+                            uuidNetworkPlayerInfoEntry.getValue().getResponseTime(),
+                            uuidNetworkPlayerInfoEntry.getValue().getGameType(),
+                            ((CustomNetworkPlayerInfo) uuidNetworkPlayerInfoEntry.getValue()).getOriginalDisplayName()
+                    ));
+                    playerInfoMap.put(uuidNetworkPlayerInfoEntry.getKey(), newInfo);
+                }
+            }
+        }
+
+        World world = Minecraft.getMinecraft().getRenderManager().worldObj;
+        if (world != null) {
+            for (AbstractClientPlayer entity : world.getEntities(AbstractClientPlayer.class, input -> true)) {
+                transform(entity);
+            }
+            for (AbstractClientPlayer player : world.getPlayers(AbstractClientPlayer.class, input -> true)) {
+                transform(player);
+            }
+        }
+
+        try {
+            List<IResourcePack> resourcePackList = ReflectionHelper.getPrivateValue(Minecraft.class, Minecraft.getMinecraft(), "defaultResourcePacks", "aA", "field_110449_ao");
+            resourcePackList.removeIf(a -> a instanceof DGTexturePack);
+            Minecraft.getMinecraft().refreshResources();
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+
+
         THREAD_GROUP.interrupt();
         THREAD_GROUP.stop();
         try {
-            Thread.sleep(1000); // This is requirement for all the threads to finish within 1 second. or reference leak.
+            Thread.sleep(2000); // This is requirement for all the threads to finish within 1 second. or reference leak.
         } catch (InterruptedException e) {
         }
         THREAD_GROUP.destroy();
+        GameSDK.cleanup();
     }
 
     @Override
diff --git a/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/commands/CommandDungeonsGuide.java b/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/commands/CommandDungeonsGuide.java
index 5c65424e..fa516bfd 100644
--- a/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/commands/CommandDungeonsGuide.java
+++ b/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/commands/CommandDungeonsGuide.java
@@ -32,8 +32,12 @@ import kr.syeyoung.dungeonsguide.mod.stomp.StompManager;
 import kr.syeyoung.dungeonsguide.mod.stomp.StompPayload;
 import kr.syeyoung.dungeonsguide.mod.wsresource.StaticResourceCache;
 import net.minecraft.client.Minecraft;
+import net.minecraft.client.gui.GuiMainMenu;
+import net.minecraft.client.gui.GuiMultiplayer;
+import net.minecraft.client.multiplayer.WorldClient;
 import net.minecraft.command.CommandBase;
 import net.minecraft.command.ICommandSender;
+import net.minecraft.realms.RealmsBridge;
 import net.minecraft.util.ChatComponentText;
 import net.minecraft.util.ChatStyle;
 import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
@@ -141,7 +145,21 @@ public class CommandDungeonsGuide extends CommandBase {
                 }
             }
         } else if (args[0].equals("reload")) {
-            Main.getMain().reloadWithoutStacktraceReference(Main.getMain().getCurrentLoader());
+            Minecraft.getMinecraft().addScheduledTask(() -> {
+                boolean flag = Minecraft.getMinecraft().isIntegratedServerRunning();
+                boolean flag1 = Minecraft.getMinecraft().isConnectedToRealms();
+                Minecraft.getMinecraft().theWorld.sendQuittingDisconnectingPacket();
+                Minecraft.getMinecraft().loadWorld((WorldClient)null);
+                if (flag) {
+                    Minecraft.getMinecraft().displayGuiScreen(new GuiMainMenu());
+                } else if (flag1) {
+                    RealmsBridge realmsbridge = new RealmsBridge();
+                    realmsbridge.switchToRealms(new GuiMainMenu());
+                } else {
+                    Minecraft.getMinecraft().displayGuiScreen(new GuiMultiplayer(new GuiMainMenu()));
+                }
+                Main.getMain().reloadWithoutStacktraceReference(Main.getMain().getCurrentLoader());
+            });
         } else {
                 sender.addChatMessage(new ChatComponentText("§eDungeons Guide §7:: §e/dg §7-§fOpens configuration gui"));
                 sender.addChatMessage(new ChatComponentText("§eDungeons Guide §7:: §e/dg gui §7-§fOpens configuration gui"));
diff --git a/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/cosmetics/CustomNetworkPlayerInfo.java b/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/cosmetics/CustomNetworkPlayerInfo.java
index 7be9a1ca..a54efaf4 100644
--- a/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/cosmetics/CustomNetworkPlayerInfo.java
+++ b/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/cosmetics/CustomNetworkPlayerInfo.java
@@ -18,17 +18,24 @@
 
 package kr.syeyoung.dungeonsguide.mod.cosmetics;
 
+import com.google.common.base.Objects;
 import com.mojang.authlib.GameProfile;
+import com.mojang.authlib.minecraft.MinecraftProfileTexture;
 import kr.syeyoung.dungeonsguide.mod.DungeonsGuide;
 import kr.syeyoung.dungeonsguide.mod.utils.TextUtils;
+import net.minecraft.client.Minecraft;
 import net.minecraft.client.network.NetworkPlayerInfo;
+import net.minecraft.client.resources.DefaultPlayerSkin;
+import net.minecraft.client.resources.SkinManager;
 import net.minecraft.network.play.server.S38PacketPlayerListItem;
 import net.minecraft.scoreboard.ScorePlayerTeam;
 import net.minecraft.util.ChatComponentText;
 import net.minecraft.util.IChatComponent;
+import net.minecraft.util.ResourceLocation;
 
 import java.util.List;
 
+// Reimplement this using ASM. Removing reference to this is too painful.
 public class CustomNetworkPlayerInfo extends NetworkPlayerInfo {
     public CustomNetworkPlayerInfo(GameProfile gameProfile) {
         super(gameProfile);
@@ -44,6 +51,10 @@ public class CustomNetworkPlayerInfo extends NetworkPlayerInfo {
     private String unformattedDisplayText;
     private String playerNameWithoutColor;
 
+    public IChatComponent getOriginalDisplayName() {
+        return displayName;
+    }
+
     @Override
     public void setDisplayName(IChatComponent displayNameIn) {
         displayName = displayNameIn;
@@ -66,6 +77,7 @@ public class CustomNetworkPlayerInfo extends NetworkPlayerInfo {
         }
     }
 
+
     public IChatComponent getDisplayName()
     {
 
diff --git a/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/discord/gamesdk/GameSDK.java b/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/discord/gamesdk/GameSDK.java
index f66043b5..cde3f7e2 100644
--- a/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/discord/gamesdk/GameSDK.java
+++ b/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/discord/gamesdk/GameSDK.java
@@ -18,12 +18,11 @@
 
 package kr.syeyoung.dungeonsguide.mod.discord.gamesdk;
 
-import com.sun.jna.Library;
-import com.sun.jna.Native;
-import com.sun.jna.Platform;
+import com.sun.jna.*;
 import kr.syeyoung.dungeonsguide.mod.discord.gamesdk.jna.GameSDKTypeMapper;
 import kr.syeyoung.dungeonsguide.mod.discord.gamesdk.jna.NativeGameSDK;
 import lombok.Getter;
+import net.minecraftforge.fml.relauncher.ReflectionHelper;
 
 import java.io.File;
 import java.io.IOException;
@@ -31,6 +30,7 @@ import java.io.InputStream;
 import java.nio.file.Files;
 import java.nio.file.StandardCopyOption;
 import java.util.Collections;
+import java.util.Map;
 
 public class GameSDK {
     @Getter
@@ -73,6 +73,20 @@ public class GameSDK {
                 Collections.singletonMap(Library.OPTION_TYPE_MAPPER, GameSDKTypeMapper.INSTANCE));
     }
 
+    public static void cleanup() {
+//        com.sun.jna.CallbackReference has reference of DiscordCallback. idk how i should approach fixing this -> I would better write native lib myself later.
+        if (System.getProperty("dg.safe") == null) return;
+
+        Map infos = ReflectionHelper.getPrivateValue(Structure.class, null, "layoutInfo");
+        infos.clear();
+        Map options = ReflectionHelper.getPrivateValue(Native.class, null, "options");
+        options.clear();
+        Map alignments = ReflectionHelper.getPrivateValue(Native.class, null, "alignments");
+        alignments.clear();
+        Map<Class, TypeMapper> typeMapperMap = ReflectionHelper.getPrivateValue(Native.class, null, "typeMappers");
+        typeMapperMap.clear();
+    }
+
     public static void writeString(byte[] bts, String str) {
         System.arraycopy(str.getBytes(), 0, bts, 0, str.getBytes().length);
         bts[str.getBytes().length] = 0;
diff --git a/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/events/listener/PacketListener.java b/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/events/listener/PacketListener.java
index acfffb7c..dcdc2c53 100644
--- a/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/events/listener/PacketListener.java
+++ b/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/events/listener/PacketListener.java
@@ -188,6 +188,11 @@ public class PacketListener extends ChannelDuplexHandler {
     }
 
     public void cleanup() {
-        thePipeline.remove("dg_packet_handler");
+        try {
+            if (thePipeline != null)
+                thePipeline.remove("dg_packet_handler");
+        } catch (Exception e) {
+
+        }
     }
 }
diff --git a/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/utils/AhUtils.java b/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/utils/AhUtils.java
index 62b11d3d..b28b97a5 100644
--- a/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/utils/AhUtils.java
+++ b/mod/src/main/java/kr/syeyoung/dungeonsguide/mod/utils/AhUtils.java
@@ -43,7 +43,7 @@ public class AhUtils {
 
     static Logger logger = LogManager.getLogger("AhUtils");
 
-    public static Timer timer = new Timer();
+//    public static Timer timer = new Timer();
 
     public static int totalAuctions = 0;