1 files changed, 78 insertions, 0 deletions
diff --git a/src/main/java/cc/woverflow/chatting/mixin/GuiNewChatMapMixin.java b/src/main/java/cc/woverflow/chatting/mixin/GuiNewChatMapMixin.java
new file mode 100644
index 0000000..5b38bda
--- /dev/null
+++ b/src/main/java/cc/woverflow/chatting/mixin/GuiNewChatMapMixin.java
@@ -0,0 +1,78 @@
+package cc.woverflow.chatting.mixin;
+
+import cc.woverflow.chatting.hook.GuiNewChatHook;
+import net.minecraft.client.gui.ChatLine;
+import net.minecraft.client.gui.GuiNewChat;
+import net.minecraft.util.IChatComponent;
+import org.spongepowered.asm.mixin.Final;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.Unique;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+import org.spongepowered.asm.mixin.injection.callback.LocalCapture;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+@Mixin(value = GuiNewChat.class, priority = Integer.MIN_VALUE)
+public abstract class GuiNewChatMapMixin implements GuiNewChatHook {
+
+    @Unique private final Map<ChatLine, ChatLine> drawnToFull = new HashMap<>();
+    @Unique private final List<ChatLine> tempDrawnLines = new ArrayList<>();
+    @Unique private ChatLine lastTempLine = null;
+
+    @Shadow @Final private List<ChatLine> drawnChatLines;
+    @Shadow @Final private List<ChatLine> chatLines;
+
+    @Inject(method = "setChatLine", at = @At("HEAD"))
+    private void handleSetChatLine(IChatComponent chatComponent, int chatLineId, int updateCounter, boolean displayOnly, CallbackInfo ci) {
+        tempDrawnLines.clear();
+    }
+
+    @Inject(method = "setChatLine", at = @At(value = "INVOKE", target = "Ljava/util/List;add(ILjava/lang/Object;)V", ordinal = 0, shift = At.Shift.AFTER, remap = false))
+    private void handleDrawnLineAdded(IChatComponent chatComponent, int chatLineId, int updateCounter, boolean displayOnly, CallbackInfo ci) {
+        if (!displayOnly) tempDrawnLines.add(drawnChatLines.get(0));
+        else if (lastTempLine != null) {
+            drawnToFull.put(drawnChatLines.get(0), lastTempLine);
+        }
+    }
+
+    @Inject(method = "setChatLine", at = @At(value = "INVOKE", target = "Ljava/util/List;remove(I)Ljava/lang/Object;", ordinal = 0, shift = At.Shift.BEFORE, remap = false))
+    private void handleDrawnLineRemoved(IChatComponent chatComponent, int chatLineId, int updateCounter, boolean displayOnly, CallbackInfo ci) {
+        drawnToFull.remove(drawnChatLines.get(drawnChatLines.size() - 1));
+    }
+
+    @Inject(method = "setChatLine", at = @At("RETURN"))
+    private void handleLineAdded(IChatComponent chatComponent, int chatLineId, int updateCounter, boolean displayOnly, CallbackInfo ci) {
+        if (!displayOnly) {
+            ChatLine masterLine = chatLines.get(0);
+            for (ChatLine tempDrawnLine : tempDrawnLines) drawnToFull.put(tempDrawnLine, masterLine);
+        }else {
+            lastTempLine = null;
+        }
+    }
+
+    @Inject(method = "refreshChat", at = @At("HEAD"))
+    private void handleRefreshedChat(CallbackInfo ci) {
+        drawnToFull.clear();
+    }
+
+    @Inject(method = "refreshChat", locals = LocalCapture.CAPTURE_FAILHARD, at = @At(value = "INVOKE", target = "Lnet/minecraft/client/gui/GuiNewChat;setChatLine(Lnet/minecraft/util/IChatComponent;IIZ)V", ordinal = 0, shift = At.Shift.BEFORE))
+    private void handleLineRefresh(CallbackInfo ci, int i, ChatLine chatline) {
+        lastTempLine = chatline;
+    }
+
+    @Inject(method = "clearChatMessages", at = @At("HEAD"))
+    private void handleChatCleared(CallbackInfo ci) {
+        drawnToFull.clear();
+    }
+
+    @Override
+    public ChatLine getFullMessage(ChatLine line) {
+        return drawnToFull.getOrDefault(line, null);
+    }
+}