org.scalatra.json.JValueResult.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scalatra-json-jakarta_3 Show documentation

JSON support for Scalatra

The newest version!

package org.scalatra
package json

import java.io.File

import org.json4s._

import scala.xml.NodeSeq

/**
 * Responsible for passing a JValue further in the render pipeline.
 */
trait JValueResult extends ScalatraBase { self: JsonSupport[?] =>

  implicit protected def jsonFormats: Formats

  override protected def renderPipeline: RenderPipeline = renderToJson orElse super.renderPipeline

  private[this] def isJValueResponse = format == "json" || format == "xml"

  private[this] def customSerializer = jsonFormats.customSerializer

  private[this] def renderToJson: RenderPipeline = {
    case JNothing =>
    case JNull => response.writer.write("null")
    case a: JValue => super.renderPipeline(a)
    case a: Any if isJValueResponse && customSerializer.isDefinedAt(a) =>
      customSerializer.lift(a) match {
        case Some(jv: JValue) => jv
        case None => super.renderPipeline(a)
      }
    case status: Int => super.renderPipeline(status)
    case bytes: Array[Byte] => super.renderPipeline(bytes)
    case is: java.io.InputStream => super.renderPipeline(is)
    case file: File => super.renderPipeline(file)
    case a: ActionResult => super.renderPipeline(a)
    case _: Unit | null =>
    case s: String => super.renderPipeline(s)
    case x: scala.xml.Node if format == "xml" =>
      contentType = formats("xml")
      response.writer.write(scala.xml.Utility.trim(x).toString())
    case x: NodeSeq if format == "xml" =>
      contentType = formats("xml")
      response.writer.write(x.toString)
    case x: NodeSeq =>
      response.writer.write(x.toString)
    case p: Product if isJValueResponse => Extraction.decompose(p)
    case p: TraversableOnce[?] if isJValueResponse => Extraction.decompose(p)
  }

}