Movement hooks

Author: Avanatiker

common/src/main/java/com/lambda/mixin/entity/ClientPlayerEntityMixin.java

@@ -0,0 +1,58 @@
+package com.lambda.mixin.entity;
+
+import com.lambda.Lambda;
+import com.lambda.event.EventFlow;
+import com.lambda.event.events.MovementEvent;
+import net.minecraft.client.MinecraftClient;
+import net.minecraft.client.network.ClientPlayerEntity;
+import net.minecraft.entity.MovementType;
+import net.minecraft.util.math.Vec3d;
+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.Redirect;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+
+@Mixin(value = ClientPlayerEntity.class, priority = Integer.MAX_VALUE)
+public abstract class ClientPlayerEntityMixin extends EntityMixin {
+
+    @Shadow protected abstract void autoJump(float dx, float dz);
+
+    @Shadow public abstract boolean isUsingItem();
+
+    @Shadow private boolean autoJumpEnabled;
+
+    @Inject(method = "move", at = @At("HEAD"), cancellable = true)
+    void onMove(MovementType movementType, Vec3d movement, CallbackInfo ci) {
+        ClientPlayerEntity self = (ClientPlayerEntity) (Object) this;
+        if (self != Lambda.getMc().player) return;
+
+        ci.cancel();
+
+        float prevX = (float) self.getX();
+        float prevZ = (float) self.getZ();
+
+        EventFlow.post(new MovementEvent.Pre());
+        super.move(movementType, self.getVelocity());
+        EventFlow.post(new MovementEvent.Post());
+
+        float currX = (float) self.getX();
+        float currZ = (float) self.getZ();
+
+        this.autoJump(currX - prevX, currZ - prevZ);
+    }
+
+    @Redirect(method = "tickMovement", at = @At(value = "INVOKE", target = "Lnet/minecraft/client/network/ClientPlayerEntity;isUsingItem()Z"))
+    boolean onSlowDown(ClientPlayerEntity entity) {
+        if (EventFlow.post(new MovementEvent.SlowDown()).isCanceled()) return false;
+        return this.isUsingItem();
+    }
+
+    @Inject(method = "sendMovementPackets", at = @At(value = "HEAD"), cancellable = true)
+    void sendBegin(CallbackInfo ci) {
+        ci.cancel();
+//        PlayerPacketManager.sendPlayerPackets();
+        autoJumpEnabled = MinecraftClient.getInstance().options.getAutoJump().getValue();
+    }
+}

common/src/main/java/com/lambda/mixin/entity/EntityMixin.java

@@ -0,0 +1,12 @@
+package com.lambda.mixin.entity;
+
+import net.minecraft.entity.Entity;
+import net.minecraft.entity.MovementType;
+import net.minecraft.util.math.Vec3d;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+
+@Mixin(Entity.class)
+public abstract class EntityMixin {
+    @Shadow public void move(MovementType movementType, Vec3d movement) {}
+}

common/src/main/java/com/lambda/mixin/entity/LivingEntityMixin.java

@@ -0,0 +1,43 @@
+package com.lambda.mixin.entity;
+
+import com.lambda.Lambda;
+import com.lambda.event.EventFlow;
+import com.lambda.event.events.MovementEvent;
+import net.minecraft.entity.LivingEntity;
+import net.minecraft.util.math.Vec3d;
+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(LivingEntity.class)
+public abstract class LivingEntityMixin extends EntityMixin {
+
+    @Shadow
+    protected abstract float getJumpVelocity();
+
+    @Inject(method = "jump", at = @At("HEAD"), cancellable = true)
+    void onJump(CallbackInfo ci) {
+        LivingEntity self = (LivingEntity) (Object) this;
+        if (self != Lambda.getMc().player) return;
+        ci.cancel();
+
+        float height = this.getJumpVelocity();
+        MovementEvent.Jump event = EventFlow.post(new MovementEvent.Jump(height));
+
+        if (event.isCanceled()) return;
+
+        if (!self.isSprinting()) {
+            Vec3d velocity = self.getVelocity();
+            self.setVelocity(velocity.x, event.getHeight(), velocity.z);
+        } else {
+            // ToDo: Implement rotation system
+//            Float yaw = RotationManager.getMovementYaw();
+//            float f = ((yaw != null) ? yaw : self.getYaw()) * ((float)Math.PI / 180);
+//            self.setVelocity(self.getVelocity().add(-MathHelper.sin(f) * 0.2f, 0.0, MathHelper.cos(f) * 0.2f));
+        }
+
+        self.velocityDirty = true;
+    }
+}

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

@@ -5,5 +5,9 @@ import com.lambda.event.cancellable.Cancellable
 import com.lambda.event.cancellable.ICancellable
 
 abstract class MovementEvent : Event {
+    class Pre : MovementEvent()
+    class Post : MovementEvent()
     class ClipAtLedge : MovementEvent(), ICancellable by Cancellable()
+    class Jump(var height: Double) : MovementEvent(), ICancellable by Cancellable()
+    class SlowDown : Event, ICancellable by Cancellable()
 }

common/src/main/resources/lambda.mixins.common.json

@@ -7,10 +7,13 @@
     "ChatInputSuggestorMixin",
     "ChatScreenMixin",
     "ClientConnectionMixin",
+    "KeyBindingMixin",
     "KeyboardMixin",
     "MinecraftClientMixin",
-    "KeyBindingMixin",
-    "PlayerEntityMixin"
+    "PlayerEntityMixin",
+    "entity.ClientPlayerEntityMixin",
+    "entity.EntityMixin",
+    "entity.LivingEntityMixin"
   ],
   "injectors": {
     "defaultRequire": 1