ast: add getPIdent (#17684)

* ast: add getPIdent * fixup
2021-04-09 01:00:13 -05:00 · 2021-04-09 01:00:13 -05:00 · cce1b24b1c
commit cce1b24b1c
parent 826ad2f36f
3 changed files with 19 additions and 16 deletions
--- a/compiler/ast.nim
+++ b/compiler/ast.nim
@ -1054,6 +1054,14 @@ const
  defaultAlignment = -1
  defaultOffset = -1

+proc getPIdent*(a: PNode): PIdent {.inline.} =
+  ## Returns underlying `PIdent` for `{nkSym, nkIdent}`, or `nil`.
+  # xxx consider whether also returning the 1st ident for {nkOpenSymChoice, nkClosedSymChoice}
+  # which may simplify code.
+  case a.kind
+  of nkSym: a.sym.name
+  of nkIdent:  a.ident
+  else: nil

 proc getnimblePkg*(a: PSym): PSym =
  result = a