org.http4k.lens.header.kt Maven / Gradle / Ivy

Go to download
package org.http4k.lens

import org.http4k.core.Accept
import org.http4k.core.ContentType
import org.http4k.core.HttpMessage
import org.http4k.core.Parameters
import org.http4k.core.Uri
import org.http4k.core.Uri.Companion.of
import org.http4k.lens.ParamMeta.EnumParam
import org.http4k.lens.ParamMeta.StringParam
import java.util.Locale.getDefault

typealias HeaderLens = Lens

object Header : BiDiLensSpec("header", StringParam,
    LensGet { name, target -> target.headerValues(name).map { it ?: "" } },
    LensSet { name, values, target -> values.fold(target.removeHeader(name)) { m, next -> m.header(name, next) } }
) {
    val CONTENT_TYPE = map(
        {
            parseValueAndDirectives(it).let {
                ContentType(it.first, it.second
                    .filter { it.first.lowercase(getDefault()) in setOf("boundary", "charset", "media-type") }
                )
            }
        },
        ContentType::toHeaderValue
    ).optional("content-type")


    val LOCATION = map(::of, Uri::toString).required("location")

    val ACCEPT = map {
        parseValueAndDirectives(it).let {
            Accept(it.first.split(",").map { it.trim() }.map(::ContentType), it.second)
        }
    }.optional("Accept")

    val LINK = map(
        {
            Regex("""<(.+?)>; rel="(.+?)"""")
                .findAll(it)
                .map {
                    it.destructured.match.groupValues[2] to
                        of(it.destructured.match.groupValues[1])
                }.toMap()
        },
        {
            it.map { """<${it.value}>; rel="${it.key}"""" }.joinToString(", ")
        }
    )
        .defaulted("Link", emptyMap())

    val AUTHORIZATION_BASIC = basicCredentials().optional("Authorization")

    internal fun parseValueAndDirectives(it: String): Pair =
        with(it.split(";").mapNotNull { it.trim().takeIf(String::isNotEmpty) }) {
            first() to drop(1).map {
                with(it.split("=")) {
                    first() to if (size == 1) null else drop(1).joinToString("=")
                }
            }
        }
}

inline fun > Header.enum(caseSensitive: Boolean = true) = mapWithNewMeta(
    if (caseSensitive) StringBiDiMappings.enum() else StringBiDiMappings.caseInsensitiveEnum(),
    EnumParam(T::class)
)