using System.Linq;
using Robust.Shared.Map;

namespace Content.Server.AI.Utils
{
    public static class Visibility
    {
        // Should this be in robust or something? Fark it
        public static IEnumerable<EntityUid> GetNearestEntities(EntityCoordinates grid, Type component, float range)
        {
            var entMan = IoCManager.Resolve<IEntityManager>();
            var inRange = GetEntitiesInRange(grid, component, range).ToList();
            var sortedInRange = inRange.OrderBy(o => (entMan.GetComponent<TransformComponent>(o).Coordinates.Position - grid.Position).Length);

            return sortedInRange;
        }

        public static IEnumerable<EntityUid> GetEntitiesInRange(EntityCoordinates grid, Type component, float range)
        {
            var entityManager = IoCManager.Resolve<IEntityManager>();
            foreach (var entity in entityManager.GetAllComponents(component).Select(c => c.Owner))
            {
                var transform = entityManager.GetComponent<TransformComponent>(entity);

                if (transform.Coordinates.GetGridEntityId(entityManager) != grid.GetGridEntityId(entityManager))
                {
                    continue;
                }

                if ((transform.Coordinates.Position - grid.Position).Length <= range)
                {
                    yield return entity;
                }
            }
        }
    }
}