SecureKit.scala

package unfilteredx.kit

import unfiltered.request._
import unfiltered.response._

object Secure {
  def secure[A,B](intent: unfiltered.Cycle.Intent[A,B]) = {
    case req@ForwardedProto(proto) if (intent.isDefinedAt(req)) => intent(wrap(req, "https" == proto.toLowerCase))
    case req if (intent.isDefinedAt(req)) => intent(req)
  }
  
  private def wrap[A](req: HttpRequest[A], secure: Boolean) = new DelegatingRequest(req) {
    override def isSecure = secure
  }
}

object ForwardedProto extends StringHeader("Forwarded-Proto")