Files
MendingBeGone/src/main/java/me/youhavetrouble/mendingbegone/MendingBeGone.java
T

128 lines
5.3 KiB
Java

package me.youhavetrouble.mendingbegone;
import com.destroystokyo.paper.event.entity.EntityAddToWorldEvent;
import org.bukkit.Material;
import org.bukkit.enchantments.Enchantment;
import org.bukkit.entity.EntityType;
import org.bukkit.event.EventHandler;
import org.bukkit.event.EventPriority;
import org.bukkit.event.Listener;
import org.bukkit.event.entity.EntityPickupItemEvent;
import org.bukkit.event.inventory.InventoryClickEvent;
import org.bukkit.event.inventory.InventoryOpenEvent;
import org.bukkit.event.player.PlayerJoinEvent;
import org.bukkit.inventory.ItemStack;
import org.bukkit.inventory.Merchant;
import org.bukkit.inventory.MerchantInventory;
import org.bukkit.inventory.MerchantRecipe;
import org.bukkit.inventory.meta.EnchantmentStorageMeta;
import org.bukkit.inventory.meta.ItemMeta;
import org.bukkit.plugin.java.JavaPlugin;
import java.util.ArrayList;
import java.util.List;
public final class MendingBeGone extends JavaPlugin implements Listener {
@Override
public void onEnable() {
getLogger().info("Mending enchantment will be replaced with unbreaking 3");
getServer().getPluginManager().registerEvents(this, this);
}
@EventHandler(priority = EventPriority.HIGH, ignoreCancelled = true)
public void onMendingItemPickup(EntityPickupItemEvent event) {
replaceMendingOnItem(event.getItem().getItemStack());
}
@EventHandler(priority = EventPriority.HIGH, ignoreCancelled = true)
public void onInvOpen(InventoryOpenEvent event) {
for (ItemStack itemStack : event.getInventory().getContents()) {
replaceMendingOnItem(itemStack);
}
}
@EventHandler(priority = EventPriority.HIGH, ignoreCancelled = true)
public void onLogin(PlayerJoinEvent event) {
for (ItemStack itemStack : event.getPlayer().getInventory()) {
replaceMendingOnItem(itemStack);
}
}
@EventHandler(priority = EventPriority.HIGH, ignoreCancelled = true)
public void onInvClick(InventoryClickEvent event) {
replaceMendingOnItem(event.getCurrentItem());
replaceMendingOnItem(event.getCursor());
}
@EventHandler(priority = EventPriority.HIGH, ignoreCancelled = true)
public void onOpenVillagerInventory(InventoryOpenEvent event) {
if (!(event.getInventory() instanceof MerchantInventory)) return;
MerchantInventory merchantInventory = (MerchantInventory) event.getInventory();
removeMendingTrade(merchantInventory.getMerchant());
}
@EventHandler(priority = EventPriority.HIGH, ignoreCancelled = true)
public void onVillagerJoinWorld(EntityAddToWorldEvent event) {
if (!event.getEntityType().equals(EntityType.VILLAGER)) return;
removeMendingTrade((Merchant) event.getEntity());
}
private void replaceMendingOnItem(ItemStack itemStack) {
if (itemStack == null) return;
if (!itemStack.hasItemMeta()) return;
if (itemStack.containsEnchantment(Enchantment.MENDING)) {
itemStack.removeEnchantment(Enchantment.MENDING);
if (itemStack.containsEnchantment(Enchantment.DURABILITY))
itemStack.removeEnchantment(Enchantment.DURABILITY);
itemStack.addEnchantment(Enchantment.DURABILITY, 3);
}
if (!Material.ENCHANTED_BOOK.equals(itemStack.getType())) return;
ItemMeta meta = itemStack.getItemMeta();
if (!(meta instanceof EnchantmentStorageMeta)) return;
EnchantmentStorageMeta storage = (EnchantmentStorageMeta) meta;
if (storage.hasStoredEnchant(Enchantment.MENDING)) {
storage.removeStoredEnchant(Enchantment.MENDING);
if (storage.hasStoredEnchant(Enchantment.DURABILITY))
storage.removeStoredEnchant(Enchantment.DURABILITY);
storage.addStoredEnchant(Enchantment.DURABILITY, 3, true);
itemStack.setItemMeta(storage);
}
}
private void removeMendingTrade(Merchant merchant) {
List<MerchantRecipe> trades = new ArrayList<>(merchant.getRecipes());
List<Integer> toReplace = new ArrayList<>();
for (int i = 0; i < trades.size(); i++) {
MerchantRecipe recipe = trades.get(i);
if (!recipe.getResult().getType().equals(Material.ENCHANTED_BOOK)) continue;
if (!(recipe.getResult().getItemMeta() instanceof EnchantmentStorageMeta)) continue;
EnchantmentStorageMeta storage = (EnchantmentStorageMeta) recipe.getResult().getItemMeta();
if (!storage.hasStoredEnchant(Enchantment.MENDING)) continue;
toReplace.add(i);
}
if (toReplace.isEmpty()) return;
for (int index : toReplace) {
MerchantRecipe oldTrade = trades.get(index);
ItemStack result = oldTrade.getResult();
replaceMendingOnItem(result);
MerchantRecipe newTrade = new MerchantRecipe(
result,
oldTrade.getUses(),
oldTrade.getMaxUses(),
oldTrade.hasExperienceReward(),
oldTrade.getVillagerExperience(),
oldTrade.getPriceMultiplier()
);
for (ItemStack ingredient : oldTrade.getIngredients()) {
newTrade.addIngredient(ingredient);
}
trades.set(index, newTrade);
}
merchant.setRecipes(trades);
}
}