using ElectricLocomotive.Exceptions;

namespace ElectricLocomotive;

public class SetGeneric <T> where T : class
{
    private readonly List<T?> _places;

    public int Count => _places.Count;
    private readonly int _maxCount;

    public SetGeneric(int count)
    {
        _maxCount = count;
        _places = new List<T>(count);
    }
    public bool Insert(T electricLocomotiv)
    {
        if (_places.Count == _maxCount)
            throw new StorageOverflowException(_maxCount);
        Insert(electricLocomotiv, 0);
        return true;
    }

    public bool Insert(T electricLocomotiv, int position)
    {
        if (_places.Count == _maxCount)
            throw new StorageOverflowException(_maxCount);
        
        if (!(position >= 0 && position <= Count))
            return false;
        _places.Insert(position, electricLocomotiv);
        return true;
    }

    public bool Remove(int position)
    {
        if (!(position >= 0 && position < Count))
            throw new LocoNotFoundException(position);
        _places.RemoveAt(position);
        return true;
    }

    public T? this[int position]
    {
        get {
            if (!(position >= 0 && position < Count))
                return null;
            return _places[position];
        }
        set {
            if (!(position >= 0 && position < Count && _places.Count < _maxCount))
                return;
            _places.Insert(position, value);
            return;
        }
    }
    public IEnumerable<T?> GetElectricLocos(int? maxElectricLocos = null)
    {
        for (int i = 0; i < _places.Count; ++i)
        {
            yield return _places[i];
            if (maxElectricLocos.HasValue && i == maxElectricLocos.Value)
            {
                yield break;
            }
        }
    }

}