WrappedHttpResponseTransform.kt

import io.ktor.client.HttpClient
import io.ktor.client.features.HttpClientFeature
import io.ktor.client.response.HttpResponse
import io.ktor.client.response.HttpResponseContainer
import io.ktor.client.response.HttpResponsePipeline
import io.ktor.util.AttributeKey

internal class WrappedHttpResponseTransform {

    companion object Feature : HttpClientFeature<Unit, WrappedHttpResponseTransform> {
        override val key: AttributeKey<WrappedHttpResponseTransform> = AttributeKey("WrappedHttpResponseTransform")
        override fun prepare(block: Unit.() -> Unit): WrappedHttpResponseTransform = WrappedHttpResponseTransform()

        override fun install(feature: WrappedHttpResponseTransform, scope: HttpClient) {
            scope.responsePipeline.intercept(HttpResponsePipeline.Transform) { (info, response) ->
                if (response !is HttpResponse) {
                    return@intercept
                }
                proceedWith(HttpResponseContainer(info, WrappedHttpResponse(response)))
            }
        }
    }
}