ba.sake.sharaf.handlers.RoutesHandler.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sharaf_3 Show documentation

Sharaf http library

There is a newer version: 0.8.0

package ba.sake.sharaf.handlers

import io.undertow.server.HttpHandler
import io.undertow.server.HttpServerExchange

import ba.sake.sharaf.*
import ba.sake.sharaf.routing.*

final class RoutesHandler private (routes: Routes, nextHandler: Option[HttpHandler]) extends HttpHandler {

  override def handleRequest(exchange: HttpServerExchange): Unit = {
    exchange.startBlocking()
    if (exchange.isInIoThread) {
      exchange.dispatch(this)
    } else {

      val request = Request.create(exchange)

      given Request = request

      val reqParams = fillReqParams(exchange)

      val resOpt = routes.definition.lift(reqParams)

      resOpt match {
        case Some(res) => ResponseWritable.writeResponse(res, exchange)
        case None =>
          nextHandler match
            case Some(next) => next.handleRequest(exchange)
            case None       =>
              // will be catched by ExceptionHandler
              throw exceptions.NotFoundException("route")
      }
    }
  }

  private def fillReqParams(exchange: HttpServerExchange): RequestParams = {
    val relPath =
      if exchange.getRelativePath.startsWith("/") then exchange.getRelativePath.drop(1)
      else exchange.getRelativePath
    val pathSegments = relPath.split("/")

    val path =
      if pathSegments.size == 1 && pathSegments.head == ""
      then Path()
      else Path(pathSegments*)

    (exchange.getRequestMethod, path)
  }

}

object RoutesHandler:
  def apply(routes: Routes): RoutesHandler =
    new RoutesHandler(routes, None)

  def apply(routes: Routes, nextHandler: HttpHandler): RoutesHandler =
    new RoutesHandler(routes, Some(nextHandler))