aboutsummaryrefslogtreecommitdiff
path: root/src/main/kotlin/moe/nea/firmament/features/texturepack/CustomModelOverrideParser.kt
blob: 458f8ff70f754c6471a0861369b24b86f7a7c9c2 (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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
/*
 * SPDX-FileCopyrightText: 2024 Linnea Gräf <nea@nea.moe>
 *
 * SPDX-License-Identifier: GPL-3.0-or-later
 */

package moe.nea.firmament.features.texturepack

import com.google.gson.JsonObject
import kotlinx.serialization.KSerializer
import kotlinx.serialization.descriptors.SerialDescriptor
import kotlinx.serialization.encoding.Decoder
import kotlinx.serialization.encoding.Encoder
import net.minecraft.item.ItemStack
import net.minecraft.util.Identifier

object CustomModelOverrideParser {
    object FirmamentRootPredicateSerializer : KSerializer<FirmamentModelPredicate> {
        val delegateSerializer = kotlinx.serialization.json.JsonObject.serializer()
        override val descriptor: SerialDescriptor
            get() = SerialDescriptor("FirmamentModelRootPredicate", delegateSerializer.descriptor)

        override fun deserialize(decoder: Decoder): FirmamentModelPredicate {
            val json = decoder.decodeSerializableValue(delegateSerializer).intoGson() as JsonObject
            return AndPredicate(parsePredicates(json).toTypedArray())
        }

        override fun serialize(encoder: Encoder, value: FirmamentModelPredicate) {
            TODO("Cannot serialize firmament predicates")
        }
    }

    val predicateParsers = mutableMapOf<Identifier, FirmamentModelPredicateParser>()


    fun registerPredicateParser(name: String, parser: FirmamentModelPredicateParser) {
        predicateParsers[Identifier("firmament", name)] = parser
    }

    init {
        registerPredicateParser("display_name", DisplayNamePredicate.Parser)
        registerPredicateParser("lore", LorePredicate.Parser)
        registerPredicateParser("all", AndPredicate.Parser)
        registerPredicateParser("any", OrPredicate.Parser)
        registerPredicateParser("not", NotPredicate.Parser)
        registerPredicateParser("item", ItemPredicate.Parser)
        registerPredicateParser("extra_attributes", ExtraAttributesPredicate.Parser)
    }

    private val neverPredicate = listOf(
        object : FirmamentModelPredicate {
            override fun test(stack: ItemStack): Boolean {
                return false
            }
        }
    )

    fun parsePredicates(predicates: JsonObject): List<FirmamentModelPredicate> {
        val parsedPredicates = mutableListOf<FirmamentModelPredicate>()
        for (predicateName in predicates.keySet()) {
            if (!predicateName.startsWith("firmament:")) continue
            val identifier = Identifier(predicateName)
            val parser = predicateParsers[identifier] ?: return neverPredicate
            val parsedPredicate = parser.parse(predicates[predicateName]) ?: return neverPredicate
            parsedPredicates.add(parsedPredicate)
        }
        return parsedPredicates
    }

    @JvmStatic
    fun parseCustomModelOverrides(jsonObject: JsonObject): Array<FirmamentModelPredicate>? {
        val predicates = (jsonObject["predicate"] as? JsonObject) ?: return null
        val parsedPredicates = parsePredicates(predicates)
        if (parsedPredicates.isEmpty())
            return null
        return parsedPredicates.toTypedArray()
    }
}