org.http4k.client.HelidonClient.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of http4k-client-helidon Show documentation
HTTP Client built on top of Helidon
There is a newer version: 5.41.0.0
Show newest version
package org.http4k.client

import io.helidon.http.HeaderNames
import io.helidon.http.Method
import io.helidon.webclient.api.HttpClientResponse
import io.helidon.webclient.api.WebClient
import org.http4k.core.BodyMode
import org.http4k.core.BodyMode.Memory
import org.http4k.core.HttpHandler
import org.http4k.core.Request
import org.http4k.core.Response
import org.http4k.core.Status
import org.http4k.core.Status.Companion.CLIENT_TIMEOUT
import org.http4k.core.Status.Companion.UNKNOWN_HOST
import org.http4k.core.queries
import org.http4k.core.toParametersMap
import java.io.UncheckedIOException
import java.net.ConnectException
import java.net.SocketTimeoutException
import java.net.UnknownHostException

object HelidonClient {
    @JvmStatic
    @JvmOverloads
    @JvmName("create")
    operator fun invoke(
        client: WebClient = WebClient.builder().followRedirects(false).build(),
        bodyMode: BodyMode = Memory,
    ): HttpHandler = object : HttpHandler {
        override fun invoke(request: Request) = try {
            client.makeHelidonRequest(request)
                .submit(request.body.payload.array())
                .asHttp4k()
        } catch (e: IllegalArgumentException) {
            when {
                e.localizedMessage.contains("Failed to get address") -> Response(UNKNOWN_HOST.toClientStatus(e))
                else -> throw e
            }
        } catch (e: UncheckedIOException) {
            when (e.cause) {
                is UnknownHostException -> Response(UNKNOWN_HOST.toClientStatus(e))
                is ConnectException -> Response(UNKNOWN_HOST.toClientStatus(e))
                is SocketTimeoutException -> Response(CLIENT_TIMEOUT.toClientStatus(e))
                else -> throw e
            }
        }

        private fun HttpClientResponse.asHttp4k() =
            headers()
                .fold(Response(Status(status().code(), status().reasonPhrase()))) { acc, header ->
                    header.allValues().fold(acc) { acc2, value ->
                        acc2.header(header.name(), value)
                    }
                }
                .body(bodyMode(inputStream()))
    }

    internal fun WebClient.makeHelidonRequest(request: Request) =
        method(Method.create(request.method.name))
            .uri(request.uri.copy(query = "").toString())
            .apply {
                request.uri.queries().toParametersMap().forEach { (name, values) ->
                    // Replacing space with '+' because unlike other http clients, Helidon encodes space as %20
                    queryParam(name, *values.map { it?.replace(' ', '+') }.toTypedArray())
                }
                request.headers.groupBy { it.first }.entries.fold(this) { acc, (key, parameters) ->
                    acc.header(HeaderNames.create(key.lowercase(), key), parameters.map { it.second })
                }
            }
}