using System.Collections.Generic;
using MRBot.Extensions;

namespace MRBot.Bot.AbilityExporter
{
    internal class AbilityList : Dictionary<string, List<ExportableAbility>>
    {
        public void AddAbility(ExportableAbility exportableAbility)
        {
            var abilityLineName = exportableAbility.Name.RemoveRomanNumeralsFromEnd();
            if (!this.ContainsKey(abilityLineName))
            {
                this.Add(abilityLineName, new List<ExportableAbility>());
            }
            
            this[abilityLineName].Add(exportableAbility);
        }
    }
}