diff options
Diffstat (limited to 'src/main/java/club')
3 files changed, 91 insertions, 0 deletions
diff --git a/src/main/java/club/eviltransgenders/autoreconnect/AutoReconnect.java b/src/main/java/club/eviltransgenders/autoreconnect/AutoReconnect.java new file mode 100644 index 0000000..d756c23 --- /dev/null +++ b/src/main/java/club/eviltransgenders/autoreconnect/AutoReconnect.java @@ -0,0 +1,45 @@ +package club.eviltransgenders.autoreconnect; + +import net.fabricmc.api.ClientModInitializer; +import net.fabricmc.fabric.api.client.screen.v1.ScreenEvents; +import net.minecraft.client.gui.components.Button; +import net.minecraft.client.gui.screens.ConnectScreen; +import net.minecraft.client.gui.screens.DisconnectedScreen; +import net.minecraft.client.gui.screens.multiplayer.JoinMultiplayerScreen; +import net.minecraft.client.gui.screens.Screen; +import net.minecraft.client.gui.screens.TitleScreen; +import net.minecraft.client.Minecraft; +import net.minecraft.client.multiplayer.resolver.ServerAddress; +import net.minecraft.client.multiplayer.ServerData; +import net.minecraft.network.chat.Component; + +public final class AutoReconnect implements ClientModInitializer { + private static AutoReconnect instance; + private long reconnectTimer; + public ServerData lastServerData; + + @Override + public void onInitializeClient() { + instance = this; + ScreenEvents.AFTER_INIT.register(this::afterScreenInit); + } + + public void afterScreenInit(Minecraft client, Screen screen, int scaledWidth, int scaledHeight) { + if (screen instanceof DisconnectedScreen) { + this.reconnectTimer = 5 * 20; + ScreenEvents.afterTick(screen).register(this::afterScreenTick); + } + } + + public void afterScreenTick(Screen screen) { + if (--this.reconnectTimer <= 0) { + ConnectScreen.startConnecting(new JoinMultiplayerScreen(new TitleScreen()), Minecraft.getInstance(), ServerAddress.parseString(this.lastServerData.ip), this.lastServerData, false, null); + } else { + ((Button) screen.children().getLast()).setMessage(Component.translatable("gui.autoreconnect.reconnect", this.reconnectTimer / 20)); + } + } + + public static AutoReconnect getInstance() { + return instance; + } +} 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; + } +} diff --git a/src/main/java/club/eviltransgenders/autoreconnect/mixin/DisconnectedScreenMixin.java b/src/main/java/club/eviltransgenders/autoreconnect/mixin/DisconnectedScreenMixin.java new file mode 100644 index 0000000..99a59f9 --- /dev/null +++ b/src/main/java/club/eviltransgenders/autoreconnect/mixin/DisconnectedScreenMixin.java @@ -0,0 +1,26 @@ +package club.eviltransgenders.autoreconnect.mixin; + +import net.minecraft.client.gui.components.Button; +import net.minecraft.client.gui.layouts.LinearLayout; +import net.minecraft.client.gui.screens.DisconnectedScreen; +import net.minecraft.network.chat.Component; +import org.spongepowered.asm.mixin.Final; +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; +import org.spongepowered.asm.mixin.Shadow; + +@Mixin(DisconnectedScreen.class) +public abstract class DisconnectedScreenMixin { + @Shadow + @Final + private LinearLayout layout; + + @Inject(at = @At(value = "INVOKE", target = "Lnet/minecraft/client/gui/layouts/LinearLayout;visitWidgets(Ljava/util/function/Consumer;)V"), method = "init") + private void addAutoReconnectButtons(CallbackInfo ci) { + Button reconnectButton = Button.builder(Component.translatable("gui.autoreconnect.reconnect"), btn -> {}).width(200).build(); + this.layout.addChild(reconnectButton); + this.layout.arrangeElements(); + } +} |