1 files changed, 19 insertions, 4 deletions
diff --git a/src/main/java/io/github/moulberry/notenoughupdates/mixins/MixinNetHandlerPlayClient.java b/src/main/java/io/github/moulberry/notenoughupdates/mixins/MixinNetHandlerPlayClient.java
index 876338da..5cb2dc8d 100644
--- a/src/main/java/io/github/moulberry/notenoughupdates/mixins/MixinNetHandlerPlayClient.java
+++ b/src/main/java/io/github/moulberry/notenoughupdates/mixins/MixinNetHandlerPlayClient.java
@@ -1,19 +1,21 @@
 package io.github.moulberry.notenoughupdates.mixins;
 
-import io.github.moulberry.notenoughupdates.miscfeatures.CustomItemEffects;
-import io.github.moulberry.notenoughupdates.miscfeatures.EnchantingSolvers;
+import io.github.moulberry.notenoughupdates.miscfeatures.*;
 import io.github.moulberry.notenoughupdates.NotEnoughUpdates;
-import io.github.moulberry.notenoughupdates.miscfeatures.ItemCooldowns;
-import io.github.moulberry.notenoughupdates.miscfeatures.MiningStuff;
 import net.minecraft.client.network.NetHandlerPlayClient;
 import net.minecraft.entity.player.EntityPlayer;
+import net.minecraft.network.Packet;
+import net.minecraft.network.play.client.C13PacketPlayerAbilities;
 import net.minecraft.network.play.server.S23PacketBlockChange;
 import net.minecraft.network.play.server.S2FPacketSetSlot;
+import net.minecraft.network.play.server.S39PacketPlayerAbilities;
+import org.lwjgl.opengl.Display;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.injection.At;
 import org.spongepowered.asm.mixin.injection.Inject;
 import org.spongepowered.asm.mixin.injection.Redirect;
 import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
 
 @Mixin(NetHandlerPlayClient.class)
 public class MixinNetHandlerPlayClient {
@@ -40,6 +42,19 @@ public class MixinNetHandlerPlayClient {
         ItemCooldowns.processBlockChangePacket(packetIn);
     }
 
+    @Inject(method="handlePlayerAbilities", at=@At("HEAD"))
+    public void handlePlayerAbilities(S39PacketPlayerAbilities packetIn, CallbackInfo ci) {
+        FlyFix.onReceiveAbilities(packetIn);
+    }
+
+    @Inject(method="addToSendQueue", at=@At("HEAD"))
+    public void addToSendQueue(Packet packet, CallbackInfo ci) {
+        if(packet instanceof C13PacketPlayerAbilities) {
+            C13PacketPlayerAbilities abilities = (C13PacketPlayerAbilities) packet;
+            FlyFix.onSendAbilities(abilities);
+        }
+    }
+
 
 
 }