package com.ustadmobile.testservercontroller.util

import com.ustadmobile.ihttp.ext.getCaseInsensitiveOrNull
import com.ustadmobile.ihttp.ext.requirePostfix
import com.ustadmobile.ihttp.headers.directives.directivesToMap
import io.ktor.util.StringValues

/**
 * Gets the protocol that the client used, using headers to adjust as needed when a reverse proxy
 * was used.
 *
 * Uses the Forwarded Header directive first, then
 * X-Forwarded-Proto, and finally, origin.scheme.
 *
 * @receiver IStringValues representing request headers as received by the server
 */
fun StringValues.clientProtocol(): String? {
    return get("Forwarded")?.let {
        directivesToMap(it)
    }?.getCaseInsensitiveOrNull("proto") ?: get("X-Forwarded-Proto")
}

/**
 * Gets the host that the client used, using headers to adjust as needed when a reverse proxy was
 * used. e.g. servername.com (if using default port for protocol) or servername.com:8087
 *
 * Uses the Forwarded header directive first, then X-Forwarded-Host, then the Host header
 *
 * @receiver IStringValues representing request headers as received by the server
 */
fun StringValues.clientHost(): String? {
    return get("Forwarded")?.let {
        directivesToMap(it)
    }?.getCaseInsensitiveOrNull("host") ?: get("X-Forwarded-Host") ?: get("Host")
}

/**
 * Get the protocol and host being used e.g. http://ip.addr:8087/ https://server.tld/ etc. This
 * will use the Forwarded header when available.
 */
fun StringValues.clientProtocolAndHost(
    defaultProtocol: String = "http"
): String {
    return "${clientProtocol() ?: defaultProtocol}://${clientHost()}".requirePostfix("/")
}