mirrors/Nim
1
0
Fork 0
mirror of https://github.com/nim-lang/Nim.git synced 2026-04-19 22:10:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow ProcTy in addPragma and friends (#12158)

Browse Source
This commit is contained in:
Yuriy Glukhov
2019-09-10 08:54:51 +03:00
committed by Andreas Rumpf
parent 60911304b4
commit 5d9c1544bf
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
lib/core/macros.nim
View File

@@ -1191,17 +1191,23 @@ proc `params=`* (someProc: NimNode; params: NimNode) {.compileTime.}=
proc pragma*(someProc: NimNode): NimNode {.compileTime.} =
## Get the pragma of a proc type
## These will be expanded
someProc.expectRoutine
result = someProc[4]
proc `pragma=`*(someProc: NimNode; val: NimNode){.compileTime.}=
if someProc.kind == nnkProcTy:
result = someProc[1]
else:
someProc.expectRoutine
result = someProc[4]
proc `pragma=`*(someProc: NimNode; val: NimNode) {.compileTime.}=
## Set the pragma of a proc type
someProc.expectRoutine
expectKind(val, {nnkEmpty, nnkPragma})
someProc[4] = val
if someProc.kind == nnkProcTy:
someProc[1] = val
else:
someProc.expectRoutine
someProc[4] = val
proc addPragma*(someProc, pragma: NimNode) {.compileTime.} =
## Adds pragma to routine definition
someProc.expectRoutine
someProc.expectKind(RoutineNodes + {nnkProcTy})
var pragmaNode = someProc.pragma
if pragmaNode.isNil or pragmaNode.kind == nnkEmpty:
pragmaNode = newNimNode(nnkPragma)