com.lightningkite.lightningserver.http.Http.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of server-core Show documentation
A set of tools to fill in/replace what Ktor is lacking in.
The newest version!
package com.lightningkite.lightningserver.http

import com.lightningkite.lightningserver.HtmlDefaults
import com.lightningkite.lightningserver.exceptions.HttpStatusException
import com.lightningkite.lightningserver.exceptions.report
import com.lightningkite.lightningserver.metrics.Metrics
import com.lightningkite.lightningserver.settings.generalSettings
import com.lightningkite.lightningserver.tasks.Tasks
import com.lightningkite.lightningserver.utils.MutableMapWithChangeHandler
import org.slf4j.LoggerFactory

object Http {
    init {
        Metrics
    }

    var fixEndingSlash: Boolean = true

    private val logger = LoggerFactory.getLogger("com.lightningkite.lightningserver.http.Http")

    val endpoints: MutableMap HttpResponse> =
        MutableMapWithChangeHandler HttpResponse> {
            _matcher = null
        }
    private var _matcher: HttpEndpointMatcher? = null
    val matcher: HttpEndpointMatcher
        get() {
            return _matcher ?: run {
                val created = HttpEndpointMatcher(endpoints.keys.asSequence())
                _matcher = created
                created
            }
        }

    var exception: suspend (HttpRequest, Exception) -> HttpResponse =
        { request, exception ->
            if (exception is HttpStatusException) {
                if (generalSettings().debug) {
                    println(exception.toLSError())
                    logger.warn(exception.toLSError().toString())
                }
                exception.toResponse(request)
            } else {
                exception.report(request)
                HttpResponse(status = HttpStatus.InternalServerError)
            }
        }
    var notFound: suspend (HttpEndpoint, HttpRequest) -> HttpResponse = { path, request ->
        HttpResponse.html(
            HttpStatus.NotFound, content = HtmlDefaults.basePage(
                """
            Not Found
            Sorry, the page you're looking for isn't here.
            ${
                    if (generalSettings().debug) {
                        "Your path is $path."
                    } else ""
                }
        """.trimIndent()
            )
        )
    }

    val onRequest = ArrayList Unit>()

    suspend fun execute(request: HttpRequest): HttpResponse {
        return endpoints[request.endpoint]?.let { handler ->
            try {
                Metrics.handlerPerformance(request.endpoint) {
                    onRequest.forEach { it(request) }
                    handler(request)
                }
            } catch (e: Exception) {
                exception(request, e)
            }
        } ?: notFound(request.endpoint, request).also {
            if (generalSettings().debug) {
                logger.warn("${request.endpoint} not found!")
            }
        }
    }
}

suspend fun HttpEndpoint.test(
    parts: Map = mapOf(),
    wildcard: String? = null,
    queryParameters: List> = listOf(),
    headers: HttpHeaders = HttpHeaders.EMPTY,
    body: HttpContent? = null,
    domain: String = generalSettings().publicUrl.substringAfter("://").substringBefore("/"),
    protocol: String = generalSettings().publicUrl.substringBefore("://"),
    sourceIp: String = "0.0.0.0"
): HttpResponse {
    Tasks.onSettingsReady()
    Tasks.onEngineReady()
    val req = HttpRequest(
        endpoint = this,
        parts = parts,
        wildcard = wildcard,
        queryParameters = queryParameters,
        headers = headers,
        body = body,
        domain = domain,
        protocol = protocol,
        sourceIp = sourceIp,
    )
    return try {
        Http.execute(req)
    } catch (e: HttpStatusException) {
        e.toResponse(req)
    }
}