cask.endpoints.ParamReader.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cask_3 Show documentation

cask

The newest version!

package cask.endpoints

import cask.router.ArgReader
import cask.model.{Cookie, Request}
import io.undertow.server.HttpServerExchange
import io.undertow.server.handlers.form.{FormData, FormParserFactory}

abstract class ParamReader[T] extends ArgReader[Unit, T, cask.model.Request]{
  def arity: Int
  def read(ctx: cask.model.Request, label: String, v: Unit): T
}
object ParamReader{
  class NilParam[T](f: (Request, String) => T) extends ParamReader[T]{
    def arity = 0
    def read(ctx: cask.model.Request, label: String, v: Unit): T = f(ctx, label)
  }
  implicit object HttpExchangeParam extends NilParam[HttpServerExchange]((ctx, label) => ctx.exchange)

  implicit object FormDataParam extends NilParam[FormData]((ctx, label) =>
    FormParserFactory.builder().build().createParser(ctx.exchange).parseBlocking()
  )

  implicit object RequestParam extends NilParam[Request]((ctx, label) => ctx)

  implicit object CookieParam extends NilParam[Cookie]((ctx, label) =>
    Cookie.fromUndertow(ctx.exchange.getRequestCookies().get(label))
  )

  implicit object QueryParams extends ParamReader[cask.model.QueryParams] {
    def arity: Int = 0

    override def unknownQueryParams = true

    def read(ctx: cask.model.Request, label: String, v: Unit) = {
      cask.model.QueryParams(ctx.queryParams)
    }
  }

  implicit object RemainingPathSegments extends ParamReader[cask.model.RemainingPathSegments] {
    def arity: Int = 0

    override def remainingPathSegments = true

    def read(ctx: cask.model.Request, label: String, v: Unit) = {
      cask.model.RemainingPathSegments(ctx.remainingPathSegments)
    }
  }
}