feat(category): add disabled key to prevent loading a category

okocraft · Apr 16, 2024 · d8015cb · d8015cb
1 parent a1450ed
commit d8015cb
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/...in/java/net/okocraft/box/feature/category/internal/category/defaults/DefaultCategory.java b/...in/java/net/okocraft/box/feature/category/internal/category/defaults/DefaultCategory.java
@@ -1,5 +1,6 @@
 package net.okocraft.box.feature.category.internal.category.defaults;
 
+import com.github.siroshun09.configapi.core.node.BooleanValue;
 import com.github.siroshun09.configapi.core.node.MapNode;
 import net.okocraft.box.api.model.manager.ItemManager;
 import net.okocraft.box.feature.category.api.category.Category;
@@ -33,6 +34,7 @@ public record DefaultCategory(@NotNull String key,
     }
 
     public void storeToMapNode(@NotNull MapNode mapNode) {
+        mapNode.set(CategoryFile.DISABLED_CATEGORY, BooleanValue.FALSE);
         mapNode.set(CategoryFile.ICON_KEY, this.icon.name());
 
         var displayName = mapNode.createMap(CategoryFile.DISPLAY_NAME_KEY);

diff --git a/.../category/src/main/java/net/okocraft/box/feature/category/internal/file/CategoryFile.java b/.../category/src/main/java/net/okocraft/box/feature/category/internal/file/CategoryFile.java
@@ -39,6 +39,7 @@ public final class CategoryFile implements AutoCloseable {
     public static final String ICON_KEY = "icon";
     public static final String DISPLAY_NAME_KEY = "display-name";
     public static final String LOCALE_DEFAULT = "default";
+    public static final String DISABLED_CATEGORY = "disabled";
 
     private final Path filepath;
     private final CategoryRegistry registry;
@@ -103,7 +104,7 @@ public CategoryFile readCategoriesIfExists() {
         for (var entry : this.loadedSource.value().entrySet()) {
             var key = String.valueOf(entry.getKey());
             if (!key.startsWith("$") && entry.getValue() instanceof MapNode section) {
-                this.registry.register(key, loadCategory(key, section, BoxAPI.api().getItemManager().getItemNameConverter(this.version)));
+                this.loadAndRegisterCategory(key, section, BoxAPI.api().getItemManager().getItemNameConverter(this.version));
             }
         }
 
@@ -165,7 +166,11 @@ public void addNewDefaultItemsIfNeeded() throws IOException {
         YamlFormat.COMMENT_PROCESSING.save(mapNode, this.filepath);
     }
 
-    private static LoadedCategory loadCategory(@NotNull String key, @NotNull MapNode source, @NotNull UnaryOperator<String> itemNameConverter) {
+    private void loadAndRegisterCategory(@NotNull String key, @NotNull MapNode source, @NotNull UnaryOperator<String> itemNameConverter) {
+        if (source.getBoolean(DISABLED_CATEGORY)) {
+            return;
+        }
+
         Material iconMaterial;
 
         try {
@@ -204,7 +209,7 @@ private static LoadedCategory loadCategory(@NotNull String key, @NotNull MapNode
 
         var category = new LoadedCategory(iconMaterial, displayNameMap);
         category.addItems(items);
-        return category;
+        this.registry.register(key, category);
     }
 
     private @NotNull Collection<BoxItem> toBoxItems(@NotNull List<String> itemNames) {