Add filtering option on creative inventory interaction

Author: minoneer

src/main/java/me/minoneer/bukkit/bookexploit/BookListener.java

@@ -1,11 +1,13 @@
 package me.minoneer.bukkit.bookexploit;
 
 import org.bukkit.Material;
+import org.bukkit.entity.HumanEntity;
 import org.bukkit.entity.Player;
 import org.bukkit.event.EventHandler;
 import org.bukkit.event.EventPriority;
 import org.bukkit.event.Listener;
 import org.bukkit.event.block.Action;
+import org.bukkit.event.inventory.InventoryCreativeEvent;
 import org.bukkit.event.player.PlayerEditBookEvent;
 import org.bukkit.event.player.PlayerInteractEvent;
 import org.bukkit.inventory.EquipmentSlot;
@@ -85,4 +87,32 @@ public void onBookRead(@NotNull final PlayerInteractEvent event) {
             }
         }
     }
+
+    @EventHandler(priority = EventPriority.LOW)
+    public void onCreativeInventory(@NotNull final InventoryCreativeEvent event) {
+        if (!config.isOnCreativeInventory()) {
+            return;
+        }
+        final HumanEntity clicker = event.getWhoClicked();
+        if (!(clicker instanceof Player)) {
+            return;
+        }
+        final Player player = (Player) clicker;
+        if (player.hasPermission("bookfilter.overridefilter")) {
+            return;
+        }
+        final ItemStack eventItem = event.getCurrentItem();
+        final ItemStack filteredEventItem = bookFilter.filterBook(eventItem);
+        if (filteredEventItem != null) {
+            logger.log(Level.WARNING, "Player {0} has a book with illegal click events!", player);
+            event.setCurrentItem(filteredEventItem);
+        }
+
+        final ItemStack cursorItem = event.getCurrentItem();
+        final ItemStack filteredCursorItem = bookFilter.filterBook(cursorItem);
+        if (filteredCursorItem != null) {
+            logger.log(Level.WARNING, "Player {0} has a book with illegal click events!", player);
+            event.setCursor(filteredCursorItem);
+        }
+    }
 }

src/main/java/me/minoneer/bukkit/bookexploit/ConfigHandler.java

@@ -14,6 +14,7 @@ public final class ConfigHandler extends AbstractConfigHandler {
     private static final String PATH_WARNING = "chat_warning";
     private static final String PATH_ON_CREATION = "activate.on_creation";
     private static final String PATH_ON_READ = "activate.on_read";
+    private static final String PATH_ON_CREATIVE_INVENTORY = "activate.on_creative_inventory";
     private static final String PATH_DEBUG = "log_debug";
 
     private static final List<String> DEF_FILTER = Arrays.asList(
@@ -38,6 +39,7 @@ public final class ConfigHandler extends AbstractConfigHandler {
     private List<String> chatWarning;
     private boolean onCreation;
     private boolean onRead;
+    private boolean onCreativeInventory;
     private boolean debug;
 
     public ConfigHandler(@NotNull final Plugin plugin) {
@@ -67,6 +69,8 @@ protected void loadData() {
 
         onRead = config.getBoolean(PATH_ON_READ);
 
+        onCreativeInventory = config.getBoolean(PATH_ON_CREATIVE_INVENTORY);
+
         debug = config.getBoolean(PATH_DEBUG);
     }
 
@@ -75,6 +79,7 @@ protected void addDefaults() {
         config.addDefault(PATH_FILTER, DEF_FILTER);
         config.addDefault(PATH_WARNING, DEF_WARNING);
         config.addDefault(PATH_ON_CREATION, true);
+        config.addDefault(PATH_ON_CREATIVE_INVENTORY, true);
         config.addDefault(PATH_ON_READ, false);
         config.addDefault(PATH_DEBUG, false);
     }
@@ -97,6 +102,10 @@ public boolean isOnRead() {
         return onRead;
     }
 
+    public boolean isOnCreativeInventory() {
+        return onCreativeInventory;
+    }
+
     public boolean isDebug() {
         return debug;
     }