138 lines
3.8 KiB
Kotlin
138 lines
3.8 KiB
Kotlin
package ru.dbotthepony.kstarbound.util
|
|
|
|
import com.google.gson.JsonArray
|
|
import com.google.gson.JsonElement
|
|
import com.google.gson.JsonNull
|
|
import com.google.gson.JsonObject
|
|
import com.google.gson.JsonPrimitive
|
|
import com.google.gson.JsonSyntaxException
|
|
import com.google.gson.TypeAdapter
|
|
import ru.dbotthepony.kstarbound.io.json.InternedJsonElementAdapter
|
|
|
|
operator fun JsonObject.set(key: String, value: JsonElement?) { add(key, value) }
|
|
operator fun JsonObject.set(key: String, value: String) { add(key, JsonPrimitive(value)) }
|
|
operator fun JsonObject.set(key: String, value: Int) { add(key, JsonPrimitive(value)) }
|
|
operator fun JsonObject.set(key: String, value: Long) { add(key, JsonPrimitive(value)) }
|
|
operator fun JsonObject.set(key: String, value: Float) { add(key, JsonPrimitive(value)) }
|
|
operator fun JsonObject.set(key: String, value: Double) { add(key, JsonPrimitive(value)) }
|
|
operator fun JsonObject.set(key: String, value: Boolean) { add(key, InternedJsonElementAdapter.of(value)) }
|
|
operator fun JsonObject.set(key: String, value: Nothing?) { add(key, JsonNull.INSTANCE) }
|
|
operator fun JsonObject.contains(key: String): Boolean = has(key)
|
|
|
|
fun JsonObject.get(key: String, default: Boolean): Boolean {
|
|
if (!has(key))
|
|
return default
|
|
|
|
val value = this[key]
|
|
|
|
if (value !is JsonPrimitive || !value.isBoolean) {
|
|
throw JsonSyntaxException("Expected boolean at $key; got $value")
|
|
}
|
|
|
|
return value.asBoolean
|
|
}
|
|
|
|
fun JsonObject.get(key: String, default: String): String {
|
|
if (!has(key))
|
|
return default
|
|
|
|
val value = this[key]
|
|
|
|
if (value !is JsonPrimitive || !value.isString) {
|
|
throw JsonSyntaxException("Expected string at $key; got $value")
|
|
}
|
|
|
|
return value.asString
|
|
}
|
|
|
|
fun JsonObject.get(key: String, default: Int): Int {
|
|
if (!has(key))
|
|
return default
|
|
|
|
val value = this[key]
|
|
|
|
if (value !is JsonPrimitive || !value.isNumber) {
|
|
throw JsonSyntaxException("Expected integer at $key; got $value")
|
|
}
|
|
|
|
return value.asInt
|
|
}
|
|
|
|
fun JsonObject.get(key: String, default: Long): Long {
|
|
if (!has(key))
|
|
return default
|
|
|
|
val value = this[key]
|
|
|
|
if (value !is JsonPrimitive || !value.isNumber) {
|
|
throw JsonSyntaxException("Expected long at $key; got $value")
|
|
}
|
|
|
|
return value.asLong
|
|
}
|
|
|
|
fun JsonObject.get(key: String, default: Double): Double {
|
|
if (!has(key))
|
|
return default
|
|
|
|
val value = this[key]
|
|
|
|
if (value !is JsonPrimitive || !value.isNumber) {
|
|
throw JsonSyntaxException("Expected double at $key; got $value")
|
|
}
|
|
|
|
return value.asDouble
|
|
}
|
|
|
|
fun JsonObject.get(key: String, default: JsonObject): JsonObject {
|
|
if (!has(key))
|
|
return default
|
|
|
|
val value = this[key]
|
|
|
|
if (value !is JsonObject)
|
|
throw JsonSyntaxException("Expected json object at $key; got $value")
|
|
|
|
return value
|
|
}
|
|
|
|
fun JsonObject.get(key: String, default: JsonArray): JsonArray {
|
|
if (!has(key))
|
|
return default
|
|
|
|
val value = this[key]
|
|
|
|
if (value !is JsonArray)
|
|
throw JsonSyntaxException("Expected json array at $key; got $value")
|
|
|
|
return value
|
|
}
|
|
|
|
fun JsonObject.get(key: String, default: JsonPrimitive): JsonElement {
|
|
if (!has(key))
|
|
return default
|
|
|
|
return this[key]
|
|
}
|
|
|
|
fun <T> JsonObject.get(key: String, type: TypeAdapter<T>): T {
|
|
return get(key, type) { throw JsonSyntaxException("Expected value at $key, got nothing") }
|
|
}
|
|
|
|
inline fun <T> JsonObject.get(key: String, type: TypeAdapter<out T>, orElse: () -> T): T {
|
|
if (!has(key))
|
|
return orElse.invoke()
|
|
|
|
return type.fromJsonTree(this[key])
|
|
}
|
|
|
|
fun JsonObject.getArray(key: String): JsonArray {
|
|
if (!has(key)) throw JsonSyntaxException("Expected array at $key, got nothing")
|
|
return this[key] as? JsonArray ?: throw JsonSyntaxException("Expected array at $key, got ${this[key]}")
|
|
}
|
|
|
|
fun JsonObject.getObject(key: String): JsonObject {
|
|
if (!has(key)) throw JsonSyntaxException("Expected object at $key, got nothing")
|
|
return this[key] as? JsonObject ?: throw JsonSyntaxException("Expected object at $key, got ${this[key]}")
|
|
}
|