spice.http.server.DefaultErrorHandler.scala Maven / Gradle / Ivy

Go to download
package spice.http.server

import cats.effect.IO
import spice.http.{CacheControl, HttpExchange, HttpStatus}
import spice.http.content.Content
import spice.http.server.dsl.string2Content
import spice.net.ContentType

object DefaultErrorHandler extends ErrorHandler {
  lazy val lastModified: Long = System.currentTimeMillis()

  def html(status: HttpStatus): Content = s"""
    
      Error ${status.code}
    
    
      ${status.code} - ${status.message}
    
  """.withContentType(ContentType.`text/html`).withLastModified(lastModified)

  override def handle(exchange: HttpExchange, t: Option[Throwable]): IO[HttpExchange] = {
    exchange.modify { response => IO {
      val status = if (response.status.isError) {
        response.status
      } else {
        HttpStatus.InternalServerError
      }
      response
        .withContent(html(status))
        .withHeader(CacheControl(CacheControl.NoCache))
    }}
  }
}