summary refs log tree commit diff stats
path: root/tests/macros/m18235.nim
blob: 6bb4547e01ac55413ece17db96b74db759f09d1f (plain) (blame)
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
import macros

# Necessary code to update the AST on a symbol across module boundaries when
# processed by a type macro. Used by a test of a corresponding name of this
# file.

macro eexport(n: typed): untyped =
  result = copyNimTree(n)
  # turn exported nnkSym -> nnkPostfix(*, nnkIdent), forcing re-sem
  result[0] = nnkPostfix.newTree(ident"*").add:
    n.name.strVal.ident

macro unexport(n: typed): untyped =
  result = copyNimTree(n)
  # turn nnkSym -> nnkIdent, forcing re-sem and dropping any exported-ness
  # that might be present
  result[0] = n.name.strVal.ident

proc foo*() {.unexport.} = discard
proc bar() {.eexport.} = discard

proc foooof*() {.unexport, eexport, unexport.} = discard
proc barrab() {.eexport, unexport, eexport.} = discard

macro eexportMulti(n: typed): untyped =
  # use the call version of `eexport` macro for one or more decls
  result = copyNimTree(n)
  for i in 0..<result.len:
    result[i] = newCall(ident"eexport", result[i])

macro unexportMulti(n: typed): untyped =
  # use the call version of `unexport` macro for one or more decls
  result = copyNimTree(n)
  for i in 0..<result.len:
    result[i] = newCall(ident"unexport", result[i])

unexportMulti:
  proc oof*() = discard

eexportMulti:
  proc rab() = discard
  proc baz*() = discard