1 files changed, 79 insertions, 0 deletions
diff --git a/src/texturePacks/java/moe/nea/firmament/features/texturepack/TreeishTextReplacer.kt b/src/texturePacks/java/moe/nea/firmament/features/texturepack/TreeishTextReplacer.kt
new file mode 100644
index 0000000..ed486f5
--- /dev/null
+++ b/src/texturePacks/java/moe/nea/firmament/features/texturepack/TreeishTextReplacer.kt
@@ -0,0 +1,79 @@
+package moe.nea.firmament.features.texturepack
+
+import java.util.regex.Matcher
+import util.json.CodecSerializer
+import kotlinx.serialization.Serializable
+import net.minecraft.network.chat.Style
+import net.minecraft.network.chat.Component
+import net.minecraft.network.chat.ComponentSerialization
+import moe.nea.firmament.util.directLiteralStringContent
+import moe.nea.firmament.util.transformEachRecursively
+
+@Serializable
+data class TreeishTextReplacer(
+	val match: StringMatcher,
+	val replacements: List<SubPartReplacement>
+) {
+	@Serializable
+	data class SubPartReplacement(
+        val match: StringMatcher,
+        val style: @Serializable(StyleSerializer::class) Style? = null,
+        val replace: @Serializable(TextSerializer::class) Component,
+	)
+
+	object TextSerializer : CodecSerializer<Component>(ComponentSerialization.CODEC)
+	object StyleSerializer : CodecSerializer<Style>(Style.Serializer.CODEC)
+	companion object {
+		val pattern = "[$]\\{(?<name>[^}]+)}".toPattern()
+		fun injectMatchResults(text: Component, matches: Matcher): Component {
+			return text.transformEachRecursively { it ->
+				val content = it.directLiteralStringContent ?: return@transformEachRecursively it
+				val matcher = pattern.matcher(content)
+				val builder = StringBuilder()
+				while (matcher.find()) {
+					matcher.appendReplacement(builder, matches.group(matcher.group("name")).toString())
+				}
+				matcher.appendTail(builder)
+				Component.literal(builder.toString()).setStyle(it.style)
+			}
+		}
+	}
+
+	fun match(text: Component): Boolean {
+		return match.matches(text)
+	}
+
+	fun replaceText(text: Component): Component {
+		return text.transformEachRecursively { part ->
+			var part: Component = part
+			for (replacement in replacements) {
+				val rawPartText = part.string
+				replacement.style?.let { expectedStyle ->
+					val parentStyle = part.style
+					val parented = expectedStyle.applyTo(parentStyle)
+					if (parented.isStrikethrough != parentStyle.isStrikethrough
+						|| parented.isObfuscated != parentStyle.isObfuscated
+						|| parented.isBold != parentStyle.isBold
+						|| parented.isUnderlined != parentStyle.isUnderlined
+						|| parented.isItalic != parentStyle.isItalic
+						|| parented.color?.value != parentStyle.color?.value)
+						continue
+				}
+				val matcher = replacement.match.asRegex.matcher(rawPartText)
+				if (!matcher.find()) continue
+				val p = Component.literal("")
+				p.setStyle(part.style)
+				var lastAppendPosition = 0
+				do {
+					p.append(rawPartText.substring(lastAppendPosition, matcher.start()))
+					lastAppendPosition = matcher.end()
+					p.append(injectMatchResults(replacement.replace, matcher))
+				} while (matcher.find())
+				p.append(rawPartText.substring(lastAppendPosition))
+				part = p
+			}
+			part
+		}
+	}
+
+}