Generate documentation from TASTY in sbt plugin

Author: Duhemm

compiler/src/dotty/tools/dotc/Driver.scala

@@ -61,7 +61,7 @@ class Driver {
   }
 
   /** Setup extra classpath and figure out class names for tasty file inputs */
-  private def fromTastySetup(fileNames0: List[String], ctx0: Context) = {
+  protected def fromTastySetup(fileNames0: List[String], ctx0: Context): (List[String], Context) = {
     if (ctx0.settings.fromTasty.value(ctx0)) {
       // Resolve classpath and class names of tasty files
       val (classPaths, classNames) = fileNames0.map { name =>

compiler/src/dotty/tools/dotc/fromtasty/TASTYCompilationUnit.scala

@@ -1,8 +1,20 @@
 package dotty.tools.dotc.fromtasty
 
 import dotty.tools.dotc.CompilationUnit
+import dotty.tools.dotc.core.Contexts.Context
 import dotty.tools.dotc.util.NoSource
+import dotty.tools.io.Path
 
 class TASTYCompilationUnit(val className: String) extends CompilationUnit(NoSource) {
   override def toString: String = s"class file $className"
 }
+object TASTYCompilationUnit {
+  def apply(className: String)(implicit ctx: Context): Option[TASTYCompilationUnit] = {
+    if (Path(className).exists) {
+      ctx.inform(s"Ignoring $className: cannot create a `TASTYCompilationUnit` for a source file.")
+      None
+    } else {
+      Some(new TASTYCompilationUnit(className))
+    }
+  }
+}

compiler/src/dotty/tools/dotc/fromtasty/TASTYRun.scala

@@ -5,8 +5,8 @@ package fromtasty
 import core.Contexts._
 
 class TASTYRun(comp: Compiler, ictx: Context) extends Run(comp, ictx) {
-  override def compile(classNames: List[String]): Unit = {
-    val units = classNames.map(new TASTYCompilationUnit(_))
+  override def compile(classNames: List[String]) = {
+    val units = classNames.flatMap(TASTYCompilationUnit(_)(ictx))
     compileUnits(units)
   }
 }

doc-tool/src/dotty/tools/dottydoc/DocDriver.scala

@@ -27,7 +27,7 @@ class DocDriver extends Driver {
     ctx.setProperty(ContextDoc, new ContextDottydoc)
 
     val fileNames = CompilerCommand.checkUsage(summary, sourcesRequired)(ctx)
-    (fileNames, ctx)
+    fromTastySetup(fileNames, ctx)
   }
 
   override def newCompiler(implicit ctx: Context): Compiler = new DocCompiler

sbt-dotty/src/dotty/tools/sbtplugin/DottyPlugin.scala

@@ -216,9 +216,19 @@ object DottyPlugin extends AutoPlugin {
           old.withCircularDependencyLevel(sbt.librarymanagement.ivy.CircularDependencyLevel.Ignore)
         } else old
       }
-    )
+    ) ++ inConfig(Compile)(docSettings) ++ inConfig(Test)(docSettings)
   }
 
+  private val docSettings = inTask(doc)(Seq(
+    sources := {
+      val _ = compile.value // Ensure that everything is compiled, so TASTy is available.
+      val prev = sources.value
+      val tastyFiles = (classDirectory.value ** "*tasty").get.map(_.getAbsoluteFile)
+      prev ++ tastyFiles
+    },
+    scalacOptions += "-from-tasty"
+  ))
+
   /** Fetch artefacts for scalaOrganization.value %% moduleName % scalaVersion.value */
   private def fetchArtifactsOf(moduleName: String) = Def.task {
     val dependencyResolution = Keys.dependencyResolution.value