Rift/src/main/java/org/dimdev/rift/mixin/hook/MixinNetHandlerPlayServer.java at d2d26ceda25e11d199c27a59519cc6e22c3e9d06 · DimensionalDevelopment/Rift · GitHub

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
44
45
46
47
48
49
50
51
52
53
54
55
56
package org.dimdev.rift.mixin.hook;

import net.minecraft.entity.player.EntityPlayerMP;
import net.minecraft.network.NetHandlerPlayServer;
import net.minecraft.network.NetworkManager;
import net.minecraft.network.PacketBuffer;
import net.minecraft.network.play.client.CPacketCustomPayload;
import net.minecraft.server.MinecraftServer;
import net.minecraft.util.ResourceLocation;
import org.dimdev.rift.injectedmethods.RiftCPacketCustomPayload;
import org.dimdev.rift.listener.CustomPayloadHandler;
import org.dimdev.rift.network.Message;
import org.dimdev.rift.network.ServerMessageContext;
import org.dimdev.riftloader.RiftLoader;
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(NetHandlerPlayServer.class)
public class MixinNetHandlerPlayServer {
    @Shadow @Final private MinecraftServer server;
    @Shadow public EntityPlayerMP player;
    @Shadow @Final public NetworkManager netManager;

    @SuppressWarnings("deprecation")
    @Inject(method = "processCustomPayload", at = @At("HEAD"), cancellable = true)
    private void handleModCustomPayload(CPacketCustomPayload packet, CallbackInfo ci) {
        ResourceLocation channelName = ((RiftCPacketCustomPayload) packet).getChannelName();
        PacketBuffer data = ((RiftCPacketCustomPayload) packet).getData();

        for (CustomPayloadHandler customPayloadHandler : RiftLoader.instance.getListeners(CustomPayloadHandler.class)) {
            if (customPayloadHandler.serverHandlesChannel(channelName)) {
                customPayloadHandler.serverHandleCustomPayload(channelName, data);
            }
        }

        if(Message.REGISTRY.isEmpty()){
            return;
        }

        Class<? extends Message> messageClass = Message.REGISTRY.get(channelName);
        if (messageClass != null) {
            try {
                Message message = RiftLoader.instance.newInstance(messageClass);
                message.read(data);
                message.process(new ServerMessageContext(server, player, netManager));
            } catch (ReflectiveOperationException e) {
                throw new RuntimeException("Error creating " + messageClass, e);
            }
            ci.cancel();
        }
    }
}