Make it possible to change the default heads in the categories menu through the config.

src/main/java/tsp/headdb/inventory/InventoryUtils.java

@@ -39,23 +39,33 @@ public class InventoryUtils {
     }
 
     public static ItemStack uiGetItem(String category, ItemStack item) {
-        // Try to use the cached value first; then the config; then the given default.
+        // Try to use the cached value first.
-        if (!uiItem.containsKey(category)) {
+        if (uiItem.containsKey(category)) return uiItem.get(category);
+
+        // Try to get a head from the config file.
+        if (HeadDB.getInstance().getCfg().contains("ui.category." + category + ".head")) {
+            int id = HeadDB.getInstance().getCfg().getInt("ui.category." + category + ".head");
+            Head head = HeadAPI.getHeadByID(id);
+            if (head != null) {
+                uiItem.put(category, head.getItemStack());
+                return uiItem.get(category);
+            }
+        }
+
+        // Try to get an item from the config file.
         if (HeadDB.getInstance().getCfg().contains("ui.category." + category + ".item")) {
             String cfg = HeadDB.getInstance().getCfg().getString("ui.category." + category + ".item");
             Material mat = Material.matchMaterial(cfg);
-                if (mat == null || mat == Material.AIR) {
+            if (mat != null && mat != Material.AIR) {
-                    // Material set in config is invalid, use the given default.
-                    uiItem.put(category, item);
-                } else {
                 uiItem.put(category, new ItemStack(mat));
-                }
-            } else {
-                uiItem.put(category, item);
-            }
-        }
                 return uiItem.get(category);
             }
+        }
+
+        // No valid head or item in the config file, return the given default.
+        uiItem.put(category, item);
+        return item;
+    }
 
     public static void openLocalMenu(Player player) {
         PagedPane pane = new PagedPane(4, 6, Utils.colorize("&c&lHeadDB &8- &aLocal Heads"));

src/main/resources/config.yml

@@ -26,26 +26,38 @@ economy:
 # UI customization options.
 ui:
   category:
+    # Head categories.
     alphabet:
       location: 20
+      head: 1788
     animals:
       location: 21
+      head: 5741
     blocks:
       location: 22
+      head: 8624
     decoration:
       location: 23
+      head: 11046
     food-drinks:
       location: 24
+      head: 17442
     humans:
       location: 29
+      head: 19361
     humanoid:
       location: 30
+      head: 28320
     miscellaneous:
       location: 31
+      head: 32746
     monsters:
       location: 32
+      head: 34819
     plants:
       location: 33
+      head: 37278
+    # Meta categories, used for UI elements.
     favorites:
       location: 39
       item: BOOK