dimgigov
4e54075078
CI / test (push) Has been cancelled
CI / verify (push) Has been cancelled
Clients CI / build-server (push) Has been cancelled
Clients CI / test-python (push) Has been cancelled
Clients CI / test-javascript (push) Has been cancelled
Clients CI / test-nim (push) Has been cancelled
Clients CI / test-rust (push) Has been cancelled
55 lines
2.1 KiB
Nim
55 lines
2.1 KiB
Nim
include system/inclrtl
|
|
import macros
|
|
|
|
proc transLastStmt(n, res, bracketExpr: NimNode): (NimNode, NimNode, NimNode) =
|
|
case n.kind
|
|
of nnkIfExpr, nnkIfStmt, nnkTryStmt, nnkCaseStmt:
|
|
result[0] = copyNimTree(n)
|
|
result[1] = copyNimTree(n)
|
|
result[2] = copyNimTree(n)
|
|
for i in ord(n.kind == nnkCaseStmt)..<n.len:
|
|
(result[0][i], result[1][^1], result[2][^1]) = transLastStmt(n[i], res, bracketExpr)
|
|
of nnkStmtList, nnkStmtListExpr, nnkBlockStmt, nnkBlockExpr, nnkWhileStmt,
|
|
nnkForStmt, nnkElifBranch, nnkElse, nnkElifExpr, nnkOfBranch, nnkExceptBranch:
|
|
result[0] = copyNimTree(n)
|
|
result[1] = copyNimTree(n)
|
|
result[2] = copyNimTree(n)
|
|
if n.len >= 1:
|
|
(result[0][^1], result[1][^1], result[2][^1]) = transLastStmt(n[^1], res, bracketExpr)
|
|
of nnkTableConstr:
|
|
result[1] = n[0][0]
|
|
result[2] = n[0][1]
|
|
if bracketExpr.len == 1:
|
|
bracketExpr.add([newCall(bindSym"typeof", newEmptyNode()), newCall(
|
|
bindSym"typeof", newEmptyNode())])
|
|
template adder(res, k, v) = res[k] = v
|
|
result[0] = getAst(adder(res, n[0][0], n[0][1]))
|
|
of nnkCurly:
|
|
result[2] = n[0]
|
|
if bracketExpr.len == 1:
|
|
bracketExpr.add(newCall(bindSym"typeof", newEmptyNode()))
|
|
template adder(res, v) = res.incl(v)
|
|
result[0] = getAst(adder(res, n[0]))
|
|
else:
|
|
result[2] = n
|
|
if bracketExpr.len == 1:
|
|
bracketExpr.add(newCall(bindSym"typeof", newEmptyNode()))
|
|
template adder(res, v) = res.add(v)
|
|
result[0] = getAst(adder(res, n))
|
|
|
|
macro collect*(init, body: untyped): untyped =
|
|
let res = genSym(nskVar, "collectResult")
|
|
expectKind init, {nnkCall, nnkIdent, nnkSym}
|
|
let bracketExpr = newTree(nnkBracketExpr,
|
|
if init.kind == nnkCall: init[0] else: init)
|
|
let (resBody, keyType, valueType) = transLastStmt(body, res, bracketExpr)
|
|
if bracketExpr.len == 3:
|
|
bracketExpr[1][1] = keyType
|
|
bracketExpr[2][1] = valueType
|
|
else:
|
|
bracketExpr[1][1] = valueType
|
|
let call = newTree(nnkCall, bracketExpr)
|
|
if init.kind == nnkCall:
|
|
for i in 1 ..< init.len:
|
|
call.add init[i]
|
|
result = newTree(nnkStmtListExpr, newVarStmt(res, call), resBody, res) |