korolev.web.Response.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of korolev-web_3 Show documentation

Collection of data classes for Web Standards support

There is a newer version: 1.16.0-M5

/*
 * Copyright 2017-2020 Aleksey Fomkin
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package korolev.web

import korolev.web.Response.Status

import scala.annotation.switch

final case class Response[Body](
    status: Status,
    body: Body,
    headers: Seq[(String, String)],
    contentLength: Option[Long]
) {
  def header(name: String): Option[String] =
    headers.collectFirst {
      case (k, v) if k.equalsIgnoreCase(name) => v
    }
}

object Response {

  final case class Status(code: Int, phrase: String) {
    val codeAsString: String = code.toString
  }

  object Status {
    val Ok: Status = new Status(200, "OK")
    val Created = new Status(201, "Created")
    val ResetContent = new Status(205, "Reset Content")
    val NotFound: Status = new Status(404, "Not Found")
    val BadRequest: Status = new Status(400, "Bad Request")
    val Gone: Status = new Status(410, "Gone")
    val SwitchingProtocols: Status = new Status(101, "Switching Protocols")
    val InternalServerError: Status = new Status(500, "Internal Server Error")

    def apply(code: Int, phrase: String): Status = {
      (code: @switch) match {
        case 200 => Ok
        case 404 => NotFound
        case 400 => BadRequest
        case 410 => Gone
        case 101 => SwitchingProtocols
        case _ => new Status(code, phrase)
      }
    }
  }
}