spice.http.cookie.Cookie.scala Maven / Gradle / Ivy

Go to download

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

Core functionality leveraged and shared by most other sub-projects of YouI.

There is a newer version: 0.7.2

package spice.http.cookie

import spice.http.DateHeaderKey

import scala.collection.mutable

sealed trait Cookie {
  def name: String
  def value: String

  def http: String
}

object Cookie {
  case class Request(name: String, value: String) extends Cookie {
    override def http: String = s"$name=$value"
  }

  case class Response(name: String,
                            value: String,
                            expires: Option[Long] = None,
                            maxAge: Option[Long] = None,
                            domain: Option[String] = None,
                            path: Option[String] = None,
                            secure: Boolean = false,
                            httpOnly: Boolean = false,
                            sameSite: SameSite = SameSite.Normal) extends Cookie {
    override def http: String = {
      val b = new mutable.StringBuilder
      b.append(s"$name=$value")
      expires.foreach(l => b.append(s"; Expires=${DateHeaderKey.format(l)}"))
      maxAge.foreach(l => b.append(s"; Max-Age=$l"))
      domain.foreach(s => b.append(s"; Domain=$s"))
      path.foreach(s => b.append(s"; Path=$s"))
      if (secure) b.append("; Secure")
      if (httpOnly) b.append("; HttpOnly")
      sameSite match {
        case SameSite.Normal => // Nothing to set
        case SameSite.Lax => b.append("; SameSite=lax")
        case SameSite.Strict => b.append("; SameSite=strict")
      }
      b.toString
    }

    override def hashCode(): Int = name.hashCode

    override def equals(obj: Any): Boolean = obj match {
      case that: Cookie => this.name == that.name
      case _ => false
    }
  }
}