harness.http.server.Route.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of harness-http-server_3 Show documentation
Miscellaneous libraries/utilities for Scala.
There is a newer version: 5.1.3
Show newest version
package harness.http.server

import cats.syntax.option.*
import harness.core.*
import harness.web.*
import harness.zio.*
import zio.*
import zio.json.*
import zio.json.ast.Json

trait Route[-R] { self =>

  final def /:(met: HttpMethod): Route[R] =
    (method, path) =>
      if (method == met) self(method, path)
      else ZIO.succeed(HttpResponse.NotFound)

  final def /:(str: String): Route[R] =
    (method, path) =>
      path match {
        case head :: tail if head == str => self(method, tail)
        case _                           => ZIO.succeed(HttpResponse.NotFound)
      }

  final def widen[R2 <: R]: Route[R2] = self

  def apply(
      method: HttpMethod,
      path: List[String],
  ): SHRIO[BuiltInRequestEnv & R, HttpResponse]

}
object Route {

  def oneOf[R](routes: Route[R]*): Route[R] = { (method, path) =>
    def rec(routes: List[Route[R]]): SHRIO[BuiltInRequestEnv & R, HttpResponse] =
      routes match {
        case head :: tail =>
          head(method, path).flatMap {
            case HttpResponse.NotFound => rec(tail)
            case response              => ZIO.succeed(response)
          }
        case Nil => ZIO.succeed(HttpResponse.NotFound)
      }

    rec(routes.toList)
  }

  private def pageHtmlResponse(harnessUiConfig: HConfig): HttpResponse =
    HttpResponse(
      s"""
        |
        |
        |
        |    
        |    Title
        |    
        |    
        |    
        |
        |
        |
        |
        |
        |""".stripMargin,
    )

  private def getFile(config: ServerConfig, after: List[String]): SHRIO[Scope, HttpResponse] = {
    val path = s"${config.resDir}${after.map(p => s"/$p").mkString}"

    if (config.useJarResource)
      HttpResponse.jarResourceOrNotFound(path)
    else
      Path(path).flatMap(HttpResponse.fileOrNotFound(_))
  }

  /**
    * {res} : resource dir, can be set via `--res-dir=VALUE`
    *
    * Expects things to be set up in the following format:
    *  - /api/{*}          ->  provided api routes
    *  - /api/health-check ->  runs health-check
    *  - /page/{*}         ->  {res}/index.html
    *  - /res/favicon.ico  ->  {res}/favicon.ico
    *  - /res/js/{*}       ->  {res}/js/{*}
    *  - /res/css/{*}      ->  {res}/css/{*}
    */
  def stdRoot[A <: HasStdClientConfig: JsonEncoder, R](harnessUiConfig: A)(apis: Route[R]*): URIO[ServerConfig, Route[R]] =
    ZIO.serviceWith[ServerConfig] { config =>
      val pageHtml = pageHtmlResponse(HConfig.unsafeFromEncodable(harnessUiConfig))

      Route.oneOf(
        "api" /: Route.oneOf(apis*),
        (HttpMethod.GET / "api" / "health-check").implement { _ =>
          for {
            _ <- Logger.log.info("health-check")
            _ <- HttpRequest.query.logAll(Logger.LogLevel.Info)
            _ <- HttpRequest.header.logAll(Logger.LogLevel.Info)
            _ <- HttpRequest.cookie.logAll(Logger.LogLevel.Info)
          } yield HttpResponse.fromHttpCode.Ok
        },
        (HttpMethod.GET / "page" / RouteMatcher.**).implement { _ =>
          ZIO.succeed(pageHtml)
        },
        HttpMethod.GET /: "res" /: Route.oneOf(
          "favicon.ico".implement { _ =>
            getFile(config, "favicon.ico" :: Nil)
          },
          ("js" / RouteMatcher.**).implement { routes =>
            getFile(config, "js" :: routes)
          },
        ),
        HttpMethod.GET.implement { _ => ZIO.succeed(HttpResponse.redirect("/page")) },
      )
    }

}