Basic websocket backend + a few api endpoints

Author: SubAt0m1c

build.gradle.kts

@@ -2,6 +2,7 @@
 
 import com.github.jengelman.gradle.plugins.shadow.tasks.ShadowJar
 import dev.deftu.gradle.utils.GameSide
+import dev.deftu.gradle.utils.includeOrShade
 
 plugins {
     java
@@ -13,6 +14,11 @@ plugins {
     id("dev.deftu.gradle.tools.shadow") // Applies the Shadow plugin, which allows us to shade our dependencies into our mod JAR. This is NOT recommended for Fabric mods, but we have an *additional* configuration for those!
     id("dev.deftu.gradle.tools.minecraft.loom") // Applies the Loom plugin, which automagically configures Essential's Architectury Loom plugin for you.
     id("dev.deftu.gradle.tools.minecraft.releases") // Applies the Minecraft auto-releasing plugin, which allows you to automatically release your mod to CurseForge and Modrinth.
+    kotlin("plugin.serialization")
+}
+
+repositories {
+    mavenCentral()
 }
 
 toolkitLoomHelper {
@@ -59,4 +65,19 @@ dependencies {
         }
     }
     modImplementation("com.github.JnCrMx:discord-game-sdk4j:v0.5.5")
+
+    implementation("io.ktor:ktor-client-core:3.3.1")
+    shade("io.ktor:ktor-client-core:3.3.1")
+    implementation("io.ktor:ktor-client-cio:3.3.1")
+    shade("io.ktor:ktor-client-cio:3.3.1")
+    implementation("io.ktor:ktor-client-content-negotiation:3.3.1")
+    shade("io.ktor:ktor-client-content-negotiation:3.3.1")
+    implementation("io.ktor:ktor-serialization-kotlinx-json:3.3.1")
+    shade("io.ktor:ktor-serialization-kotlinx-json:3.3.1")
+}
+
+tasks {
+    named("build") {
+        dependsOn("fatJar")
+    }
 }

gradle.properties

@@ -3,7 +3,7 @@ org.gradle.daemon=true
 org.gradle.parallel=true
 org.gradle.configureoncommand=true
 org.gradle.parallel.threads=4
-org.gradle.jvmargs=-Xmx2G
+org.gradle.jvmargs=-Xmx5G
 loom.ignoreDependencyLoomVersionValidation=true
 
 # gradle.properties file -- CHANGE THE VALUES STARTING WITH `mod.*` AND REMOVE THIS COMMENT.

settings.gradle.kts

@@ -24,6 +24,7 @@ pluginManagement {
 
     plugins {
         kotlin("jvm") version("2.2.10")
+        kotlin("plugin.serialization") version("2.2.10")
         id("dev.deftu.gradle.multiversion-root") version("2.51.0")
     }
 }

src/main/java/org/polyfrost/polyplus/client/mixin/Mixin_ReplaceCapeTexture.java

@@ -20,6 +20,6 @@ public class Mixin_ReplaceCapeTexture {
 
     @Inject(method = "getLocationCape", at = @org.spongepowered.asm.mixin.injection.At("HEAD"))
     private void polyplus$onGetCape(CallbackInfoReturnable<ResourceLocation> cir) {
-        this.locationCape = new ResourceLocation(PolyPlus.ID, "randomcape2.png");
+        this.locationCape = new ResourceLocation(PolyPlus.ID, "64px_poly.png");
     }
 }

src/main/kotlin/org/polyfrost/polyplus/PolyPlus.kt

@@ -1,12 +1,20 @@
 package org.polyfrost.polyplus
 
+import io.ktor.client.HttpClient
+import io.ktor.client.engine.cio.CIO
+import io.ktor.client.plugins.contentnegotiation.ContentNegotiation
+import io.ktor.client.plugins.defaultRequest
+import io.ktor.http.userAgent
+import io.ktor.serialization.kotlinx.json.json
 import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.SupervisorJob
-import net.minecraft.client.Minecraft
 import org.polyfrost.oneconfig.api.commands.v1.CommandManager
 import org.polyfrost.polyplus.client.ExampleCommand
 import org.polyfrost.polyplus.client.Config
+import org.polyfrost.polyplus.discordrpc.RPC
+import java.time.Instant
+import java.util.logging.Logger
 
 //#if FABRIC
 //$$ import net.fabricmc.api.ModInitializer
@@ -23,9 +31,6 @@ import org.polyfrost.polyplus.client.Config
 //#else
 import net.minecraftforge.fml.common.Mod
 import net.minecraftforge.fml.common.event.FMLInitializationEvent
-import org.polyfrost.polyplus.discordrpc.RPC
-import java.time.Instant
-
 //#endif
 //#elseif NEOFORGE
 //$$ import net.neoforged.bus.api.IEventBus
@@ -37,7 +42,6 @@ import java.time.Instant
 
 
 //#if FORGE-LIKE
-//$$ import org.polyfrost.example.ExampleConstants
 //#if MC >= 1.16.5
 //$$ @Mod(PolyPlus.ID)
 //#else
@@ -91,7 +95,7 @@ class PolyPlus
         //#endif
         //#endif
     ) {
-        //#if MC <= 1.12.2
+        //#if FORGE-LIKE && MC <= 1.12.2
         if (!event.side.isClient) return
         //#endif
 
@@ -109,6 +113,17 @@ class PolyPlus
     //#endif
 
     companion object {
+        val logger = Logger.getLogger(NAME)
+
+        val client = HttpClient(CIO) {
+            defaultRequest {
+                userAgent("todo")
+            }
+
+            install(ContentNegotiation) {
+                json()
+            }
+        }
         var launch: Instant? = null
 
         val scope = CoroutineScope(SupervisorJob() + Dispatchers.Default)

src/main/kotlin/org/polyfrost/polyplus/client/Config.kt

@@ -4,6 +4,7 @@ import org.polyfrost.oneconfig.api.config.v1.Config
 import org.polyfrost.oneconfig.api.config.v1.annotations.Dropdown
 import org.polyfrost.oneconfig.api.config.v1.annotations.Slider
 import org.polyfrost.oneconfig.api.config.v1.annotations.Switch
+import org.polyfrost.oneconfig.api.config.v1.annotations.Text
 import org.polyfrost.polyplus.PolyPlus
 
 object Config : Config("${PolyPlus.ID}.json", PolyPlus.NAME, Category.OTHER) {
@@ -12,11 +13,13 @@ object Config : Config("${PolyPlus.ID}.json", PolyPlus.NAME, Category.OTHER) {
     @Switch(title = "Discord RPC")
     var rpcEnabled = true // The default value for the boolean Switch
 
-    @JvmStatic
-    @Slider(title = "Example Slider", min = 0f, max = 100f, step = 10f)
-    var exampleSlider = 50f // The default value for the float Slider
-
-    @Dropdown(title = "Example Dropdown", options = ["Option 1", "Option 2", "Option 3", "Option 4"])
-    var exampleDropdown = 1 // Default option (in this case, "Option 2")
+//    @JvmStatic
+//    @Slider(title = "Example Slider", min = 0f, max = 100f, step = 10f)
+//    var exampleSlider = 50f // The default value for the float Slider
+//
+//    @Dropdown(title = "Example Dropdown", options = ["Option 1", "Option 2", "Option 3", "Option 4"])
+//    var exampleDropdown = 1 // Default option (in this case, "Option 2")
 
+    @Text(title = "API URL", description = "The url for the polyplus api. Only change if you know what you're doing.", placeholder = "https://plus.polyfrost.org/")
+    var apiUrl: String = "https://plus-staging.polyfrost.org/"
 }

src/main/kotlin/org/polyfrost/polyplus/discordrpc/DownloadSDK.kt

@@ -1,8 +1,14 @@
 package org.polyfrost.polyplus.discordrpc
 
+import io.ktor.client.call.body
+import io.ktor.client.request.get
+import io.ktor.client.statement.bodyAsChannel
+import io.ktor.client.statement.request
+import io.ktor.utils.io.jvm.javaio.toInputStream
 import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.withContext
 import org.polyfrost.oneconfig.utils.v1.dsl.mc
+import org.polyfrost.polyplus.PolyPlus.Companion.client
 import java.io.BufferedInputStream
 import java.io.File
 import java.net.URI
@@ -31,11 +37,9 @@ object DownloadSDK {
     val url = URI.create(SDK_URL).toURL()
 
     suspend fun download(): File? = withContext(Dispatchers.IO) {
-        val connection = url.openConnection().apply {
-            setRequestProperty("User-Agent", "todo")
-        }
+        val test = client.get(url)
 
-        ZipInputStream(BufferedInputStream(connection.getInputStream())).use { stream ->
+        ZipInputStream(BufferedInputStream(test.bodyAsChannel().toInputStream())).use { stream ->
             for (entry in generateSequence { stream.nextEntry }) {
                 if (entry.name != path) continue
 

src/main/kotlin/org/polyfrost/polyplus/discordrpc/RPC.kt

@@ -7,6 +7,7 @@ import de.jcm.discordgamesdk.activity.Activity
 import kotlinx.coroutines.launch
 import kotlinx.coroutines.time.delay
 import org.polyfrost.polyplus.PolyPlus
+import org.polyfrost.polyplus.PolyPlus.Companion.logger
 import org.polyfrost.polyplus.client.Config.rpcEnabled
 import java.time.Duration
 import java.util.concurrent.atomic.AtomicBoolean
@@ -19,7 +20,7 @@ object RPC {
     fun start() {
         if (!rpcEnabled || running.getAndSet(true)) return
         PolyPlus.scope.launch { run().onFailure {
-            println("[PolyPlus] Failed to start Discord RPC: ${it.message}")
+            logger.warning("Failed to start Discord RPC: ${it.message}")
             running.set(false)
         } }
     }

src/main/kotlin/org/polyfrost/polyplus/network/plus/Auth.kt

@@ -0,0 +1,112 @@
+package org.polyfrost.polyplus.network.plus
+
+import dev.deftu.omnicore.api.client.client
+import dev.deftu.omnicore.api.client.player.uuid
+import io.ktor.client.HttpClient
+import io.ktor.client.call.body
+import io.ktor.client.request.HttpRequestBuilder
+import io.ktor.client.request.bearerAuth
+import io.ktor.client.request.post
+import io.ktor.client.request.request
+import io.ktor.client.request.url
+import io.ktor.client.statement.HttpResponse
+import io.ktor.http.HttpMethod
+import io.ktor.http.HttpStatusCode
+import io.ktor.http.Url
+import kotlinx.coroutines.CoroutineStart
+import kotlinx.coroutines.Deferred
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.async
+import kotlinx.coroutines.sync.Mutex
+import kotlinx.coroutines.sync.withLock
+import kotlinx.coroutines.withContext
+import org.polyfrost.polyplus.PolyPlus
+import org.polyfrost.polyplus.PolyPlus.Companion.logger
+import org.polyfrost.polyplus.client.Config
+import org.polyfrost.polyplus.network.plus.responses.AuthResponse
+import kotlin.concurrent.atomics.ExperimentalAtomicApi
+
+@OptIn(ExperimentalAtomicApi::class)
+object Auth {
+    val authLock = Mutex()
+    var authRes: AuthResponse? = null
+    var authJob: Deferred<AuthResponse>? = null
+
+    suspend fun getToken(): String {
+        return authLock.withLock {
+            authRes?.token ?: refreshToken()
+        }
+    }
+
+    suspend fun refreshToken(): String {
+        authJob?.let { return it.await().token }
+
+        return authLock.withLock {
+            authJob?.let { return it.await().token }
+
+            val job = PolyPlus.scope.async(start = CoroutineStart.LAZY) {
+                runAuth().also {
+                    authLock.withLock {
+                        authRes = it
+                        authJob = null
+                    }
+                }
+            }
+
+            authJob = job
+            job.start()
+            job.await().token
+        }
+    }
+
+    suspend inline fun HttpClient.authRequest(method: HttpMethod, url: String, noinline builder: HttpRequestBuilder.() -> Unit): Result<HttpResponse> = runCatching {
+        var response = this.request {
+            apply(builder)
+            url(url)
+            this.method = method
+            bearerAuth(getToken())
+        }
+
+        if (response.status == HttpStatusCode.Unauthorized) {
+            response = this.request {
+                apply(builder)
+                url(url)
+                this.method = method
+                bearerAuth(refreshToken())
+            }
+        }
+
+        return Result.success(response)
+    }
+
+    suspend fun runAuth() = withContext(Dispatchers.IO) {
+        val serverId = genServerId()
+        mojangAuth(serverId)
+        val auth: AuthResponse = PolyPlus.client.post(Url(Config.apiUrl + "account/login?server_id=$serverId&username=${client.session.username}")).body()
+        logger.info("Successfully authenticated with PolyPlus")
+        auth
+    }
+
+    suspend inline fun <reified T> HttpClient.authRequest(method: HttpMethod, url: String) = runCatching {
+        val response = this.authRequest(method, url) {}
+        response.getOrElse { return Result.failure<T>(it) }.body<T>()
+    }
+}
+
+fun genServerId(): String {
+    val chars = ('a'..'z') + ('A'..'Z')
+    return (0..<32).joinToString("") { "${chars.random()}" }
+}
+
+fun mojangAuth(serverId: String) {
+    try {
+        //#if MC >= 1.20.4
+        //$$ val profile = client.uuid
+        //#else
+        val profile = client.session.profile
+        //#endif
+        client.sessionService.joinServer(profile, client.session.token, serverId)
+    } catch (e: Exception) {
+        logger.warning("Failed to authenticate with Mojang: ${e.message}")
+    }
+}

src/main/kotlin/org/polyfrost/polyplus/network/plus/Cosmetics.kt

@@ -0,0 +1,43 @@
+package org.polyfrost.polyplus.network.plus
+
+import io.ktor.client.call.body
+import io.ktor.client.request.get
+import io.ktor.client.request.request
+import io.ktor.client.request.setBody
+import io.ktor.client.statement.bodyAsText
+import io.ktor.http.ContentType
+import io.ktor.http.HttpMethod
+import io.ktor.http.HttpStatusCode
+import io.ktor.http.contentType
+import kotlinx.coroutines.async
+import kotlinx.coroutines.launch
+import org.polyfrost.polyplus.PolyPlus
+import org.polyfrost.polyplus.PolyPlus.Companion.logger
+import org.polyfrost.polyplus.client.Config
+import org.polyfrost.polyplus.network.plus.Auth.authRequest
+import org.polyfrost.polyplus.network.plus.responses.Cosmetic
+import org.polyfrost.polyplus.network.plus.responses.PlayerCosmetics
+import org.polyfrost.polyplus.network.plus.responses.PutCosmetics
+
+object Cosmetics {
+    fun getOwned() = PolyPlus.scope.async {
+        val cosmetics: Result<PlayerCosmetics> = PolyPlus.client.authRequest(HttpMethod.Get, "${Config.apiUrl}/cosmetics/player")
+        cosmetics.onFailure { logger.warning("Failed to fetch owned cosmetics: ${it.message}") }
+        // put on player probably?
+    }
+
+    fun setOwned(cosmetics: PutCosmetics) = PolyPlus.scope.launch {
+        val response = PolyPlus.client.authRequest(HttpMethod.Put, "${Config.apiUrl}/cosmetics/player") {
+            contentType(ContentType.Application.Json)
+            setBody(cosmetics)
+        }.onFailure { logger.warning("Failed to set owned cosmetic: ${it.message}") }.getOrNull() ?: return@launch
+
+        if (response.status != HttpStatusCode.OK) logger.warning("Failed to set owned cosmetic: ${response.status}, ${response.bodyAsText()}")
+    }
+
+    fun getAll() = PolyPlus.scope.async {
+        val cosmetics: Result<List<Cosmetic>> = runCatching { PolyPlus.client.get("${Config.apiUrl}/cosmetics") }.map { it.body() }
+        cosmetics.onFailure { logger.warning("Failed to fetch all cosmetics: ${it.message}") }
+        // store somewhere/do something with them
+    }
+}