wwdpublic/Content.Shared/Prototypes/CharacterItemGroupPrototype.cs

using System.Diagnostics.CodeAnalysis;
using System.Linq;
using Content.Shared.Clothing.Loadouts.Prototypes;
using Content.Shared.Clothing.Loadouts.Systems;
using Content.Shared.Preferences;
using Content.Shared.Traits;
using Robust.Shared.Prototypes;
using Robust.Shared.Utility;

namespace Content.Shared.Prototypes;

[Prototype]
public sealed partial class CharacterItemGroupPrototype : IPrototype
{
    [IdDataField]
    public string ID { get; } = default!;

    /// How many items from this group can be selected at once
    [DataField]
    public int MaxItems = 1;

    /// An arbitrary list of traits, loadouts, etc
    [DataField]
    public List<CharacterItemGroupItem> Items = new();
}

[DataDefinition]
public sealed partial class CharacterItemGroupItem
{
    [DataField(required: true)]
    public string Type;

    [DataField("id", required: true)]
    public string ID;

    /// Tries to get Value from whatever Type maps to on a character profile
    //TODO: Make a test for this
    public bool TryGetValue(HumanoidCharacterProfile profile, IPrototypeManager protoMan, [NotNullWhen(true)] out object? value)
    {
        value = null;

        // This sucks
        switch (Type)
        {
            case "trait":
                // RobustToolbox my beloved
                value = profile.TraitPreferences.FirstOrNull(
                    p => protoMan.Index<TraitPrototype>((string) p).ID == ID);
                return value != null;
            case "loadout":
                return profile.LoadoutPreferences.TryFirstOrDefault(
                    p => protoMan.Index<LoadoutPrototype>(((Loadout) p).LoadoutName).ID == ID, out value);
            default:
                DebugTools.Assert($"Invalid CharacterItemGroupItem Type: {Type}");
                return false;
        }
    }
}