io.finch.endpoint.cookie.scala Maven / Gradle / Ivy

Go to download
package io.finch.endpoint

import cats.effect.Sync
import com.twitter.finagle.http.{Cookie => FinagleCookie}
import io.finch._

abstract private[finch] class Cookie[F[_], A](name: String)(implicit
    protected val F: Sync[F]
) extends Endpoint[F, A] {

  protected def missing(name: String): F[Output[A]]
  protected def present(value: FinagleCookie): F[Output[A]]

  def apply(input: Input): EndpointResult[F, A] = {
    val output = F.defer {
      input.request.cookies.get(name) match {
        case None        => missing(name)
        case Some(value) => present(value)
      }
    }

    EndpointResult.Matched(input, Trace.empty, output)
  }

  final override def toString: String = s"cookie($name)"
}

private[finch] object Cookie {

  trait Optional[F[_]] { _: Cookie[F, Option[FinagleCookie]] =>
    protected def missing(name: String): F[Output[Option[FinagleCookie]]] = F.pure(Output.None)
    protected def present(value: FinagleCookie): F[Output[Option[FinagleCookie]]] =
      F.pure(Output.payload(Some(value)))
  }

  trait Required[F[_]] { _: Cookie[F, FinagleCookie] =>
    protected def missing(name: String): F[Output[FinagleCookie]] =
      F.raiseError(Error.CookieNotPresent(name))
    protected def present(value: FinagleCookie): F[Output[FinagleCookie]] =
      F.pure(Output.payload(value))
  }
}