dev.pellet.server.routing.http.HTTPRouteResponse.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pellet-server Show documentation

An opinionated Kotlin web framework, with best-practices built-in

There is a newer version: 0.0.16

package dev.pellet.server.routing.http

import dev.pellet.server.codec.http.HTTPEntity
import dev.pellet.server.codec.http.HTTPHeader
import dev.pellet.server.codec.http.HTTPHeaders
import kotlinx.serialization.encodeToString
import kotlinx.serialization.json.Json
import java.nio.ByteBuffer

data class HTTPRouteResponse(
    val statusCode: Int,
    val headers: HTTPHeaders,
    val entity: HTTPEntity
) {

    class Builder {

        private var statusCode: Int = 0
        private val headers = HTTPHeaders()
        private var entity: HTTPEntity = HTTPEntity.NoContent

        fun statusCode(code: Int): Builder {
            statusCode = code
            return this
        }

        fun header(name: String, value: String): Builder {
            headers.add(
                HTTPHeader(name, value)
            )
            return this
        }

        fun noContent(): Builder {
            statusCode = 204
            entity = HTTPEntity.NoContent
            return this
        }

        fun notFound(): Builder {
            statusCode = 404
            entity = HTTPEntity.NoContent
            return this
        }

        fun internalServerError(): Builder {
            statusCode = 500
            entity = HTTPEntity.NoContent
            return this
        }

        fun entity(entity: HTTPEntity): Builder {
            this.entity = entity
            return this
        }

        fun entity(
            byteBuffer: ByteBuffer,
            contentType: String
        ): Builder {
            this.entity = HTTPEntity.Content.of(byteBuffer, contentType)
            return this
        }

        fun entity(
            entity: String,
            contentType: String
        ): Builder {
            this.entity = HTTPEntity.Content.of(entity, contentType = contentType)
            return this
        }

        inline fun  jsonEntity(
            encoder: Json,
            value: T
        ): Builder {
            val encodedResponse = encoder.encodeToString(value)
            val contentType = "application/json; charset=utf-8"
            this.entity(encodedResponse, contentType)
            return this
        }

        // todo: what to do with failure to build? Result<>?
        fun build(): HTTPRouteResponse {
            val statusCode = this.statusCode

            val response = HTTPRouteResponse(
                statusCode = statusCode,
                headers = headers,
                entity = entity
            )
            return response
        }
    }
}