Merge pull request #5193 from dotty-staging/semanticdb-infrastructure

Create infrastructure for SemanticDB

Author: nicolasstucki

build.sbt

@@ -12,6 +12,7 @@ val `dotty-sbt-bridge-bootstrapped` = Build.`dotty-sbt-bridge-bootstrapped`
 val `dotty-language-server` = Build.`dotty-language-server`
 val `dotty-bench` = Build.`dotty-bench`
 val `dotty-bench-bootstrapped` = Build.`dotty-bench-bootstrapped`
+val `dotty-semanticdb` = Build.`dotty-semanticdb`
 val `scala-library` = Build.`scala-library`
 val `scala-compiler` = Build.`scala-compiler`
 val `scala-reflect` = Build.`scala-reflect`

compiler/src/dotty/tools/dotc/consumetasty/ConsumeTasty.scala

@@ -0,0 +1,23 @@
+package dotty.tools.dotc.consumetasty
+
+import dotty.tools.dotc
+import dotty.tools.dotc.core.Contexts._
+import dotty.tools.dotc.quoted.QuoteDriver
+
+import scala.tasty.file.TastyConsumer
+
+object ConsumeTasty {
+  def apply(classpath: String, classes: List[String], tastyConsumer: TastyConsumer): Unit = {
+    if (classes.isEmpty)
+      throw new IllegalArgumentException("Parameter classes should no be empty")
+
+    class Consume extends dotc.Driver {
+      override protected def newCompiler(implicit ctx: Context): dotc.Compiler =
+        new TastyFromClass(tastyConsumer)
+    }
+
+    val currentClasspath = QuoteDriver.currentClasspath
+    val args = "-from-tasty" +: "-classpath" +: s"$classpath:$currentClasspath" +: classes
+    (new Consume).process(args.toArray)
+  }
+}

compiler/src/dotty/tools/dotc/consumetasty/TastyConsumerPhase.scala

@@ -0,0 +1,18 @@
+package dotty.tools.dotc.consumetasty
+
+import dotty.tools.dotc.core.Contexts._
+import dotty.tools.dotc.core.Phases.Phase
+import dotty.tools.dotc.tastyreflect.TastyImpl
+
+import scala.tasty.file.TastyConsumer
+
+class TastyConsumerPhase(consumer: TastyConsumer) extends Phase {
+
+  override def phaseName: String = "tastyConsumer"
+
+  override def run(implicit ctx: Context): Unit = {
+    val tasty = new TastyImpl(ctx)
+    consumer(tasty)(ctx.compilationUnit.tpdTree)
+  }
+
+}

compiler/src/dotty/tools/dotc/consumetasty/TastyFromClass.scala

@@ -0,0 +1,20 @@
+package dotty.tools.dotc.consumetasty
+
+import dotty.tools.dotc.core.Phases.Phase
+import dotty.tools.dotc.fromtasty._
+
+import scala.tasty.file.TastyConsumer
+
+class TastyFromClass(consumer: TastyConsumer) extends TASTYCompiler {
+
+  override protected def frontendPhases: List[List[Phase]] =
+    List(new ReadTastyTreesFromClasses) :: // Load classes from tasty
+    Nil
+
+  override protected def picklerPhases: List[List[Phase]] = Nil
+  override protected def transformPhases: List[List[Phase]] = Nil
+
+  override protected def backendPhases: List[List[Phase]] =
+    List(new TastyConsumerPhase(consumer)) ::  // Print all loaded classes
+    Nil
+}

compiler/src/dotty/tools/dotc/quoted/QuoteDriver.scala

@@ -75,17 +75,24 @@ class QuoteDriver extends Driver {
 
   override def initCtx: Context = {
     val ictx = contextBase.initialCtx
-    var classpath = System.getProperty("java.class.path")
+    ictx.settings.classpath.update(QuoteDriver.currentClasspath)(ictx)
+    ictx
+  }
+
+}
+
+object QuoteDriver {
+
+  def currentClasspath: String = {
+    val classpath0 = System.getProperty("java.class.path")
     this.getClass.getClassLoader match {
       case cl: URLClassLoader =>
         // Loads the classes loaded by this class loader
         // When executing `run` or `test` in sbt the classpath is not in the property java.class.path
         val newClasspath = cl.getURLs.map(_.getFile())
-        classpath = newClasspath.mkString("", java.io.File.pathSeparator, if (classpath == "") "" else java.io.File.pathSeparator + classpath)
-      case _ =>
+        newClasspath.mkString("", java.io.File.pathSeparator, if (classpath0 == "") "" else java.io.File.pathSeparator + classpath0)
+      case _ => classpath0
     }
-    ictx.settings.classpath.update(classpath)(ictx)
-    ictx
   }
 
 }

library/src/scala/tasty/file/ConsumeTasty.scala

@@ -0,0 +1,26 @@
+package scala.tasty.file
+
+object ConsumeTasty {
+
+  /** Load and process TASTy files using TASTy reflect
+   *
+   * @param classpath Classpath where the classes are located
+   * @param classes classes to be consumed
+   * @param tastyConsumer consumer that will process the tasty trees
+   */
+  def apply(classpath: String, classes: List[String], tastyConsumer: TastyConsumer): Unit = {
+    val cl = getClass.getClassLoader
+    try {
+      val dottyConsumeTastyCls = cl.loadClass("dotty.tools.dotc.consumetasty.ConsumeTasty")
+      val makeMeth = dottyConsumeTastyCls.getMethod("apply", classOf[String], classOf[List[_]], classOf[TastyConsumer])
+      makeMeth.invoke(null, classpath, classes, tastyConsumer)
+    }
+    catch {
+      case ex: ClassNotFoundException =>
+        throw new Exception(
+          s"""Could not load the dotty.tools.dotc.consumetasty.ConsumeTasty class `${ex.getMessage}` from the JVM classpath. Make sure that the compiler is on the JVM classpath.""",
+          ex
+        )
+    }
+  }
+}

library/src/scala/tasty/file/TastyConsumer.scala

@@ -0,0 +1,7 @@
+package scala.tasty.file
+
+import scala.tasty.Tasty
+
+trait TastyConsumer {
+  def apply(tasty: Tasty)(root: tasty.Tree): Unit
+}

project/Build.scala

@@ -381,6 +381,12 @@ object Build {
     dottyLib + File.pathSeparator + dottyInterfaces + File.pathSeparator + otherDeps
   }
 
+  lazy val semanticDBSettings = Seq(
+    baseDirectory in (Compile, run) := baseDirectory.value / "..",
+    baseDirectory in Test := baseDirectory.value / "..",
+    libraryDependencies += "com.novocode" % "junit-interface" % "0.11" % Test
+  )
+
   // Settings shared between dotty-doc and dotty-doc-bootstrapped
   lazy val dottyDocSettings = Seq(
     baseDirectory in (Compile, run) := baseDirectory.value / "..",
@@ -904,6 +910,8 @@ object Build {
   lazy val `dotty-bench` = project.in(file("bench")).asDottyBench(NonBootstrapped)
   lazy val `dotty-bench-bootstrapped` = project.in(file("bench")).asDottyBench(Bootstrapped)
 
+  lazy val `dotty-semanticdb` = project.in(file("semanticdb")).asDottySemanticDB(Bootstrapped)
+
   // Depend on dotty-library so that sbt projects using dotty automatically
   // depend on the dotty-library
   lazy val `scala-library` = project.
@@ -1297,6 +1305,10 @@ object Build {
       settings(commonBenchmarkSettings).
       enablePlugins(JmhPlugin)
 
+    def asDottySemanticDB(implicit mode: Mode): Project = project.withCommonSettings.
+      dependsOn(dottyCompiler).
+      settings(semanticDBSettings)
+
     def asDist(implicit mode: Mode): Project = project.
       enablePlugins(PackPlugin).
       withCommonSettings.

semanticdb/src/dotty/semanticdb/DBConsumer.scala

@@ -0,0 +1,27 @@
+package dotty.semanticdb
+
+import scala.tasty.Tasty
+import scala.tasty.file.TastyConsumer
+import scala.tasty.util.TreeTraverser
+
+class DBConsumer extends TastyConsumer {
+
+  final def apply(tasty: Tasty)(root: tasty.Tree): Unit = {
+    import tasty._
+    object Traverser extends TreeTraverser[tasty.type](tasty) {
+
+      override def traverseTree(tree: Tree)(implicit ctx: Context): Unit = tree match {
+        case IsDefinition(tree) =>
+          println(tree.name)
+          super.traverseTree(tree)
+        case tree =>
+          super.traverseTree(tree)
+      }
+
+    }
+    Traverser.traverseTree(root)(tasty.rootContext)
+  }
+
+  def println(x: Any): Unit = Predef.println(x)
+
+}

semanticdb/src/dotty/semanticdb/Main.scala

@@ -0,0 +1,18 @@
+package dotty.semanticdb
+
+import scala.tasty.Tasty
+import scala.tasty.util.TreeTraverser
+import scala.tasty.file._
+
+object Main {
+  def main(args: Array[String]): Unit = {
+    val extraClasspath = "." // TODO allow to set it from the args with -classpath XYZ
+    val classes = args.toList
+    if (args.isEmpty) {
+      println("Dotty Semantic DB: No classes where passed as argument")
+    } else {
+      println("Running Dotty Semantic DB on: " + args.mkString(" "))
+      ConsumeTasty(extraClasspath, classes, new DBConsumer)
+    }
+  }
+}