Update SkillEngine API

src/main/java/net/fretux/skillengine/api/SkillEngineAPI.java

@@ -3,6 +3,10 @@ package net.fretux.skillengine.api;
 import net.fretux.skillengine.SkillEngine;
 import net.fretux.skillengine.capability.PlayerSkillData;
 import net.fretux.skillengine.capability.SkillEngineCapabilities;
+import net.fretux.skillengine.client.SkilltreeClientState;
+import net.fretux.skillengine.skilltree.AbilityNode;
+import net.fretux.skillengine.skilltree.AbilityNodeRegistry;
+import net.fretux.skillengine.skilltree.SkillLogic;
 import net.fretux.skillengine.skilltree.SkillNode;
 import net.fretux.skillengine.skilltree.SkillNodeRegistry;
 import net.minecraft.resources.ResourceLocation;
@@ -10,44 +14,57 @@ import net.minecraft.world.entity.player.Player;
 
 import java.util.List;
 import java.util.Set;
+
 public final class SkillEngineAPI {
 
     private SkillEngineAPI() {}
+
     public static ResourceLocation id(String path) {
         return new ResourceLocation(SkillEngine.MODID, path);
     }
+
     public static ResourceLocation rl(String namespace, String path) {
         return new ResourceLocation(namespace, path);
     }
-    public static SkillNode getNode(ResourceLocation id) {
+
+    public static SkillNode getSkillNode(ResourceLocation id) {
         return SkillNodeRegistry.get(id);
     }
-    public static boolean hasNode(ResourceLocation id) {
+
+    public static AbilityNode getAbilityNode(ResourceLocation id) {
+        return AbilityNodeRegistry.get(id);
+    }
+
+    public static boolean hasSkillNode(ResourceLocation id) {
         return SkillNodeRegistry.get(id) != null;
     }
-    public static List<SkillNode> getAllNodes() {
+
+    public static boolean hasAbilityNode(ResourceLocation id) {
+        return AbilityNodeRegistry.get(id) != null;
+    }
+
+    public static List<SkillNode> getAllSkillNodes() {
         return List.copyOf(SkillNodeRegistry.all());
     }
+
+    public static List<AbilityNode> getAllAbilityNodes() {
+        return List.copyOf(AbilityNodeRegistry.all());
+    }
+
     public static PlayerSkillData getPlayerData(Player player) {
         return player.getCapability(SkillEngineCapabilities.PLAYER_SKILLS).orElse(null);
     }
-    public static Set<ResourceLocation> getUnlockedNodes(Player player) {
-        PlayerSkillData data = getPlayerData(player);
-        return data != null ? data.getUnlockedNodes() : Set.of();
-    }
-    
+
     public static int getSkillPoints(Player player) {
         PlayerSkillData data = getPlayerData(player);
         return data != null ? data.getSkillPoints() : 0;
     }
-    
+
     public static void addSkillPoints(Player player, int amount) {
         PlayerSkillData data = getPlayerData(player);
-        if (data != null) {
-            data.addSkillPoints(amount);
-        }
+        if (data != null) data.addSkillPoints(amount);
     }
-    
+
     public static boolean consumeSkillPoints(Player player, int amount) {
         PlayerSkillData data = getPlayerData(player);
         if (data != null && data.getSkillPoints() >= amount) {
@@ -57,26 +74,70 @@ public final class SkillEngineAPI {
         return false;
     }
     
-    public static boolean isUnlocked(Player player, ResourceLocation nodeId) {
+    public static Set<ResourceLocation> getUnlockedSkillNodes(Player player) {
         PlayerSkillData data = getPlayerData(player);
-        return data != null && data.isUnlocked(nodeId);
+        return data != null ? data.getUnlockedNodes() : Set.of();
     }
-    
-    public static boolean unlockNode(Player player, ResourceLocation nodeId) {
+
+    public static boolean isSkillUnlocked(Player player, ResourceLocation id) {
+        PlayerSkillData data = getPlayerData(player);
+        return data != null && data.isUnlocked(id);
+    }
+
+    public static boolean unlockSkill(Player player, ResourceLocation id) {
         PlayerSkillData data = getPlayerData(player);
         if (data == null) return false;
-
-        SkillNode node = SkillNodeRegistry.get(nodeId);
+        SkillNode node = SkillNodeRegistry.get(id);
         if (node == null) return false;
-
-        if (!net.fretux.skillengine.skilltree.SkillLogic.canUnlock(data, player, node)) {
-            return false;
-        }
-
+        if (!SkillLogic.canUnlock(data, player, node)) return false;
         data.unlockNode(node);
         return true;
     }
-    public static boolean clientHasNode(ResourceLocation nodeId) {
-        return net.fretux.skillengine.client.SkilltreeClientState.isUnlocked(nodeId);
+
+    public static boolean isAbilityUnlocked(Player player, ResourceLocation id) {
+        PlayerSkillData data = getPlayerData(player);
+        return data != null && data.isAbilityUnlocked(id);
     }
-}
+
+    public static Set<ResourceLocation> getUnlockedAbilities(Player player) {
+        PlayerSkillData data = getPlayerData(player);
+        return data != null ? data.getUnlockedAbilities() : Set.of();
+    }
+
+    public static boolean unlockAbility(Player player, ResourceLocation id) {
+        PlayerSkillData data = getPlayerData(player);
+        if (data == null) return false;
+        AbilityNode ability = AbilityNodeRegistry.get(id);
+        if (ability == null) return false;
+        data.unlockAbility(ability);
+        return true;
+    }
+
+    public static void bindAbility(Player player, int slot, ResourceLocation abilityId) {
+        PlayerSkillData data = getPlayerData(player);
+        if (data == null) return;
+        data.bindAbility(slot, abilityId);
+    }
+
+    public static ResourceLocation getAbilityInSlot(Player player, int slot) {
+        PlayerSkillData data = getPlayerData(player);
+        return data != null ? data.getAbilityInSlot(slot) : null;
+    }
+
+    public static ResourceLocation[] getAbilitySlots(Player player) {
+        PlayerSkillData data = getPlayerData(player);
+        return data != null ? data.getAbilitySlots() : new ResourceLocation[3];
+    }
+
+    public static boolean clientHasSkill(ResourceLocation id) {
+        return SkilltreeClientState.isUnlocked(id);
+    }
+
+    public static int clientGetAbilitySlot(ResourceLocation id) {
+        return SkilltreeClientState.getSlotOfAbility(id);
+    }
+
+    public static ResourceLocation[] clientGetAbilitySlots() {
+        return SkilltreeClientState.getAbilitySlots();
+    }
+}