package httpz

import scalaz.{Maybe, Endo}

final case class Request(
  url: String,
  method: String = "GET",
  body: Option[Array[Byte]] = None,
  params: Map[String, String] = Map.empty,
  headers: Map[String, String] = Map.empty,
  basicAuth: Option[(String, String)] = None
) {

  def bodyString(string: String): Request =
    copy(body = Some(string.getBytes("UTF-8")))

  def addParam(k: String, v: String): Request =
    copy(params = this.params + (k -> v))

  def addParams(p: (String, String) *): Request =
    copy(params = this.params ++ p.toMap)

  def addParamOpt(k: String, v: Option[String]): Request =
    v match {
      case Some(value) =>
        copy(params = this.params + (k -> value))
      case None =>
        this
    }

  def addParamsOpt(p: (String, Option[String]) *): Request =
    copy(params = this.params ++ p.collect{case (k, Some(v)) => k -> v})

  def addParamsMaybe(p: (String, Maybe[String]) *): Request =
    copy(params = this.params ++ p.collect{case (k, Maybe.Just(v)) => k -> v})

  def addParamMaybe(k: String, v: Maybe[String]): Request =
    v match {
      case Maybe.Just(value) =>
        copy(params = this.params + (k -> value))
      case Maybe.Empty() =>
        this
    }

  def addHeader(k: String, v: String): Request =
    copy(headers = this.headers + (k -> v))

  def addHeaders(p: (String, String) *): Request =
    copy(headers = this.headers ++ p.toMap)

  /** Basic authentication */
  def auth(user: String, pass: String): Request =
    copy(basicAuth = Some(user -> pass))

  /** @see [[http://tools.ietf.org/html/rfc6750]] */
  def bearer(token: String): Request =
    copy(headers = this.headers + ("Authorization" -> ("Bearer " + token)))
}

object Request {

  def bearer(token: String): Config =
    Endo(_.bearer(token))

  def header(k: String, v: String): Config =
    Endo(_.addHeader(k, v))

  def headers(p: (String, String) *): Config =
    Endo(_.addHeaders(p: _*))

  def param(k: String, v: String): Config =
    Endo(_.addParam(k, v))

  def params(p: (String, String) *): Config =
    Endo(_.addParams(p: _*))

  def paramOpt(k: String, v: Option[String]): Config =
    Endo(_.addParamOpt(k, v))

  def paramsOpt(p: (String, Option[String])*): Config =
    Endo(_.addParamsOpt(p: _*))

  def paramMaybe(k: String, v: Maybe[String]): Config =
    Endo(_.addParamMaybe(k, v))

  def paramsMaybe(p: (String, Maybe[String])*): Config =
    Endo(_.addParamsMaybe(p: _*))

  def auth(user: String, pass: String): Config =
    Endo(_.auth(user, pass))

  def method(methodName: String): Config =
    Endo(_.copy(method = methodName))

  def body(bytes: Array[Byte]): Config =
    Endo(_.copy(body = Option(bytes)))

  def bodyString(string: String): Config =
    Endo(_.bodyString(string))
}