mirror of
https://github.com/alexcrea/CustomAnvil.git
synced 2026-06-23 16:16:17 +02:00
117 lines
4.6 KiB
Kotlin
117 lines
4.6 KiB
Kotlin
package io.delilaheve.util
|
|
|
|
import io.delilaheve.CustomAnvil
|
|
import org.bukkit.entity.HumanEntity
|
|
import org.bukkit.inventory.ItemStack
|
|
import xyz.alexcrea.cuanvil.config.ConfigHolder
|
|
import xyz.alexcrea.cuanvil.enchant.CAEnchantment
|
|
import xyz.alexcrea.cuanvil.group.ConflictType
|
|
import xyz.alexcrea.cuanvil.util.ItemTypeUtil.itemType
|
|
import kotlin.math.max
|
|
import kotlin.math.min
|
|
|
|
/**
|
|
* Enchantment manipulation utilities
|
|
*/
|
|
object EnchantmentUtil {
|
|
|
|
/**
|
|
* Enchantment name without namespace
|
|
*/
|
|
val CAEnchantment.enchantmentName: String
|
|
get() = key.key
|
|
|
|
/**
|
|
* Combine 2 sets of enchantments according to our configuration
|
|
*/
|
|
fun Map<CAEnchantment, Int>.combineWith(
|
|
other: Map<CAEnchantment, Int>,
|
|
item: ItemStack,
|
|
player: HumanEntity
|
|
) = mutableMapOf<CAEnchantment, Int>().apply {
|
|
putAll(this@combineWith)
|
|
|
|
val bypassFuse = player.hasPermission(CustomAnvil.bypassFusePermission)
|
|
val bypassLevel = player.hasPermission(CustomAnvil.bypassLevelPermission)
|
|
|
|
var maxEnchantCount = ConfigOptions.getEnchantCountLimit(item.itemType)
|
|
if(maxEnchantCount == null || maxEnchantCount < 0) maxEnchantCount = Int.MAX_VALUE
|
|
|
|
other.forEach { (enchantment, level) ->
|
|
if(!enchantment.isAllowed(player)) return@forEach
|
|
|
|
// Get max level or 255 if player can bypass
|
|
val maxLevel = if (bypassLevel) { 255 }
|
|
else { ConfigOptions.enchantLimit(enchantment) }
|
|
CustomAnvil.verboseLog("Max level of ${enchantment.key} is $maxLevel (bypassLevel is $bypassLevel)")
|
|
|
|
val cappedLevel = min(level, maxLevel)
|
|
|
|
// Enchantment not yet in result list
|
|
if (!containsKey(enchantment)) {
|
|
// Do not allow new enchantment if above maximum
|
|
if(this.size >= maxEnchantCount) return@forEach
|
|
|
|
// Add the enchantment if it doesn't have conflicts, or if player is allowed to bypass enchantment restrictions
|
|
this[enchantment] = cappedLevel
|
|
if(bypassFuse){
|
|
CustomAnvil.verboseLog("Bypassed conflict check for ${enchantment.key}")
|
|
return@forEach
|
|
}
|
|
|
|
val conflictType = ConfigHolder.CONFLICT_HOLDER.conflictManager
|
|
.isConflicting(this, item, enchantment)
|
|
|
|
if (conflictType != ConflictType.NO_CONFLICT) {
|
|
CustomAnvil.verboseLog("Enchantment not yet in result list, but there is conflict (${enchantment.key}, conflict: $conflictType)")
|
|
this.remove(enchantment)
|
|
}
|
|
|
|
}
|
|
// Enchantment already in result list
|
|
else {
|
|
val oldLevel = this[enchantment]!! // <- should not be null. (enchantment already in result list)
|
|
|
|
if(bypassFuse){
|
|
CustomAnvil.verboseLog("Bypassed conflict check for ${enchantment.key}")
|
|
} else {
|
|
val conflictType = ConfigHolder.CONFLICT_HOLDER.conflictManager
|
|
.isConflicting(this, item, enchantment)
|
|
|
|
// ... and they are conflicting
|
|
if(conflictType != ConflictType.NO_CONFLICT){
|
|
CustomAnvil.verboseLog(
|
|
"Enchantment already in result list, and they are conflicting (${enchantment.key}, conflict: $conflictType)")
|
|
return@forEach
|
|
}
|
|
}
|
|
|
|
// ... and they're not the same level
|
|
if (oldLevel != cappedLevel) {
|
|
// apply the greater of the two or left one if right is above max
|
|
this[enchantment] = max(oldLevel, cappedLevel)
|
|
|
|
}
|
|
// ... and they're the same level
|
|
else {
|
|
// We test if it is allowed to merge at this level
|
|
if(!bypassLevel){
|
|
val maxBeforeDisabled = ConfigOptions.maxBeforeMergeDisabled(enchantment)
|
|
if((maxBeforeDisabled > 0) && (oldLevel >= maxBeforeDisabled)) {
|
|
CustomAnvil.verboseLog(
|
|
"Reached max merge before disable for ${enchantment.key}: $oldLevel/$maxBeforeDisabled)")
|
|
return@forEach
|
|
}
|
|
}
|
|
|
|
// Now we increase the enchantment level by 1
|
|
var newLevel = oldLevel + 1
|
|
newLevel = max(min(newLevel, maxLevel), oldLevel)
|
|
this[enchantment] = newLevel
|
|
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|