using UnityEngine;

namespace Kryz.CharacterStats.Examples
{
	public enum EquipmentType
	{
		Helmet,
		Chest,
		Gloves,
		Boots,
		MainWeapon,
		OffWeapon,
		Accessory,
		Amulet,
	}

	[CreateAssetMenu]
	public class EquippableItem : Item
	{
		public int StrengthBonus;
		public int AgilityBonus;
		public int IntelligenceBonus;
		public int SpiritBonus;
		public int VitalityBonus;
		[Space]
		public float StrengthPercentBonus;
		public float AgilityPercentBonus;
		public float IntelligencePercentBonus;
		public float SpiritPercentBonus;
		public float VitalityPercentBonus;
		[Space]
		public EquipmentType EquipmentType;

		public void Equip(PlayerCharacterStats c)
		{
			if (StrengthBonus != 0)
				c.Strength.AddModifier(new StatModifier(StrengthBonus, StatModType.Flat, this));
			if (AgilityBonus != 0)
				c.Agility.AddModifier(new StatModifier(AgilityBonus, StatModType.Flat, this));
			if (IntelligenceBonus != 0)
				c.Intelligence.AddModifier(new StatModifier(IntelligenceBonus, StatModType.Flat, this));
			if (SpiritBonus != 0)
				c.Spirit.AddModifier(new StatModifier(SpiritBonus, StatModType.Flat, this));
			if (VitalityBonus != 0)
				c.Vitality.AddModifier(new StatModifier(VitalityBonus, StatModType.Flat, this));

			if (StrengthPercentBonus != 0)
				c.Strength.AddModifier(new StatModifier(StrengthPercentBonus, StatModType.PercentMult, this));
			if (AgilityPercentBonus != 0)
				c.Agility.AddModifier(new StatModifier(AgilityPercentBonus, StatModType.PercentMult, this));
			if (IntelligencePercentBonus != 0)
				c.Intelligence.AddModifier(new StatModifier(IntelligencePercentBonus, StatModType.PercentMult, this));
			if (SpiritPercentBonus != 0)
				c.Spirit.AddModifier(new StatModifier(SpiritPercentBonus, StatModType.PercentMult, this));
			if (VitalityPercentBonus != 0)
				c.Vitality.AddModifier(new StatModifier(VitalityPercentBonus, StatModType.PercentMult, this));
		}

		public void Unequip(PlayerCharacterStats c)
		{
			c.Strength.RemoveAllModifiersFromSource(this);
			c.Agility.RemoveAllModifiersFromSource(this);
			c.Intelligence.RemoveAllModifiersFromSource(this);
			c.Spirit.RemoveAllModifiersFromSource(this);
			c.Vitality.RemoveAllModifiersFromSource(this);
		}
	}
}