1 files changed, 20 insertions, 0 deletions
diff --git a/src/main/java/club/eviltransgenders/autoreconnect/mixin/ConnectScreenMixin.java b/src/main/java/club/eviltransgenders/autoreconnect/mixin/ConnectScreenMixin.java
new file mode 100644
index 0000000..dd2e4a1
--- /dev/null
+++ b/src/main/java/club/eviltransgenders/autoreconnect/mixin/ConnectScreenMixin.java
@@ -0,0 +1,20 @@
+package club.eviltransgenders.autoreconnect.mixin;
+
+import club.eviltransgenders.autoreconnect.AutoReconnect;
+import net.minecraft.client.gui.screens.ConnectScreen;
+import net.minecraft.client.Minecraft;
+import net.minecraft.client.multiplayer.resolver.ServerAddress;
+import net.minecraft.client.multiplayer.ServerData;
+import net.minecraft.client.multiplayer.TransferState;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.Mixin;
+
+@Mixin(ConnectScreen.class)
+public abstract class ConnectScreenMixin {
+	@Inject(at = @At("HEAD"), method = "connect")
+	private void updateLastServerData(Minecraft client, ServerAddress hostAndPort, ServerData serverData, TransferState transferState, CallbackInfo ci) {
+		AutoReconnect.getInstance().lastServerData = serverData;
+	}
+}