Скомпилировать строку в AST внутри CompilerPlugin? - PullRequest
Новая
       52

Скомпилировать строку в AST внутри CompilerPlugin?

6 голосов
/ 27 января 2011

Я хотел бы создать шаблонный плагин и в качестве первого шага преобразовать произвольную строку в ее «скомпилированное» представление AST (как, наверное, делает интерпретатор scala).Поэтому плагин компилятора может, например, назначить someString для «HELLO WORLD»: 

  @StringAnnotation("""("hello world").toString.toUpperCase""")
  var someString = ""

Мой текущий плагин первого выстрела делает коротко:

  • runafter parser
  • createновый компилятор только для представления и VirtualFile с содержанием аннотации
  • скомпилируйте и распечатайте unit.body

см: http://paste.pocoo.org/show/326025/

a) Прямо сейчас, "object o{val x = 0}" возвращает AST, но, например, "var x = 1+ 2" нет, потому что это не будет действительный файл .scala.Как я могу это исправить?

б) Является ли OnlyPresentation хорошим выбором?Должен ли я вместо этого переопределить computeInternalPhases с соответствующими фазами или использовать -Ystop: phase?

c) Можно ли связать окружение внешнего компилятора с внутренним, чтобы, например, 

  var x = _
  (...)
  @StringAnnotation("x += 3")

будет работать?

Я нашел следующий код [1] с использованием интерпретатора и одной переменной, которая делает нечто похожее: 

  Interpreter interpreter = new Interpreter(settings);
  String[] context = { "FOO" };
  interpreter.bind("context", "Array[String]", context);
  interpreter
    .interpret("de.tutorials.scala2.Test.main(context)");
  context[0] = "BAR";
  interpreter
    .interpret("de.tutorials.scala2.Test.main(context)");

[1] http://www.tutorials.de/java/320639-beispiel-zur-einbindung-des-scala-interpreters-kompilierte-scala-anwendungen.html#post1653884

спасибо

Полный код: 

class AnnotationsPI(val global: Global) extends Plugin {
  import global._
  val name = "a_plugins::AnnotationsPI" //a_ to run before namer
  val description = "AST Trans PI"
  val components = List[PluginComponent](Component)

  private object Component extends PluginComponent with Transform with TypingTransformers with TreeDSL {
    val global: AnnotationsPI.this.global.type = AnnotationsPI.this.global
    val runsAfter = List[String]("parser");
    val phaseName = AnnotationsPI.this.name

    def newTransformer(unit: CompilationUnit) = {
      new AnnotationsTransformer(unit)
    }

    val SaTpe = "StringAnnotation".toTypeName

    class AnnotationsTransformer(unit: CompilationUnit) extends TypingTransformer(unit) {

      /** When using <code>preTransform</code>, each node is
       *  visited before its children.
       */
      def preTransform(tree: Tree): Tree = tree match {
        case anno@ValDef(Modifiers(_, _, List(Apply(Select(New(Ident(SaTpe)), _), List(Literal(Constant(a))))), _), b, c, d) => //Apply(Select(New(Ident(SaTpe)), /*nme.CONSTRUCTOR*/_), /*List(x)*/x)
          val str = a.toString
          val strArr = str.getBytes("UTF-8")
          import scala.tools.nsc.{ Global, Settings, SubComponent }
          import scala.tools.nsc.reporters.{ ConsoleReporter, Reporter }

          val settings = new Settings()
          val compiler = new Global(settings, new ConsoleReporter(settings)) {
            override def onlyPresentation = true
          }

          val run = new compiler.Run
          val vfName = "Script.scala"
          var vfile = new scala.tools.nsc.io.VirtualFile(vfName)

          val os = vfile.output
          os.write(strArr, 0, str.size) // void  write(byte[] b, int off, int len) 
          os.close
          new scala.tools.nsc.util.BatchSourceFile(vfName, str)
          run.compileFiles(vfile :: Nil)
          for (unit <- run.units) {
            println("Unit: " + unit)
            println("Body:\n" + unit.body)
          }
          tree

        case _ =>
          tree
      }

      override def transform(tree: Tree): Tree = {
        super.transform(preTransform(tree))
      }
    }
  }

1 Ответ

1 голос
/ 27 января 2011

Я не знаю, поможет ли это вам, но вместо того, чтобы возиться с интерпретатором, вы можете использовать treeFrom (aString), который является частью проекта scala-refactoring (http://scala -refactoring.org / ). не отвечает на ваш вопрос о перекрестных связях, хотя ...

...