Cielonos/Assets/Scripts/MainGame/Managers/BattleManager/EnemySubmodule/EnemySubmodule.cs

using System.Collections.Generic;
using System.Linq;
using Cielonos.MainGame.Characters;
using UnityEngine;

namespace Cielonos.MainGame
{
    public partial class BattleManager
    {
        public partial class EnemySubmodule : SubmoduleBase<BattleManager>
        {
            public enum SortingType
            {
                Nearest,
                Farthest
            }
            
            public List<CharacterBase> activeEnemiesList;
            
            public EnemySubmodule(BattleManager owner) : base(owner)
            {
                activeEnemiesList = new List<CharacterBase>();
            }
            
            public void Reset()
            {
                activeEnemiesList.Clear();
            }
            
            public CharacterBase GetNearestEnemy(float maximumDistance = float.MaxValue, Transform origin = null)
            {
                if (origin == null)
                {
                    origin = Player.transform;
                }
            
                List<CharacterBase> nearestEnemies = GetEnemiesInRadius(origin.position, maximumDistance);
                CharacterBase nearestEnemy = nearestEnemies.Count > 0 ? nearestEnemies[0] : null;

                return nearestEnemy;
            }

            public List<CharacterBase> GetNearestEnemies(float maximumDistance = float.MaxValue, int enemyCount = 1, Transform origin = null)
            {
                if (origin == null)
                {
                    origin = Player.transform;
                }
            
                List<CharacterBase> nearestEnemies = GetEnemiesInRadius(origin.position, maximumDistance).Take(enemyCount).ToList();

                return nearestEnemies;
            }
            
            public List<CharacterBase> GetEnemiesInRadius(Vector3 origin, float radius, SortingType sortingType = SortingType.Nearest)
            {
                List<CharacterBase> enemiesInRadius = new List<CharacterBase>();
                foreach (CharacterBase enemy in activeEnemiesList)
                {
                    float enemyRadius = enemy.collisionSc.useCharacterController
                        ? enemy.collisionSc.characterController.radius
                        : enemy.collisionSc.mainColliderRadius;
                    
                    float distance = Vector3.Distance(origin, enemy.transform.position) - enemyRadius;
                    if (distance <= radius)
                    {
                        enemiesInRadius.Add(enemy);
                    }
                }

                if (sortingType == SortingType.Nearest)
                {
                    enemiesInRadius = enemiesInRadius.OrderBy(x => Vector3.Distance(origin, x.transform.position)).ToList();
                }
                else if (sortingType == SortingType.Farthest)
                {
                    enemiesInRadius = enemiesInRadius.OrderByDescending(x => Vector3.Distance(origin, x.transform.position)).ToList();
                }

                return enemiesInRadius;
            }
        }
    }
}