jp.co.cyberagent.aeromock.server.http.JsonApiResponseWriter.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of aeromock-server_2.11 Show documentation
Aeromock server daemon.
There is a newer version: 0.2.5
package jp.co.cyberagent.aeromock.server.http

import jp.co.cyberagent.aeromock.AeromockLoadDataException
import jp.co.cyberagent.aeromock.config.Project
import jp.co.cyberagent.aeromock.core.el.VariableHelper
import jp.co.cyberagent.aeromock.data.CommonDataHelper
import jp.co.cyberagent.aeromock.helper.DeepTraversal._

import scalaz.{Failure, NonEmptyList, Success}

/**
 * Trait create response object for JSON API.
 * @author stormcat24
 */
trait JsonApiResponseWriter {

  /**
   * Returns response object.
   * @return response object
   */
  def write(): (AnyRef, Option[CustomResponse])

}

/**
 * [[jp.co.cyberagent.aeromock.server.http.JsonApiResponseWriter]] for root element of array object.
 * @param project [[Project]]
 * @param variableHelper [[VariableHelper]]
 * @param data sequential data
 */
class JsonApiArrayResponseWriter(project: Project, variableHelper: VariableHelper, data: Seq[Any]) extends JsonApiResponseWriter {

  // Don't merge common data files, because structur of root element is array.
  /**
   * @inheritdoc
   */
  override def write(): (AnyRef, Option[CustomResponse]) = (scanSeq(data)(variableHelper.variableConversion), None)
}

/**
 * [[jp.co.cyberagent.aeromock.server.http.JsonApiResponseWriter]] for root element of map object.
 * @param project [[Project]]
 * @param variableHelper [[VariableHelper]]
 * @param data map data
 */
class JsonApiMapResponseWriter(project: Project, variableHelper: VariableHelper, data: Map[Any, Any])
  extends JsonApiResponseWriter with ResponseDataSupport{

  /**
   * @inheritdoc
   */
  override def write(): (AnyRef, Option[CustomResponse]) = {

    val commonDataHelper = new CommonDataHelper(project._naming)
    val commonMergeMap = commonDataHelper.getMergedDataMap(project._ajax.root, project.ajaxScript)
    val mergedMap = commonDataHelper.mergeAdditional(commonMergeMap, data)

    val customResponse = createCustomResponse(project._naming, mergedMap) match {
      case Success(value) => value
      case Failure(e) => {
        val errors = e.list.map(_.getMessage)
        throw new AeromockLoadDataException(NonEmptyList(errors.head, errors.drop(1): _*))
      }
    }

    val reducedMap = mergedMap - project._naming.response
    (scanMap(reducedMap)(variableHelper.variableConversion), customResponse)
  }
}

/**
 * Factory of [[jp.co.cyberagent.aeromock.server.http.JsonApiResponseWriter]]
 * @author stormcat24
 */
object JsonApiResponseWriterFactory {

  /**
   * Create [[jp.co.cyberagent.aeromock.server.http.JsonApiResponseWriter]].
   * @param project [[Project]]
   * @param variableHelper [[VariableHelper]]
   * @param data Iterable object
   * @return [[jp.co.cyberagent.aeromock.server.http.JsonApiResponseWriter]]
   */
  def create(project: Project, variableHelper: VariableHelper, data: Iterable[_]): JsonApiResponseWriter = {
    data match {
      case m: Map[Any, Any] @unchecked => new JsonApiMapResponseWriter(project, variableHelper, m)
      case l: Seq[Any] @unchecked => new JsonApiArrayResponseWriter(project, variableHelper, l)
    }
  }

}