A statemachinebehaviour to toggle rootmotion when entering or exiting animation states in Unity

UpdateRootMotion.cs

using UnityEngine;

public class UpdateRootMotion : StateMachineBehaviour
{
    public bool onStateEnter;
    public bool applyRootMotionOnEnter;
    
    public bool onStateExit;
    public bool applyRootMotionOnExit;

    public override void OnStateEnter(Animator animator, AnimatorStateInfo stateInfo, int layerIndex)
    {
        if (onStateEnter)
            animator.applyRootMotion = applyRootMotionOnEnter;

    }

    public override void OnStateExit(Animator animator, AnimatorStateInfo stateInfo, int layerIndex)
    {
        if (onStateExit)
            animator.applyRootMotion = applyRootMotionOnExit;
    }
}