Initial commit

src/main/kotlin/io/delilaheve/util/EnchantmentUtil.kt

@@ -0,0 +1,64 @@
+package io.delilaheve.util
+
+import io.delilaheve.UnsafeEnchants
+import org.bukkit.enchantments.Enchantment
+import kotlin.math.max
+import kotlin.math.min
+
+/**
+ * Enchantment manipulation utilities
+ */
+object EnchantmentUtil {
+
+    val Enchantment.enchantmentName: String
+        get() = key.key
+    /**
+     * Combine 2 sets of enchantments according to our configuration
+     */
+    fun MutableMap<Enchantment, Int>.combineWith(
+        other: MutableMap<Enchantment, Int>
+    ) = mutableMapOf<Enchantment, Int>().apply {
+        putAll(this@combineWith)
+        other.forEach { (enchantment, level) ->
+            when {
+                // Enchantment not yet in result list
+                !containsKey(enchantment) -> {
+                    // Add the enchantment if it doesn't have conflicts, or, if we're allowing unsafe enchantments
+                    if (!keys.any { enchantment.conflictsWith(it) } || ConfigOptions.allowUnsafe) {
+                        this[enchantment] = level
+                    }
+                }
+                // Enchantment already in result list...
+                else -> when {
+                    // ... and they're not the same level
+                    this[enchantment] != other[enchantment] -> {
+                        val newLevel = max(this[enchantment] ?: 0, other[enchantment] ?: 0)
+                        // apply the greater of the two if non-zero
+                        if (newLevel > 0) { this[enchantment] = newLevel }
+                    }
+                    // ... and they're the same level
+                    else -> {
+                        // try to increase the enchantment level by 1
+                        var newLevel = this[enchantment]?.plus(1) ?: 0
+                        val maxLevel = ConfigOptions.enchantLimit(enchantment)
+                        newLevel = min(newLevel, maxLevel)
+                        if (newLevel > 0) { this[enchantment] = newLevel }
+                    }
+                }
+            }
+        }
+    }
+
+    /**
+     * Calculate the value of a set of enchantments
+     */
+    fun Map<Enchantment, Int>.calculateValue(
+        fromBook: Boolean
+    ) = entries.sumOf { (enchantment, level) ->
+        val enchantmentMultiplier = ConfigOptions.enchantmentValue(enchantment, fromBook)
+        val value = level * enchantmentMultiplier
+        UnsafeEnchants.log("Value for ${enchantment.enchantmentName} is $value")
+        value
+    }
+
+}