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 List<RibbonInfo> GetRibbonInfo(PKM pk)
    {
        // Get a list of all Ribbon Attributes in the PKM
        var riblist = new List<RibbonInfo>();
        var names = ReflectUtil.GetPropertiesStartWithPrefix(pk.GetType(), PropertyPrefix);
        foreach (var name in names)
        {
            object? RibbonValue = ReflectUtil.GetValue(pk, 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;
    }
}