mirror of
https://github.com/MewoLab/AquaDX.git
synced 2026-02-06 14:37:38 +08:00
38 lines
1.5 KiB
Kotlin
38 lines
1.5 KiB
Kotlin
package ext
|
|
|
|
import icu.samnyan.aqua.sega.util.ZLib
|
|
import java.net.URI
|
|
import java.net.http.HttpClient
|
|
import java.net.http.HttpRequest
|
|
import java.net.http.HttpResponse
|
|
import java.time.Duration
|
|
|
|
val client = HttpClient.newBuilder().build()
|
|
|
|
fun HttpRequest.Builder.send() = client.send(this.build(), HttpResponse.BodyHandlers.ofByteArray())
|
|
fun HttpRequest.Builder.header(pair: Pair<Any, Any>) = this.header(pair.first.toString(), pair.second.toString())
|
|
fun String.request() = HttpRequest.newBuilder(URI.create(this)).timeout(Duration.ofMinutes(5))
|
|
|
|
fun HttpRequest.Builder.post(body: Any? = null) = this.POST(when (body) {
|
|
is ByteArray -> HttpRequest.BodyPublishers.ofByteArray(body)
|
|
is String -> HttpRequest.BodyPublishers.ofString(body)
|
|
is HttpRequest.BodyPublisher -> body
|
|
else -> throw IllegalArgumentException("Unsupported body type")
|
|
}).send()
|
|
|
|
|
|
inline fun <reified T> HttpResponse<String>.json(): T? = body()?.json()
|
|
|
|
fun HttpRequest.Builder.postZ(body: String) = run {
|
|
header("Content-Type" to "application/json")
|
|
header("Content-Encoding" to "deflate")
|
|
post(ZLib.compress(body.toByteArray()))
|
|
}
|
|
|
|
fun <T> HttpResponse<T>.header(key: String) = headers().firstValue(key).orElse(null)
|
|
fun HttpResponse<ByteArray>.bodyString() = body()?.toString(Charsets.UTF_8)
|
|
fun HttpResponse<ByteArray>.bodyZ() = body()?.let { ZLib.decompress(it)?.decodeToString() }
|
|
fun HttpResponse<ByteArray>.bodyMaybeZ() =
|
|
if (body().first().let { it != '{'.code.toByte() && it != '['.code.toByte() }) bodyZ()
|
|
else bodyString()
|