categorize thingies

1 files changed, 69 insertions, 0 deletions

diff --git a/src/main/kotlin/util/minecrat/TextUtil.kt b/src/main/kotlin/util/minecrat/TextUtil.kt
index 2a09e13..0640021 100644
--- a/src/main/kotlin/util/minecrat/TextUtil.kt
+++ b/src/main/kotlin/util/minecrat/TextUtil.kt
@@ -1 +1,70 @@
 package moe.nea.ultranotifier.util.minecrat
+import net.minecraft.text.Text
+
+//#if MC > 1.16
+import net.minecraft.text.TextColor
+import net.minecraft.util.Formatting
+//#endif
+//#if MC > 1.20
+import net.minecraft.text.MutableText
+import net.minecraft.text.PlainTextContent
+//#endif
+
+fun Text.getDirectlyContainedText() =
+//#if MC < 1.16
+//$$	this.unformattedComponentText
+//#elseif MC < 1.20
+//$$    this.asString()
+//#else
+        (this.content as? PlainTextContent)?.string().orEmpty()
+//#endif
+
+fun Text?.getFormattedTextCompat(): String =
+//#if MC < 1.16
+//$$	this?.formattedText.orEmpty()
+//#else
+run {
+    this ?: return@run ""
+    val sb = StringBuilder()
+    for (component in iterator()) {
+        sb.append(component.style.color?.toChatFormatting()?.toString() ?: "§r")
+        sb.append(component.getDirectlyContainedText())
+        sb.append("§r")
+    }
+    sb.toString()
+}
+
+private val textColorLUT = Formatting.entries
+    .mapNotNull { formatting -> formatting.colorValue?.let { it to formatting } }
+    .toMap()
+
+fun TextColor.toChatFormatting(): Formatting? {
+    return textColorLUT[this.rgb]
+}
+
+fun Text.iterator(): Sequence<Text> {
+    return sequenceOf(this) + siblings.asSequence().flatMap { it.iterator() } // TODO: in theory we want to properly inherit styles here
+}
+//#endif
+
+//#if MC > 1.20
+fun MutableText.withColor(formatting: Formatting): Text {
+    return this.styled { it.withColor(formatting) }
+}
+//#endif
+
+fun CharSequence.removeFormattingCodes(): String {
+	var nextParagraph = indexOf('§')
+	if (nextParagraph < 0) return this.toString()
+	val stringBuffer = StringBuilder(this.length)
+	var readIndex = 0
+	while (nextParagraph >= 0) {
+		stringBuffer.append(this, readIndex, nextParagraph)
+		readIndex = nextParagraph + 2
+		nextParagraph = indexOf('§', startIndex = readIndex)
+		if (readIndex > this.length)
+			readIndex = this.length
+	}
+	stringBuffer.append(this, readIndex, this.length)
+	return stringBuffer.toString()
+}