1 files changed, 65 insertions, 0 deletions
diff --git a/src/main/java/ganarchy/friendcode/mixin/DirectConnectScreenMixin.java b/src/main/java/ganarchy/friendcode/mixin/DirectConnectScreenMixin.java
new file mode 100644
index 0000000..6b3a182
--- /dev/null
+++ b/src/main/java/ganarchy/friendcode/mixin/DirectConnectScreenMixin.java
@@ -0,0 +1,65 @@
+package ganarchy.friendcode.mixin;
+
+import ganarchy.friendcode.client.FriendCodeScreen;
+import ganarchy.friendcode.client.FriendConnectScreen;
+import it.unimi.dsi.fastutil.booleans.BooleanConsumer;
+import net.minecraft.client.font.TextRenderer;
+import net.minecraft.client.gui.screen.DirectConnectScreen;
+import net.minecraft.client.gui.screen.OpenToLanScreen;
+import net.minecraft.client.gui.screen.Screen;
+import net.minecraft.client.gui.widget.ButtonWidget;
+import net.minecraft.client.network.ServerInfo;
+import net.minecraft.client.util.math.MatrixStack;
+import net.minecraft.text.Text;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.Redirect;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+
+@Mixin(DirectConnectScreen.class)
+public abstract class DirectConnectScreenMixin extends Screen {
+    protected DirectConnectScreenMixin(Text title) {
+        super(title);
+    }
+
+    @Shadow
+    private Screen parent;
+    @Shadow
+    private BooleanConsumer callback;
+    @Shadow
+    private ServerInfo serverEntry;
+
+    @Inject(at = @At("HEAD"), method = "init()V")
+    private void init(CallbackInfo info) {
+        // title, meet button
+        this.addDrawableChild(new ButtonWidget(
+                this.width / 2 - 100,
+                20,
+                200,
+                20,
+                this.title,
+                button -> this.client.setScreen(new FriendConnectScreen(this.parent, this.callback, this.serverEntry))
+        ));
+    }
+
+    @Redirect(
+            method = "render(Lnet/minecraft/client/util/math/MatrixStack;IIF)V",
+            at = @At(
+                    value = "INVOKE",
+                    target = "Lnet/minecraft/client/gui/screen/DirectConnectScreen;drawCenteredText(Lnet/minecraft/client/util/math/MatrixStack;Lnet/minecraft/client/font/TextRenderer;Lnet/minecraft/text/Text;III)V",
+                    ordinal = 0
+            )
+    )
+    private void removeTitle(
+            MatrixStack matrices,
+            TextRenderer renderer,
+            Text text,
+            int centerX,
+            int y,
+            int color
+    ) {
+        // literally does nothing
+    }
+}