SPT-Server-Build/Libraries/SPTarkov.Server.Core/Extensions/TemplateItemExtensions.cs

using SPTarkov.Server.Core.Models.Common;
using SPTarkov.Server.Core.Models.Eft.Common.Tables;

namespace SPTarkov.Server.Core.Extensions
{
    public static class TemplateItemExtensions
    {
        public static IEnumerable<TemplateItem> OfClass(
            this Dictionary<MongoId, TemplateItem> templates,
            params MongoId[] baseClasses
        )
        {
            return templates.Where(x => baseClasses.Contains(x.Value.Parent)).Select(x => x.Value);
        }

        public static IEnumerable<TemplateItem> OfClass(
            this Dictionary<MongoId, TemplateItem> templates,
            Func<TemplateItem, bool> pred,
            params MongoId[] baseClasses
        )
        {
            return templates
                .Where(x => baseClasses.Contains(x.Value.Parent) && pred(x.Value))
                .Select(x => x.Value);
        }

        /// <summary>
        ///     Check if item is quest item
        /// </summary>
        /// <param name="tpl">Items tpl to check quest status of</param>
        /// <returns>true if item is flagged as quest item</returns>
        public static bool IsQuestItem(this TemplateItem templateItem)
        {
            if (templateItem.Properties.QuestItem.GetValueOrDefault(false))
            {
                return true;
            }

            return false;
        }

        /// <summary>
        ///     Get a weapons default magazine template id
        /// </summary>
        /// <param name="weaponTemplate">Weapon to get default magazine for</param>
        /// <returns>Tpl of magazine</returns>
        public static string? GetWeaponsDefaultMagazineTpl(this TemplateItem weaponTemplate)
        {
            return weaponTemplate.Properties.DefMagType;
        }

        /// <summary>
        ///     Get the default plate an armor has in its db item
        /// </summary>
        /// <param name="armorItem">Item to look up default plate</param>
        /// <param name="modSlot">front/back</param>
        /// <returns>Tpl of plate</returns>
        public static string? GetDefaultPlateTpl(this TemplateItem armorItem, string modSlot)
        {
            var relatedItemDbModSlot = armorItem.Properties.Slots?.FirstOrDefault(slot =>
                string.Equals(slot.Name, modSlot, StringComparison.OrdinalIgnoreCase)
            );

            return relatedItemDbModSlot?.Props?.Filters?.FirstOrDefault()?.Plate;
        }
    }
}