1 files changed, 60 insertions, 0 deletions
diff --git a/src/main/java/com/raeids/stratus/mixin/GuiChatMixin.java b/src/main/java/com/raeids/stratus/mixin/GuiChatMixin.java
new file mode 100644
index 0000000..be04586
--- /dev/null
+++ b/src/main/java/com/raeids/stratus/mixin/GuiChatMixin.java
@@ -0,0 +1,60 @@
+package com.raeids.stratus.mixin;
+
+import com.raeids.stratus.config.StratusConfig;
+import com.raeids.stratus.hook.ChatHookKt;
+import net.minecraft.client.gui.GuiChat;
+import net.minecraft.client.gui.GuiScreen;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+
+@Mixin(GuiChat.class)
+public abstract class GuiChatMixin extends GuiScreen {
+
+    @Inject(method = "initGui", at = @At("TAIL"))
+    private void init(CallbackInfo ci) {
+        if (StratusConfig.INSTANCE.getChatSearch()) {
+            ChatHookKt.initGui();
+        }
+    }
+
+    @Inject(method = "drawScreen", at = @At(value = "INVOKE", target = "Lnet/minecraft/client/gui/GuiTextField;drawTextBox()V", shift = At.Shift.AFTER))
+    private void yeah(int mouseX, int mouseY, float partialTicks, CallbackInfo ci) {
+        if (ChatHookKt.getInputField() != null) {
+            ChatHookKt.getInputField().drawTextBox();
+        }
+    }
+
+    @Inject(method = "onGuiClosed", at = @At("TAIL"))
+    private void onGuiClosed(CallbackInfo ci) {
+        ChatHookKt.setInputField(null);
+        ChatHookKt.setPrevText("");
+    }
+
+    @Inject(method = "updateScreen", at = @At("HEAD"))
+    private void updateScreen(CallbackInfo ci) {
+        ChatHookKt.updateScreen();
+    }
+
+    @Inject(method = "keyTyped", at = @At("HEAD"), cancellable = true)
+    private void keyTyped(char typedChar, int keyCode, CallbackInfo ci) {
+        if (ChatHookKt.getInputField() != null) {
+            if (ChatHookKt.getInputField().isFocused()) {
+                ci.cancel();
+                if (keyCode == 1 && ChatHookKt.getInputField().isFocused()) {
+                    ChatHookKt.getInputField().setFocused(false);
+                    return;
+                }
+                ChatHookKt.getInputField().textboxKeyTyped(typedChar, keyCode);
+            }
+        }
+    }
+
+    @Inject(method = "mouseClicked", at = @At("HEAD"))
+    private void mouseClicked(int mouseX, int mouseY, int mouseButton, CallbackInfo ci) {
+        if (ChatHookKt.getInputField() != null) {
+            ChatHookKt.getInputField().mouseClicked(mouseX, mouseY, mouseButton);
+        }
+    }
+}