mirror of https://github.com/waku-org/nwaku.git
101 lines
3.1 KiB
Nim
101 lines
3.1 KiB
Nim
when (NimMajor, NimMinor) < (1, 4):
|
|
{.push raises: [Defect].}
|
|
else:
|
|
{.push raises: [].}
|
|
|
|
import
|
|
stew/results,
|
|
stew/shims/net,
|
|
chronicles,
|
|
chronos,
|
|
presto
|
|
|
|
|
|
type RestServerResult*[T] = Result[T, string]
|
|
|
|
|
|
### Configuration
|
|
|
|
type RestServerConf* = object
|
|
cacheSize*: Natural ## \
|
|
## The maximum number of recently accessed states that are kept in \
|
|
## memory. Speeds up requests obtaining information for consecutive
|
|
## slots or epochs.
|
|
|
|
cacheTtl*: Natural ## \
|
|
## The number of seconds to keep recently accessed states in memory
|
|
|
|
requestTimeout*: Natural ## \
|
|
## The number of seconds to wait until complete REST request will be received
|
|
|
|
maxRequestBodySize*: Natural ## \
|
|
## Maximum size of REST request body (kilobytes)
|
|
|
|
maxRequestHeadersSize*: Natural ## \
|
|
## Maximum size of REST request headers (kilobytes)
|
|
|
|
proc default*(T: type RestServerConf): T =
|
|
RestServerConf(
|
|
cacheSize: 3,
|
|
cacheTtl: 60,
|
|
requestTimeout: 0,
|
|
maxRequestBodySize: 16_384,
|
|
maxRequestHeadersSize: 64
|
|
)
|
|
|
|
|
|
### Initialization
|
|
|
|
proc getRouter(allowedOrigin: Option[string]): RestRouter =
|
|
# TODO: Review this `validate` method. Check in nim-presto what is this used for.
|
|
proc validate(pattern: string, value: string): int =
|
|
## This is rough validation procedure which should be simple and fast,
|
|
## because it will be used for query routing.
|
|
if pattern.startsWith("{") and pattern.endsWith("}"): 0
|
|
else: 1
|
|
|
|
RestRouter.init(validate, allowedOrigin = allowedOrigin)
|
|
|
|
proc init*(T: type RestServerRef,
|
|
ip: ValidIpAddress, port: Port,
|
|
allowedOrigin=none(string),
|
|
conf=RestServerConf.default(),
|
|
requestErrorHandler: RestRequestErrorHandler = nil): RestServerResult[T] =
|
|
let address = initTAddress(ip, port)
|
|
let serverFlags = {
|
|
HttpServerFlags.QueryCommaSeparatedArray,
|
|
HttpServerFlags.NotifyDisconnect
|
|
}
|
|
|
|
let
|
|
headersTimeout = if conf.requestTimeout == 0: chronos.InfiniteDuration
|
|
else: seconds(int64(conf.requestTimeout))
|
|
maxHeadersSize = conf.maxRequestHeadersSize * 1024
|
|
maxRequestBodySize = conf.maxRequestBodySize * 1024
|
|
|
|
let router = getRouter(allowedOrigin)
|
|
|
|
var res: RestResult[RestServerRef]
|
|
try:
|
|
res = RestServerRef.new(
|
|
router,
|
|
address,
|
|
serverFlags = serverFlags,
|
|
httpHeadersTimeout = headersTimeout,
|
|
maxHeadersSize = maxHeadersSize,
|
|
maxRequestBodySize = maxRequestBodySize,
|
|
requestErrorHandler = requestErrorHandler
|
|
)
|
|
except CatchableError:
|
|
return err(getCurrentExceptionMsg())
|
|
|
|
# RestResult error type is cstring, so we need to map it to string
|
|
res.mapErr(proc(err: cstring): string = $err)
|
|
|
|
proc newRestHttpServer*(ip: ValidIpAddress, port: Port,
|
|
allowedOrigin=none(string),
|
|
conf=RestServerConf.default(),
|
|
requestErrorHandler: RestRequestErrorHandler = nil):
|
|
RestServerResult[RestServerRef] =
|
|
RestServerRef.init(ip, port, allowedOrigin, conf, requestErrorHandler)
|