com.emarsys.escher.akka.http.EscherHttpRequest.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of escher-akka-http_2.12 Show documentation

escher-akka-http

The newest version!

package com.emarsys.escher.akka.http

import java.net.URI

import akka.http.scaladsl.model.headers.{Host, RawHeader}
import akka.http.scaladsl.model.{HttpHeader, HttpRequest}
import com.emarsys.escher.EscherRequest

import scala.jdk.CollectionConverters._

class EscherHttpRequest(var request: HttpRequest, body: String) extends EscherRequest {

  def addHeader(fieldName: String, fieldValue: String): Unit = {
    if (fieldName != "host") {
      request = request.addHeader(RawHeader(fieldName, fieldValue))
    } else {
      val parts = fieldValue.split(':')
      if (parts.length > 1) request = request.addHeader(Host(parts(0), parts(1).toInt))
      else request = request.addHeader(Host(parts(0)))
    }
  }

  def hasHeader(fieldName: String): Boolean = request.headers.exists((elem: HttpHeader) => elem.name == fieldName)

  def getRequestHeaders = request.headers.map(header => new EscherRequest.Header(header.name, header.value)).asJava

  def getBody: String = body

  def getHttpMethod: String = request.method.value

  def getURI: URI = new URI(request.getUri().toString)

  def getHttpRequest: HttpRequest = request

}