org.jetbrains.kotlinx.kandy.util.serialization.specSerialization.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kandy-util Show documentation
Kotlin plotting DSL.
The newest version!
/*
* Copyright 2020-2023 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
*/

package org.jetbrains.kotlinx.kandy.util.serialization

import kotlinx.serialization.json.*

public typealias LetsPlotSpec = Map

public fun serializeSpec(spec: LetsPlotSpec): JsonElement {
    return serialize(spec)
}

private fun serializeAny(obj: Any?): JsonElement {
    return when (obj) {
        null -> JsonNull
        is Map<*, *> -> serialize(obj)
        is List<*> -> serialize(obj)
        is String -> JsonPrimitive(obj)
        is Boolean -> JsonPrimitive(obj)
        is Number -> JsonPrimitive(obj)
        else -> error("Don't know how to parse object [$obj] of class ${obj::class}")
    }
}

private fun serialize(map: Map<*, *>): JsonObject {
    return buildJsonObject {
        for ((key, value) in map) {
            if (key !is String) error("Map key [$key] is of type ${key?.let { it::class }}. Don't know how to serialize it.")
            put(key, serializeAny(value))
        }
    }
}

private fun serialize(list: List<*>): JsonArray {
    return buildJsonArray {
        for (value in list) {
            add(serializeAny(value))
        }
    }
}

public fun deserializeSpec(json: JsonElement): LetsPlotSpec {
    if (json !is JsonObject) error("LetsPlot spec should be a key-value object, but it's $json")
    val map = deserializeMap(json)

    for (value in map.values) {
        if (value == null) error("LetsPlot spec shouldn't have null values on the top level")
    }

    @Suppress("UNCHECKED_CAST")
    return map as LetsPlotSpec
}

private fun deserializeAny(json: JsonElement): Any? {
    return when (json) {
        is JsonObject -> deserializeMap(json)
        is JsonArray -> deserializeList(json)
        is JsonPrimitive -> deserializePrimitive(json)
    }
}

private fun deserializePrimitive(json: JsonPrimitive): Any? {
    return when {
        json is JsonNull -> null
        json.isString -> json.content
        else -> {
            json.booleanOrNull ?: json.doubleOrNull ?: error("Unknown JSON primitive type: [$json]")
        }
    }
}

private fun deserializeMap(json: JsonObject): Map {
    return buildMap {
        for ((key, value) in json) {
            put(key, deserializeAny(value))
        }
    }
}

private fun deserializeList(jsonArray: JsonArray): List {
    return buildList {
        for (el in jsonArray) {
            add(deserializeAny(el))
        }
    }
}