com.nrinaudo.fetch.Response.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fetch_2.11 Show documentation
fetch
The newest version!
package com.nrinaudo.fetch

import java.util.Date
import Headers._
import java.io.{FilterInputStream, InputStream}

object Response {
  // It appears that some stream implementation (decompressing ones, for instance, such as GZipInputStream) stop reading
  // before read returns -1 - which seems to prevent keeping connections alive.
  // The following is a nasty workaround, but it works.
  private def closing(stream: InputStream): InputStream = new FilterInputStream(stream) {
    override def close(): Unit = {
      in.read()
      super.close()
    }
  }

  def fromStream(status: Status, headers: Headers, stream: InputStream): Response[ResponseEntity] =
    Response(status, headers,
      new ResponseEntity(headers.getOpt[MediaType]("Content-Type"),
        headers.getOpt[Seq[Encoding]]("Content-Encoding").fold(stream) { values =>
          values.foldRight(closing(stream)) { _ decode _ }
        }))
}

/** Represents an HTTP response. */
case class Response[A](status: Status, headers: Headers, body: A) {
  // - Monadic operations ----------------------------------------------------------------------------------------------
  // -------------------------------------------------------------------------------------------------------------------
  def map[B](f: A => B): Response[B] = copy(body = f(body))
  def flatMap[B](f: A => Response[B]): Response[B] = copy(body = f(body).body)
  def foreach[U](f: A => U): U = f(body)



  // - Header helpers ----------------------------------------------------------------------------------------------
  // -------------------------------------------------------------------------------------------------------------------
  def date: Option[Date] = headers.getOpt[Date]("Date")
  def contentEncoding: Option[Seq[Encoding]] = headers.getOpt[Seq[Encoding]]("Content-Encoding")
  def contentLanguage: Option[Seq[Language]] = headers.getOpt[Seq[Language]]("Content-Language")
  def contentType: Option[MediaType] = headers.getOpt[MediaType]("Content-Type")
  def lastModified: Option[Date] = headers.getOpt[Date]("Last-Modified")
  def expires: Option[Date] = headers.getOpt[Date]("Expires")
  def etag: Option[ETag] = headers.getOpt[ETag]("ETag")
  def server: Option[String] = headers.getOpt[String]("Server")
  def allow: Option[Seq[Method]] = headers.getOpt[Seq[Method]]("Allow")
  def age: Option[Int] = headers.getOpt[Int]("Age")
  def wwwAuthenticate: Option[String] = headers.getOpt[String]("WWW-Authenticate")

  // TODO: implement Accept-Ranges
  // TODO: implement Cache-Control
  // TODO: implement Connection
  // TODO: implement Content-Length
  // TODO: implement Content-Location
  // TODO: implement Content-Disposition
  // TODO: implement Content-Range
  // TODO: implement Location
  // TODO: implement Pragma
  // TODO: implement Retry-After
  // TODO: implement Set-Cookie
  // TODO: implement Vary
}