From f2aa75f7c193513215ba4619c66bcc1449b0bc27 Mon Sep 17 00:00:00 2001 From: nea Date: Thu, 4 May 2023 15:12:56 +0200 Subject: Make REI aware of SkyBlock item ids --- .../moe/nea/notenoughupdates/rei/NEUReiPlugin.kt | 27 ++++++++++++-- .../notenoughupdates/rei/SBItemEntryDefinition.kt | 31 ++++++++++++---- .../moe/nea/notenoughupdates/repo/RepoManager.kt | 3 ++ .../moe/nea/notenoughupdates/util/SkyblockId.kt | 42 ++++++++++++++++++++++ 4 files changed, 94 insertions(+), 9 deletions(-) create mode 100644 src/main/kotlin/moe/nea/notenoughupdates/util/SkyblockId.kt (limited to 'src/main/kotlin/moe') diff --git a/src/main/kotlin/moe/nea/notenoughupdates/rei/NEUReiPlugin.kt b/src/main/kotlin/moe/nea/notenoughupdates/rei/NEUReiPlugin.kt index 475a5c9..f4a01fd 100644 --- a/src/main/kotlin/moe/nea/notenoughupdates/rei/NEUReiPlugin.kt +++ b/src/main/kotlin/moe/nea/notenoughupdates/rei/NEUReiPlugin.kt @@ -1,15 +1,23 @@ package moe.nea.notenoughupdates.rei +import dev.architectury.event.CompoundEventResult import io.github.moulberry.repo.data.NEUItem +import me.shedaniel.math.Point import me.shedaniel.rei.api.client.plugins.REIClientPlugin import me.shedaniel.rei.api.client.registry.entry.EntryRegistry +import me.shedaniel.rei.api.client.registry.screen.FocusedStackProvider +import me.shedaniel.rei.api.client.registry.screen.ScreenRegistry import me.shedaniel.rei.api.common.entry.EntryStack import me.shedaniel.rei.api.common.entry.type.EntryTypeRegistry import me.shedaniel.rei.api.common.entry.type.VanillaEntryTypes -import moe.nea.notenoughupdates.repo.ItemCache.asItemStack -import moe.nea.notenoughupdates.repo.RepoManager +import net.minecraft.client.gui.screen.Screen +import net.minecraft.client.gui.screen.ingame.HandledScreen import net.minecraft.item.ItemStack import net.minecraft.util.Identifier +import moe.nea.notenoughupdates.mixins.accessor.AccessorHandledScreen +import moe.nea.notenoughupdates.repo.ItemCache.asItemStack +import moe.nea.notenoughupdates.repo.RepoManager +import moe.nea.notenoughupdates.util.skyBlockId class NEUReiPlugin : REIClientPlugin { @@ -26,6 +34,21 @@ class NEUReiPlugin : REIClientPlugin { registry.register(SKYBLOCK_ITEM_TYPE_ID, SBItemEntryDefinition) } + override fun registerScreens(registry: ScreenRegistry) { + registry.registerFocusedStack(object : FocusedStackProvider { + override fun provide(screen: Screen?, mouse: Point?): CompoundEventResult> { + if (screen !is HandledScreen<*>) return CompoundEventResult.pass() + screen as AccessorHandledScreen + val focusedSlot = screen.focusedSlot_NEU ?: return CompoundEventResult.pass() + val item = focusedSlot.stack ?: return CompoundEventResult.pass() + val skyblockId = item.skyBlockId ?: return CompoundEventResult.pass() + val neuItem = RepoManager.getNEUItem(skyblockId) ?: return CompoundEventResult.interrupt(false, null) + return CompoundEventResult.interruptTrue(EntryStack.of(SBItemEntryDefinition, neuItem)) + } + + override fun getPriority(): Double = 1_000_000.0 + }) + } override fun registerEntries(registry: EntryRegistry) { RepoManager.neuRepo.items?.items?.values?.forEach { diff --git a/src/main/kotlin/moe/nea/notenoughupdates/rei/SBItemEntryDefinition.kt b/src/main/kotlin/moe/nea/notenoughupdates/rei/SBItemEntryDefinition.kt index 6e88b42..22835e6 100644 --- a/src/main/kotlin/moe/nea/notenoughupdates/rei/SBItemEntryDefinition.kt +++ b/src/main/kotlin/moe/nea/notenoughupdates/rei/SBItemEntryDefinition.kt @@ -1,6 +1,7 @@ package moe.nea.notenoughupdates.rei import io.github.moulberry.repo.data.NEUItem +import java.util.stream.Stream import me.shedaniel.math.Rectangle import me.shedaniel.rei.api.client.entry.renderer.EntryRenderer import me.shedaniel.rei.api.client.gui.widgets.Tooltip @@ -11,15 +12,17 @@ import me.shedaniel.rei.api.common.entry.comparison.ComparisonContext import me.shedaniel.rei.api.common.entry.type.EntryDefinition import me.shedaniel.rei.api.common.entry.type.EntryType import me.shedaniel.rei.api.common.entry.type.VanillaEntryTypes -import moe.nea.notenoughupdates.rei.NEUReiPlugin.Companion.asItemEntry -import moe.nea.notenoughupdates.repo.ItemCache.asItemStack -import moe.nea.notenoughupdates.repo.ItemCache.getIdentifier import net.minecraft.client.util.math.MatrixStack import net.minecraft.item.ItemStack +import net.minecraft.nbt.NbtCompound +import net.minecraft.registry.tag.TagKey import net.minecraft.text.Text import net.minecraft.util.Identifier -import java.util.stream.Stream -import net.minecraft.registry.tag.TagKey +import moe.nea.notenoughupdates.rei.NEUReiPlugin.Companion.asItemEntry +import moe.nea.notenoughupdates.repo.ItemCache.asItemStack +import moe.nea.notenoughupdates.repo.ItemCache.getIdentifier +import moe.nea.notenoughupdates.repo.RepoManager +import moe.nea.notenoughupdates.util.SkyblockId object SBItemEntryDefinition : EntryDefinition { override fun equals(o1: NEUItem?, o2: NEUItem?, context: ComparisonContext?): Boolean { @@ -57,8 +60,21 @@ object SBItemEntryDefinition : EntryDefinition { } - override fun getSerializer(): EntrySerializer? { - return null + override fun getSerializer(): EntrySerializer { + return object : EntrySerializer { + override fun supportSaving(): Boolean = true + override fun supportReading(): Boolean = true + + override fun read(tag: NbtCompound): NEUItem? { + return RepoManager.getNEUItem(SkyblockId(tag.getString("SKYBLOCK_ID"))) + } + + override fun save(entry: EntryStack, value: NEUItem?): NbtCompound { + return NbtCompound().apply { + putString("SKYBLOCK_ID", value?.skyblockItemId ?: "null") + } + } + } } override fun getTagsFor(entry: EntryStack?, value: NEUItem?): Stream>? { @@ -70,6 +86,7 @@ object SBItemEntryDefinition : EntryDefinition { } override fun hash(entry: EntryStack, value: NEUItem, context: ComparisonContext): Long { + // Repo items are immutable, and get replaced entirely when loaded from disk return System.identityHashCode(value) * 31L } diff --git a/src/main/kotlin/moe/nea/notenoughupdates/repo/RepoManager.kt b/src/main/kotlin/moe/nea/notenoughupdates/repo/RepoManager.kt index 06a8a87..6470914 100644 --- a/src/main/kotlin/moe/nea/notenoughupdates/repo/RepoManager.kt +++ b/src/main/kotlin/moe/nea/notenoughupdates/repo/RepoManager.kt @@ -14,6 +14,7 @@ import net.fabricmc.fabric.api.client.event.lifecycle.v1.ClientTickEvents import net.minecraft.client.MinecraftClient import net.minecraft.network.packet.s2c.play.SynchronizeRecipesS2CPacket import net.minecraft.text.Text +import moe.nea.notenoughupdates.util.SkyblockId object RepoManager : DataHolder(serializer(), "repo", ::Config) { @Serializable @@ -55,6 +56,8 @@ object RepoManager : DataHolder(serializer(), "repo", ::Conf }) } + fun getNEUItem(skyblockId: SkyblockId) = neuRepo.items.getItemBySkyblockId(skyblockId.neuItem) + fun launchAsyncUpdate(force: Boolean = false) { NotEnoughUpdates.coroutineScope.launch { progressBar.reportProgress("Downloading", 0, null) diff --git a/src/main/kotlin/moe/nea/notenoughupdates/util/SkyblockId.kt b/src/main/kotlin/moe/nea/notenoughupdates/util/SkyblockId.kt new file mode 100644 index 0000000..8d7ef91 --- /dev/null +++ b/src/main/kotlin/moe/nea/notenoughupdates/util/SkyblockId.kt @@ -0,0 +1,42 @@ +package moe.nea.notenoughupdates.util + +import io.github.moulberry.repo.data.Rarity +import kotlinx.serialization.Serializable +import kotlinx.serialization.decodeFromString +import kotlinx.serialization.json.Json +import net.minecraft.item.ItemStack +import net.minecraft.nbt.NbtCompound + +@JvmInline +value class SkyblockId(val neuItem: String) + +@Serializable +data class HypixelPetInfo( + val type: String, + val tier: Rarity, +) { + val skyblockId get() = SkyblockId("${type.uppercase()};${tier}") +} + +private val jsonparser = Json { ignoreUnknownKeys = true } + +val ItemStack.extraAttributes: NbtCompound + get() = getOrCreateSubNbt("ExtraAttributes") + +val ItemStack.skyBlockId: SkyblockId? + get() { + when (val id = extraAttributes.getString("id")) { + "PET" -> { + val jsonString = extraAttributes.getString("petInfo") + if (jsonString.isNullOrBlank()) return null + val petInfo = + runCatching { jsonparser.decodeFromString(jsonString) } + .getOrElse { return null } + return petInfo.skyblockId + } + // TODO: RUNE, ENCHANTED_BOOK, PARTY_HAT_CRAB{,_ANIMATED}, ABICASE + else -> { + return SkyblockId(id) + } + } + } -- cgit