PKHeX/PKHeX.Core/Ribbons/RibbonInfo.cs

using System.Collections.Generic;

namespace PKHeX.Core
{
    /// <summary>
    /// Provides ribbon information about the state of a given ribbon.
    /// </summary>
    public sealed class RibbonInfo
    {
        public const string PropertyPrefix = "Ribbon";

        public readonly string Name;
        public bool HasRibbon { get; set; }
        public int RibbonCount { get; set; }

        private RibbonInfo(string name, bool hasRibbon)
        {
            Name = name;
            HasRibbon = hasRibbon;
            RibbonCount = -1;
        }

        private RibbonInfo(string name, int count)
        {
            Name = name;
            HasRibbon = false;
            RibbonCount = count;
        }

        public int MaxCount
        {
            get
            {
                if (RibbonCount < 0)
                    return -1;
                return Name switch
                {
                    nameof(IRibbonSetCommon6.RibbonCountMemoryContest) => 40,
                    nameof(IRibbonSetCommon6.RibbonCountMemoryBattle) => 8,
                    _ => 4,
                };
            }
        }

        public static IReadOnlyList<RibbonInfo> GetRibbonInfo(PKM pkm)
        {
            // Get a list of all Ribbon Attributes in the PKM
            var riblist = new List<RibbonInfo>();
            var names = ReflectUtil.GetPropertiesStartWithPrefix(pkm.GetType(), PropertyPrefix);
            foreach (var name in names)
            {
                object? RibbonValue = ReflectUtil.GetValue(pkm, name);
                if (RibbonValue is int x)
                    riblist.Add(new RibbonInfo(name, x));
                if (RibbonValue is bool b)
                    riblist.Add(new RibbonInfo(name, b));
            }

            return riblist;
        }
    }
}