Added ability to End Leggings Item

src/main/java/com/majruszs_difficulty/items/EndLeggingsItem.java

@@ -1,10 +1,131 @@
 package com.majruszs_difficulty.items;
 
+import com.majruszs_difficulty.Instances;
+import com.mlib.TimeConverter;
+import com.mlib.attributes.AttributeHandler;
+import com.mlib.config.ConfigGroup;
+import com.mlib.config.DoubleConfig;
+import net.minecraft.client.util.ITooltipFlag;
+import net.minecraft.enchantment.Enchantment;
+import net.minecraft.enchantment.EnchantmentHelper;
+import net.minecraft.entity.LivingEntity;
+import net.minecraft.entity.ai.attributes.AttributeModifier;
+import net.minecraft.entity.ai.attributes.Attributes;
+import net.minecraft.entity.player.PlayerEntity;
 import net.minecraft.inventory.EquipmentSlotType;
+import net.minecraft.item.ItemStack;
+import net.minecraft.nbt.CompoundNBT;
+import net.minecraft.util.text.*;
+import net.minecraft.world.World;
+import net.minecraft.world.server.ServerWorld;
+import net.minecraftforge.api.distmarker.Dist;
+import net.minecraftforge.api.distmarker.OnlyIn;
+import net.minecraftforge.event.entity.living.LivingEquipmentChangeEvent;
+import net.minecraftforge.event.entity.living.LivingEvent;
+import net.minecraftforge.eventbus.api.SubscribeEvent;
+import net.minecraftforge.fml.common.Mod;
+
+import javax.annotation.Nullable;
+import java.util.List;
+import java.util.Map;
+
+import static com.majruszs_difficulty.MajruszsDifficulty.FEATURES_GROUP;
 
 /** New late game leggings. */
+@Mod.EventBusSubscriber
 public class EndLeggingsItem extends EndArmorItem {
+	private static final String ARMOR_TAG = "EndLeggingsArmorTag";
+	private static final String ARMOR_BONUS_TAG = "EndLeggingsArmorBonus";
+	private static final String ARMOR_COUNTER_TAG = "EndLeggingsCounter";
+	protected static final AttributeHandler ARMOR_ATTRIBUTE = new AttributeHandler( "4290dea8-ef26-40bc-8f7f-28912d31329a", "EndLeggingsArmorBonus",
+		Attributes.ARMOR, AttributeModifier.Operation.ADDITION
+	);
+	protected final ConfigGroup configGroup;
+	protected final DoubleConfig armorBonus;
+
 	public EndLeggingsItem() {
 		super( EquipmentSlotType.LEGS );
+
+		this.configGroup = new ConfigGroup( "EndLeggings", "Config for End Leggings ability." );
+		FEATURES_GROUP.addGroup( this.configGroup );
+
+		String comment = "Armor bonus for every enchantment level.";
+		this.armorBonus = new DoubleConfig( "armor_bonus", comment, false, 0.2, 0.0, 10.0 );
+		this.configGroup.addConfig( this.armorBonus );
+	}
+
+	@Override
+	@OnlyIn( Dist.CLIENT )
+	public void addInformation( ItemStack stack, @Nullable World world, List< ITextComponent > toolTip, ITooltipFlag flag ) {
+		super.addInformation( stack, world, toolTip, flag );
+
+		CompoundNBT tag = stack.getOrCreateChildTag( ARMOR_TAG );
+		double bonus = tag.getDouble( ARMOR_BONUS_TAG );
+
+		if( bonus > 0.0 )
+			toolTip.add( getArmorBonusTooltip( bonus ) );
+	}
+
+	@SubscribeEvent
+	public static void applyArmorBonus( LivingEquipmentChangeEvent event ) {
+		LivingEntity entity = event.getEntityLiving();
+		double armorBonus = calculateEnchantmentBonus( entity );
+		ItemStack leggingsItemStack = entity.getItemStackFromSlot( EquipmentSlotType.LEGS );
+		boolean hasEndLeggings = leggingsItemStack.getItem() instanceof EndLeggingsItem;
+
+		updateLeggingsTag( entity.getItemStackFromSlot( EquipmentSlotType.LEGS ), armorBonus );
+		ARMOR_ATTRIBUTE.setValueAndApply( entity, hasEndLeggings ? armorBonus : 0.0 );
+	}
+
+	@SubscribeEvent
+	public static void updateArmorBonusOnAllItems( LivingEvent.LivingUpdateEvent event ) {
+		LivingEntity entity = event.getEntityLiving();
+		if( !( entity.world instanceof ServerWorld ) || !( entity instanceof PlayerEntity ) )
+			return;
+
+		CompoundNBT data = entity.getPersistentData();
+		int counter = ( data.getInt( ARMOR_COUNTER_TAG ) + 1 ) % TimeConverter.secondsToTicks( 1.0 );
+		data.putInt( ARMOR_COUNTER_TAG, counter );
+
+		if( counter == 0 )
+			for( ItemStack itemStack : ( ( PlayerEntity )entity ).inventory.mainInventory )
+				updateLeggingsTag( itemStack, 0.0 );
+	}
+
+	/** Calculates current armor bonus depending on enchantment levels. */
+	private static double calculateEnchantmentBonus( LivingEntity entity ) {
+		double armorBonus = Instances.END_LEGGINGS_ITEM.armorBonus.get();
+		int enchantmentLevelSum = 0;
+		ItemStack leggings = entity.getItemStackFromSlot( EquipmentSlotType.LEGS );
+		if( leggings.getItem() instanceof EndLeggingsItem ) {
+			for( ItemStack armorPiece : entity.getArmorInventoryList() ) {
+				if( armorPiece.isEmpty() )
+					continue;
+
+				Map< Enchantment, Integer > enchantments = EnchantmentHelper.getEnchantments( armorPiece );
+				for( Map.Entry< Enchantment, Integer > enchantment : enchantments.entrySet() )
+					enchantmentLevelSum += enchantment.getValue();
+			}
+		}
+
+		return enchantmentLevelSum * armorBonus;
+	}
+
+	/** Updates armor bonus value stored in leggings data. */
+	private static void updateLeggingsTag( ItemStack leggings, double armorBonus ) {
+		if( leggings.isEmpty() || !( leggings.getItem() instanceof EndLeggingsItem ) )
+			return;
+
+		CompoundNBT tag = leggings.getOrCreateChildTag( ARMOR_TAG );
+		tag.putDouble( ARMOR_BONUS_TAG, armorBonus );
+		leggings.setTagInfo( ARMOR_TAG, tag );
+	}
+
+	/** Returns text with current armor bonus. */
+	@OnlyIn( Dist.CLIENT )
+	private IFormattableTextComponent getArmorBonusTooltip( double armorBonus ) {
+		IFormattableTextComponent text = new StringTextComponent( "+" ).appendString( armorBonus + " " );
+		return text.append( new TranslationTextComponent( "item.majruszs_difficulty.end_leggings.item_tooltip" ) )
+			.mergeStyle( TextFormatting.GRAY );
 	}
 }

src/main/resources/assets/majruszs_difficulty/lang/en_us.json

@@ -37,9 +37,13 @@
   "item.majruszs_difficulty.end_axe": "End Axe",
   "item.majruszs_difficulty.end_hoe": "End Hoe",
   "item.majruszs_difficulty.end_helmet": "End Helmet",
+  "item.majruszs_difficulty.end_helmet.item_tooltip": "armor",
   "item.majruszs_difficulty.end_chestplate": "End Chestplate",
+  "item.majruszs_difficulty.end_chestplate.item_tooltip": "armor",
   "item.majruszs_difficulty.end_leggings": "End Leggings",
+  "item.majruszs_difficulty.end_leggings.item_tooltip": "armor bonus from enchantment levels",
   "item.majruszs_difficulty.end_boots": "End Boots",
+  "item.majruszs_difficulty.end_boots.item_tooltip": "armor",
   "item.majruszs_difficulty.tattered_cloth": "Tattered Cloth",
   "item.majruszs_difficulty.end_shard_locator": "End Shard Locator",
   "item.majruszs_difficulty.end_shard_locator.item_tooltip": "Informs the player how far it is from the End Shard Ore.",