spice.http.client.JSHttpClientInstance.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spice-client_sjs1_2.13 Show documentation
spice-client
The newest version!
package spice.http.client

import cats.effect.IO
import moduload.Moduload
import spice.ajax.{AjaxAction, AjaxRequest}
import spice.http.content.{Content, StringContent}
import spice.http.{Headers, HttpRequest, HttpResponse, HttpStatus, WebSocket}
import spice.net.{ContentType, URL}

import scala.concurrent.duration.FiniteDuration
import scala.util.{Failure, Success, Try}

class JSHttpClientInstance(client: HttpClient) extends HttpClientInstance {
  assert(client.proxy.isEmpty, "Proxy is not supported in JSHttpClientImplementation")

  private val HeaderRegex = """(.+)[:](.+)""".r

  override def send(request: HttpRequest): IO[Try[HttpResponse]] = {
    val manager = client.connectionPool.asInstanceOf[JSConnectionPool].manager
    val contentString = request.content match {
      case Some(content) => content.asString.map(Some.apply)
      case None => IO.pure(None)
    }
    contentString.flatMap { data =>
      val ajaxRequest = new AjaxRequest(
        url = request.url,
        data = data,
        timeout = 0,
        headers = request.headers.map.flatMap(t => t._2.map(value => t._1 -> value)),
        withCredentials = true,
        responseType = ""
      )
      val action = new AjaxAction(ajaxRequest)
      manager.enqueue(action).map {
        case Failure(err) => Failure(err)
        case Success(xmlHttpRequest) =>
          val headers: Map[String, List[String]] = xmlHttpRequest.getAllResponseHeaders().split('\n').map(_.trim).map {
            case HeaderRegex(key, value) => key.trim -> value.trim
            case s => throw new RuntimeException(s"Invalid Header: [$s]")
          }.groupBy(_._1).map {
            case (key, array) => key -> array.toList.map(_._2)
          }
          val content = xmlHttpRequest.responseType match {
            case null => None
            case _ =>
              val `type` = if (xmlHttpRequest.responseType == "") ContentType.`text/plain` else ContentType.parse(xmlHttpRequest.responseType)
              Some(Content.string(xmlHttpRequest.responseText, `type`))
          }
          Success(HttpResponse(
            status = HttpStatus(xmlHttpRequest.status, xmlHttpRequest.statusText),
            headers = Headers(headers),
            content = content
          ))
      }
    }
  }

  override def webSocket(url: URL): WebSocket = new JSWebSocketClient(url)

  override def dispose(): IO[Unit] = IO.unit
}