test: Add sack util test
This commit is contained in:
Linnea Gräf
@@ -1,5 +1,3 @@
|
||||
|
||||
|
||||
package moe.nea.firmament.repo
|
||||
|
||||
import com.mojang.serialization.Dynamic
|
||||
@@ -30,186 +28,194 @@ import moe.nea.firmament.Firmament
|
||||
import moe.nea.firmament.gui.config.HudMeta
|
||||
import moe.nea.firmament.gui.config.HudPosition
|
||||
import moe.nea.firmament.gui.hud.MoulConfigHud
|
||||
import moe.nea.firmament.repo.RepoManager.initialize
|
||||
import moe.nea.firmament.util.LegacyTagParser
|
||||
import moe.nea.firmament.util.MC
|
||||
import moe.nea.firmament.util.SkyblockId
|
||||
import moe.nea.firmament.util.TestUtil
|
||||
import moe.nea.firmament.util.mc.appendLore
|
||||
import moe.nea.firmament.util.mc.modifyLore
|
||||
import moe.nea.firmament.util.mc.setCustomName
|
||||
import moe.nea.firmament.util.mc.setSkullOwner
|
||||
import moe.nea.firmament.util.mc.modifyLore
|
||||
import moe.nea.firmament.util.skyblockId
|
||||
|
||||
object ItemCache : IReloadable {
|
||||
private val cache: MutableMap<String, ItemStack> = ConcurrentHashMap()
|
||||
private val df = Schemas.getFixer()
|
||||
val logger = LogManager.getLogger("${Firmament.logger.name}.ItemCache")
|
||||
var isFlawless = true
|
||||
private set
|
||||
private val cache: MutableMap<String, ItemStack> = ConcurrentHashMap()
|
||||
private val df = Schemas.getFixer()
|
||||
val logger = LogManager.getLogger("${Firmament.logger.name}.ItemCache")
|
||||
var isFlawless = true
|
||||
private set
|
||||
|
||||
private fun NEUItem.get10809CompoundTag(): NbtCompound = NbtCompound().apply {
|
||||
put("tag", LegacyTagParser.parse(nbttag))
|
||||
putString("id", minecraftItemId)
|
||||
putByte("Count", 1)
|
||||
putShort("Damage", damage.toShort())
|
||||
}
|
||||
private fun NEUItem.get10809CompoundTag(): NbtCompound = NbtCompound().apply {
|
||||
put("tag", LegacyTagParser.parse(nbttag))
|
||||
putString("id", minecraftItemId)
|
||||
putByte("Count", 1)
|
||||
putShort("Damage", damage.toShort())
|
||||
}
|
||||
|
||||
private fun NbtCompound.transformFrom10809ToModern(): NbtCompound? =
|
||||
try {
|
||||
df.update(
|
||||
TypeReferences.ITEM_STACK,
|
||||
Dynamic(NbtOps.INSTANCE, this),
|
||||
-1,
|
||||
SharedConstants.getGameVersion().saveVersion.id
|
||||
).value as NbtCompound
|
||||
} catch (e: Exception) {
|
||||
isFlawless = false
|
||||
logger.error("Could not data fix up $this", e)
|
||||
null
|
||||
}
|
||||
private fun NbtCompound.transformFrom10809ToModern(): NbtCompound? =
|
||||
try {
|
||||
df.update(
|
||||
TypeReferences.ITEM_STACK,
|
||||
Dynamic(NbtOps.INSTANCE, this),
|
||||
-1,
|
||||
SharedConstants.getGameVersion().saveVersion.id
|
||||
).value as NbtCompound
|
||||
} catch (e: Exception) {
|
||||
isFlawless = false
|
||||
logger.error("Could not data fix up $this", e)
|
||||
null
|
||||
}
|
||||
|
||||
fun brokenItemStack(neuItem: NEUItem?, idHint: SkyblockId? = null): ItemStack {
|
||||
return ItemStack(Items.PAINTING).apply {
|
||||
setCustomName(Text.literal(neuItem?.displayName ?: idHint?.neuItem ?: "null"))
|
||||
appendLore(
|
||||
listOf(
|
||||
Text.stringifiedTranslatable(
|
||||
"firmament.repo.brokenitem",
|
||||
(neuItem?.skyblockItemId ?: idHint)
|
||||
)
|
||||
)
|
||||
)
|
||||
}
|
||||
}
|
||||
fun brokenItemStack(neuItem: NEUItem?, idHint: SkyblockId? = null): ItemStack {
|
||||
return ItemStack(Items.PAINTING).apply {
|
||||
setCustomName(Text.literal(neuItem?.displayName ?: idHint?.neuItem ?: "null"))
|
||||
appendLore(
|
||||
listOf(
|
||||
Text.stringifiedTranslatable(
|
||||
"firmament.repo.brokenitem",
|
||||
(neuItem?.skyblockItemId ?: idHint)
|
||||
)
|
||||
)
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
private fun NEUItem.asItemStackNow(): ItemStack {
|
||||
try {
|
||||
val oldItemTag = get10809CompoundTag()
|
||||
val modernItemTag = oldItemTag.transformFrom10809ToModern()
|
||||
?: return brokenItemStack(this)
|
||||
val itemInstance =
|
||||
ItemStack.fromNbt(MC.defaultRegistries, modernItemTag).getOrNull() ?: return brokenItemStack(this)
|
||||
val extraAttributes = oldItemTag.getCompound("tag").getCompound("ExtraAttributes")
|
||||
if (extraAttributes != null)
|
||||
itemInstance.set(DataComponentTypes.CUSTOM_DATA, NbtComponent.of(extraAttributes))
|
||||
return itemInstance
|
||||
} catch (e: Exception) {
|
||||
e.printStackTrace()
|
||||
return brokenItemStack(this)
|
||||
}
|
||||
}
|
||||
private fun NEUItem.asItemStackNow(): ItemStack {
|
||||
try {
|
||||
val oldItemTag = get10809CompoundTag()
|
||||
val modernItemTag = oldItemTag.transformFrom10809ToModern()
|
||||
?: return brokenItemStack(this)
|
||||
val itemInstance =
|
||||
ItemStack.fromNbt(MC.defaultRegistries, modernItemTag).getOrNull() ?: return brokenItemStack(this)
|
||||
val extraAttributes = oldItemTag.getCompound("tag").getCompound("ExtraAttributes")
|
||||
if (extraAttributes != null)
|
||||
itemInstance.set(DataComponentTypes.CUSTOM_DATA, NbtComponent.of(extraAttributes))
|
||||
return itemInstance
|
||||
} catch (e: Exception) {
|
||||
e.printStackTrace()
|
||||
return brokenItemStack(this)
|
||||
}
|
||||
}
|
||||
|
||||
fun NEUItem?.asItemStack(idHint: SkyblockId? = null, loreReplacements: Map<String, String>? = null): ItemStack {
|
||||
if (this == null) return brokenItemStack(null, idHint)
|
||||
var s = cache[this.skyblockItemId]
|
||||
if (s == null) {
|
||||
s = asItemStackNow()
|
||||
cache[this.skyblockItemId] = s
|
||||
}
|
||||
if (!loreReplacements.isNullOrEmpty()) {
|
||||
s = s.copy()!!
|
||||
s.applyLoreReplacements(loreReplacements)
|
||||
s.setCustomName(s.name.applyLoreReplacements(loreReplacements))
|
||||
}
|
||||
return s
|
||||
}
|
||||
fun NEUItem?.asItemStack(idHint: SkyblockId? = null, loreReplacements: Map<String, String>? = null): ItemStack {
|
||||
if (this == null) return brokenItemStack(null, idHint)
|
||||
var s = cache[this.skyblockItemId]
|
||||
if (s == null) {
|
||||
s = asItemStackNow()
|
||||
cache[this.skyblockItemId] = s
|
||||
}
|
||||
if (!loreReplacements.isNullOrEmpty()) {
|
||||
s = s.copy()!!
|
||||
s.applyLoreReplacements(loreReplacements)
|
||||
s.setCustomName(s.name.applyLoreReplacements(loreReplacements))
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
fun ItemStack.applyLoreReplacements(loreReplacements: Map<String, String>) {
|
||||
modifyLore { lore ->
|
||||
lore.map {
|
||||
it.applyLoreReplacements(loreReplacements)
|
||||
}
|
||||
}
|
||||
}
|
||||
fun ItemStack.applyLoreReplacements(loreReplacements: Map<String, String>) {
|
||||
modifyLore { lore ->
|
||||
lore.map {
|
||||
it.applyLoreReplacements(loreReplacements)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fun Text.applyLoreReplacements(loreReplacements: Map<String, String>): Text {
|
||||
assert(this.siblings.isEmpty())
|
||||
var string = this.string
|
||||
loreReplacements.forEach { (find, replace) ->
|
||||
string = string.replace("{$find}", replace)
|
||||
}
|
||||
return Text.literal(string).styled { this.style }
|
||||
}
|
||||
fun Text.applyLoreReplacements(loreReplacements: Map<String, String>): Text {
|
||||
assert(this.siblings.isEmpty())
|
||||
var string = this.string
|
||||
loreReplacements.forEach { (find, replace) ->
|
||||
string = string.replace("{$find}", replace)
|
||||
}
|
||||
return Text.literal(string).styled { this.style }
|
||||
}
|
||||
|
||||
fun NEUItem.getIdentifier() = skyblockId.identifier
|
||||
var job: Job? = null
|
||||
|
||||
var job: Job? = null
|
||||
object ReloadProgressHud : MoulConfigHud(
|
||||
"repo_reload", HudMeta(HudPosition(0.0, 0.0, 1F), Text.literal("Repo Reload"), 180, 18)) {
|
||||
object ReloadProgressHud : MoulConfigHud(
|
||||
"repo_reload", HudMeta(HudPosition(0.0, 0.0, 1F), Text.literal("Repo Reload"), 180, 18)) {
|
||||
|
||||
|
||||
var isEnabled = false
|
||||
override fun shouldRender(): Boolean {
|
||||
return isEnabled
|
||||
}
|
||||
var isEnabled = false
|
||||
override fun shouldRender(): Boolean {
|
||||
return isEnabled
|
||||
}
|
||||
|
||||
@get:Bind("current")
|
||||
var current: Double = 0.0
|
||||
@get:Bind("current")
|
||||
var current: Double = 0.0
|
||||
|
||||
@get:Bind("label")
|
||||
var label: String = ""
|
||||
@get:Bind("label")
|
||||
var label: String = ""
|
||||
|
||||
@get:Bind("max")
|
||||
var max: Double = 0.0
|
||||
@get:Bind("max")
|
||||
var max: Double = 0.0
|
||||
|
||||
fun reportProgress(label: String, current: Int, max: Int) {
|
||||
this.label = label
|
||||
this.current = current.toDouble()
|
||||
this.max = max.toDouble()
|
||||
}
|
||||
}
|
||||
fun reportProgress(label: String, current: Int, max: Int) {
|
||||
this.label = label
|
||||
this.current = current.toDouble()
|
||||
this.max = max.toDouble()
|
||||
}
|
||||
}
|
||||
|
||||
override fun reload(repository: NEURepository) {
|
||||
val j = job
|
||||
if (j != null && j.isActive) {
|
||||
j.cancel()
|
||||
}
|
||||
cache.clear()
|
||||
isFlawless = true
|
||||
override fun reload(repository: NEURepository) {
|
||||
val j = job
|
||||
if (j != null && j.isActive) {
|
||||
j.cancel()
|
||||
}
|
||||
cache.clear()
|
||||
isFlawless = true
|
||||
if (TestUtil.isInTest) return
|
||||
job = Firmament.coroutineScope.launch {
|
||||
val items = repository.items?.items
|
||||
if (items == null) {
|
||||
ReloadProgressHud.isEnabled = false
|
||||
return@launch
|
||||
}
|
||||
val recacheItems = I18n.translate("firmament.repo.cache")
|
||||
ReloadProgressHud.reportProgress(recacheItems, 0, items.size)
|
||||
ReloadProgressHud.isEnabled = true
|
||||
var i = 0
|
||||
items.values.forEach {
|
||||
it.asItemStack() // Rebuild cache
|
||||
ReloadProgressHud.reportProgress(recacheItems, i++, items.size)
|
||||
}
|
||||
ReloadProgressHud.isEnabled = false
|
||||
}
|
||||
}
|
||||
|
||||
job = Firmament.coroutineScope.launch {
|
||||
val items = repository.items?.items
|
||||
if (items == null) {
|
||||
ReloadProgressHud.isEnabled = false
|
||||
return@launch
|
||||
}
|
||||
val recacheItems = I18n.translate("firmament.repo.cache")
|
||||
ReloadProgressHud.reportProgress(recacheItems, 0, items.size)
|
||||
ReloadProgressHud.isEnabled = true
|
||||
var i = 0
|
||||
items.values.forEach {
|
||||
it.asItemStack() // Rebuild cache
|
||||
ReloadProgressHud.reportProgress(recacheItems, i++, items.size)
|
||||
}
|
||||
ReloadProgressHud.isEnabled = false
|
||||
}
|
||||
}
|
||||
fun coinItem(coinAmount: Int): ItemStack {
|
||||
var uuid = UUID.fromString("2070f6cb-f5db-367a-acd0-64d39a7e5d1b")
|
||||
var texture =
|
||||
"http://textures.minecraft.net/texture/538071721cc5b4cd406ce431a13f86083a8973e1064d2f8897869930ee6e5237"
|
||||
if (coinAmount >= 100000) {
|
||||
uuid = UUID.fromString("94fa2455-2881-31fe-bb4e-e3e24d58dbe3")
|
||||
texture =
|
||||
"http://textures.minecraft.net/texture/c9b77999fed3a2758bfeaf0793e52283817bea64044bf43ef29433f954bb52f6"
|
||||
}
|
||||
if (coinAmount >= 10000000) {
|
||||
uuid = UUID.fromString("0af8df1f-098c-3b72-ac6b-65d65fd0b668")
|
||||
texture =
|
||||
"http://textures.minecraft.net/texture/7b951fed6a7b2cbc2036916dec7a46c4a56481564d14f945b6ebc03382766d3b"
|
||||
}
|
||||
val itemStack = ItemStack(Items.PLAYER_HEAD)
|
||||
itemStack.setCustomName(Text.literal("§r§6" + NumberFormat.getInstance().format(coinAmount) + " Coins"))
|
||||
itemStack.setSkullOwner(uuid, texture)
|
||||
return itemStack
|
||||
}
|
||||
|
||||
init {
|
||||
if (TestUtil.isInTest) {
|
||||
initialize()
|
||||
}
|
||||
}
|
||||
|
||||
fun coinItem(coinAmount: Int): ItemStack {
|
||||
var uuid = UUID.fromString("2070f6cb-f5db-367a-acd0-64d39a7e5d1b")
|
||||
var texture =
|
||||
"http://textures.minecraft.net/texture/538071721cc5b4cd406ce431a13f86083a8973e1064d2f8897869930ee6e5237"
|
||||
if (coinAmount >= 100000) {
|
||||
uuid = UUID.fromString("94fa2455-2881-31fe-bb4e-e3e24d58dbe3")
|
||||
texture =
|
||||
"http://textures.minecraft.net/texture/c9b77999fed3a2758bfeaf0793e52283817bea64044bf43ef29433f954bb52f6"
|
||||
}
|
||||
if (coinAmount >= 10000000) {
|
||||
uuid = UUID.fromString("0af8df1f-098c-3b72-ac6b-65d65fd0b668")
|
||||
texture =
|
||||
"http://textures.minecraft.net/texture/7b951fed6a7b2cbc2036916dec7a46c4a56481564d14f945b6ebc03382766d3b"
|
||||
}
|
||||
val itemStack = ItemStack(Items.PLAYER_HEAD)
|
||||
itemStack.setCustomName(Text.literal("§r§6" + NumberFormat.getInstance().format(coinAmount) + " Coins"))
|
||||
itemStack.setSkullOwner(uuid, texture)
|
||||
return itemStack
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
operator fun NbtCompound.set(key: String, value: String) {
|
||||
putString(key, value)
|
||||
putString(key, value)
|
||||
}
|
||||
|
||||
operator fun NbtCompound.set(key: String, value: NbtElement) {
|
||||
put(key, value)
|
||||
put(key, value)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user