package com.kunthean.pensell.inventory.gui; import com.kunthean.pensell.PenSell; import com.kunthean.pensell.util.ColorUtil; import com.kunthean.pensell.util.FormatUtil; import io.netty.channel.Channel; import io.netty.channel.ChannelDuplexHandler; import io.netty.channel.ChannelHandlerContext; import io.netty.channel.ChannelPromise; import org.bukkit.Bukkit; import org.bukkit.ChatColor; import org.bukkit.GameMode; import org.bukkit.Material; import org.bukkit.block.ShulkerBox; import org.bukkit.enchantments.Enchantment; import org.bukkit.entity.Player; import org.bukkit.event.inventory.InventoryType; import org.bukkit.inventory.InventoryView; import org.bukkit.inventory.ItemStack; import org.bukkit.inventory.meta.BlockStateMeta; import org.bukkit.inventory.meta.EnchantmentStorageMeta; import org.bukkit.inventory.meta.ItemMeta; import org.bukkit.inventory.meta.PotionMeta; import java.lang.reflect.Constructor; import java.lang.reflect.Field; import java.lang.reflect.Method; import java.util.*; import java.util.concurrent.ConcurrentHashMap; import java.util.stream.Collectors; public final class WorthLorePacketService { private static final String HANDLER_PREFIX = "pensell_worth_lore_"; private static final String PKT_SET_SLOT = "ClientboundContainerSetSlotPacket"; private static final String PKT_SET_CONTENT = "ClientboundContainerSetContentPacket"; private final PenSell plugin; private final ReflectionBridge bridge; private final Map handlerNames = new ConcurrentHashMap<>(); private final Map worthEnabledCache = new ConcurrentHashMap<>(); private final Map gameModeCache = new ConcurrentHashMap<>(); private final Set hasCursorItemCache = ConcurrentHashMap.newKeySet(); private final Set noWorthOpen = ConcurrentHashMap.newKeySet(); private final Set bedrockPlayers = ConcurrentHashMap.newKeySet(); private volatile boolean worthLoreEnabled = true; private volatile List loreTemplate = List.of(); private volatile List lorePlainPrefixes = List.of(); private volatile List disabledGuiNames = List.of(); private volatile String worthLoreFormat = "&7Worth: fb00$%price%"; private volatile Set blacklistCache = Set.of(); private volatile Map itemPriceCache = Map.of(); private volatile Map enchantPriceCache = Map.of(); private volatile Map potionPriceCache = Map.of(); public WorthLorePacketService(PenSell plugin) { this.plugin = plugin; this.bridge = new ReflectionBridge(plugin); loadConfig(); } public void loadConfig() { var cm = plugin.getConfigManager(); worthLoreEnabled = cm.isWorthLoreEnabled(); String raw = cm.getWorthLoreFormat(); worthLoreFormat = raw; loreTemplate = List.of(raw); lorePlainPrefixes = loreTemplate.stream() .map(line -> ChatColor.stripColor( ColorUtil.translate(line.replace("%price%", ""))) .toLowerCase(Locale.ROOT).trim()) .collect(Collectors.toUnmodifiableList()); disabledGuiNames = cm.getWorthLoreDisabledGuis().stream() .map(s -> ColorUtil.strip(s).toLowerCase(Locale.ROOT)) .collect(Collectors.toUnmodifiableList()); blacklistCache = Set.copyOf(cm.getBlacklist()); itemPriceCache = Map.copyOf(cm.getItemPrices()); enchantPriceCache = Map.copyOf(cm.getEnchantmentPrices()); potionPriceCache = Map.copyOf(cm.getPotionPrices()); } public void updateWorthCache(UUID uuid, boolean enabled) { worthEnabledCache.put(uuid, enabled); } public void evictWorthCache(UUID uuid) { worthEnabledCache.remove(uuid); } public void updateGameModeCache(UUID uuid, int ordinal) { gameModeCache.put(uuid, ordinal); } public void updateCursorCache(Player player) { ItemStack cursor = player.getItemOnCursor(); boolean has = cursor != null && cursor.getType() != Material.AIR && cursor.getAmount() > 0; if (has) hasCursorItemCache.add(player.getUniqueId()); else hasCursorItemCache.remove(player.getUniqueId()); } public void inject(Player player) { boolean enabled = plugin.getPlayerDataManager() .getPlayerData(player.getUniqueId()).isWorthEnabled(); worthEnabledCache.put(player.getUniqueId(), enabled); gameModeCache.put(player.getUniqueId(), player.getGameMode().ordinal()); updateCursorCache(player); String name = HANDLER_PREFIX + player.getUniqueId(); handlerNames.put(player.getUniqueId(), name); Channel ch = bridge.channelOf(player); if (ch == null || ch.pipeline().get(name) != null) return; ChannelDuplexHandler handler = new ChannelDuplexHandler() { @Override public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception { super.write(ctx, rewritePacket(player, msg), promise); } }; ch.eventLoop().execute(() -> { try { if (ch.pipeline().get(name) == null) ch.pipeline().addBefore("packet_handler", name, handler); } catch (Throwable t) { plugin.getLogger().warning("Failed to inject worth-lore handler for " + player.getName() + ": " + t.getMessage()); } }); } public void uninject(Player player) { evictWorthCache(player.getUniqueId()); gameModeCache.remove(player.getUniqueId()); hasCursorItemCache.remove(player.getUniqueId()); String name = handlerNames.remove(player.getUniqueId()); noWorthOpen.remove(player.getUniqueId()); bedrockPlayers.remove(player.getUniqueId()); if (name == null) return; Channel ch = bridge.channelOf(player); if (ch == null) return; ch.eventLoop().execute(() -> { try { if (ch.pipeline().get(name) != null) ch.pipeline().remove(name); } catch (Throwable ignored) {} }); } public void injectOnlinePlayers() { for (Player p : Bukkit.getOnlinePlayers()) inject(p); } public void shutdown() { for (Player p : Bukkit.getOnlinePlayers()) uninject(p); } public void setNoWorthOpen(Player player, boolean disabled) { if (disabled) noWorthOpen.add(player.getUniqueId()); else noWorthOpen.remove(player.getUniqueId()); } public void setBedrockPlayer(UUID uuid, boolean bedrock) { if (bedrock) bedrockPlayers.add(uuid); else bedrockPlayers.remove(uuid); } public boolean isDisabledGui(InventoryView view) { if (view == null || view.getTopInventory() == null) return false; if (view.getTopInventory().getType() == InventoryType.PLAYER) return false; String title = ChatColor.stripColor(view.getTitle()); if (title == null) return false; String lower = title.toLowerCase(Locale.ROOT); return disabledGuiNames.stream().anyMatch(lower::contains); } Object rewritePacket(Player player, Object packet) { if (packet == null) return packet; UUID uuid = player.getUniqueId(); int gm = gameModeCache.getOrDefault(uuid, 0); boolean show = worthLoreEnabled && Boolean.TRUE.equals(worthEnabledCache.getOrDefault(uuid, Boolean.TRUE)) && gm != GameMode.CREATIVE.ordinal() && gm != GameMode.SPECTATOR.ordinal() && !hasCursorItemCache.contains(uuid) && !bedrockPlayers.contains(uuid); boolean guiDisabled = noWorthOpen.contains(uuid); String simple = packet.getClass().getSimpleName(); if (PKT_SET_SLOT.equals(simple)) return rewriteSetSlot(packet, show, guiDisabled); if (PKT_SET_CONTENT.equals(simple)) return rewriteSetContent(packet, show, guiDisabled); return packet; } Object rewriteSetSlot(Object packet, boolean show, boolean guiDisabled) { try { Object nmsItem = bridge.findNmsItemField(packet); if (nmsItem == null) return packet; boolean slotInTop = bridge.isSetSlotInTopInventory(packet); boolean addWorth = show && !(guiDisabled && slotInTop); Object replaced = addWorth ? withWorthLore(nmsItem) : stripWorthLore(nmsItem); Object cloned = bridge.cloneSetSlotPacket(packet, replaced); return cloned == null ? packet : cloned; } catch (Throwable t) { return packet; } } Object rewriteSetContent(Object packet, boolean show, boolean guiDisabled) { try { List