ctucx.git: oeffi-web

[nimlang] oeffisearch fork that works without javascript

1 
2 
3 
4 
5 
6 
7 
8 
9 
10 
11 
12 
13 
14 
15 
16 
17 
18 
19 
20 
21 
22 
23 
24 
25 
26 
27 
28 
29 
30 
31 
32 
33 
34 
35 
36 
37 
38 
39 
40 
41 
42 
43 
44 
45 
46 
47 
48 
49 
50 
51 
52 
53 
54 
55 
56 
57 
58 
59 
60 
61 
62 
63 
64 
65 
66 
67 
68 
69 
70 
71 
72 
73 
74 
75 
76 
77 
78 
79 
80 
81 
82 
83 
84 
85 
86 
87 
88 
89 
90 
91 
92 
93 
94 
95 
96 
97 
98 
99 
100 
101 
102 
103 
104 
105 
106 
107 
108 
109 
110 
111 
112 
113 
114 
115 
116 
117 
118 
119 
120 
121 
122 
123 
124 
125 
126 
127 
128 
129 
130 
131 
132 
133 
134 
135 
136 
137 
138 
139 
140 
141 
142 
143 
144 
145 
146 
147 
148 
149 
150 
151 
152 
153 
154 
155 
156 
157 
158 
159 
160 
161 
162 
163 
164 
165 
166 
167 
168 
169 
170 
171 
172 
173 
174 
175 
176 
177 
178 
179 
180 
181 
182 
183 
184 
185 
186 
187 
188 
189 
190 
191 
192 
193 
194 
195 
196 
197 
198 
199 
200 
201 
202 
203 
204 
205 
206 
207 
208 
209 
210 
211 
212 
213 
214 
import asyncdispatch, asynchttpserver, macros
import strutils, parseutils
import cookies, uri, times, re, json
import tables

export asyncdispatch, asynchttpserver, cookies, uri
export strutils, re
export tables, json


type
  ServerRef* = ref object
    port*: int
    address*: string
    server*: AsyncHttpServer

proc newServer* (address: string = "0.0.0.0", port: int = 5000): ServerRef =
  return ServerRef(
    address: address,
    port: port,
    server: newAsyncHttpServer()
  )

proc cookies*(req: Request): Table[string, string] =
  result = initTable[string, string]()

  if (let cookie = req.headers.getOrDefault("Cookie"); cookie != ""):
    var i = 0
    while true:
      i += skipWhile(cookie, {' ', '\t'}, i)
      var keystart = i
      i += skipUntil(cookie, {'='}, i)
      var keyend = i-1
      if i >= len(cookie): break
      inc(i) # skip '='
      var valstart = i
      i += skipUntil(cookie, {';'}, i)
      result[substr(cookie, keystart, keyend)] = substr(cookie, valstart, i-1)
      if i >= len(cookie): break
      inc(i) # skip ';'

proc parseQuery*(data: string): Table[string, string] =
  result = initTable[string, string]()

  let data = data.split("&")
  for i in data:
    let timed = i.split("=")
    if timed.len > 1:
      result[decodeUrl(timed[0])] = decodeUrl(timed[1])

proc daysForward*(days: int): DateTime =
  return getTime().utc + initTimeInterval(days = days)

template setCookie* (key: string, value: string, time: DateTime) =
  headers.add("Set-Cookie", setCookie(key, value, time, request.headers["host"], noName=true))

proc redirect* (request: Request, url: string) {.async.}=
  await request.respond(Http303, "", newHttpHeaders([("Location", url)]))

proc respondJson* (request: Request, httpCode: HttpCode, status: string, message: string, data: JsonNode) {.async.} = 
  let response = %* {
      "status": status,
      "msg":    message,
      "data":   data
    }

  await request.respond(httpCode, $response, newHttpHeaders([("Content-Type","application/json")]))

macro pages*(server: ServerRef, body: untyped): untyped =
  ## This macro provides convenient page adding.
  ##
  ## `body` should be StmtList.
  ## page type can be:
  ## -   ``equals``
  ## -   ``startsWith``
  ## -   ``endsWith``
  ## -   ``regex``
  ## -   ``notfound`` - this page uses without URL argument.
  ##
  ## When a new request to the server is received, variables are automatically created:
  ## -   ``request`` - new Request.
  ## -   ``url`` - matched URL.
  ##     -   ``equals`` - URL is request.url.path
  ##     -   ``startsWith`` - URL is text after `startswith`.
  ##     -   ``endsWith`` - URL is text before `endswith`.
  ##     -   ``regex`` - `url` param not created.
  ##     -   ``notfound`` - `url` param not created.
  ## -   ``decoded_url`` - URL always is request.url.path
  var
    stmtlist = newStmtList()
    notfound_declaration = false

  stmtlist.add(
    newNimNode(nnkLetSection).add(  # let urlParams: JsonNode = await parseQuery(request)
      newNimNode(nnkIdentDefs).add(  # let decode_url: string = decodeUrl(request.url.path)
        ident("decoded_url"),
        ident("string"),
        newCall(
          "decodeUrl",
          newNimNode(nnkDotExpr).add(
            newNimNode(nnkDotExpr).add(
              ident("request"), ident("url")
            ),
            ident("path")
          )
        )
      )
    )
  )
  stmtlist.add(newNimNode(nnkIfStmt))
  var ifstmtlist = stmtlist[1]

  for i in body:  # for each page in statment list.
    let
      current = if i.len == 3: $i[0] else: "equals"
      path = if i.len == 3: i[1] else: i[0]
      slist = if i.len == 3: i[2] else: i[1]
    if (i.kind == nnkCall and
        (path.kind == nnkStrLit or path.kind == nnkCallStrLit or path.kind == nnkEmpty) and
        slist.kind == nnkStmtList):
      if current == "equals":
        slist.insert(0,  # let url: string = `path`
          newNimNode(nnkLetSection).add(
            newNimNode(nnkIdentDefs).add(
              ident("url"), ident("string"), path
            )
          )
        )
        ifstmtlist.add(  # decoded_url == `path`
          newNimNode(nnkElifBranch).add(
            newCall("==", path, ident("decoded_url")),
            slist
          )
        )
      elif current == "startsWith":
        slist.insert(0,  # let url = decoded_url[`path`.len..^1]
          newNimNode(nnkLetSection).add(
            newNimNode(nnkIdentDefs).add(
              ident("url"),
              ident("string"),
              newCall(
                "[]",
                ident("decoded_url"),
                newCall("..^", newCall("len", path), newLit(1))
              )
            )
          )
        )
        ifstmtlist.add(  # decode_url.startsWith(`path`)
          newNimNode(nnkElifBranch).add(
            newCall("startsWith", ident("decoded_url"), path),
            slist
            )
          )
      elif current == "endsWith":
        slist.insert(0,  # let url: string = decoded_url[0..^`path`.len]
          newNimNode(nnkLetSection).add(
            newNimNode(nnkIdentDefs).add(
              ident("url"),
              ident("string"),
              newCall(
                "[]",
                ident("decoded_url"),
                newCall(
                  "..^", newLit(0), newCall("+", newLit(1), newCall("len", path))
                )
              )
            )
          )
        )
        ifstmtlist.add(  # decode_url.endsWith(`path`)
          newNimNode(nnkElifBranch).add(
            newCall("endsWith", ident("decoded_url"), path),
            slist
          )
        )
      elif current == "regex":
        ifstmtlist.add(  # decode_url.match(`path`)
          newNimNode(nnkElifBranch).add(
            newCall("match", ident("decoded_url"), path),
            slist))

  ifstmtlist.add(
    newNimNode(nnkElse).add(
      newCall(  # await request.respond(Http404, "Not found")
        "await",
        newCall("respond", ident("request"), ident("Http404"), newLit("Not found"), newCall("newHttpHeaders"))
      )
    )
  )

  result = newNimNode(nnkProcDef).add(
    ident("receivepages"),  # procedure name.
    newEmptyNode(),  # for template and macros
    newEmptyNode(),  # generics
    newNimNode(nnkFormalParams).add(  # proc params
      newEmptyNode(),  # return type
      newNimNode(nnkIdentDefs).add(  # param
        ident("request"),  # param name
        ident("Request"),  # param type
        newEmptyNode()  # param default value
      )
    ),
    newNimNode(nnkPragma).add(  # pragma declaration
      ident("async"),
      ident("gcsafe")
    ),
    newEmptyNode(),
    stmtlist)

macro start*(server: ServerRef): untyped =
  result = quote do:
    echo "Server starts on http://", `server`.address, ":", `server`.port
    waitFor `server`.server.serve(Port(`server`.port), receivepages, `server`.address)