93 lines
2.7 KiB
Nim
93 lines
2.7 KiB
Nim
import json, tables, os, httpcore, asynchttpserver, asyncdispatch, uri, strutils, posix
|
|
import types, endpoints/[suggestions, journeys, moreJourneys, refreshJourney]
|
|
|
|
when not defined(release):
|
|
echo "running in development mode, for production compile with -d:release"
|
|
import fileserver
|
|
|
|
var endpoints {.threadvar.}: Table[string, proc(data: JsonNode): Future[JsonNode] {.gcsafe.}]
|
|
|
|
proc removePidFile() =
|
|
if getEnv("PID_FILE") != "":
|
|
removeFile(getEnv("PID_FILE"))
|
|
|
|
proc CtrlCHook() {.noconv.} =
|
|
echo "Ctrl+C fired! \nStopping Server now!"
|
|
removePidFile()
|
|
quit()
|
|
|
|
proc handleRequest(req: Request) {.async,gcsafe.} =
|
|
var resp = NimHttpResponse(
|
|
code: Http404,
|
|
msg: $(%* {"status": "error", "msg": "ENDPOINT_NOT_FOUND", "data": {}}),
|
|
headers: {"Content-Type": "application/json", "Access-Control-Allow-Origin": "*"}.newHttpHeaders,
|
|
)
|
|
|
|
try:
|
|
if endpoints.hasKey($req.reqMethod & req.url.path):
|
|
resp.code = Http200
|
|
resp.msg = $(%* {
|
|
"status": "success",
|
|
"msg": "",
|
|
"data": (await endpoints[$req.reqMethod & req.url.path](parseJson(decodeUrl(req.url.query))))
|
|
})
|
|
|
|
else:
|
|
when not defined(release):
|
|
# development server
|
|
resp = servePath(req)
|
|
|
|
except notFoundException:
|
|
resp.code = Http404
|
|
resp.msg = $(%* {"status": "error", "msg": getCurrentExceptionMsg().split("\n")[0], "data": {}})
|
|
except:
|
|
resp.code = Http500
|
|
resp.msg = $(%* {"status": "error", "msg": getCurrentExceptionMsg().split("\n")[0], "data": {}})
|
|
echo repr(getCurrentException())
|
|
|
|
await req.respond(resp.code, resp.msg, resp.headers)
|
|
|
|
proc main() =
|
|
setControlCHook(CtrlCHook)
|
|
|
|
onSignal(SIGTERM):
|
|
echo "Got SIGTERM! \nStopping Server now!"
|
|
removePidFile()
|
|
quit()
|
|
|
|
if getEnv("CACHE_PATH") == "":
|
|
echo "CACHE_PATH not set! Bye...."
|
|
quit(1)
|
|
else:
|
|
discard existsOrCreateDir(getEnv("CACHE_PATH"))
|
|
|
|
var port = 8081
|
|
if getEnv("PORT") != "":
|
|
port = parseInt(getEnv("PORT"))
|
|
|
|
if getEnv("PID_FILE") != "":
|
|
try:
|
|
writeFile(getEnv("PID_FILE"), $getpid() & "\n")
|
|
echo "pid-file \"" & getEnv("PID_FILE") & "\" crated"
|
|
except:
|
|
echo "Can't write pid file!"
|
|
quit(1)
|
|
|
|
|
|
endpoints = initTable[string, proc(data: JsonNode): Future[JsonNode] {.gcsafe.}]()
|
|
|
|
endpoints[$HttpGet & "/suggestions"] = suggestionsEndpoint
|
|
endpoints[$HttpGet & "/journeys"] = journeysEndpoint
|
|
endpoints[$HttpGet & "/moreJourneys"] = moreJourneysEndpoint
|
|
endpoints[$HttpGet & "/refreshJourney"] = refreshJourneyEndpoint
|
|
|
|
let server = newAsyncHttpServer()
|
|
|
|
when not defined(release):
|
|
initFileserver()
|
|
|
|
echo "starting webserver on port " & $port
|
|
waitFor server.serve(Port(port), handleRequest)
|
|
|
|
main()
|