nim-chronos/chronos/apps/http/httpcommon.nim

163 lines
5.4 KiB
Nim
Raw Normal View History

2021-01-27 06:14:17 +00:00
#
# Chronos HTTP/S common types
# (c) Copyright 2021-Present
2021-01-27 06:14:17 +00:00
# Status Research & Development GmbH
#
# Licensed under either of
# Apache License, version 2.0, (LICENSE-APACHEv2)
# MIT license (LICENSE-MIT)
2021-01-27 19:39:14 +00:00
import stew/results, httputils, strutils, uri
import ../../asyncloop, ../../asyncsync
import ../../streams/[asyncstream, boundstream]
2021-01-27 19:39:14 +00:00
export results, httputils, strutils
2021-01-27 06:14:17 +00:00
const
HeadersMark* = @[byte(0x0D), byte(0x0A), byte(0x0D), byte(0x0A)]
2021-01-30 18:18:06 +00:00
PostMethods* = {MethodPost, MethodPatch, MethodPut, MethodDelete}
2021-01-27 06:14:17 +00:00
type
HttpResult*[T] = Result[T, string]
HttpResultCode*[T] = Result[T, HttpCode]
HttpDefect* = object of Defect
2021-01-27 06:14:17 +00:00
HttpError* = object of CatchableError
2021-01-27 19:39:14 +00:00
HttpCriticalError* = object of HttpError
code*: HttpCode
2021-01-27 19:39:14 +00:00
HttpRecoverableError* = object of HttpError
code*: HttpCode
2021-01-27 19:39:14 +00:00
TransferEncodingFlags* {.pure.} = enum
Identity, Chunked, Compress, Deflate, Gzip
ContentEncodingFlags* {.pure.} = enum
Identity, Br, Compress, Deflate, Gzip
2021-01-27 06:14:17 +00:00
HttpBodyReader* = ref object of AsyncStreamReader
streams*: seq[AsyncStreamReader]
proc newHttpBodyReader*(streams: varargs[AsyncStreamReader]): HttpBodyReader =
## HttpBodyReader is AsyncStreamReader which holds references to all the
## ``streams``. Also on close it will close all the ``streams``.
##
## First stream in sequence will be used as a source.
doAssert(len(streams) > 0, "At least one stream must be added")
var res = HttpBodyReader(streams: @streams)
res.init(streams[0])
res
proc closeWait*(bstream: HttpBodyReader) {.async.} =
## Close and free resource allocated by body reader.
if len(bstream.streams) > 0:
var res = newSeq[Future[void]]()
for item in bstream.streams.items():
res.add(item.closeWait())
await allFutures(res)
await procCall(AsyncStreamReader(bstream).closeWait())
2021-02-10 13:13:36 +00:00
proc atBound*(bstream: HttpBodyReader): bool {.
raises: [Defect].} =
## Returns ``true`` if lowest stream is at EOF.
let lreader = bstream.streams[^1]
doAssert(lreader of BoundedStreamReader)
let breader = cast[BoundedStreamReader](lreader)
breader.atEof() and (breader.bytesLeft() == 0)
2021-02-10 13:13:36 +00:00
proc newHttpDefect*(msg: string): ref HttpDefect {.
raises: [HttpDefect].} =
newException(HttpDefect, msg)
2021-01-27 19:39:14 +00:00
2021-02-10 13:13:36 +00:00
proc newHttpCriticalError*(msg: string,
code = Http400): ref HttpCriticalError {.
raises: [HttpCriticalError].} =
var tre = newException(HttpCriticalError, msg)
tre.code = code
tre
2021-01-27 19:39:14 +00:00
proc newHttpRecoverableError*(msg: string,
2021-02-10 13:13:36 +00:00
code = Http400): ref HttpRecoverableError {.
raises: [HttpRecoverableError].} =
var tre = newException(HttpRecoverableError, msg)
tre.code = code
tre
2021-01-27 19:39:14 +00:00
2021-02-10 13:13:36 +00:00
iterator queryParams*(query: string): tuple[key: string, value: string] {.
raises: [Defect].} =
2021-01-27 19:39:14 +00:00
## Iterate over url-encoded query string.
for pair in query.split('&'):
let items = pair.split('=', maxsplit = 1)
let k = items[0]
if len(k) > 0:
let v = if len(items) > 1: items[1] else: ""
yield (decodeUrl(k), decodeUrl(v))
2021-01-27 19:39:14 +00:00
func getTransferEncoding*(ch: openarray[string]): HttpResult[
2021-02-10 13:13:36 +00:00
set[TransferEncodingFlags]] {.
raises: [Defect].} =
2021-01-27 19:39:14 +00:00
## Parse value of multiple HTTP headers ``Transfer-Encoding`` and return
## it as set of ``TransferEncodingFlags``.
var res: set[TransferEncodingFlags] = {}
if len(ch) == 0:
res.incl(TransferEncodingFlags.Identity)
ok(res)
else:
for header in ch:
for item in header.split(","):
case strip(item.toLowerAscii())
of "identity":
res.incl(TransferEncodingFlags.Identity)
of "chunked":
res.incl(TransferEncodingFlags.Chunked)
of "compress":
res.incl(TransferEncodingFlags.Compress)
of "deflate":
res.incl(TransferEncodingFlags.Deflate)
of "gzip":
res.incl(TransferEncodingFlags.Gzip)
2021-02-10 13:13:36 +00:00
of "x-gzip":
res.incl(TransferEncodingFlags.Gzip)
2021-01-27 19:39:14 +00:00
of "":
res.incl(TransferEncodingFlags.Identity)
else:
return err("Incorrect Transfer-Encoding value")
ok(res)
func getContentEncoding*(ch: openarray[string]): HttpResult[
2021-02-10 13:13:36 +00:00
set[ContentEncodingFlags]] {.
raises: [Defect].} =
2021-01-27 19:39:14 +00:00
## Parse value of multiple HTTP headers ``Content-Encoding`` and return
## it as set of ``ContentEncodingFlags``.
var res: set[ContentEncodingFlags] = {}
if len(ch) == 0:
res.incl(ContentEncodingFlags.Identity)
ok(res)
else:
for header in ch:
for item in header.split(","):
case strip(item.toLowerAscii()):
of "identity":
res.incl(ContentEncodingFlags.Identity)
of "br":
res.incl(ContentEncodingFlags.Br)
of "compress":
res.incl(ContentEncodingFlags.Compress)
of "deflate":
res.incl(ContentEncodingFlags.Deflate)
of "gzip":
res.incl(ContentEncodingFlags.Gzip)
2021-02-10 13:13:36 +00:00
of "x-gzip":
res.incl(ContentEncodingFlags.Gzip)
2021-01-27 19:39:14 +00:00
of "":
res.incl(ContentEncodingFlags.Identity)
else:
return err("Incorrect Content-Encoding value")
ok(res)
2021-02-10 13:13:36 +00:00
func getContentType*(ch: openarray[string]): HttpResult[string] {.
raises: [Defect].} =
2021-01-27 19:39:14 +00:00
## Check and prepare value of ``Content-Type`` header.
if len(ch) > 1:
err("Multiple Content-Type values found")
else:
let mparts = ch[0].split(";")
ok(strip(mparts[0]).toLowerAscii())