src/main/kotlin/moe/nea/firmament/rei/NEUItemEntrySerializer.kt


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

package moe.nea.firmament.rei

import me.shedaniel.rei.api.common.entry.EntrySerializer
import me.shedaniel.rei.api.common.entry.EntryStack
import net.minecraft.nbt.NbtCompound
import moe.nea.firmament.util.SkyblockId

object NEUItemEntrySerializer : EntrySerializer<SBItemStack> {
    const val SKYBLOCK_ID_ENTRY = "SKYBLOCK_ID"
    const val SKYBLOCK_ITEM_COUNT = "SKYBLOCK_ITEM_COUNT"

    override fun supportSaving(): Boolean = true
    override fun supportReading(): Boolean = true

    override fun read(tag: NbtCompound): SBItemStack {
        val id = SkyblockId(tag.getString(SKYBLOCK_ID_ENTRY))
        val count = if (tag.contains(SKYBLOCK_ITEM_COUNT)) tag.getInt(SKYBLOCK_ITEM_COUNT) else 1
        return SBItemStack(id, count)
    }

    override fun save(entry: EntryStack<SBItemStack>, value: SBItemStack): NbtCompound {
        return NbtCompound().apply {
            putString(SKYBLOCK_ID_ENTRY, value.skyblockId.neuItem)
            putInt(SKYBLOCK_ITEM_COUNT, value.getStackSize())
        }
    }
}