jyapayne/nim-extensions
1
0
Fork 0
Code Issues 3 Pull requests Projects Releases 1 Packages Wiki Activity Actions

Added ability for macros, converters, and templates to be added in a class.

Browse source
This commit is contained in:
Joey Payne 2015-10-07 23:35:17 -06:00
commit 324834dc23
1 changed files with 12 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

13
oop_macro.nim
View file

@ -174,10 +174,12 @@ macro class*(head: untyped, body: untyped): untyped =
setNodeName(n2, procName, typeName) setNodeName(n2, procName, typeName)
# add the base pragma when it's a base method # add the base pragma when it's a base method
n2[4] = newNimNode(nnkPragma).add(ident"base") if node.kind == nnkMethodDef:
n2[4] = newNimNode(nnkPragma).add(ident"base")
else: else:
discard discard
var numRes = result.len()
# Iterate over the statements, adding `self: T` # Iterate over the statements, adding `self: T`
# to the parameters of functions # to the parameters of functions
for node in body.children: for node in body.children:
@ -217,6 +219,14 @@ macro class*(head: untyped, body: untyped): untyped =
result.add(n) result.add(n)
of nnkIteratorDef, nnkConverterDef, nnkMacroDef, nnkTemplateDef:
# inject `self: T` into the arguments
let n = copyNimTree(node)
echo n.treeRepr
n[3].insert(1, newIdentDefs(ident(objReference), typeName))
result.insert(numRes, n)
numRes += 1
of nnkVarSection: of nnkVarSection:
# variables get turned into fields of the type. # variables get turned into fields of the type.
for n in node.children: for n in node.children:
@ -281,6 +291,7 @@ macro class*(head: untyped, body: untyped): untyped =
# Empty <= We want to replace self # Empty <= We want to replace self
typeDecl[0][0][2][0][2] = recList typeDecl[0][0][2][0][2] = recList
result.insert(0, typeDecl) result.insert(0, typeDecl)
echo result.toStrLit()
when isMainModule: when isMainModule:
class Animal of RootObj: class Animal of RootObj: