using Godot;
using System.Collections.Generic;

namespace Rokojori
{  
  public static class PhysicsBodies
  {
    public static bool IsInAir( this PhysicsBody3D physicsBody3D )
    {
      if ( physicsBody3D is RigidBody3D rb)
      {
        return false;
      }

      if ( physicsBody3D is CharacterBody3D cb)
      {
        return ! cb.IsOnFloor();
      }

      if ( physicsBody3D is PhysicalBone3D pb )
      {
        return false;
      }

      if ( physicsBody3D is StaticBody3D sb )
      {
        return false;
      }

      return false;
    }

    public static Vector3 GetVelocity( this PhysicsBody3D physicsBody3D )
    {
      if ( physicsBody3D is RigidBody3D rb)
      {
        return rb.LinearVelocity;
      }

      if ( physicsBody3D is CharacterBody3D cb)
      {
        return cb.Velocity;
      }

      if ( physicsBody3D is PhysicalBone3D pb )
      {
        return pb.LinearVelocity;
      }

      if ( physicsBody3D is StaticBody3D sb )
      {
        return sb.ConstantLinearVelocity;
      }

      return Vector3.Zero;
    }
  }
}