org.elasticmq.rest.sqs.directives.RespondDirectives.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of elasticmq-rest-sqs_2.11 Show documentation

elasticmq-rest-sqs

The newest version!

package org.elasticmq.rest.sqs.directives

import akka.http.scaladsl.server.{Route, Directives, RequestContext}

import scala.xml.{Null, UnprefixedAttribute, Elem}
import org.elasticmq.rest.sqs.Constants._

trait RespondDirectives {
  this: Directives =>

  def respondWith(elem: Elem): Route = namespace { ns => (ctx: RequestContext) =>
    {
      val result = elem % ns
      ctx.complete(result.toString())
    }
  }

  def respondWith(statusCode: Int)(elem: Elem): Route = namespace { ns => (ctx: RequestContext) =>
    {
      val result = elem % ns
      ctx.complete((statusCode, result.toString()))
    }
  }

  private def namespace(route: UnprefixedAttribute => Route): Route =
    parameter("Version" ?) { versionOpt =>
      val version = versionOpt match {
        case Some(v) if !v.isEmpty => v
        case _                     => SqsDefaultVersion
      }

      route(new UnprefixedAttribute("xmlns", "http://queue.amazonaws.com/doc/%s/".format(version), Null))
    }
}