Connection events

Author: Avanatiker

common/src/main/java/com/lambda/mixin/ClientConnectionMixin.java

@@ -1,11 +1,15 @@
 package com.lambda.mixin;
 
 import com.lambda.event.EventFlow;
+import com.lambda.event.events.ConnectionEvent;
 import com.lambda.event.events.PacketEvent;
 import io.netty.channel.ChannelHandlerContext;
 import net.minecraft.network.ClientConnection;
 import net.minecraft.network.NetworkSide;
+import net.minecraft.network.listener.PacketListener;
 import net.minecraft.network.packet.Packet;
+import net.minecraft.network.packet.c2s.handshake.ConnectionIntent;
+import net.minecraft.text.Text;
 import org.spongepowered.asm.mixin.Final;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.Shadow;
@@ -53,4 +57,20 @@ private void receivingPacketPost(
 
         EventFlow.post(new PacketEvent.Receive.Post(packet));
     }
+
+    @Inject(method = "connect(Ljava/lang/String;ILnet/minecraft/network/listener/PacketListener;Lnet/minecraft/network/packet/c2s/handshake/ConnectionIntent;)V", at = @At("HEAD"))
+    private void onConnect(
+            String address,
+            int port,
+            PacketListener listener,
+            ConnectionIntent intent,
+            CallbackInfo ci
+    ) {
+        EventFlow.post(new ConnectionEvent.Connect(address, port, listener, intent));
+    }
+
+    @Inject(method = "disconnect(Lnet/minecraft/text/Text;)V", at = @At("HEAD"))
+    private void onDisconnect(Text reason, CallbackInfo ci) {
+        EventFlow.post(new ConnectionEvent.Disconnect(reason));
+    }
 }

common/src/main/kotlin/com/lambda/event/events/ConnectionEvent.kt

@@ -0,0 +1,16 @@
+package com.lambda.event.events
+
+import com.lambda.event.Event
+import net.minecraft.network.listener.PacketListener
+import net.minecraft.network.packet.c2s.handshake.ConnectionIntent
+import net.minecraft.text.Text
+
+abstract class ConnectionEvent : Event {
+    class Connect(
+        val host: String,
+        port: Int,
+        listener: PacketListener,
+        intent: ConnectionIntent
+    ) : ConnectionEvent()
+    class Disconnect(val reason: Text) : ConnectionEvent()
+}