spice.http.server.handler.ValidatorHttpHandler.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spice-server_3 Show documentation

spice-server

There is a newer version: 0.6.1

package spice.http.server.handler

import cats.effect.IO
import scribe.mdc.MDC
import spice.http.content.Content
import spice.http.{HttpExchange, HttpStatus}
import spice.http.server.validation.ValidationResult.{Continue, Redirect}
import spice.http.server.validation.{ValidationResult, Validator}

class ValidatorHttpHandler(validators: List[Validator]) extends HttpHandler {
  override def handle(exchange: HttpExchange)(implicit mdc: MDC): IO[HttpExchange] = {
    ValidatorHttpHandler.validate(exchange, validators).map(_.exchange)
  }
}

object ValidatorHttpHandler {
  def validate(exchange: HttpExchange,
               validators: List[Validator]): IO[ValidationResult] = if (validators.isEmpty) {
    IO.pure(ValidationResult.Continue(exchange))
  } else {
    val validator = validators.head
    validator.validate(exchange).flatMap {
      case ValidationResult.Continue(c) => validate(c, validators.tail)
      case v: ValidationResult.Redirect =>
        HttpHandler.redirect(v.exchange, v.location).map(v.copy(_))
      case v: ValidationResult.Error => exchange.modify { response =>
        IO(response.withStatus(HttpStatus(v.status, v.message)).withContent(Content.empty))
      }.map(_.finish()).map { exchange =>
        v.copy(exchange)
      }
    }
  }
}