aboutsummaryrefslogtreecommitdiff
path: root/src/main/kotlin/gui/entity/ModifyEquipment.kt
blob: 73e450e71dd381ec9819c0687be4da951cb93436 (plain)
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
package moe.nea.firmament.gui.entity

import com.google.gson.JsonObject
import net.minecraft.component.DataComponentTypes
import net.minecraft.component.type.DyedColorComponent
import net.minecraft.entity.EquipmentSlot
import net.minecraft.entity.LivingEntity
import net.minecraft.item.ArmorItem
import net.minecraft.item.Item
import net.minecraft.item.ItemStack
import net.minecraft.item.Items
import moe.nea.firmament.rei.SBItemStack
import moe.nea.firmament.util.SkyblockId
import moe.nea.firmament.util.item.setEncodedSkullOwner
import moe.nea.firmament.util.item.zeroUUID

object ModifyEquipment : EntityModifier {
    val names = mapOf(
        "hand" to EquipmentSlot.MAINHAND,
        "helmet" to EquipmentSlot.HEAD,
        "chestplate" to EquipmentSlot.CHEST,
        "leggings" to EquipmentSlot.LEGS,
        "feet" to EquipmentSlot.FEET,
    )

    override fun apply(entity: LivingEntity, info: JsonObject): LivingEntity {
        names.forEach { (key, slot) ->
            info[key]?.let {
                entity.equipStack(slot, createItem(it.asString))
            }
        }
        return entity
    }

    private fun createItem(item: String): ItemStack {
        val split = item.split("#")
        if (split.size != 2) return SBItemStack(SkyblockId(item)).asImmutableItemStack()
        val (type, data) = split
        return when (type) {
            "SKULL" -> ItemStack(Items.PLAYER_HEAD).also { it.setEncodedSkullOwner(zeroUUID, data) }
            "LEATHER_LEGGINGS" -> coloredLeatherArmor(Items.LEATHER_LEGGINGS, data)
            "LEATHER_BOOTS" -> coloredLeatherArmor(Items.LEATHER_BOOTS, data)
            "LEATHER_HELMET" -> coloredLeatherArmor(Items.LEATHER_HELMET, data)
            "LEATHER_CHESTPLATE" -> coloredLeatherArmor(Items.LEATHER_CHESTPLATE, data)
            else -> error("Unknown leather piece: $type")
        }
    }

    private fun coloredLeatherArmor(leatherArmor: Item, data: String): ItemStack {
        val stack = ItemStack(leatherArmor)
        stack.set(DataComponentTypes.DYED_COLOR, DyedColorComponent(data.toInt(16), false))
        return stack
    }
}