fix: mixin disconnect bug for atm9

Author: dsx137

forge-1.20.1/src/main/java/club/asynclab/asyncraft/asyncauth/mixin/MixinServerLoginPacketListenerImpl.java

@@ -1,11 +1,16 @@
 package club.asynclab.asyncraft.asyncauth.mixin;
 
 import club.asynclab.asyncraft.asyncauth.util.UtilComponent;
+import net.minecraft.network.Connection;
 import net.minecraft.network.chat.Component;
+import net.minecraft.network.protocol.login.ClientboundLoginDisconnectPacket;
 import net.minecraft.server.network.ServerLoginPacketListenerImpl;
 import org.objectweb.asm.Opcodes;
+import org.slf4j.Logger;
+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;
@@ -15,17 +20,40 @@ public abstract class MixinServerLoginPacketListenerImpl {
     @Shadow
     private int tick;
 
+    @Final
+    @Shadow
+    Connection connection;
+
+    @Final
+    @Shadow
+    static Logger LOGGER;
+
     @Shadow
     public static int MAX_TICKS_BEFORE_LOGIN = 600;
 
     @Shadow
-    public abstract void disconnect(Component reason);
+    public abstract String getUserName();
+
+//    @Shadow
+//    public abstract void disconnect(Component reason);
 
     @Inject(method = "tick", at = @At(value = "FIELD", target = "Lnet/minecraft/server/network/ServerLoginPacketListenerImpl;tick:I", opcode = Opcodes.GETFIELD), cancellable = true)
     private void onTick(CallbackInfo ci) {
         if (MAX_TICKS_BEFORE_LOGIN != 0 && this.tick++ == MAX_TICKS_BEFORE_LOGIN) {
             this.disconnect(UtilComponent.getTranslatableComponent("multiplayer.disconnect.slow_login"));
         }
         ci.cancel();
+
+    }
+
+    @Unique
+    private void disconnect(Component pReason) {
+        try {
+            LOGGER.info("Disconnecting {}: {}", this.getUserName(), pReason.getString());
+            this.connection.send(new ClientboundLoginDisconnectPacket(pReason));
+            this.connection.disconnect(pReason);
+        } catch (Exception exception) {
+            LOGGER.error("Error whilst disconnecting player", exception);
+        }
     }
 }