package com.kodari.omnitool; import com.cryptomorin.xseries.XEnchantment; import com.cryptomorin.xseries.XItemFlag; import com.cryptomorin.xseries.XMaterial; import org.bukkit.NamespacedKey; import org.bukkit.inventory.ItemStack; import org.bukkit.inventory.meta.ItemMeta; import org.bukkit.persistence.PersistentDataContainer; import org.bukkit.persistence.PersistentDataType; import java.util.*; import java.util.stream.Collectors; public class OmniToolManager { public static final Map> ENCHANTS; static { ENCHANTS = new EnumMap<>(ToolMode.class); ENCHANTS.put(ToolMode.SWORD, Arrays.asList( new EnchantOption("SHARPNESS", "Sharpness", 5, "DIAMOND_SWORD"), new EnchantOption("SMITE", "Smite", 5, "BONE"), new EnchantOption("BANE_OF_ARTHROPODS", "Bane of Arthropods", 5, "SPIDER_EYE"), new EnchantOption("KNOCKBACK", "Knockback", 2, "SLIME_BALL"), new EnchantOption("FIRE_ASPECT", "Fire Aspect", 2, "FIRE_CHARGE"), new EnchantOption("LOOTING", "Looting", 3, "GOLD_INGOT"), new EnchantOption("SWEEPING_EDGE", "Sweeping Edge", 3, "IRON_SWORD") )); ENCHANTS.put(ToolMode.PICKAXE, Arrays.asList( new EnchantOption("EFFICIENCY", "Efficiency", 5, "FEATHER"), new EnchantOption("UNBREAKING", "Unbreaking", 3, "OBSIDIAN"), new EnchantOption("FORTUNE", "Fortune", 3, "EMERALD"), new EnchantOption("SILK_TOUCH", "Silk Touch", 1, "STRING") )); ENCHANTS.put(ToolMode.HOE, Arrays.asList( new EnchantOption("EFFICIENCY", "Efficiency", 5, "FEATHER"), new EnchantOption("UNBREAKING", "Unbreaking", 3, "OBSIDIAN"), new EnchantOption("FORTUNE", "Fortune", 3, "EMERALD"), new EnchantOption("SILK_TOUCH", "Silk Touch", 1, "STRING") )); ENCHANTS.put(ToolMode.FISHING_ROD, Arrays.asList( new EnchantOption("LUCK_OF_THE_SEA", "Luck of the Sea", 3, "TROPICAL_FISH"), new EnchantOption("LURE", "Lure", 3, "NAUTILUS_SHELL"), new EnchantOption("UNBREAKING", "Unbreaking", 3, "OBSIDIAN"), new EnchantOption("MENDING", "Mending", 1, "EXPERIENCE_BOTTLE") )); } private final NamespacedKey keyIsOmniTool; private final NamespacedKey keyMode; private final Map enchantKeys; public OmniToolManager(OmniToolPlugin plugin) { this.keyIsOmniTool = new NamespacedKey(plugin, "is_omnitool"); this.keyMode = new NamespacedKey(plugin, "mode"); this.enchantKeys = new EnumMap<>(ToolMode.class); for (ToolMode mode : ToolMode.values()) { enchantKeys.put(mode, new NamespacedKey(plugin, "enchants_" + mode.name().toLowerCase())); } } public ItemStack createOmniTool() { ItemStack item = XMaterial.matchXMaterial("DIAMOND_SWORD") .map(XMaterial::parseItem) .orElse(null); if (item == null) return null; ItemMeta meta = item.getItemMeta(); if (meta == null) return item; meta.setDisplayName("§6§l✦ §r§6OmniTool §7- §eSword"); meta.setLore(Collections.singletonList("§7§oShift + Right Click to customize")); meta.setUnbreakable(true); XItemFlag.of("HIDE_UNBREAKABLE").ifPresent(f -> f.set(meta)); XItemFlag.of("HIDE_ATTRIBUTES").ifPresent(f -> f.set(meta)); PersistentDataContainer pdc = meta.getPersistentDataContainer(); pdc.set(keyIsOmniTool, PersistentDataType.BYTE, (byte) 1); pdc.set(keyMode, PersistentDataType.STRING, ToolMode.SWORD.name()); for (ToolMode mode : ToolMode.values()) { pdc.set(enchantKeys.get(mode), PersistentDataType.STRING, ""); } item.setItemMeta(meta); return item; } public boolean isOmniTool(ItemStack item) { if (item == null || !item.hasItemMeta()) return false; return item.getItemMeta().getPersistentDataContainer().has(keyIsOmniTool, PersistentDataType.BYTE); } public ToolMode getMode(ItemStack item) { if (!item.hasItemMeta()) return ToolMode.SWORD; String raw = item.getItemMeta().getPersistentDataContainer() .getOrDefault(keyMode, PersistentDataType.STRING, "SWORD"); try { return ToolMode.valueOf(raw); } catch (IllegalArgumentException e) { return ToolMode.SWORD; } } public Map getStoredEnchants(ItemStack item, ToolMode mode) { if (!item.hasItemMeta()) return new LinkedHashMap<>(); String data = item.getItemMeta().getPersistentDataContainer() .getOrDefault(enchantKeys.get(mode), PersistentDataType.STRING, ""); Map result = new LinkedHashMap<>(); if (data.isEmpty()) return result; for (String part : data.split(",")) { String[] kv = part.split(":"); if (kv.length == 2) { try { result.put(kv[0], Integer.parseInt(kv[1])); } catch (NumberFormatException ignored) {} } } return result; } public void saveStoredEnchants(ItemStack item, ToolMode mode, Map enchants) { if (!item.hasItemMeta()) return; ItemMeta meta = item.getItemMeta(); StringBuilder sb = new StringBuilder(); enchants.forEach((name, level) -> { if (sb.length() > 0) sb.append(","); sb.append(name).append(":").append(level); }); meta.getPersistentDataContainer().set(enchantKeys.get(mode), PersistentDataType.STRING, sb.toString()); item.setItemMeta(meta); } public ItemStack setMode(ItemStack oldItem, ToolMode newMode) { ItemStack newItem = XMaterial.matchXMaterial(newMode.getMaterialName()) .map(XMaterial::parseItem) .orElse(null); if (newItem == null) return oldItem; ItemMeta oldMeta = oldItem.getItemMeta(); ItemMeta newMeta = newItem.getItemMeta(); if (oldMeta == null || newMeta == null) return oldItem; // Copy all PDC keys from old item to new item PersistentDataContainer oldPdc = oldMeta.getPersistentDataContainer(); PersistentDataContainer newPdc = newMeta.getPersistentDataContainer(); copyPDC(oldPdc, newPdc); // Update mode key newPdc.set(keyMode, PersistentDataType.STRING, newMode.name()); newMeta.setUnbreakable(true); XItemFlag.of("HIDE_UNBREAKABLE").ifPresent(f -> f.set(newMeta)); XItemFlag.of("HIDE_ATTRIBUTES").ifPresent(f -> f.set(newMeta)); newItem.setItemMeta(newMeta); // Apply stored enchants for new mode Map enchants = getStoredEnchants(newItem, newMode); enchants.forEach((name, level) -> XEnchantment.matchXEnchantment(name).ifPresent(xe -> newItem.addUnsafeEnchantment(xe.getEnchant(), level) ) ); updateItemDisplay(newItem); return newItem; } public void cycleEnchant(ItemStack item, EnchantOption option, boolean remove) { ToolMode mode = getMode(item); Map enchants = getStoredEnchants(item, mode); String name = option.getXEnchantName(); if (remove) { enchants.remove(name); } else { int current = enchants.getOrDefault(name, 0); int next = current + 1; if (next > option.getMaxLevel()) { enchants.remove(name); } else { // Silk Touch / Fortune are mutually exclusive if (name.equals("SILK_TOUCH")) enchants.remove("FORTUNE"); if (name.equals("FORTUNE")) enchants.remove("SILK_TOUCH"); enchants.put(name, next); } } saveStoredEnchants(item, mode, enchants); // Sync actual item enchants with stored data new HashSet<>(item.getEnchantments().keySet()).forEach(item::removeEnchantment); enchants.forEach((eName, level) -> XEnchantment.matchXEnchantment(eName).ifPresent(xe -> item.addUnsafeEnchantment(xe.getEnchant(), level) ) ); } public void updateItemDisplay(ItemStack item) { if (!item.hasItemMeta()) return; ItemMeta meta = item.getItemMeta(); ToolMode mode = getMode(item); meta.setDisplayName("§6§l✦ §r§6OmniTool §7- §e" + mode.getDisplayName()); List lore = new ArrayList<>(); lore.add("§8§m §r §7Mode: §e" + mode.getDisplayName() + " §8§m "); Map enchants = getStoredEnchants(item, mode); if (enchants.isEmpty()) { lore.add("§7No enchantments applied"); } else { enchants.forEach((name, level) -> { String pretty = formatEnchantName(name); String levelStr = level == 1 ? "" : " " + toRoman(level); lore.add("§b " + pretty + levelStr); }); } lore.add(""); lore.add("§7§oShift + Right Click to customize"); meta.setLore(lore); item.setItemMeta(meta); } private void copyPDC(PersistentDataContainer from, PersistentDataContainer to) { if (from.has(keyIsOmniTool, PersistentDataType.BYTE)) { to.set(keyIsOmniTool, PersistentDataType.BYTE, Objects.requireNonNull(from.get(keyIsOmniTool, PersistentDataType.BYTE))); } if (from.has(keyMode, PersistentDataType.STRING)) { to.set(keyMode, PersistentDataType.STRING, Objects.requireNonNull(from.get(keyMode, PersistentDataType.STRING))); } for (ToolMode mode : ToolMode.values()) { NamespacedKey key = enchantKeys.get(mode); if (from.has(key, PersistentDataType.STRING)) { to.set(key, PersistentDataType.STRING, Objects.requireNonNull(from.get(key, PersistentDataType.STRING))); } } } private String formatEnchantName(String name) { return Arrays.stream(name.split("_")) .map(w -> Character.toUpperCase(w.charAt(0)) + w.substring(1).toLowerCase()) .collect(Collectors.joining(" ")); } private String toRoman(int num) { String[] r = {"", "I", "II", "III", "IV", "V"}; return num >= 0 && num < r.length ? r[num] : String.valueOf(num); } }