Step Height Attribute

EntityMixin.java

package io.github.daomephsta.mixin;

import org.spongepowered.asm.mixin.Debug;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.gen.Accessor;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;

import net.minecraft.entity.Entity;

@Debug(export = true)
@Mixin(Entity.class)
public abstract class EntityMixin
{
    @Accessor("stepHeight")
    protected abstract void setStepHeightDirect(float stepHeight);

    @Inject(method = "getStepHeight", at = @At("TAIL"))
    protected abstract void scratch_getStepHeightShim(CallbackInfoReturnable<Float> info);

    @Inject(method = "setStepHeight", at = @At("TAIL"))
    protected abstract void scratch_setStepHeightShim(float stepHeight, CallbackInfo info);
}

LivingEntityMixin.java

package io.github.daomephsta.mixin;

import org.spongepowered.asm.mixin.Debug;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Shadow;

import net.minecraft.entity.LivingEntity;
import net.minecraft.entity.attribute.EntityAttribute;
import net.minecraft.entity.attribute.EntityAttributeInstance;

@Debug(export = true)
@Mixin(LivingEntity.class)
public abstract class LivingEntityMixin extends EntityMixin
{
    @Shadow
    public abstract double getAttributeValue(EntityAttribute attribute);

    @Shadow
    public abstract EntityAttributeInstance getAttributeInstance(EntityAttribute attribute);
}

PlayerEntityMixin.java

package io.github.daomephsta.mixin;

import org.spongepowered.asm.mixin.Debug;
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;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;

import io.github.daomephsta.api.StepHeightApi;
import net.minecraft.entity.attribute.DefaultAttributeContainer;
import net.minecraft.entity.player.PlayerEntity;

@Debug(export = true)
@Mixin(PlayerEntity.class)
public abstract class PlayerEntityMixin extends LivingEntityMixin
{
    @Inject(method = "createPlayerAttributes", at = @At("RETURN"))
    private static void scratch_addStepHeightAttribute(CallbackInfoReturnable<DefaultAttributeContainer.Builder> info)
    {
        info.getReturnValue().add(StepHeightApi.PLAYER_STEP_HEIGHT, 0.6);
    }

    @Override
    protected void scratch_getStepHeightShim(CallbackInfoReturnable<Float> info)
    {
        setStepHeightDirect((float) getAttributeValue(StepHeightApi.PLAYER_STEP_HEIGHT));
    }

    @Override
    protected void scratch_setStepHeightShim(float stepHeight, CallbackInfo info)
    {
        this.getAttributeInstance(StepHeightApi.PLAYER_STEP_HEIGHT).setBaseValue(stepHeight);
    }
}

StepHeightApi.java

package io.github.daomephsta.api;

import io.github.daomephsta.Scratch;
import net.minecraft.entity.attribute.ClampedEntityAttribute;
import net.minecraft.entity.attribute.EntityAttribute;
import net.minecraft.registry.Registries;
import net.minecraft.registry.Registry;

public class StepHeightApi
{
    public static final EntityAttribute PLAYER_STEP_HEIGHT = Registry.register(Registries.ATTRIBUTE, Scratch.id("player.step_height"),
        new ClampedEntityAttribute("attribute.name.player.step_height", 1.0, 0.6, 1024.0));

    public static void initialise()
    {
        // Force static init
    }
}