mirror of
https://github.com/nim-lang/Nim.git
synced 2026-07-25 10:01:56 +00:00
49 lines
983 B
Nim
49 lines
983 B
Nim
discard """
|
|
msg: '''"muhaha"
|
|
proc poo(x, y: int) =
|
|
let y = x
|
|
echo ["poo"]'''
|
|
"""
|
|
|
|
import macros
|
|
|
|
const
|
|
foo = "muhaha"
|
|
|
|
proc poo(x, y: int) =
|
|
let y = x
|
|
echo "poo"
|
|
|
|
macro m(x: typed): untyped =
|
|
echo repr x.getImpl
|
|
|
|
m(foo)
|
|
m(poo)
|
|
|
|
#------------
|
|
|
|
macro checkOwner(x: typed, check_id: static[int]): untyped =
|
|
let sym = case check_id:
|
|
of 0: x
|
|
of 1: x.getImpl.body[0][0][0]
|
|
of 2: x.getImpl.body[0][0][^1]
|
|
of 3: x.getImpl.body[1][0]
|
|
else: x
|
|
result = newStrLitNode($sym.owner.symKind)
|
|
|
|
macro isSameOwner(x, y: typed): untyped =
|
|
result =
|
|
if x.owner == y.owner: bindSym"true"
|
|
else: bindSym"false"
|
|
|
|
|
|
static:
|
|
doAssert checkOwner(foo, 0) == "nskModule"
|
|
doAssert checkOwner(poo, 0) == "nskModule"
|
|
doAssert checkOwner(poo, 1) == "nskProc"
|
|
doAssert checkOwner(poo, 2) == "nskProc"
|
|
doAssert checkOwner(poo, 3) == "nskModule"
|
|
doAssert isSameOwner(foo, poo)
|
|
doAssert isSameOwner(foo, echo) == false
|
|
doAssert isSameOwner(poo, len) == false
|