src/main/java/moe/ymc/acron/mixin/ServerPlayNetworkHandlerMixin.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

package moe.ymc.acron.mixin;

import moe.ymc.acron.s2c.Entity;
import moe.ymc.acron.s2c.EventQueue;
import moe.ymc.acron.s2c.event.EventDisconnected;
import moe.ymc.acron.s2c.event.EventPlayerMessage;
import net.minecraft.network.ClientConnection;
import net.minecraft.server.filter.TextStream;
import net.minecraft.server.network.ServerPlayNetworkHandler;
import net.minecraft.server.network.ServerPlayerEntity;
import net.minecraft.text.Text;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.spongepowered.asm.mixin.Final;
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.callback.CallbackInfo;

@Mixin(ServerPlayNetworkHandler.class)
public class ServerPlayNetworkHandlerMixin {
    private static final Logger AC_LOGGER = LogManager.getLogger();

    @Shadow
    public ServerPlayerEntity player;

    @Shadow
    @Final
    public ClientConnection connection;

    @Inject(at = @At("RETURN"), method = "handleMessage")
    private void handleMessage(TextStream.Message message, CallbackInfo ci) {
        EventQueue.enqueue(new EventPlayerMessage(new Entity(player),
                message.getRaw()));
    }

    @Inject(at = @At("RETURN"), method = "onDisconnected")
    private void onDisconnected(Text reason, CallbackInfo ci) {
        EventQueue.enqueue(new EventDisconnected(new Entity(player),
                reason.getString()));
    }
}