1

Src/Unity/Scripts/Entity/Components/EntityMovement/RigidbodyBasedMovement.cs

@@ -6,7 +6,7 @@ using UnityEngine.InputSystem;
 
 namespace BITKit.Entities.Movement
 {
-	public class RigidbodyBasedMovement : StateBasedComponent<IEntityMovementState>,IEntityMovement
+	public class RigidbodyBasedMovement : StateBasedBehavior<IEntityMovementState>,IEntityMovement
 	{
 		[SerializeField] private new Rigidbody rigidbody;
 		[SerializeField] private Animator animator;
@@ -20,6 +20,11 @@ namespace BITKit.Entities.Movement
 		public Vector3 GroundVelocity { get; }
 		public Vector3 AngularVelocity { get; }
 		public bool IsGrounded { get; }
+		
+		[Inject(true)]
+		private IHealth health;
+		
+		
 		public void SyncMovement(Vector3 velocity, Vector3 position, Quaternion rotation, bool isGrounded)
 		{
 		}
@@ -37,6 +42,20 @@ namespace BITKit.Entities.Movement
 		}
 
 		public event Action<object> OnCommand;
+		public override void OnAwake()
+		{
+			base.OnAwake();
+			if (health is not null)
+			{
+				health.OnSetAlive += OnSetAlive;
+			}
+		}
+
+		private void OnSetAlive(bool obj)
+		{
+			rigidbody.isKinematic = !obj;
+		}
+
 		public override void OnFixedUpdate(float deltaTime)
 		{
 			rigidbody.MovePosition(rigidbody.position + Velocity * deltaTime