From c94c2b1b0a2970448bf24bde5c62277c672110a0 Mon Sep 17 00:00:00 2001
From: Bella <bella.who.two@gmail.com>
Date: Wed, 15 Apr 2020 13:29:02 -0400
Subject: [PATCH] add head and hover listeners

---
 .../module/modules/movement/ElytraFlight.java | 28 +++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/src/main/java/me/zeroeightsix/kami/module/modules/movement/ElytraFlight.java b/src/main/java/me/zeroeightsix/kami/module/modules/movement/ElytraFlight.java
index 60352afa5..b7c61f1c3 100644
--- a/src/main/java/me/zeroeightsix/kami/module/modules/movement/ElytraFlight.java
+++ b/src/main/java/me/zeroeightsix/kami/module/modules/movement/ElytraFlight.java
@@ -33,6 +33,34 @@ public class ElytraFlight extends Module {
     private ElytraFlightMode enabledMode;
     private boolean hasDoneWarning;
 
+    /* Control mode states */
+    private double hoverTarget = -1.0;
+    public float packetYaw = 0.0f;
+
+    /* Control Mode */
+    @EventHandler
+    private Listener<PacketEvent.Send> sendListener = new Listener<>(event -> {
+        if (!mode.getValue().equals(ElytraFlightMode.CONTROL) || mc.player == null) return;
+        if (event.getPacket() instanceof CPacketPlayer) {
+            if (!mc.player.isElytraFlying()) return;
+            CPacketPlayer packet = (CPacketPlayer) event.getPacket();
+            packet.pitch = 0.0f;
+            packet.yaw = packetYaw;
+        }
+        if (event.getPacket() instanceof CPacketEntityAction && ((CPacketEntityAction) event.getPacket()).getAction() == CPacketEntityAction.Action.START_FALL_FLYING) {
+            hoverTarget = mc.player.posY + 0.35;
+        }
+    });
+
+    @EventHandler
+    private Listener<PacketEvent.Receive> receiveListener = new Listener<>(event -> {
+        if (!mode.getValue().equals(ElytraFlightMode.CONTROL) || mc.player == null || !mc.player.isElytraFlying()) return;
+        if (event.getPacket() instanceof SPacketPlayerPosLook) {
+            SPacketPlayerPosLook packet = (SPacketPlayerPosLook) event.getPacket();
+            packet.pitch = ElytraFlight.mc.player.rotationPitch;
+        }
+    });
+
     @Override
     public void onUpdate() {
         if (mc.player == null) return;