Added DataGen to the build.gradle (not done yet)

4 files changed, 138 insertions, 5 deletions

diff --git a/src/main/java/dev/mayaqq/ygasi/gui/BranchGui.java b/src/main/java/dev/mayaqq/ygasi/gui/BranchGui.java
index c7b9950..ac19478 100644
--- a/src/main/java/dev/mayaqq/ygasi/gui/BranchGui.java
+++ b/src/main/java/dev/mayaqq/ygasi/gui/BranchGui.java
@@ -1,9 +1,8 @@
 package dev.mayaqq.ygasi.gui;
 
-import dev.mayaqq.ygasi.registry.ConfigRegistry;
-import dev.mayaqq.ygasi.registry.PlayerDataRegistry;
 import eu.pb4.sgui.api.elements.*;
 import eu.pb4.sgui.api.gui.SimpleGui;
+
 import net.minecraft.item.ItemStack;
 import net.minecraft.item.Items;
 import net.minecraft.screen.ScreenHandlerType;
@@ -17,6 +16,9 @@ import java.util.UUID;
 
 import static dev.mayaqq.ygasi.registry.StatRegistry.SKILL_POINTS;
 
+import dev.mayaqq.ygasi.registry.ConfigRegistry;
+import dev.mayaqq.ygasi.registry.PlayerDataRegistry;
+
 public class BranchGui {
     public static void gui(ServerPlayerEntity player) {
         try {
@@ -65,7 +67,7 @@ public class BranchGui {
                         .glow()
                         .setName(Text.literal("Mercenary")
                                 .setStyle(Style.EMPTY.withBold(true).withFormatting(Formatting.RED)))
-                        .setCallback((index, clickType, actionType) -> gui.close())
+                        .setCallback((index, clickType, actionType) -> MercenaryGui.gui(player))
                 );
             }
 
@@ -83,7 +85,7 @@ public class BranchGui {
                         .glow()
                         .setName(Text.literal("Wizardry")
                                 .setStyle(Style.EMPTY.withBold(true).withFormatting(Formatting.DARK_PURPLE)))
-                        .setCallback((index, clickType, actionType) -> gui.close())
+                        .setCallback((index, clickType, actionType) -> WizardryGui.gui(player))
                 );
             }
 
@@ -101,7 +103,7 @@ public class BranchGui {
                         .glow()
                         .setName(Text.literal("Druidry")
                                 .setStyle(Style.EMPTY.withBold(true).withFormatting(Formatting.GREEN)))
-                        .setCallback((index, clickType, actionType) -> gui.close())
+                        .setCallback((index, clickType, actionType) -> DruidryGui.gui(player))
                 );
             }
             gui.open();
@@ -116,6 +118,14 @@ public class BranchGui {
             PlayerDataRegistry.PLAYERDATA.branches.put(branch, true);
             PlayerDataRegistry.save(player.getUuid());
             player.closeHandledScreen();
+            if (branch.equals("mercenary")) {
+                MercenaryGui.gui(player);
+            } else if (branch.equals("wizardry")) {
+                WizardryGui.gui(player);
+            } else if (branch.equals("druidry")) {
+                DruidryGui.gui(player);
+            }
+
         } else {
             player.sendMessage(Text.literal("You don't have enough skill points to unlock this branch!").setStyle(Style.EMPTY.withBold(true).withFormatting(Formatting.RED)), false);
             player.closeHandledScreen();
diff --git a/src/main/java/dev/mayaqq/ygasi/gui/DruidryGui.java b/src/main/java/dev/mayaqq/ygasi/gui/DruidryGui.java
new file mode 100644
index 0000000..4054ead
--- /dev/null
+++ b/src/main/java/dev/mayaqq/ygasi/gui/DruidryGui.java
@@ -0,0 +1,41 @@
+package dev.mayaqq.ygasi.gui;
+
+import dev.mayaqq.ygasi.registry.PlayerDataRegistry;
+import eu.pb4.sgui.api.elements.GuiElementBuilder;
+import eu.pb4.sgui.api.gui.SimpleGui;
+import net.minecraft.item.Items;
+import net.minecraft.screen.ScreenHandlerType;
+import net.minecraft.server.network.ServerPlayerEntity;
+import net.minecraft.stat.Stats;
+import net.minecraft.text.Text;
+
+import java.util.UUID;
+
+import static dev.mayaqq.ygasi.registry.StatRegistry.SKILL_POINTS;
+
+public class DruidryGui {
+    public static void gui(ServerPlayerEntity player) {
+        try {
+            UUID playerUUID = player.getUuid();
+            PlayerDataRegistry.load(playerUUID);
+            if (PlayerDataRegistry.PLAYERDATA.branches.get("druidry") == null || !PlayerDataRegistry.PLAYERDATA.branches.get("druidry")) {
+                player.sendMessage(Text.of("§cYou have not unlocked this branch yet!"), false);
+                BranchGui.gui(player);
+            } else {
+                SimpleGui gui = new SimpleGui(ScreenHandlerType.GENERIC_9X6, player, false) {};
+
+                gui.setTitle(Text.of( "§a§lDruidry " + "§3Skill Points: " + player.getStatHandler().getStat(Stats.CUSTOM.getOrCreateStat(SKILL_POINTS))));
+
+                for (int x = 0; x <= 53; x++) {
+                    gui.setSlot(x, new GuiElementBuilder()
+                            .setItem(Items.GRAY_STAINED_GLASS_PANE)
+                            .setName(Text.of(" "))
+                    );
+                }
+                gui.open();
+            }
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+    }
+}
diff --git a/src/main/java/dev/mayaqq/ygasi/gui/MercenaryGui.java b/src/main/java/dev/mayaqq/ygasi/gui/MercenaryGui.java
new file mode 100644
index 0000000..5c95eb2
--- /dev/null
+++ b/src/main/java/dev/mayaqq/ygasi/gui/MercenaryGui.java
@@ -0,0 +1,41 @@
+package dev.mayaqq.ygasi.gui;
+
+import dev.mayaqq.ygasi.registry.PlayerDataRegistry;
+import eu.pb4.sgui.api.elements.GuiElementBuilder;
+import eu.pb4.sgui.api.gui.SimpleGui;
+import net.minecraft.item.Items;
+import net.minecraft.screen.ScreenHandlerType;
+import net.minecraft.server.network.ServerPlayerEntity;
+import net.minecraft.stat.Stats;
+import net.minecraft.text.Text;
+
+import java.util.UUID;
+
+import static dev.mayaqq.ygasi.registry.StatRegistry.SKILL_POINTS;
+
+public class MercenaryGui {
+    public static void gui(ServerPlayerEntity player) {
+        try {
+            UUID playerUUID = player.getUuid();
+            PlayerDataRegistry.load(playerUUID);
+            if (PlayerDataRegistry.PLAYERDATA.branches.get("mercenary") == null || !PlayerDataRegistry.PLAYERDATA.branches.get("mercenary")) {
+                player.sendMessage(Text.of("§cYou have not unlocked this branch yet!"), false);
+                BranchGui.gui(player);
+            } else {
+                SimpleGui gui = new SimpleGui(ScreenHandlerType.GENERIC_9X6, player, false) {};
+
+                gui.setTitle(Text.of( "§c§lMercenary " + "§3Skill Points: " + player.getStatHandler().getStat(Stats.CUSTOM.getOrCreateStat(SKILL_POINTS))));
+
+                for (int x = 0; x <= 53; x++) {
+                    gui.setSlot(x, new GuiElementBuilder()
+                            .setItem(Items.GRAY_STAINED_GLASS_PANE)
+                            .setName(Text.of(" "))
+                    );
+                }
+                gui.open();
+            }
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+    }
+}
diff --git a/src/main/java/dev/mayaqq/ygasi/gui/WizardryGui.java b/src/main/java/dev/mayaqq/ygasi/gui/WizardryGui.java
new file mode 100644
index 0000000..c48b0c1
--- /dev/null
+++ b/src/main/java/dev/mayaqq/ygasi/gui/WizardryGui.java
@@ -0,0 +1,41 @@
+package dev.mayaqq.ygasi.gui;
+
+import dev.mayaqq.ygasi.registry.PlayerDataRegistry;
+import eu.pb4.sgui.api.elements.GuiElementBuilder;
+import eu.pb4.sgui.api.gui.SimpleGui;
+import net.minecraft.item.Items;
+import net.minecraft.screen.ScreenHandlerType;
+import net.minecraft.server.network.ServerPlayerEntity;
+import net.minecraft.stat.Stats;
+import net.minecraft.text.Text;
+
+import java.util.UUID;
+
+import static dev.mayaqq.ygasi.registry.StatRegistry.SKILL_POINTS;
+
+public class WizardryGui {
+    public static void gui(ServerPlayerEntity player) {
+        try {
+            UUID playerUUID = player.getUuid();
+            PlayerDataRegistry.load(playerUUID);
+            if (PlayerDataRegistry.PLAYERDATA.branches.get("wizardry") == null || !PlayerDataRegistry.PLAYERDATA.branches.get("wizardry")) {
+                player.sendMessage(Text.of("§cYou have not unlocked this branch yet!"), false);
+                BranchGui.gui(player);
+            } else {
+                SimpleGui gui = new SimpleGui(ScreenHandlerType.GENERIC_9X6, player, false) {};
+
+                gui.setTitle(Text.of( "§5§lWizardry " + "§3Skill Points: " + player.getStatHandler().getStat(Stats.CUSTOM.getOrCreateStat(SKILL_POINTS))));
+
+                for (int x = 0; x <= 53; x++) {
+                    gui.setSlot(x, new GuiElementBuilder()
+                            .setItem(Items.GRAY_STAINED_GLASS_PANE)
+                            .setName(Text.of(" "))
+                    );
+                }
+                gui.open();
+            }
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+    }
+}