Replace redirect with wrap operation

Author: Ocelot5836

common/src/main/java/dev/ryanhcode/sable/mixin/player_standup/PlayerMixin.java

@@ -1,32 +1,33 @@
 package dev.ryanhcode.sable.mixin.player_standup;
 
+import com.llamalad7.mixinextras.injector.wrapoperation.Operation;
+import com.llamalad7.mixinextras.injector.wrapoperation.WrapOperation;
 import dev.ryanhcode.sable.mixinhelpers.CanFallAtleastHelper;
 import net.minecraft.world.entity.Entity;
 import net.minecraft.world.entity.player.Player;
 import net.minecraft.world.level.Level;
 import net.minecraft.world.phys.AABB;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.injection.At;
-import org.spongepowered.asm.mixin.injection.Redirect;
 
 @Mixin(Player.class)
 public class PlayerMixin {
 
-    @Redirect(
+    @WrapOperation(
             method = "canPlayerFitWithinBlocksAndEntitiesWhen",
             at = @At(
                     value = "INVOKE",
                     target = "Lnet/minecraft/world/level/Level;noCollision(Lnet/minecraft/world/entity/Entity;Lnet/minecraft/world/phys/AABB;)Z"
             )
     )
-    private boolean sable$noCollisionWithSubLevels(final Level level, final Entity entity, final AABB aabb) {
-        final boolean original = level.noCollision(entity, aabb);
-
-        if (!original) return false;
+    private boolean sable$noCollisionWithSubLevels(final Level instance, final Entity entity, final AABB aabb, final Operation<Boolean> original) {
+        if (!original.call(instance, entity, aabb)) {
+            return false;
+        }
 
         // If vanilla says no collision, also check sublevel blocks.
         // canFallAtleastWithSubLevels returns non-null when there IS a collision,
         // meaning the player does NOT fit → return false.
-        return CanFallAtleastHelper.canFallAtleastWithSubLevels(level, aabb) == null;
+        return CanFallAtleastHelper.canFallAtleastWithSubLevels(instance, aabb) == null;
     }
 }