Added support for Chat Image.

Author: NoMathExpectation

build.gradle.kts

@@ -22,6 +22,10 @@ repositories {
         url = uri("https://thedarkcolour.github.io/KotlinForForge/")
         content { includeGroup("thedarkcolour") }
     }
+    maven {
+        name = "kituinMavenReleases"
+        url = uri("https://maven.kituin.fun/releases")
+    }
 }
 
 base {
@@ -148,6 +152,10 @@ dependencies {
     externalImplementation("io.ktor:ktor-utils", "[2.0, 4.0)", "2.3.13")
     externalImplementation("io.ktor:ktor-network", "[2.0, 4.0)", "2.3.13")
 
+    // chatimage
+    compileOnly("io.github.kituin:ChatImageCode:0.12.1")
+    compileOnly("io.github.kituin:ChatImage:1.4.6+1.21.0+neoforge")
+
     // Example mod dependency with JEI
     // The JEI API is declared for compile time use, while the full JEI artifact is used at runtime
     // compileOnly "mezz.jei:jei-${mc_version}-common-api:${jei_version}"

src/main/kotlin/NoMathExpectation/chatExchange/neoForged/ExchangeServer.kt

@@ -1,5 +1,6 @@
 package NoMathExpectation.chatExchange.neoForged
 
+import NoMathExpectation.chatExchange.neoForged.chatImage.tryParseCICodeFileToData
 import io.ktor.network.selector.*
 import io.ktor.network.sockets.*
 import io.ktor.utils.io.*
@@ -122,11 +123,18 @@ class ExchangeServer(
     }
 
     suspend fun sendEvent(event: ExchangeEvent) {
-        logger.info("Sending event: $event")
+        var finalEvent = event
+        if (finalEvent is MessageEvent) {
+            finalEvent = finalEvent.copy(
+                content = finalEvent.content.tryParseCICodeFileToData()
+            )
+        }
+
+        logger.info("Sending event: $finalEvent")
         channelMutex.withLock {
             sendChannels.forEach {
                 kotlin.runCatching {
-                    it.writeExchangeEvent(event)
+                    it.writeExchangeEvent(finalEvent)
                 }.onFailure {
                     logger.error("Failed to send message to a client.", it)
                 }

src/main/kotlin/NoMathExpectation/chatExchange/neoForged/chatImage/ChatImageSupport.kt

@@ -0,0 +1,13 @@
+package NoMathExpectation.chatExchange.neoForged.chatImage
+
+import net.neoforged.fml.ModList
+
+internal val chatImageLoaded
+    get() = ModList.get().isLoaded("chatimage")
+
+suspend fun String.tryParseCICodeFileToData(): String {
+    if (!chatImageLoaded) {
+        return this
+    }
+    return tryParseCICodeFileToData0(this)
+}

src/main/kotlin/NoMathExpectation/chatExchange/neoForged/chatImage/ChatImageSupportInternal.kt

@@ -0,0 +1,65 @@
+package NoMathExpectation.chatExchange.neoForged.chatImage
+
+import NoMathExpectation.chatExchange.neoForged.ChatExchange
+import io.github.kituin.ChatImageCode.ChatImageCode
+import io.github.kituin.ChatImageCode.ServerStorage
+import kotlinx.coroutines.delay
+import org.apache.logging.log4j.LogManager
+import kotlin.time.Duration.Companion.seconds
+
+private val logger = LogManager.getLogger(ChatExchange.ID)
+
+private val ciCodeRegex = ChatImageCode.pattern.toRegex()
+
+private suspend fun findImageData(query: String): String? {
+    repeat(10) {
+        delay(0.5.seconds)
+
+        ServerStorage.SERVER_BLOCK_CACHE.getImage(query)?.let {
+            return it
+        }
+    }
+    //logger.info(ServerStorage.SERVER_BLOCK_CACHE.blockCache)
+    return null
+}
+
+internal suspend fun tryParseCICodeFileToData0(string: String) = buildString {
+    var codeEnd = 0
+    ciCodeRegex.findAll(string).forEach { match ->
+        val start = match.range.first
+        val end = match.range.last + 1
+
+        if (codeEnd < start) {
+            append(string.substring(codeEnd, start))
+        }
+
+        val data = match.value
+            .removeSurrounding("[[", "]]")
+            .trim()
+            .split(",")
+            .filter { "=" in it }
+            .map { it.split("=", limit = 2) }
+            .associate { it[0].trim() to it[1].trim() }
+            .toMutableMap()
+
+        var url = data["url"]
+        if (url?.startsWith("file:///") == true) {
+            val query = url.removePrefix("file:///")
+            url = "invalid"
+            findImageData(query)?.let {
+                url = "base64://$it"
+            }
+        }
+        data["url"] = url ?: "invalid"
+        if (url == null || url == "invalid") {
+            logger.warn("Unable to resolve CICode: ${match.value}")
+        }
+
+        val code = data.map { (k, v) -> "$k=$v" }
+            .joinToString(",", "[[CICode,", "]]")
+        append(code)
+
+        codeEnd = end
+    }
+    append(string.substring(codeEnd))
+}

src/main/templates/META-INF/neoforge.mods.toml

@@ -72,6 +72,13 @@ versionRange = "${minecraft_version_range}"
 ordering = "NONE"
 side = "BOTH"
 
+[[dependencies."${mod_id}"]]
+modId = "chatimage"
+type = "optional"
+versionRange = "[1.4.6, 2)"
+ordering = "AFTER"
+side = "BOTH"
+
 # Features are specific properties of the game environment, that you may want to declare you require. This example declares
 # that your mod requires GL version 3.2 or higher. Other features will be added. They are side aware so declaring this won't
 # stop your mod loading on the server for example.