org.elasticmq.rest.sqs.model.RequestPayload.scala Maven / Gradle / Ivy

Go to download
package org.elasticmq.rest.sqs.model

import org.apache.pekko.http.scaladsl.server.Directive0
import org.apache.pekko.http.scaladsl.server.Directives._
import org.elasticmq.rest.sqs.{Action, FlatParamsReader, SQSException}
import spray.json.{JsObject, RootJsonFormat}

sealed trait RequestPayload {
  def action(requiredAction: Action.Value): Directive0 = if (requiredAction.toString == action) pass else reject

  def xRayTracingHeader: Option[String]
  def action: String

  def as[A: RootJsonFormat: FlatParamsReader]: A = this match {
    case json: RequestPayload.JsonParams   => json.readAs[A]
    case query: RequestPayload.QueryParams => query.readAs[A]
  }
}

object RequestPayload {
  final case class QueryParams(params: Map[String, String], xRayTracingHeader: Option[String] = None)
      extends RequestPayload {
    def readAs[A](implicit fpr: FlatParamsReader[A]) = fpr.read(params)
    override def action: String = params.getOrElse("Action", throw SQSException.missingAction)
  }

  final case class JsonParams(params: JsObject, action: String, xRayTracingHeader: Option[String] = None)
      extends RequestPayload {
    def readAs[A: RootJsonFormat] = params.convertTo[A]
  }
}