From a6c2d5f1486b256ca9008013759e2dd73dc7a0a5 Mon Sep 17 00:00:00 2001
From: alexcrea <42614139+alexcrea@users.noreply.github.com>
Date: Wed, 24 Jul 2024 19:13:23 +0200
Subject: [PATCH] Sort new enchantment in gui on add.

---
 .../global/AbstractEnchantConfigGui.java      | 40 +++++++++++++++++++
 .../gui/config/list/ElementListConfigGui.java |  4 +-
 2 files changed, 42 insertions(+), 2 deletions(-)

diff --git a/src/main/java/xyz/alexcrea/cuanvil/gui/config/global/AbstractEnchantConfigGui.java b/src/main/java/xyz/alexcrea/cuanvil/gui/config/global/AbstractEnchantConfigGui.java
index 5ba5691..f9e2903 100644
--- a/src/main/java/xyz/alexcrea/cuanvil/gui/config/global/AbstractEnchantConfigGui.java
+++ b/src/main/java/xyz/alexcrea/cuanvil/gui/config/global/AbstractEnchantConfigGui.java
@@ -53,6 +53,46 @@ public abstract class AbstractEnchantConfigGui<T extends SettingGui.SettingGuiFa
         );
     }
 
+    @Override
+    public void updateValueForGeneric(CAEnchantment generic, boolean shouldUpdate) {
+        updateValueForGeneric(generic, shouldUpdate, true);
+    }
+
+    public void updateValueForGeneric(CAEnchantment generic, boolean shouldUpdate, boolean prepareSorting) {
+        if(!prepareSorting) {
+            super.updateValueForGeneric(generic, shouldUpdate);
+            return;
+        }
+
+        if(!this.factoryMap.containsKey(generic)){
+            // We need to sort elements again
+            super.updateValueForGeneric(generic, false);
+
+            // Clear page then refill all of them
+            this.firstPage.clear();
+            this.pages.clear();
+            this.pages.add(this.firstPage);
+
+            for (CAEnchantment enchantment : getEveryDisplayableInstanceOfGeneric()) {
+                GuiItem item = this.guiItemMap.get(enchantment);
+
+                if(item == null) {
+                    updateValueForGeneric(enchantment, false, false);
+                }else {
+                    addToPage(item);
+                }
+
+            }
+
+            if(shouldUpdate) update();
+
+        }else{
+            super.updateValueForGeneric(generic, shouldUpdate);
+
+        }
+
+    }
+
 
     // Unused methods
     @Override
diff --git a/src/main/java/xyz/alexcrea/cuanvil/gui/config/list/ElementListConfigGui.java b/src/main/java/xyz/alexcrea/cuanvil/gui/config/list/ElementListConfigGui.java
index f11cdf5..8e7d189 100644
--- a/src/main/java/xyz/alexcrea/cuanvil/gui/config/list/ElementListConfigGui.java
+++ b/src/main/java/xyz/alexcrea/cuanvil/gui/config/list/ElementListConfigGui.java
@@ -118,8 +118,8 @@ public abstract class ElementListConfigGui< T > extends ChestGui implements Valu
         this.pages.clear();
         this.pages.add(this.firstPage);
 
-        for (T conflict : getEveryDisplayableInstanceOfGeneric()) {
-            updateValueForGeneric(conflict, false);
+        for (T generic : getEveryDisplayableInstanceOfGeneric()) {
+            updateValueForGeneric(generic, false);
         }
 
         update();