using System;
using System.Linq;

namespace Barotrauma.Abilities
{
    class CharacterAbilityAtmosMachine : CharacterAbility
    {
        private readonly float addedValue;
        private readonly Identifier[] tags;
        private readonly int maxMultiplyCount;
        
        public CharacterAbilityAtmosMachine(CharacterAbilityGroup characterAbilityGroup, ContentXElement abilityElement) : base(characterAbilityGroup, abilityElement)
        {
            addedValue = abilityElement.GetAttributeFloat("addedvalue", 0f);
            tags = abilityElement.GetAttributeIdentifierArray("tags", Array.Empty<Identifier>());
            maxMultiplyCount = abilityElement.GetAttributeInt("maxmultiplycount", int.MaxValue);
        }

        protected override void ApplyEffect(AbilityObject abilityObject)
        {
            if (abilityObject is IAbilityValue abilityValue)
            {
                int multiplyCount = 0;

                foreach (Item item in Item.ItemList)
                {
                    if (item.Prefab.Tags.Any(t => tags.Contains(t)))
                    {
                        multiplyCount++;
                        if (multiplyCount == maxMultiplyCount)
                        {
                            break;
                        }
                    }
                }
                abilityValue.Value += addedValue * multiplyCount;
            }
        }
    }
}