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

Go to download
package com.nrinaudo

import java.io._
import java.nio.charset.Charset
import scala.language.implicitConversions
import java.net.URI
import com.nrinaudo.fetch.ResponseEntity.EntityParser
import java.util.Locale
import com.nrinaudo.fetch.Request.HttpEngine

package object fetch {
  // - Package constants -----------------------------------------------------------------------------------------------
  // -------------------------------------------------------------------------------------------------------------------
  /** Charset used for request or response entities when none is specified. */
  val         DefaultCharset = Charset.forName("UTF-8")
  /** Size of the buffer used when processing streams. */
  private val BufferSize     = 4096



  // - IO helper methods -----------------------------------------------------------------------------------------------
  // -------------------------------------------------------------------------------------------------------------------
  /** Writes the content of the specified input stream to the specified output stream.
    * Note that this method does not wrap its arguments in buffered streams, nor will it close either stream.
    */
  def writeBytes(in: InputStream, out: OutputStream): Unit = {
    def loop(buffer: Array[Byte]): Unit =
      in.read(buffer, 0, buffer.length) match {
        case count if count > 0 =>
          out.write(buffer, 0, count)
          loop(buffer)
        case _ =>
      }
    loop(new Array[Byte](BufferSize))
  }

  /** Writes the content of the specified reader to the specified writer.
    * Note that this method does not wrap its arguments in buffered implementations, nor will it close either stream.
    */
  def writeChars(in: Reader, out: Writer) = {
    def loop(buffer: Array[Char]): Unit =
      in.read(buffer, 0, buffer.length) match {
        case count if count > 0 =>
          out.write(buffer, 0, count)
          loop(buffer)
        case _ =>
      }
    loop(new Array[Char](BufferSize))
  }



  // - Implicit conversions --------------------------------------------------------------------------------------------
  // -------------------------------------------------------------------------------------------------------------------
  implicit def urlToRequest(url: Url)(implicit engine: HttpEngine): Request[Response[ResponseEntity]] = Request(url)(engine)

  // Request entities.
  implicit def stringToEntity(str: String): RequestEntity     = RequestEntity(str)
  implicit def readerToEntity(reader: Reader): RequestEntity  = RequestEntity(reader)
  implicit def streamToEntity(in: InputStream): RequestEntity = RequestEntity(in)
  implicit def fileToEntity(file: File): RequestEntity        = RequestEntity(file)

  // Content negotiation headers.
  implicit def mediaTypeToConneg(mediaType: MediaType): Conneg[MediaType] = Conneg(mediaType)
  implicit def encodingToConneg(encoding: Encoding): Conneg[Encoding]     = Conneg(encoding)
  implicit def charsetToConneg(charset: Charset): Conneg[Charset]         = Conneg(charset)
  implicit def localeToConneg(locale: Locale): Conneg[Locale]             = Conneg(locale)

  // Response entities.
  implicit val TextEntityParser: EntityParser[String] = (entity: ResponseEntity) => {
    val writer = new StringWriter()
    entity.withReader(writeChars(_, writer))
    writer.toString
  }

  implicit val IntEntityParser: EntityParser[Int] = TextEntityParser.andThen {_.toInt}
}