item ability utils

Author: mrsterner

Diff for: src/main/kotlin/dev/sterner/api/VoidBoundApi.kt

@@ -3,6 +3,8 @@ package dev.sterner.api
 import com.sammy.malum.client.VoidRevelationHandler
 import com.sammy.malum.common.container.WeaversWorkbenchContainer.component
 import com.sammy.malum.core.systems.recipe.SpiritWithCount
+import dev.sterner.api.item.ItemAbility
+import dev.sterner.api.item.ItemAbilityWithLevel
 import dev.sterner.listener.EnchantSpiritDataReloadListener
 import dev.sterner.registry.VoidBoundComponentRegistry
 import dev.sterner.registry.VoidBoundItemRegistry
@@ -13,6 +15,7 @@ import net.minecraft.core.registries.BuiltInRegistries
 import net.minecraft.network.chat.Component
 import net.minecraft.world.entity.EquipmentSlot
 import net.minecraft.world.entity.player.Player
+import net.minecraft.world.item.ItemStack
 import net.minecraft.world.item.enchantment.Enchantment
 import net.minecraft.world.level.Level
 import team.lodestar.lodestone.helpers.TrinketsHelper
@@ -139,4 +142,55 @@ object VoidBoundApi {
         }
         return false
     }
+
+    fun getItemAbility(stack: ItemStack): List<ItemAbilityWithLevel> {
+        val abilities = mutableListOf<ItemAbilityWithLevel>()
+        val tag = stack.tag ?: return abilities // Return empty if no NBT
+
+        val abilitiesTag = tag.getList("Abilities", 10) // 10 is the NBT type for CompoundTag
+        for (i in 0 until abilitiesTag.size) {
+            val abilityTag = abilitiesTag.getCompound(i)
+            val ability = ItemAbilityWithLevel.readNbt(abilityTag)
+            abilities.add(ability)
+        }
+        return abilities
+    }
+
+    // Function to add an ItemAbilityWithLevel to an ItemStack's NBT
+    fun addItemAbility(stack: ItemStack, abilityWithLevel: ItemAbilityWithLevel) {
+        val tag = stack.orCreateTag // Ensures the stack has NBT
+        val abilitiesTag = tag.getList("Abilities", 10) // Fetch or create list
+
+        // Check if ability already exists, if so, skip adding a duplicate
+        for (i in 0 until abilitiesTag.size) {
+            val abilityTag = abilitiesTag.getCompound(i)
+            val existingAbility = ItemAbilityWithLevel.readNbt(abilityTag)
+            if (existingAbility.itemAbility == abilityWithLevel.itemAbility) {
+                return // Ability already exists, exit without adding
+            }
+        }
+
+        // Add new ability
+        abilitiesTag.add(abilityWithLevel.writeNbt())
+        tag.put("Abilities", abilitiesTag)
+    }
+
+    // Function to modify the level of an existing ItemAbility in NBT
+    fun modifyItemAbilityLevel(stack: ItemStack, itemAbility: ItemAbility, newLevel: Int) {
+        val tag = stack.tag ?: return // No NBT, nothing to modify
+        val abilitiesTag = tag.getList("Abilities", 10)
+
+        // Find the ability and modify its level
+        for (i in 0 until abilitiesTag.size) {
+            val abilityTag = abilitiesTag.getCompound(i)
+            val ability = ItemAbilityWithLevel.readNbt(abilityTag)
+            if (ability.itemAbility == itemAbility) {
+                // Modify the level and update the NBT
+                abilityTag.putInt("Level", newLevel)
+                abilitiesTag[i] = abilityTag // Replace the modified ability in the list
+                tag.put("Abilities", abilitiesTag)
+                return
+            }
+        }
+    }
 }

Diff for: src/main/kotlin/dev/sterner/api/item/ItemAbility.kt

@@ -17,7 +17,7 @@ enum class ItemAbility(private val equipmentSlot: EquipmentSlot?, private val cl
         return this.name.lowercase()
     }
 
-    fun getAbilityFromItem(item: Item): List<ItemAbility> {
+    fun getAvailableAbilitiesFromItem(item: Item): List<ItemAbility> {
         val list = mutableListOf<ItemAbility>()
         for (ability in entries) {
             // Check if the item matches the clazz or the equipmentSlot