From 602112724d8236c1ec6671e1893128862c9f5815 Mon Sep 17 00:00:00 2001
From: Linnea Gräf <nea@nea.moe>
Date: Thu, 22 Feb 2024 00:30:05 +0100
Subject: Add custom model predicates

Add regex support

Add and and or predicates
---
 .../features/texturepack/DisplayNamePredicate.kt   | 28 ++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 src/main/kotlin/moe/nea/firmament/features/texturepack/DisplayNamePredicate.kt

(limited to 'src/main/kotlin/moe/nea/firmament/features/texturepack/DisplayNamePredicate.kt')

diff --git a/src/main/kotlin/moe/nea/firmament/features/texturepack/DisplayNamePredicate.kt b/src/main/kotlin/moe/nea/firmament/features/texturepack/DisplayNamePredicate.kt
new file mode 100644
index 0000000..373910a
--- /dev/null
+++ b/src/main/kotlin/moe/nea/firmament/features/texturepack/DisplayNamePredicate.kt
@@ -0,0 +1,28 @@
+/*
+ * 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.JsonElement
+import net.minecraft.item.ItemStack
+import net.minecraft.nbt.NbtElement
+import net.minecraft.nbt.NbtString
+
+data class DisplayNamePredicate(val stringMatcher: StringMatcher) : FirmamentModelPredicate {
+    override fun test(stack: ItemStack): Boolean {
+        val display = stack.getOrCreateSubNbt(ItemStack.DISPLAY_KEY)
+        return if (display.contains(ItemStack.NAME_KEY, NbtElement.STRING_TYPE.toInt()))
+            stringMatcher.matches(display.get(ItemStack.NAME_KEY) as NbtString)
+        else
+            false
+    }
+
+    object Parser : FirmamentModelPredicateParser {
+        override fun parse(jsonElement: JsonElement): FirmamentModelPredicate {
+            return DisplayNamePredicate(StringMatcher.parse(jsonElement))
+        }
+    }
+}
-- 
cgit