Scala 3宏能否引入未定义标识符并完成定义？

问题背景 #

在Scala 3.1.3中编写解析器时，希望通过宏简化语法规则的定义：

• 原写法（用String表示非终结符）：

new Grammar() {
  val rules = Set(
    Rule("tuple", List('(', "as", ')')),
    Rule("as", List('a', "more")),
    Rule("as", Nil),
    Rule("more", List(',', "as")),
    Rule("more", Nil),
  )
}

• 期望的简洁写法（用标识符代替String）：

new Grammar() {
  rule(tuple, List('(', as, ')'))
  rule(as, List('a', more), Nil)
  rule(more, List(',', as), Nil)
}

• 期望宏自动生成的代码：

new Grammar() {
  val rules = mutable.Set.empty[Rule]

  val tuple = NonTerminal("tuple")
  val as = NonTerminal("as")
  val more = NonTerminal("more")
  
  rules.add(Rule(tuple, List('(', as, ')')))
  rules.add(Rule(as, List('a', more)))
  rules.add(Rule(as, Nil))
  rules.add(Rule(more, List(',', as)))
  rules.add(Rule(more, Nil))
}

答案 #

Scala 3的宏完全支持这种需求，核心要点如下：

• 需结合inline def与编译期宏实现（基于scala.quoted.*库），利用Quasiquotes生成目标代码。
• 宏可以捕获调用点中未定义的标识符，通过其符号信息提取名称，生成对应的NonTerminal实例定义（如val tuple = NonTerminal("tuple")）。
• 要处理同一非终结符多次调用rule的场景，需在宏中收集所有非终结符名称并去重，避免生成重复的val定义。
• 宏能够在编译期遍历所有rule调用，将每个备选规则转换为rules.add(Rule(...))的代码，并将所有生成的代码插入到匿名类的作用域内。

具体实现时，宏可以访问调用上下文的语法树，识别未定义标识符的名称，即使这些标识符在当前作用域中尚未声明——这是因为Scala 3的宏在编译期的早期阶段执行，能够捕获这些符号的元信息。

内容的提问来源于stack exchange，提问作者Todd O'Bryan