Menu.cs

namespace SDP_Project;

public class Menu(string name) : MenuComponent
{
    public List<MenuComponent> Components { get; set; } = [];
    public string Name { get; set; } = name;
    public int ItemCount
    {
        get
        {
            var count = 0;
            var iter = CreateIterator();
            while (iter.HasNext())
            {
                if ((MenuComponent)iter.Next() is MenuItem) count++;
                
            }
            return count;
        }
    }

    public override void Add(MenuComponent component)
    {
        Components.Add(component);
    }

    public override void Remove(MenuComponent component)
    {
        Components.Remove(component);
    }

    public override MenuComponent GetChild(int index)
    {
        return Components[index];
    }

    public MenuItem GetItem(int index)
    {
        var iter = CreateIterator();
        while (iter.HasNext())
        {
            var component = (MenuComponent)iter.Next();
            if (component is MenuItem item && index-- == 0) return item;
        }

        throw new ArgumentOutOfRangeException();
    }

    public override string ToString() => Name;

    public override Iterator CreateIterator()
    {
        return new CompositeMenuIterator(new FlatMenuIterator(Components));
    }
}