intellij无法识别使用scalameta生成的代码

时间:2017-12-03 16:38:05

标签: scala intellij-idea scala-meta

我试图创建一个scalameta注释,它将在现有对象中定义一个新的case类。

我的目标是改造:

object MyObj {
  case class A()
}

成:

object MyObj {
  case class A()
  case class B(b: Int, bb: Int)
}

为此我创建了以下注释:

import scala.annotation.StaticAnnotation
import scala.meta._

class schema extends StaticAnnotation {
  inline def apply(defn: Any): Any = meta {
    val q"..$mods object $ename extends $template" = defn

    val generatedClass: Defn.Class =
      q"""
         case class B(b: Int, bb: Int)
       """

    //the stats that were defined in the annotated object
    val existingStats: scala.collection.immutable.Seq[Stat] = template.stats.get

    //new stats
    val stats = Some(existingStats :+ generatedClass)

    //the new template
    val newT = template.copy(stats=stats)

    val res =
      q"""
         ..$mods object $ename extends $newT
       """

    println("============== result ==============")
    println("res: " +res)
    println("====================================")
    res
  }
}

和这个主要类来测试它:

object Main {

  def main(args: Array[String]): Unit = {
    println("Hello")

    val b = new MyObj.B(2,22)
    val q = MyObj.A()
    println(b)

  }
}


@schema
object MyObj {
  case class A()
}

当我编译并运行主类时,代码的行为与预期的一样 在编译期间它打印:

============== result ==============
res: object MyObj {
  case class A()
  case class B(b: Int, bb: Int)
}
====================================

运行代码后:

Hello
B(2,22)

我的问题是当我尝试使用intellij这个代码时 代码从intellij编译并运行,但新类B无法被识别,因此代码完成和提示不起作用

enter image description here

我正在使用最新的intellij和scala插件

我已将整个项目上传到git: https://github.com/lev112/scalameta_intellij/tree/master

这是我第一次尝试使用scalameta,我试图了解它是否是intellij中的错误,或者我做错了什么

1 个答案:

答案 0 :(得分:0)

就像老式宏一样,需要在单独的编译单元中定义scalameta代码,以便首先对其进行编译,然后在编译其他源代码时将其调用。

感谢https://www.michaelpollmeier.com/2016/12/01/scalameta-code-generation-tutorial