harness.http.server.Cookie.scala Maven / Gradle / Ivy

Go to download
package harness.http.server

import cats.syntax.option.*
import harness.core.*
import zio.json.*

final case class Cookie private (
    private val _name: String,
    private val _value: String,
    private val _secure: Boolean,
    private val _maxAge: Option[Int],
    private val _path: Option[String],
    private val _sameSite: Option[Cookie.SameSite],
) { self =>

  def secure(s: Boolean): Cookie = self.copy(_secure = s)
  inline def secure: Cookie = self.secure(true)

  def maxAge(a: Option[Int]): Cookie = self.copy(_maxAge = a)
  inline def maxAge(a: Int): Cookie = self.maxAge(a.some)

  def path(p: Option[String]): Cookie = self.copy(_path = p)
  inline def path(p: String): Cookie = self.path(p.some)
  inline def rootPath: Cookie = self.path("/")

  def sameSite(ss: Option[Cookie.SameSite]): Cookie = self.copy(_sameSite = ss)
  inline def sameSite(ss: Cookie.SameSite): Cookie = self.sameSite(ss.some)

  // TODO (KR) : Make sure things are encoded properly
  def cookieString: String =
    List[Option[String]](
      s"${_name}=${_value}".some,
      Option.when(_secure)("Secure"),
      _maxAge.map { a => s"Max-Age=$a" },
      _path.map { p => s"Path=$p" },
      _sameSite.map { ss => s"SameSite=$ss" },
    ).flatten.mkString("; ")

  def isUnset: Boolean = _maxAge.contains(0) && _value.isEmpty

  def name: String = _name
  def value: String = _value

}
object Cookie {

  def apply[V](name: String, value: V)(implicit encoder: StringEncoder[V]): Cookie =
    new Cookie(
      _name = name,
      _value = encoder.encode(value),
      _secure = false,
      _maxAge = None,
      _path = None,
      _sameSite = None,
    )

  def json[V](name: String, value: V)(implicit encoder: JsonEncoder[V]): Cookie =
    new Cookie(
      _name = name,
      _value = encoder.encodeJson(value, None).toString,
      _secure = false,
      _maxAge = None,
      _path = None,
      _sameSite = None,
    )

  def unset(name: String): Cookie =
    new Cookie(
      _name = name,
      _value = "",
      _secure = false,
      _maxAge = 0.some,
      _path = None,
      _sameSite = None,
    )

  enum SameSite { case Lax, Strict, None }

}