jvmMain.kr.jadekim.jext.ktor.module.DefaultPluginsModule.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jext-ktor-jvm Show documentation

Jext Libraries

There is a newer version: 2.1.4

package kr.jadekim.jext.ktor.module

import io.ktor.http.*
import io.ktor.http.content.*
import io.ktor.server.application.*
import io.ktor.server.plugins.autohead.*
import io.ktor.server.plugins.cachingheaders.*
import io.ktor.server.plugins.callid.*
import io.ktor.server.plugins.doublereceive.*
import io.ktor.server.plugins.forwardedheaders.*
import io.ktor.util.*
import io.ktor.util.date.*

private val ATTRIBUTE_CACHE_OPTION: AttributeKey = AttributeKey("CachingHeader.max_age")

fun ApplicationCall.cacheOption(option: CacheControl? = null, expires: GMTDate? = null) {
    attributes.put(ATTRIBUTE_CACHE_OPTION, CachingOptions(option, expires))
}

object DefaultFeatureModule : KtorModuleFactory {

    class Configuration : KtorModuleConfiguration

    override fun create(config: Configuration): KtorModule = ktorModule {
        install(XForwardedHeaders) {
            forHeaders += "CF-Connecting-IP"
        }

        install(DoubleReceive)

        install(AutoHeadResponse)

        install(CachingHeaders) {
            options { call, _ ->
                call.attributes.getOrNull(ATTRIBUTE_CACHE_OPTION)
            }
        }

        install(CallId) {
            retrieveFromHeader("traceparent")
        }
    }

    override fun createDefaultConfiguration(): Configuration = Configuration()
}