semtempl.nim 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831
  1. #
  2. #
  3. # The Nim Compiler
  4. # (c) Copyright 2015 Andreas Rumpf
  5. #
  6. # See the file "copying.txt", included in this
  7. # distribution, for details about the copyright.
  8. #
  9. # included from sem.nim
  10. discard """
  11. hygienic templates:
  12. template `||` (a, b: untyped): untyped =
  13. let aa = a
  14. if aa: aa else: b
  15. var
  16. a, b: T
  17. echo a || b || a
  18. Each evaluation context has to be different and we need to perform
  19. some form of preliminary symbol lookup in template definitions. Hygiene is
  20. a way to achieve lexical scoping at compile time.
  21. """
  22. const
  23. errImplOfXNotAllowed = "implementation of '$1' is not allowed"
  24. type
  25. TSymBinding = enum
  26. spNone, spGenSym, spInject
  27. proc symBinding(n: PNode): TSymBinding =
  28. for i in 0..<n.len:
  29. var it = n[i]
  30. var key = if it.kind == nkExprColonExpr: it[0] else: it
  31. if key.kind == nkIdent:
  32. case whichKeyword(key.ident)
  33. of wGensym: return spGenSym
  34. of wInject: return spInject
  35. else: discard
  36. type
  37. TSymChoiceRule = enum
  38. scClosed, scOpen, scForceOpen
  39. proc symChoice(c: PContext, n: PNode, s: PSym, r: TSymChoiceRule;
  40. isField = false): PNode =
  41. var
  42. a: PSym
  43. o: TOverloadIter
  44. var i = 0
  45. a = initOverloadIter(o, c, n)
  46. while a != nil:
  47. if a.kind != skModule:
  48. inc(i)
  49. if i > 1: break
  50. a = nextOverloadIter(o, c, n)
  51. let info = getCallLineInfo(n)
  52. if i <= 1 and r != scForceOpen:
  53. # XXX this makes more sense but breaks bootstrapping for now:
  54. # (s.kind notin routineKinds or s.magic != mNone):
  55. # for instance 'nextTry' is both in tables.nim and astalgo.nim ...
  56. if not isField or sfGenSym notin s.flags:
  57. result = newSymNode(s, info)
  58. markUsed(c, info, s)
  59. onUse(info, s)
  60. else:
  61. result = n
  62. else:
  63. # semantic checking requires a type; ``fitNode`` deals with it
  64. # appropriately
  65. let kind = if r == scClosed or n.kind == nkDotExpr: nkClosedSymChoice
  66. else: nkOpenSymChoice
  67. result = newNodeIT(kind, info, newTypeS(tyNone, c))
  68. a = initOverloadIter(o, c, n)
  69. while a != nil:
  70. if a.kind != skModule and (not isField or sfGenSym notin a.flags):
  71. incl(a.flags, sfUsed)
  72. markOwnerModuleAsUsed(c, a)
  73. result.add newSymNode(a, info)
  74. onUse(info, a)
  75. a = nextOverloadIter(o, c, n)
  76. proc semBindStmt(c: PContext, n: PNode, toBind: var IntSet): PNode =
  77. result = copyNode(n)
  78. for i in 0..<n.len:
  79. var a = n[i]
  80. # If 'a' is an overloaded symbol, we used to use the first symbol
  81. # as a 'witness' and use the fact that subsequent lookups will yield
  82. # the same symbol!
  83. # This is however not true anymore for hygienic templates as semantic
  84. # processing for them changes the symbol table...
  85. let s = qualifiedLookUp(c, a, {checkUndeclared})
  86. if s != nil:
  87. # we need to mark all symbols:
  88. let sc = symChoice(c, n, s, scClosed)
  89. if sc.kind == nkSym:
  90. toBind.incl(sc.sym.id)
  91. result.add sc
  92. else:
  93. for x in items(sc):
  94. toBind.incl(x.sym.id)
  95. result.add x
  96. else:
  97. illFormedAst(a, c.config)
  98. proc semMixinStmt(c: PContext, n: PNode, toMixin: var IntSet): PNode =
  99. result = copyNode(n)
  100. var count = 0
  101. for i in 0..<n.len:
  102. toMixin.incl(considerQuotedIdent(c, n[i]).id)
  103. let x = symChoice(c, n[i], nil, scForceOpen)
  104. inc count, x.len
  105. result.add x
  106. if count == 0:
  107. result = newNodeI(nkEmpty, n.info)
  108. proc replaceIdentBySym(c: PContext; n: var PNode, s: PNode) =
  109. case n.kind
  110. of nkPostfix: replaceIdentBySym(c, n[1], s)
  111. of nkPragmaExpr: replaceIdentBySym(c, n[0], s)
  112. of nkIdent, nkAccQuoted, nkSym: n = s
  113. else: illFormedAst(n, c.config)
  114. type
  115. TemplCtx = object
  116. c: PContext
  117. toBind, toMixin, toInject: IntSet
  118. owner: PSym
  119. cursorInBody: bool # only for nimsuggest
  120. scopeN: int
  121. noGenSym: int
  122. inTemplateHeader: int
  123. proc getIdentNode(c: var TemplCtx, n: PNode): PNode =
  124. case n.kind
  125. of nkPostfix: result = getIdentNode(c, n[1])
  126. of nkPragmaExpr: result = getIdentNode(c, n[0])
  127. of nkIdent:
  128. result = n
  129. let s = qualifiedLookUp(c.c, n, {})
  130. if s != nil:
  131. if s.owner == c.owner and s.kind == skParam:
  132. result = newSymNode(s, n.info)
  133. of nkAccQuoted, nkSym: result = n
  134. else:
  135. illFormedAst(n, c.c.config)
  136. result = n
  137. proc isTemplParam(c: TemplCtx, n: PNode): bool {.inline.} =
  138. result = n.kind == nkSym and n.sym.kind == skParam and
  139. n.sym.owner == c.owner and sfTemplateParam in n.sym.flags
  140. proc semTemplBody(c: var TemplCtx, n: PNode): PNode
  141. proc openScope(c: var TemplCtx) =
  142. openScope(c.c)
  143. proc closeScope(c: var TemplCtx) =
  144. closeScope(c.c)
  145. proc semTemplBodyScope(c: var TemplCtx, n: PNode): PNode =
  146. openScope(c)
  147. result = semTemplBody(c, n)
  148. closeScope(c)
  149. proc onlyReplaceParams(c: var TemplCtx, n: PNode): PNode =
  150. result = n
  151. if n.kind == nkIdent:
  152. let s = qualifiedLookUp(c.c, n, {})
  153. if s != nil:
  154. if s.owner == c.owner and s.kind == skParam:
  155. incl(s.flags, sfUsed)
  156. result = newSymNode(s, n.info)
  157. onUse(n.info, s)
  158. else:
  159. for i in 0..<n.safeLen:
  160. result[i] = onlyReplaceParams(c, n[i])
  161. proc newGenSym(kind: TSymKind, n: PNode, c: var TemplCtx): PSym =
  162. result = newSym(kind, considerQuotedIdent(c.c, n), nextSymId c.c.idgen, c.owner, n.info)
  163. incl(result.flags, sfGenSym)
  164. incl(result.flags, sfShadowed)
  165. proc addLocalDecl(c: var TemplCtx, n: var PNode, k: TSymKind) =
  166. # locals default to 'gensym':
  167. if n.kind == nkPragmaExpr and symBinding(n[1]) == spInject:
  168. # even if injected, don't produce a sym choice here:
  169. #n = semTemplBody(c, n)
  170. var x = n[0]
  171. while true:
  172. case x.kind
  173. of nkPostfix: x = x[1]
  174. of nkPragmaExpr: x = x[0]
  175. of nkIdent: break
  176. of nkAccQuoted:
  177. # consider: type `T TemplParam` {.inject.}
  178. # it suffices to return to treat it like 'inject':
  179. n = onlyReplaceParams(c, n)
  180. return
  181. else:
  182. illFormedAst(x, c.c.config)
  183. let ident = getIdentNode(c, x)
  184. if not isTemplParam(c, ident):
  185. c.toInject.incl(x.ident.id)
  186. else:
  187. replaceIdentBySym(c.c, n, ident)
  188. else:
  189. if (n.kind == nkPragmaExpr and n.len >= 2 and n[1].kind == nkPragma):
  190. let pragmaNode = n[1]
  191. for i in 0..<pragmaNode.len:
  192. let ni = pragmaNode[i]
  193. # see D20210801T100514
  194. var found = false
  195. if ni.kind == nkIdent:
  196. for a in templatePragmas:
  197. if ni.ident == getIdent(c.c.cache, $a):
  198. found = true
  199. break
  200. if not found:
  201. openScope(c)
  202. pragmaNode[i] = semTemplBody(c, pragmaNode[i])
  203. closeScope(c)
  204. let ident = getIdentNode(c, n)
  205. if not isTemplParam(c, ident):
  206. if n.kind != nkSym:
  207. let local = newGenSym(k, ident, c)
  208. addPrelimDecl(c.c, local)
  209. styleCheckDef(c.c, n.info, local)
  210. onDef(n.info, local)
  211. replaceIdentBySym(c.c, n, newSymNode(local, n.info))
  212. if k == skParam and c.inTemplateHeader > 0:
  213. local.flags.incl sfTemplateParam
  214. else:
  215. replaceIdentBySym(c.c, n, ident)
  216. proc semTemplSymbol(c: PContext, n: PNode, s: PSym; isField: bool): PNode =
  217. incl(s.flags, sfUsed)
  218. # bug #12885; ideally sem'checking is performed again afterwards marking
  219. # the symbol as used properly, but the nfSem mechanism currently prevents
  220. # that from happening, so we mark the module as used here already:
  221. markOwnerModuleAsUsed(c, s)
  222. # we do not call onUse here, as the identifier is not really
  223. # resolved here. We will fixup the used identifiers later.
  224. case s.kind
  225. of skUnknown:
  226. # Introduced in this pass! Leave it as an identifier.
  227. result = n
  228. of OverloadableSyms-{skTemplate,skMacro}:
  229. result = symChoice(c, n, s, scOpen, isField)
  230. of skTemplate, skMacro:
  231. result = symChoice(c, n, s, scOpen, isField)
  232. if result.kind == nkSym:
  233. # template/macro symbols might need to be semchecked again
  234. # prepareOperand etc don't do this without setting the type to nil
  235. result.typ = nil
  236. of skGenericParam:
  237. if isField and sfGenSym in s.flags: result = n
  238. else: result = newSymNodeTypeDesc(s, c.idgen, n.info)
  239. of skParam:
  240. result = n
  241. of skType:
  242. if isField and sfGenSym in s.flags: result = n
  243. else: result = newSymNodeTypeDesc(s, c.idgen, n.info)
  244. else:
  245. if isField and sfGenSym in s.flags: result = n
  246. else: result = newSymNode(s, n.info)
  247. # Issue #12832
  248. when defined(nimsuggest):
  249. suggestSym(c.graph, n.info, s, c.graph.usageSym, false)
  250. # field access (dot expr) will be handled by builtinFieldAccess
  251. if not isField:
  252. styleCheckUse(c, n.info, s)
  253. proc semRoutineInTemplName(c: var TemplCtx, n: PNode): PNode =
  254. result = n
  255. if n.kind == nkIdent:
  256. let s = qualifiedLookUp(c.c, n, {})
  257. if s != nil:
  258. if s.owner == c.owner and (s.kind == skParam or sfGenSym in s.flags):
  259. incl(s.flags, sfUsed)
  260. result = newSymNode(s, n.info)
  261. onUse(n.info, s)
  262. else:
  263. for i in 0..<n.safeLen:
  264. result[i] = semRoutineInTemplName(c, n[i])
  265. proc semRoutineInTemplBody(c: var TemplCtx, n: PNode, k: TSymKind): PNode =
  266. result = n
  267. checkSonsLen(n, bodyPos + 1, c.c.config)
  268. # routines default to 'inject':
  269. if n.kind notin nkLambdaKinds and symBinding(n[pragmasPos]) == spGenSym:
  270. let ident = getIdentNode(c, n[namePos])
  271. if not isTemplParam(c, ident):
  272. var s = newGenSym(k, ident, c)
  273. s.ast = n
  274. addPrelimDecl(c.c, s)
  275. styleCheckDef(c.c, n.info, s)
  276. onDef(n.info, s)
  277. n[namePos] = newSymNode(s, n[namePos].info)
  278. else:
  279. n[namePos] = ident
  280. else:
  281. n[namePos] = semRoutineInTemplName(c, n[namePos])
  282. # open scope for parameters
  283. openScope(c)
  284. for i in patternPos..paramsPos-1:
  285. n[i] = semTemplBody(c, n[i])
  286. if k == skTemplate: inc(c.inTemplateHeader)
  287. n[paramsPos] = semTemplBody(c, n[paramsPos])
  288. if k == skTemplate: dec(c.inTemplateHeader)
  289. for i in paramsPos+1..miscPos:
  290. n[i] = semTemplBody(c, n[i])
  291. # open scope for locals
  292. inc c.scopeN
  293. openScope(c)
  294. n[bodyPos] = semTemplBody(c, n[bodyPos])
  295. # close scope for locals
  296. closeScope(c)
  297. dec c.scopeN
  298. # close scope for parameters
  299. closeScope(c)
  300. proc semTemplSomeDecl(c: var TemplCtx, n: PNode, symKind: TSymKind; start = 0) =
  301. for i in start..<n.len:
  302. var a = n[i]
  303. case a.kind:
  304. of nkCommentStmt: continue
  305. of nkIdentDefs, nkVarTuple, nkConstDef:
  306. checkMinSonsLen(a, 3, c.c.config)
  307. when defined(nimsuggest):
  308. inc c.c.inTypeContext
  309. a[^2] = semTemplBody(c, a[^2])
  310. when defined(nimsuggest):
  311. dec c.c.inTypeContext
  312. a[^1] = semTemplBody(c, a[^1])
  313. for j in 0..<a.len-2:
  314. addLocalDecl(c, a[j], symKind)
  315. else:
  316. illFormedAst(a, c.c.config)
  317. proc semPattern(c: PContext, n: PNode; s: PSym): PNode
  318. proc semTemplBodySons(c: var TemplCtx, n: PNode): PNode =
  319. result = n
  320. for i in 0..<n.len:
  321. result[i] = semTemplBody(c, n[i])
  322. proc semTemplBody(c: var TemplCtx, n: PNode): PNode =
  323. result = n
  324. semIdeForTemplateOrGenericCheck(c.c.config, n, c.cursorInBody)
  325. case n.kind
  326. of nkIdent:
  327. if n.ident.id in c.toInject: return n
  328. let s = qualifiedLookUp(c.c, n, {})
  329. if s != nil:
  330. if s.owner == c.owner and s.kind == skParam and sfTemplateParam in s.flags:
  331. incl(s.flags, sfUsed)
  332. result = newSymNode(s, n.info)
  333. onUse(n.info, s)
  334. elif contains(c.toBind, s.id):
  335. result = symChoice(c.c, n, s, scClosed, c.noGenSym > 0)
  336. elif contains(c.toMixin, s.name.id):
  337. result = symChoice(c.c, n, s, scForceOpen, c.noGenSym > 0)
  338. elif s.owner == c.owner and sfGenSym in s.flags and c.noGenSym == 0:
  339. # template tmp[T](x: var seq[T]) =
  340. # var yz: T
  341. incl(s.flags, sfUsed)
  342. result = newSymNode(s, n.info)
  343. onUse(n.info, s)
  344. else:
  345. result = semTemplSymbol(c.c, n, s, c.noGenSym > 0)
  346. of nkBind:
  347. result = semTemplBody(c, n[0])
  348. of nkBindStmt:
  349. result = semBindStmt(c.c, n, c.toBind)
  350. of nkMixinStmt:
  351. if c.scopeN > 0: result = semTemplBodySons(c, n)
  352. else: result = semMixinStmt(c.c, n, c.toMixin)
  353. of nkEmpty, nkSym..nkNilLit, nkComesFrom:
  354. discard
  355. of nkIfStmt:
  356. for i in 0..<n.len:
  357. var it = n[i]
  358. if it.len == 2:
  359. openScope(c)
  360. it[0] = semTemplBody(c, it[0])
  361. it[1] = semTemplBody(c, it[1])
  362. closeScope(c)
  363. else:
  364. n[i] = semTemplBodyScope(c, it)
  365. of nkWhileStmt:
  366. openScope(c)
  367. for i in 0..<n.len:
  368. n[i] = semTemplBody(c, n[i])
  369. closeScope(c)
  370. of nkCaseStmt:
  371. openScope(c)
  372. n[0] = semTemplBody(c, n[0])
  373. for i in 1..<n.len:
  374. var a = n[i]
  375. checkMinSonsLen(a, 1, c.c.config)
  376. for j in 0..<a.len-1:
  377. a[j] = semTemplBody(c, a[j])
  378. a[^1] = semTemplBodyScope(c, a[^1])
  379. closeScope(c)
  380. of nkForStmt, nkParForStmt:
  381. openScope(c)
  382. n[^2] = semTemplBody(c, n[^2])
  383. for i in 0..<n.len - 2:
  384. if n[i].kind == nkVarTuple:
  385. for j in 0..<n[i].len-1:
  386. addLocalDecl(c, n[i][j], skForVar)
  387. else:
  388. addLocalDecl(c, n[i], skForVar)
  389. openScope(c)
  390. n[^1] = semTemplBody(c, n[^1])
  391. closeScope(c)
  392. closeScope(c)
  393. of nkBlockStmt, nkBlockExpr, nkBlockType:
  394. checkSonsLen(n, 2, c.c.config)
  395. openScope(c)
  396. if n[0].kind != nkEmpty:
  397. addLocalDecl(c, n[0], skLabel)
  398. when false:
  399. # labels are always 'gensym'ed:
  400. let s = newGenSym(skLabel, n[0], c)
  401. addPrelimDecl(c.c, s)
  402. styleCheckDef(c.c, s)
  403. onDef(n[0].info, s)
  404. n[0] = newSymNode(s, n[0].info)
  405. n[1] = semTemplBody(c, n[1])
  406. closeScope(c)
  407. of nkTryStmt, nkHiddenTryStmt:
  408. checkMinSonsLen(n, 2, c.c.config)
  409. n[0] = semTemplBodyScope(c, n[0])
  410. for i in 1..<n.len:
  411. var a = n[i]
  412. checkMinSonsLen(a, 1, c.c.config)
  413. openScope(c)
  414. for j in 0..<a.len-1:
  415. if a[j].isInfixAs():
  416. addLocalDecl(c, a[j][2], skLet)
  417. a[j][1] = semTemplBody(c, a[j][1])
  418. else:
  419. a[j] = semTemplBody(c, a[j])
  420. a[^1] = semTemplBodyScope(c, a[^1])
  421. closeScope(c)
  422. of nkVarSection: semTemplSomeDecl(c, n, skVar)
  423. of nkLetSection: semTemplSomeDecl(c, n, skLet)
  424. of nkFormalParams:
  425. checkMinSonsLen(n, 1, c.c.config)
  426. semTemplSomeDecl(c, n, skParam, 1)
  427. n[0] = semTemplBody(c, n[0])
  428. of nkConstSection: semTemplSomeDecl(c, n, skConst)
  429. of nkTypeSection:
  430. for i in 0..<n.len:
  431. var a = n[i]
  432. if a.kind == nkCommentStmt: continue
  433. if (a.kind != nkTypeDef): illFormedAst(a, c.c.config)
  434. checkSonsLen(a, 3, c.c.config)
  435. addLocalDecl(c, a[0], skType)
  436. for i in 0..<n.len:
  437. var a = n[i]
  438. if a.kind == nkCommentStmt: continue
  439. if (a.kind != nkTypeDef): illFormedAst(a, c.c.config)
  440. checkSonsLen(a, 3, c.c.config)
  441. if a[1].kind != nkEmpty:
  442. openScope(c)
  443. a[1] = semTemplBody(c, a[1])
  444. a[2] = semTemplBody(c, a[2])
  445. closeScope(c)
  446. else:
  447. a[2] = semTemplBody(c, a[2])
  448. of nkProcDef, nkLambdaKinds:
  449. result = semRoutineInTemplBody(c, n, skProc)
  450. of nkFuncDef:
  451. result = semRoutineInTemplBody(c, n, skFunc)
  452. of nkMethodDef:
  453. result = semRoutineInTemplBody(c, n, skMethod)
  454. of nkIteratorDef:
  455. result = semRoutineInTemplBody(c, n, skIterator)
  456. of nkTemplateDef:
  457. result = semRoutineInTemplBody(c, n, skTemplate)
  458. of nkMacroDef:
  459. result = semRoutineInTemplBody(c, n, skMacro)
  460. of nkConverterDef:
  461. result = semRoutineInTemplBody(c, n, skConverter)
  462. of nkPragmaExpr:
  463. result[0] = semTemplBody(c, n[0])
  464. of nkPostfix:
  465. result[1] = semTemplBody(c, n[1])
  466. of nkPragma:
  467. for x in n:
  468. if x.kind == nkExprColonExpr:
  469. x[1] = semTemplBody(c, x[1])
  470. of nkBracketExpr:
  471. result = newNodeI(nkCall, n.info)
  472. result.add newIdentNode(getIdent(c.c.cache, "[]"), n.info)
  473. for i in 0..<n.len: result.add(n[i])
  474. result = semTemplBodySons(c, result)
  475. of nkCurlyExpr:
  476. result = newNodeI(nkCall, n.info)
  477. result.add newIdentNode(getIdent(c.c.cache, "{}"), n.info)
  478. for i in 0..<n.len: result.add(n[i])
  479. result = semTemplBodySons(c, result)
  480. of nkAsgn, nkFastAsgn:
  481. checkSonsLen(n, 2, c.c.config)
  482. let a = n[0]
  483. let b = n[1]
  484. let k = a.kind
  485. case k
  486. of nkBracketExpr:
  487. result = newNodeI(nkCall, n.info)
  488. result.add newIdentNode(getIdent(c.c.cache, "[]="), n.info)
  489. for i in 0..<a.len: result.add(a[i])
  490. result.add(b)
  491. let a0 = semTemplBody(c, a[0])
  492. result = semTemplBodySons(c, result)
  493. of nkCurlyExpr:
  494. result = newNodeI(nkCall, n.info)
  495. result.add newIdentNode(getIdent(c.c.cache, "{}="), n.info)
  496. for i in 0..<a.len: result.add(a[i])
  497. result.add(b)
  498. result = semTemplBodySons(c, result)
  499. else:
  500. result = semTemplBodySons(c, n)
  501. of nkCallKinds-{nkPostfix}:
  502. # do not transform runnableExamples (bug #9143)
  503. if not isRunnableExamples(n[0]):
  504. result = semTemplBodySons(c, n)
  505. of nkDotExpr, nkAccQuoted:
  506. # dotExpr is ambiguous: note that we explicitly allow 'x.TemplateParam',
  507. # so we use the generic code for nkDotExpr too
  508. let s = qualifiedLookUp(c.c, n, {})
  509. if s != nil:
  510. # do not symchoice a quoted template parameter (bug #2390):
  511. if s.owner == c.owner and s.kind == skParam and
  512. n.kind == nkAccQuoted and n.len == 1:
  513. incl(s.flags, sfUsed)
  514. onUse(n.info, s)
  515. return newSymNode(s, n.info)
  516. elif contains(c.toBind, s.id):
  517. return symChoice(c.c, n, s, scClosed, c.noGenSym > 0)
  518. elif contains(c.toMixin, s.name.id):
  519. return symChoice(c.c, n, s, scForceOpen, c.noGenSym > 0)
  520. else:
  521. return symChoice(c.c, n, s, scOpen, c.noGenSym > 0)
  522. if n.kind == nkDotExpr:
  523. result = n
  524. result[0] = semTemplBody(c, n[0])
  525. inc c.noGenSym
  526. result[1] = semTemplBody(c, n[1])
  527. dec c.noGenSym
  528. else:
  529. result = semTemplBodySons(c, n)
  530. of nkExprColonExpr, nkExprEqExpr:
  531. if n.len == 2:
  532. inc c.noGenSym
  533. result[0] = semTemplBody(c, n[0])
  534. dec c.noGenSym
  535. result[1] = semTemplBody(c, n[1])
  536. else:
  537. result = semTemplBodySons(c, n)
  538. of nkTableConstr:
  539. # also transform the keys (bug #12595)
  540. for i in 0..<n.len:
  541. result[i] = semTemplBodySons(c, n[i])
  542. else:
  543. result = semTemplBodySons(c, n)
  544. proc semTemplBodyDirty(c: var TemplCtx, n: PNode): PNode =
  545. result = n
  546. semIdeForTemplateOrGenericCheck(c.c.config, n, c.cursorInBody)
  547. case n.kind
  548. of nkIdent:
  549. let s = qualifiedLookUp(c.c, n, {})
  550. if s != nil:
  551. if s.owner == c.owner and s.kind == skParam:
  552. result = newSymNode(s, n.info)
  553. elif contains(c.toBind, s.id):
  554. result = symChoice(c.c, n, s, scClosed)
  555. of nkBind:
  556. result = semTemplBodyDirty(c, n[0])
  557. of nkBindStmt:
  558. result = semBindStmt(c.c, n, c.toBind)
  559. of nkEmpty, nkSym..nkNilLit, nkComesFrom:
  560. discard
  561. else:
  562. # dotExpr is ambiguous: note that we explicitly allow 'x.TemplateParam',
  563. # so we use the generic code for nkDotExpr too
  564. if n.kind == nkDotExpr or n.kind == nkAccQuoted:
  565. let s = qualifiedLookUp(c.c, n, {})
  566. if s != nil and contains(c.toBind, s.id):
  567. return symChoice(c.c, n, s, scClosed)
  568. result = n
  569. for i in 0..<n.len:
  570. result[i] = semTemplBodyDirty(c, n[i])
  571. # in semstmts.nim:
  572. proc semProcAnnotation(c: PContext, prc: PNode; validPragmas: TSpecialWords): PNode
  573. proc semTemplateDef(c: PContext, n: PNode): PNode =
  574. result = semProcAnnotation(c, n, templatePragmas)
  575. if result != nil: return result
  576. result = n
  577. var s: PSym
  578. if isTopLevel(c):
  579. s = semIdentVis(c, skTemplate, n[namePos], {sfExported})
  580. incl(s.flags, sfGlobal)
  581. else:
  582. s = semIdentVis(c, skTemplate, n[namePos], {})
  583. assert s.kind == skTemplate
  584. styleCheckDef(c, s)
  585. onDef(n[namePos].info, s)
  586. # check parameter list:
  587. #s.scope = c.currentScope
  588. pushOwner(c, s)
  589. openScope(c)
  590. n[namePos] = newSymNode(s)
  591. pragmaCallable(c, s, n, templatePragmas)
  592. implicitPragmas(c, s, n.info, templatePragmas)
  593. setGenericParamsMisc(c, n)
  594. # process parameters:
  595. var allUntyped = true
  596. if n[paramsPos].kind != nkEmpty:
  597. semParamList(c, n[paramsPos], n[genericParamsPos], s)
  598. # a template's parameters are not gensym'ed even if that was originally the
  599. # case as we determine whether it's a template parameter in the template
  600. # body by the absence of the sfGenSym flag:
  601. for i in 1..<s.typ.n.len:
  602. let param = s.typ.n[i].sym
  603. param.flags.incl sfTemplateParam
  604. param.flags.excl sfGenSym
  605. if param.typ.kind != tyUntyped: allUntyped = false
  606. else:
  607. s.typ = newTypeS(tyProc, c)
  608. # XXX why do we need tyTyped as a return type again?
  609. s.typ.n = newNodeI(nkFormalParams, n.info)
  610. rawAddSon(s.typ, newTypeS(tyTyped, c))
  611. s.typ.n.add newNodeIT(nkType, n.info, s.typ[0])
  612. if n[genericParamsPos].safeLen == 0:
  613. # restore original generic type params as no explicit or implicit were found
  614. n[genericParamsPos] = n[miscPos][1]
  615. n[miscPos] = c.graph.emptyNode
  616. if allUntyped: incl(s.flags, sfAllUntyped)
  617. if n[patternPos].kind != nkEmpty:
  618. n[patternPos] = semPattern(c, n[patternPos], s)
  619. var ctx: TemplCtx
  620. ctx.toBind = initIntSet()
  621. ctx.toMixin = initIntSet()
  622. ctx.toInject = initIntSet()
  623. ctx.c = c
  624. ctx.owner = s
  625. if sfDirty in s.flags:
  626. n[bodyPos] = semTemplBodyDirty(ctx, n[bodyPos])
  627. else:
  628. n[bodyPos] = semTemplBody(ctx, n[bodyPos])
  629. # only parameters are resolved, no type checking is performed
  630. semIdeForTemplateOrGeneric(c, n[bodyPos], ctx.cursorInBody)
  631. closeScope(c)
  632. popOwner(c)
  633. # set the symbol AST after pragmas, at least. This stops pragma that have
  634. # been pushed (implicit) to be explicitly added to the template definition
  635. # and misapplied to the body. see #18113
  636. s.ast = n
  637. if sfCustomPragma in s.flags:
  638. if n[bodyPos].kind != nkEmpty:
  639. localError(c.config, n[bodyPos].info, errImplOfXNotAllowed % s.name.s)
  640. elif n[bodyPos].kind == nkEmpty:
  641. localError(c.config, n.info, "implementation of '$1' expected" % s.name.s)
  642. var (proto, comesFromShadowscope) = searchForProc(c, c.currentScope, s)
  643. if proto == nil:
  644. addInterfaceOverloadableSymAt(c, c.currentScope, s)
  645. elif not comesFromShadowscope:
  646. if {sfTemplateRedefinition, sfGenSym} * s.flags == {}:
  647. #wrongRedefinition(c, n.info, proto.name.s, proto.info)
  648. message(c.config, n.info, warnTemplateRedefinition, s.name.s)
  649. symTabReplace(c.currentScope.symbols, proto, s)
  650. if n[patternPos].kind != nkEmpty:
  651. c.patterns.add(s)
  652. proc semPatternBody(c: var TemplCtx, n: PNode): PNode =
  653. template templToExpand(s: untyped): untyped =
  654. s.kind == skTemplate and (s.typ.len == 1 or sfAllUntyped in s.flags)
  655. proc newParam(c: var TemplCtx, n: PNode, s: PSym): PNode =
  656. # the param added in the current scope is actually wrong here for
  657. # macros because they have a shadowed param of type 'PNimNode' (see
  658. # semtypes.addParamOrResult). Within the pattern we have to ensure
  659. # to use the param with the proper type though:
  660. incl(s.flags, sfUsed)
  661. onUse(n.info, s)
  662. let x = c.owner.typ.n[s.position+1].sym
  663. assert x.name == s.name
  664. result = newSymNode(x, n.info)
  665. proc handleSym(c: var TemplCtx, n: PNode, s: PSym): PNode =
  666. result = n
  667. if s != nil:
  668. if s.owner == c.owner and s.kind == skParam:
  669. result = newParam(c, n, s)
  670. elif contains(c.toBind, s.id):
  671. result = symChoice(c.c, n, s, scClosed)
  672. elif templToExpand(s):
  673. result = semPatternBody(c, semTemplateExpr(c.c, n, s, {efNoSemCheck}))
  674. else:
  675. discard
  676. # we keep the ident unbound for matching instantiated symbols and
  677. # more flexibility
  678. proc expectParam(c: var TemplCtx, n: PNode): PNode =
  679. let s = qualifiedLookUp(c.c, n, {})
  680. if s != nil and s.owner == c.owner and s.kind == skParam:
  681. result = newParam(c, n, s)
  682. else:
  683. localError(c.c.config, n.info, "invalid expression")
  684. result = n
  685. result = n
  686. case n.kind
  687. of nkIdent:
  688. let s = qualifiedLookUp(c.c, n, {})
  689. result = handleSym(c, n, s)
  690. of nkBindStmt:
  691. result = semBindStmt(c.c, n, c.toBind)
  692. of nkEmpty, nkSym..nkNilLit: discard
  693. of nkCurlyExpr:
  694. # we support '(pattern){x}' to bind a subpattern to a parameter 'x';
  695. # '(pattern){|x}' does the same but the matches will be gathered in 'x'
  696. if n.len != 2:
  697. localError(c.c.config, n.info, "invalid expression")
  698. elif n[1].kind == nkIdent:
  699. n[0] = semPatternBody(c, n[0])
  700. n[1] = expectParam(c, n[1])
  701. elif n[1].kind == nkPrefix and n[1][0].kind == nkIdent:
  702. let opr = n[1][0]
  703. if opr.ident.s == "|":
  704. n[0] = semPatternBody(c, n[0])
  705. n[1][1] = expectParam(c, n[1][1])
  706. else:
  707. localError(c.c.config, n.info, "invalid expression")
  708. else:
  709. localError(c.c.config, n.info, "invalid expression")
  710. of nkStmtList, nkStmtListExpr:
  711. if stupidStmtListExpr(n):
  712. result = semPatternBody(c, n.lastSon)
  713. else:
  714. for i in 0..<n.len:
  715. result[i] = semPatternBody(c, n[i])
  716. of nkCallKinds:
  717. let s = qualifiedLookUp(c.c, n[0], {})
  718. if s != nil:
  719. if s.owner == c.owner and s.kind == skParam: discard
  720. elif contains(c.toBind, s.id): discard
  721. elif templToExpand(s):
  722. return semPatternBody(c, semTemplateExpr(c.c, n, s, {efNoSemCheck}))
  723. if n.kind == nkInfix and (let id = considerQuotedIdent(c.c, n[0]); id != nil):
  724. # we interpret `*` and `|` only as pattern operators if they occur in
  725. # infix notation, so that '`*`(a, b)' can be used for verbatim matching:
  726. if id.s == "*" or id.s == "**":
  727. result = newNodeI(nkPattern, n.info, n.len)
  728. result[0] = newIdentNode(id, n.info)
  729. result[1] = semPatternBody(c, n[1])
  730. result[2] = expectParam(c, n[2])
  731. return
  732. elif id.s == "|":
  733. result = newNodeI(nkPattern, n.info, n.len)
  734. result[0] = newIdentNode(id, n.info)
  735. result[1] = semPatternBody(c, n[1])
  736. result[2] = semPatternBody(c, n[2])
  737. return
  738. if n.kind == nkPrefix and (let id = considerQuotedIdent(c.c, n[0]); id != nil):
  739. if id.s == "~":
  740. result = newNodeI(nkPattern, n.info, n.len)
  741. result[0] = newIdentNode(id, n.info)
  742. result[1] = semPatternBody(c, n[1])
  743. return
  744. for i in 0..<n.len:
  745. result[i] = semPatternBody(c, n[i])
  746. else:
  747. # dotExpr is ambiguous: note that we explicitly allow 'x.TemplateParam',
  748. # so we use the generic code for nkDotExpr too
  749. case n.kind
  750. of nkDotExpr, nkAccQuoted:
  751. let s = qualifiedLookUp(c.c, n, {})
  752. if s != nil:
  753. if contains(c.toBind, s.id):
  754. return symChoice(c.c, n, s, scClosed)
  755. else:
  756. return newIdentNode(s.name, n.info)
  757. of nkPar:
  758. if n.len == 1: return semPatternBody(c, n[0])
  759. else: discard
  760. for i in 0..<n.len:
  761. result[i] = semPatternBody(c, n[i])
  762. proc semPattern(c: PContext, n: PNode; s: PSym): PNode =
  763. openScope(c)
  764. var ctx: TemplCtx
  765. ctx.toBind = initIntSet()
  766. ctx.toMixin = initIntSet()
  767. ctx.toInject = initIntSet()
  768. ctx.c = c
  769. ctx.owner = getCurrOwner(c)
  770. result = flattenStmts(semPatternBody(ctx, n))
  771. if result.kind in {nkStmtList, nkStmtListExpr}:
  772. if result.len == 1:
  773. result = result[0]
  774. elif result.len == 0:
  775. localError(c.config, n.info, "a pattern cannot be empty")
  776. closeScope(c)
  777. addPattern(c, LazySym(sym: s))