Scala 3编译器插件生成预期代码，但在运行时失败

Question

我正在尝试开始为Scala 3编写编译器插件。在此阶段，它主要基于 https://github.com/liufengyun/scala3-plugin-example/blob/blob/master/master/plugin/plugin/src/main/main/scala/scala/phases.scala < /a>（以及随附的YouTube视频解释其工作原理）。

到目前为止，这是一个有趣的过程，我对编译器的某些方面有些感觉。

作为第一步，我只是试图将方法主体包裹在一个块中，打印出返回的对象的任何内容，然后返回对象。

这与原始插件不同，主要是因为在每个方法中添加了单个副作用方法调用 - 这也分配了局部变量（我认为这可能是问题的原因），并且将方法主体移动到一个街区。

我在这里的叉子中尽可能地以最小的示例生产：

该插件编译良好，似乎是按预期的汇编的一部分运行，然后在运行时爆炸。我不确定这是我做错了什么（不太可能）或编译器中的错误（较小的可能性，但可能性很大！）。

有人可以阐明为什么这不起作用吗？我不知道创建新符号时应该设置什么标志，所以这是一种可能性，但是有很多东西似乎可以正常工作，所以我滚动了它。

这是我在（有趣的位）的位置：

...
 override def prepareForUnit(tree: Tree)(using ctx: Context): Context =
    //find the printLn method
    val predef = requiredModule("scala.Predef")
    printlnSym = predef.requiredMethod("println", List(defn.AnyType))
    ctx

  override def transformDefDef(tree: DefDef)(using ctx: Context): Tree =
    val sym = tree.symbol

    // ignore abstract and synthetic methods
    if tree.rhs.isEmpty|| sym.isOneOf(Synthetic | Deferred | Private | Accessor)
    then return tree
    try {
      println("\n\n\n\n")
      println("========================== tree ==========================")
      println(tree.show)

      // val body = {tree.rhs}
      val body = ValDef(
        newSymbol(
        tree.symbol, termName("body"), tree.symbol.flags, tree.rhs.tpe),
        Block(Nil, tree.rhs)
      )
      
      // println(body)
      val bodyRef  = ref(body.symbol)
      val printRes = ref(printlnSym).appliedTo(bodyRef)

      // shove it all together in a block
      val rhs1 = tpd.Block(body :: printRes :: Nil, bodyRef)

      //replace RHS with new
      val newDefDef = cpy.DefDef(tree)(rhs = rhs1)
      println("====================== transformed ======================")
      println(newDefDef.show)
      newDefDef
    } catch {
      case e =>
        println("====================== error ===========================")
        println(e)
        println(e.printStackTrace)
        tree
    }
...

用于编译器插件在使用插件的编译器插件

object Test extends App:
    def foo: String = "forty two"
    def bar(x: String): Int = x.length
    def baz(x: String, y: Int): String = x + y

    baz(foo, bar(foo))

输出的测试程序（确切地说是我想要的！我此时非常兴奋）

========================== tree ==========================
def foo: String = "forty two"
====================== transformed ======================
def foo: String = 
  {
    val body: ("forty two" : String) = 
      {
        "forty two"
      }
    println(body)
    body
  }


========================== tree ==========================
def bar(x: String): Int = x.length()
====================== transformed ======================
def bar(x: String): Int = 
  {
    val body: Int = 
      {
        x.length()
      }
    println(body)
    body
  }


========================== tree ==========================
def baz(x: String, y: Int): String = x.+(y)
====================== transformed ======================
def baz(x: String, y: Int): String = 
  {
    val body: String = 
      {
        x.+(y)
      }
    println(body)
    body
  }

在运行时输出：'（（这取决于代码，都会更改运行，但总是相同的主题）

Exception in thread "main" java.lang.VerifyError: Bad local variable type
Exception Details:
  Location:
    testing/Test$.body$2()I @0: aload_1
  Reason:
    Type top (current frame, locals[1]) is not assignable to reference type
  Current Frame:
    bci: @0
    flags: { }
    locals: { 'testing/Test$' }
    stack: { }
  Bytecode:
    0000000: 2bb6 007d ac

    at testing.Test.main(Example.scala)

编辑：我正在使用Scala 3.1.2

Answer 1

创建新符号时，我正在使用现有标志。相反，我需要使用 local 标志，我想这很有意义。