com.websudos.util.http.package.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of util-http_2.10 Show documentation

util-http

The newest version!

package com.websudos.util

import java.net.{URL, URLDecoder, URLEncoder}
import scala.collection.breakOut
import org.jboss.netty.handler.codec.http.HttpResponse

import com.twitter.finagle.http.RequestBuilder
import com.twitter.io.Charsets.Utf8


package object http extends HttpExtractor {

  implicit class RichResponseBuilder[X, Y](val builder: RequestBuilder[X, Y]) extends AnyVal {
    def asJson(): RequestBuilder[X, Y] = {
      builder.setHeader("Content-Type", "application/json")
    }
  }

  /**
   * Implicit value class used to simplify extracting responses from a Netty HTTP response.
   * @param resp The Http response to augment.
   */
  implicit class RichHttpResponse(val resp: HttpResponse) extends AnyVal {

    def buffer: Array[Byte] = {
      val channelBuffer = resp.getContent

      val length = channelBuffer.readableBytes()
      val bytes = new Array[Byte](length)
      channelBuffer.getBytes(channelBuffer.readerIndex(), bytes, 0, length)
      bytes
    }

    def responseBody: String = {
      new String(buffer)
    }
  }

  implicit class URLBuilder(val url: String) extends AnyVal {

    final def /(path: String): String = if (url.last != '/') url + "/" + path else url + path

    def < 0)
        url + "?" + param._1 + "=" + param._2
      else if (url.length > 0) {
        url + "&" + param._1 + "=" + param._2
      } else {
        url + param._1 + "=" + param._2
      }

    final def < { "&" + param._1 + "=" + param._2 })(breakOut)

    final def asUri: URL = new URL(url)

    final def utf8: String = URLEncoder.encode(url, Utf8.displayName())

    final def fromUtf8: String = URLDecoder.decode(url, Utf8.displayName())

  }

  implicit class RichURL(val url: URL) extends AnyVal {
    final def /(path: String): URL = new URL(url.toString + "/" + path)

    final def <