Merge pull request #6028 from dotty-staging/tasty-comment-api

Add tastyreflect comment API

Author: nicolasstucki

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

@@ -1,5 +1,8 @@
 package dotty.tools.dotc.consumetasty
 
+import dotty.tools.dotc.Run
+import dotty.tools.dotc.core.Mode
+import dotty.tools.dotc.core.Contexts.Context
 import dotty.tools.dotc.core.Phases.Phase
 import dotty.tools.dotc.fromtasty._
 
@@ -17,4 +20,9 @@ class TastyFromClass(consumer: TastyConsumer) extends TASTYCompiler {
   override protected def backendPhases: List[List[Phase]] =
     List(new TastyConsumerPhase(consumer)) ::  // Print all loaded classes
     Nil
+
+  override def newRun(implicit ctx: Context): Run = {
+    reset()
+    new TASTYRun(this, ctx.fresh.addMode(Mode.ReadPositions).addMode(Mode.ReadComments))
+  }
 }

compiler/src/dotty/tools/dotc/tastyreflect/CommentOpsImpl.scala

@@ -0,0 +1,10 @@
+package dotty.tools.dotc.tastyreflect
+
+trait CommentOpsImpl extends scala.tasty.reflect.CommentOps with CoreImpl {
+
+  implicit def CommentDeco(com: Comment): CommentAPI = new CommentAPI {
+    override def raw: String = com.raw
+    override def expanded: Option[String] = com.expanded
+    override def usecases: List[(String, Option[DefDef])] = com.usecases.map { uc => (uc.code, uc.tpdCode) }
+  }
+}

compiler/src/dotty/tools/dotc/tastyreflect/CoreImpl.scala

@@ -115,6 +115,8 @@ trait CoreImpl extends scala.tasty.reflect.Core {
 
   type Position = util.SourcePosition
 
+  type Comment = core.Comments.Comment
+
   type Constant = Constants.Constant
 
   type Symbol = core.Symbols.Symbol

compiler/src/dotty/tools/dotc/tastyreflect/ReflectionImpl.scala

@@ -8,6 +8,7 @@ class ReflectionImpl(val rootContext: Contexts.Context)
     with CaseDefOpsImpl
     with ConstantOpsImpl
     with ContextOpsImpl
+    with CommentOpsImpl
     with FlagsOpsImpl
     with IdOpsImpl
     with ImportSelectorOpsImpl

compiler/src/dotty/tools/dotc/tastyreflect/SymbolOpsImpl.scala

@@ -28,6 +28,16 @@ trait SymbolOpsImpl extends scala.tasty.reflect.SymbolOps with CoreImpl {
 
     def pos(implicit ctx: Context): Position = symbol.sourcePos
 
+    def comment(implicit ctx: Context): Option[Comment] = {
+      import dotty.tools.dotc.core.Comments.CommentsContext
+      val docCtx = ctx.docCtx.getOrElse {
+        throw new RuntimeException(
+          "DocCtx could not be found and comments are unavailable. This is a compiler-internal error."
+        )
+      }
+      docCtx.docstring(symbol)
+    }
+
     def owner(implicit ctx: Context): Symbol = symbol.owner
 
     def isLocalDummy(implicit ctx: Context): Boolean = symbol.isLocalDummy

library/src/scala/tasty/Reflection.scala

@@ -7,6 +7,7 @@ abstract class Reflection
     with CaseDefOps
     with ConstantOps
     with ContextOps
+    with CommentOps
     with FlagsOps
     with IdOps
     with ImportSelectorOps

library/src/scala/tasty/reflect/CommentOps.scala

@@ -0,0 +1,19 @@
+package scala.tasty.reflect
+
+trait CommentOps extends Core {
+
+  trait CommentAPI {
+
+    /** Raw comment string */
+    def raw: String
+
+    /** Expanded comment string, if any */
+    def expanded: Option[String]
+
+    /** List of usecases and their corresponding trees, if any */
+    def usecases: List[(String, Option[DefDef])]
+
+  }
+  implicit def CommentDeco(com: Comment): CommentAPI
+
+}

library/src/scala/tasty/reflect/Core.scala

@@ -410,6 +410,9 @@ trait Core {
   /** Source position */
   type Position <: AnyRef
 
+  /** Comment */
+  type Comment <: AnyRef
+
   /** Constant value represented as the constant itself */
   type Constant <: AnyRef
 

library/src/scala/tasty/reflect/SymbolOps.scala

@@ -37,6 +37,9 @@ trait SymbolOps extends Core {
     /** The position of this symbol */
     def pos(implicit ctx: Context): Position
 
+    /** The comment for this symbol, if any */
+    def comment(implicit ctx: Context): Option[Comment]
+
     def localContext(implicit ctx: Context): Context
 
     /** Unsafe cast as to PackageSymbol. Use IsPackageSymbol to safly check and cast to PackageSymbol */

tests/run-with-compiler/tasty-comment-consumer.check

@@ -0,0 +1,7 @@
+/** Object with comment */
+/** Object with comment */
+
+
+
+/** Val with comment */
+