Control Baritones rotations

Author: Avanatiker

common/build.gradle.kts

@@ -28,6 +28,7 @@ dependencies {
 
     // Baritone
     modImplementation("baritone-api:baritone-api:1.10.2")
+    modImplementation("baritone-api:baritone-unoptimized-fabric:1.10.2")
 }
 
 // Avoid nested jars

common/src/main/java/com/lambda/mixin/baritone/MixinBaritonePlayerContext.java

@@ -0,0 +1,27 @@
+package com.lambda.mixin.baritone;
+
+import baritone.Baritone;
+import baritone.api.utils.Rotation;
+import baritone.utils.player.BaritonePlayerContext;
+import com.lambda.interaction.RotationManager;
+import com.lambda.util.BaritoneUtils;
+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.CallbackInfoReturnable;
+
+@Mixin(value = BaritonePlayerContext.class, remap = false) // fix compileJava warning
+public class MixinBaritonePlayerContext {
+    @Shadow @Final private Baritone baritone;
+
+    // Let baritone know the actual rotation
+    @Inject(method = "playerRotations", at = @At("HEAD"), cancellable = true, remap = false)
+    void syncRotationWithBaritone(CallbackInfoReturnable<Rotation> cir) {
+        if (baritone != BaritoneUtils.getPrimary()) return;
+
+        RotationManager rm = RotationManager.INSTANCE;
+        cir.setReturnValue(new Rotation((float) rm.getCurrentRotation().getYaw(), (float) rm.getCurrentRotation().getPitch()));
+    }
+}

common/src/main/java/com/lambda/mixin/baritone/MixinLookBehavior.java

@@ -0,0 +1,41 @@
+package com.lambda.mixin.baritone;
+
+import baritone.api.event.events.PlayerUpdateEvent;
+import baritone.api.event.events.RotationMoveEvent;
+import baritone.api.utils.Rotation;
+import baritone.behavior.LookBehavior;
+import com.lambda.interaction.RotationManager;
+import com.lambda.util.BaritoneUtils;
+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.callback.CallbackInfo;
+
+@Mixin(LookBehavior.class)
+public class MixinLookBehavior {
+    // Redirect baritone's rotations into our rotation engine
+    @Inject(method = "updateTarget", at = @At("HEAD"), remap = false, cancellable = true)
+    void onTargetUpdate(Rotation rotation, boolean blockInteract, CallbackInfo ci) {
+        LookBehavior instance = ((LookBehavior) (Object) this);
+        if (instance.baritone != BaritoneUtils.getPrimary()) return;
+
+        RotationManager.BaritoneProcessor.handleBaritoneRotation(rotation.getYaw(), rotation.getPitch());
+        ci.cancel();
+    }
+
+    @Inject(method = "onPlayerUpdate", at = @At("HEAD"), remap = false, cancellable = true)
+    void onUpdate(PlayerUpdateEvent event, CallbackInfo ci) {
+        LookBehavior instance = ((LookBehavior) (Object) this);
+        if (instance.baritone != BaritoneUtils.getPrimary()) return;
+
+        ci.cancel();
+    }
+
+    @Inject(method = "onPlayerRotationMove", at = @At("HEAD"), remap = false, cancellable = true)
+    void onMovementUpdate(RotationMoveEvent event, CallbackInfo ci) {
+        LookBehavior instance = ((LookBehavior) (Object) this);
+        if (instance.baritone != BaritoneUtils.getPrimary()) return;
+
+        ci.cancel();
+    }
+}

common/src/main/kotlin/com/lambda/util/BaritoneUtils.kt

@@ -0,0 +1,19 @@
+package com.lambda.util
+
+import baritone.api.BaritoneAPI
+import baritone.api.IBaritone
+import baritone.api.Settings
+
+object BaritoneUtils {
+    private val baritone = BaritoneAPI.getProvider()
+    val settings: Settings = BaritoneAPI.getSettings()
+    @JvmStatic val primary: IBaritone = baritone.primaryBaritone
+
+    val isPathing: Boolean
+        get() = primary.pathingBehavior.isPathing
+
+    val isActive: Boolean
+        get() = primary.customGoalProcess.isActive || primary.pathingBehavior.isPathing || primary.pathingControlManager.mostRecentInControl().orElse(null)?.isActive == true
+
+    fun cancel() = primary.pathingBehavior.cancelEverything()
+}

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

@@ -6,6 +6,8 @@
   "client": [
     "ClientConnectionMixin",
     "MinecraftClientMixin",
+    "baritone.MixinLookBehavior",
+    "baritone.MixinBaritonePlayerContext",
     "entity.ClientPlayerEntityMixin",
     "entity.EntityMixin",
     "entity.LivingEntityMixin",