1 files changed, 1 insertions, 20 deletions
diff --git a/src/main/kotlin/moe/nea/notenoughupdates/rei/NEUReiPlugin.kt b/src/main/kotlin/moe/nea/notenoughupdates/rei/NEUReiPlugin.kt
index f4a01fd..e89b007 100644
--- a/src/main/kotlin/moe/nea/notenoughupdates/rei/NEUReiPlugin.kt
+++ b/src/main/kotlin/moe/nea/notenoughupdates/rei/NEUReiPlugin.kt
@@ -1,23 +1,16 @@
 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 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 {
@@ -35,19 +28,7 @@ class NEUReiPlugin : REIClientPlugin {
     }
 
     override fun registerScreens(registry: ScreenRegistry) {
-        registry.registerFocusedStack(object : FocusedStackProvider {
-            override fun provide(screen: Screen?, mouse: Point?): CompoundEventResult<EntryStack<*>> {
-                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
-        })
+        registry.registerFocusedStack(SkyblockItemIdFocusedStackProvider)
     }
 
     override fun registerEntries(registry: EntryRegistry) {