com.wordnik.swagger.client.SwaggerConfig.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of swagger-async-httpclient_2.9.1 Show documentation
swagger-async-httpclient
The newest version!
package com.wordnik.swagger.client

import akka.util.Duration
import akka.util.duration._
import java.io.StringWriter
import org.json4s.jackson.JsonMethods
import SwaggerConfig.DataFormat
import org.json4s._
import org.json4s.Xml._
import io.Codec

object SwaggerConfig {
  sealed trait DataFormat {

    def name: String
    def contentType: String

    def serialize[T](obj: T): String
    def deserialize[T:Manifest](json: String): T
  }
  object DataFormat {
    object Json {
      def apply(fmts: Formats) = new Json(fmts)
    }
    class Json(fmts: Formats) extends DataFormat {
      implicit protected val jsonFormats: Formats = fmts
      val contentType: String = "application/json;charset=utf-8"
      val name: String = "json"
      def deserialize[T: Manifest](json: String): T = JsonMethods.parse(json, fmts.wantsBigDecimal).extract[T]
      def serialize[T](obj: T): String = JsonMethods.compact(JsonMethods.render(Extraction.decompose(obj)))
    }

    object XML {
      def apply(fmts: Formats) = new XML(fmts)
    }
    class XML(fmts: Formats) extends DataFormat {
      implicit protected val jsonFormats: Formats = fmts
      val contentType: String = "application/xml"

      val name: String = "xml"

      def deserialize[T: Manifest](json: String): T = {
        val JObject(JField(_, jv) :: Nil) = toJson(scala.xml.XML.loadString(json))
        jv.extract[T]
      }

      protected lazy val xmlRootNode = 

      def serialize[T](obj: T): String = {
        val json = Extraction.decompose(obj)
        val sw = new StringWriter()
        scala.xml.XML.write(sw, xmlRootNode.copy(child = toXml(json)), Codec.UTF8.name, true, null)
        sw.toString
      }
    }
  }

  private case class DefaultSwaggerConfig(
     locator: ServiceLocator,
     override val userAgent: String = RestClient.DefaultUserAgent,
     override val dataFormat: SwaggerConfig.DataFormat = DataFormat.Json(DefaultFormats),
     override val idleTimeout: Duration = 5.minutes,
     override val connectTimeout: Duration = 5.seconds,
     override val maxMessageSize: Int = 8912,
     override val enableCompression: Boolean = true,
     override val followRedirects: Boolean = true,
     override val identity: String = "0",
     override val name: String = "no-name") extends SwaggerConfig

  def forUrl(
      baseUrl: String,
      userAgent: String = RestClient.DefaultUserAgent,
      dataFormat: SwaggerConfig.DataFormat = DataFormat.Json(DefaultFormats),
      idleTimeout: Duration = 5.minutes,
      connectTimeout: Duration = 5.seconds,
      maxMessageSize: Int = 8912,
      enableCompression: Boolean = true,
      followRedirects: Boolean = true,
      identity: String = "0"): SwaggerConfig =
    new DefaultSwaggerConfig(BaseUrl(baseUrl), userAgent, dataFormat, idleTimeout, connectTimeout, maxMessageSize, enableCompression, followRedirects, identity)

  def forLocator(
      locator: ServiceLocator,
      name: String,
      userAgent: String = RestClient.DefaultUserAgent,
      dataFormat: SwaggerConfig.DataFormat = DataFormat.Json(DefaultFormats),
      idleTimeout: Duration = 5.minutes,
      connectTimeout: Duration = 5.seconds,
      maxMessageSize: Int = 8912,
      enableCompression: Boolean = true,
      followRedirects: Boolean = true,
      identity: String = "0"): SwaggerConfig =
    new DefaultSwaggerConfig(locator, userAgent, dataFormat, idleTimeout, connectTimeout, maxMessageSize, enableCompression, followRedirects, identity, name)
}

trait SwaggerConfig {
  def locator: ServiceLocator
  def userAgent: String = RestClient.DefaultUserAgent
  def dataFormat: SwaggerConfig.DataFormat = DataFormat.Json(DefaultFormats)
  def idleTimeout: Duration = 5.minutes
  def connectTimeout: Duration = 5.seconds
  def maxMessageSize: Int = 8912
  def enableCompression: Boolean = true
  def followRedirects: Boolean = true
  def identity: String = "0"
  def name: String = "no-name"
}