CinderLoE/src (1)/main/java/com/zivilon/cinder_loe/mixins/MixinLOTREnchantment.java

package com.zivilon.cinder_loe.mixins;

import com.zivilon.cinder_loe.enchants.LOTREnchantmentWeakProtectionRanged;
import com.zivilon.cinder_loe.enchants.LOTREnchantmentArmorSpecial;
import com.zivilon.cinder_loe.util.Utilities;
import lotr.common.enchant.*;
import net.minecraft.item.ItemStack;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Overwrite;
import org.spongepowered.asm.mixin.Shadow;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;

import java.lang.reflect.Field;
import java.util.Map;

@Mixin(LOTREnchantment.class)
public class MixinLOTREnchantment {
    @Shadow
    private boolean bypassAnvilLimit;

    @Inject(method = "<clinit>", at = @At("TAIL"))
    private static void onStaticInit(CallbackInfo ci) {
        // Add your new enchantments here
        try {
            LOTREnchantment protectRangedWeak1 = new LOTREnchantmentWeakProtectionRanged("protectRangedWeak1", -1).setEnchantWeight(0);
            LOTREnchantment protectRangedWeak2 = new LOTREnchantmentWeakProtectionRanged("protectRangedWeak2", -2).setEnchantWeight(0);
            LOTREnchantment rangedWeak3 = new LOTREnchantmentRangedDamage("rangedWeak3", 0.25f);
            LOTREnchantment weak4 = new LOTREnchantmentDamage("weak4", -3.0f).setEnchantWeight(0);
            LOTREnchantment chill = new LOTREnchantmentDamage("strong5", 3.25f).setEnchantWeight(0);
            LOTREnchantment rangedStrong4 = new LOTREnchantmentRangedDamage("rangedStrong4", 1.4f).setEnchantWeight(0).setSkilful();
            LOTREnchantment meleeReach2 = new LOTREnchantmentMeleeReach("meleeReach2", 1.33f).setEnchantWeight(0).setSkilful();
            LOTREnchantment meleeSpeed2 = new LOTREnchantmentMeleeSpeed("meleeSpeed2", 1.33f).setEnchantWeight(0).setSkilful();
            LOTREnchantment Shinare = new LOTREnchantmentArmorSpecial("Shinare").setEnchantWeight(0).setSkilful();
            LOTREnchantment swiftness = new LOTREnchantmentArmorSpecial("swiftness").setEnchantWeight(0).setSkilful();
            LOTREnchantment fireRepair = new LOTREnchantmentArmorSpecial("fireRepair").setEnchantWeight(0).setSkilful();
            LOTREnchantment mountArmor = new LOTREnchantmentArmorSpecial("mountArmor").setEnchantWeight(0).setSkilful();
            LOTREnchantment stealth = new LOTREnchantmentArmorSpecial("stealth").setEnchantWeight(0).setSkilful();


            LOTREnchantment.allEnchantments.add(protectRangedWeak1);
            LOTREnchantment.allEnchantments.add(protectRangedWeak2);
            rangedWeak3.allEnchantments.add(rangedWeak3);
            LOTREnchantment.allEnchantments.add(weak4);
            LOTREnchantment.allEnchantments.add(chill);
            LOTREnchantment.allEnchantments.add(rangedStrong4);
            LOTREnchantment.allEnchantments.add(meleeReach2);
            LOTREnchantment.allEnchantments.add(meleeSpeed2);
            LOTREnchantment.allEnchantments.add(Shinare);
            LOTREnchantment.allEnchantments.add(swiftness);
            LOTREnchantment.allEnchantments.add(fireRepair);
            LOTREnchantment.allEnchantments.add(mountArmor);
            LOTREnchantment.allEnchantments.add(stealth);

            Field enchantsByNameField = LOTREnchantment.class.getDeclaredField("enchantsByName");
            enchantsByNameField.setAccessible(true);
            @SuppressWarnings("unchecked")
            Map<String, LOTREnchantment> enchantsByName = (Map<String, LOTREnchantment>) enchantsByNameField.get(null);

            enchantsByName.put(protectRangedWeak1.enchantName, protectRangedWeak1);
            enchantsByName.put(protectRangedWeak2.enchantName, protectRangedWeak2);
            enchantsByName.put(rangedWeak3.enchantName, rangedWeak3);
            enchantsByName.put(weak4.enchantName, weak4);
            enchantsByName.put(chill.enchantName, chill);
            enchantsByName.put(rangedStrong4.enchantName, rangedStrong4);
            enchantsByName.put(meleeReach2.enchantName, meleeReach2);
            enchantsByName.put(meleeSpeed2.enchantName, meleeSpeed2);
            enchantsByName.put(Shinare.enchantName, Shinare);
            enchantsByName.put(swiftness.enchantName, swiftness);
            enchantsByName.put(fireRepair.enchantName, fireRepair);
            enchantsByName.put(mountArmor.enchantName, mountArmor);
            enchantsByName.put(stealth.enchantName, stealth);

        } catch (NoSuchFieldException | IllegalAccessException e) {
            e.printStackTrace();
        }
    }

    /**
    * @author MrJeep20
    * @reason Bad enchantments will not count towards 3 modifier limit now
    **/
    @Overwrite(remap = false)
    public boolean bypassAnvilLimit() {
        if (Utilities.isBadEnch((LOTREnchantment)(Object)this))
            return true;
        return this.bypassAnvilLimit;
    }

}