Add semantic-db project

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-semantic-db` = Build.`dotty-semantic-db`
 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

@@ -2,6 +2,7 @@ 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
 
@@ -15,7 +16,7 @@ object ConsumeTasty {
         new TastyFromClass(tastyConsumer)
     }
 
-    val currentClasspath = System.getProperty("java.class.path")
+    val currentClasspath = QuoteDriver.currentClasspath
     val args = "-from-tasty" +: "-classpath" +: s"$classpath:$currentClasspath" +: classes
     (new Consume).process(args.toArray)
   }

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
   }
 
 }

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-semantic-db` = project.in(file("semantic-db")).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.

semantic-db/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)
+
+}

semantic-db/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)
+    }
+  }
+}

semantic-db/test/dotty/semanticdb/Tests.scala

@@ -0,0 +1,33 @@
+package dotty.semanticdb
+
+import scala.tasty.Tasty
+import scala.tasty.util.TreeTraverser
+import scala.tasty.file._
+
+import org.junit.Test
+import org.junit.Assert._
+
+class Tests {
+
+  // TODO: update scala-0.10 on version change (or resolve automatically)
+  final def testClasspath = "out/bootstrap/dotty-semantic-db/scala-0.10/test-classes"
+
+  @Test def testMain(): Unit = {
+    testOutput(
+      "tests.SimpleClass",
+      "SimpleClass;<init>;"
+    )
+    testOutput(
+      "tests.SimpleDef",
+      "SimpleDef;<init>;foo;"
+    )
+  }
+
+  def testOutput(className: String, expected: String): Unit = {
+    val out = new StringBuilder
+    ConsumeTasty(testClasspath, List(className), new DBConsumer {
+      override def println(x: Any): Unit = out.append(x).append(";")
+    })
+    assertEquals(expected, out.result())
+  }
+}

semantic-db/test/tests/SimpleClass.scala

@@ -0,0 +1,3 @@
+package tests
+
+class SimpleClass

semantic-db/test/tests/SimpleDef.scala

@@ -0,0 +1,5 @@
+package tests
+
+class SimpleDef {
+  def foo(): Int = 0
+}