org.http4k.jsonrpc.internal.kt Maven / Gradle / Ivy

Go to download
package org.http4k.jsonrpc

import org.http4k.format.Json
import org.http4k.format.JsonType

internal class ParamMappingJsonRequestHandler(json: Json,
                                                                         paramsFieldNames: Iterable,
                                                                         paramsLens: Mapping,
                                                                         function: (IN) -> OUT,
                                                                         resultLens: Mapping) : JsonRpcHandler {
    private val handler: (NODE) -> NODE = {
        val input = when (json.typeOf(it)) {
            JsonType.Array -> {
                val elements = json.elements(it).toList()
                paramsFieldNames.mapIndexed { index: Int, name: String ->
                    name to elements.getOrElse(index) { json.nullNode() }
                }.takeUnless { it.isEmpty() }
                        ?.let { json.obj(it) }
                        ?: json.nullNode()
            }
            else -> it
        }
        paramsLens(input).let(function).let(resultLens)
    }

    override fun invoke(request: NODE): NODE = handler(request)
}

internal class NoParamsJsonRequestHandler(function: () -> OUT, resultLens: Mapping) :
        JsonRpcHandler {

    private val handler: (NODE) -> NODE = { function().let(resultLens) }

    override fun invoke(request: NODE): NODE = handler(request)
}