io.javalin.openapi.experimental.processor.shared.JsonExtensions.kt Maven / Gradle / Ivy

Go to download
package io.javalin.openapi.experimental.processor.shared

import com.google.gson.Gson
import com.google.gson.GsonBuilder
import com.google.gson.JsonArray
import com.google.gson.JsonObject
import io.javalin.openapi.NULL_STRING

private val gson: Gson = GsonBuilder()
    .setPrettyPrinting()
    .create()

fun JsonObject.toPrettyString(): String =
    gson.toJson(this)

fun Map.toJsonObject(): JsonObject {
    val jsonObject = JsonObject()
    forEach { (key, value) -> jsonObject.addProperty(key, value) }
    return jsonObject
}

fun  List.toJsonArray(accumulator: JsonArray.(T) -> Unit): JsonArray {
    val jsonArray = JsonArray(size)
    forEach { accumulator(jsonArray, it) }
    return jsonArray
}

fun  Array.toJsonArray(mapper: (T) -> String = { it.toString() }): JsonArray {
    val jsonArray = JsonArray(size)
    map(mapper).forEach { jsonArray.add(it) }
    return jsonArray
}

fun JsonObject.computeIfAbsent(key: String, value: () -> JsonObject): JsonObject {
    if (!has(key)) {
        add(key, value())
    }

    return getAsJsonObject(key)
}

fun JsonObject.addString(key: String, value: String?): JsonObject = also {
    if (NULL_STRING != value) {
        addProperty(key, value)
    }
}

fun createJsonObjectOf(key: String, value: String): JsonObject {
    val jsonObject = JsonObject()
    jsonObject.addProperty(key, value)
    return jsonObject
}