Merge pull request #5235 from dotty-staging/tasty-reflect-add-settings

Add TASTy reflect settings

Author: nicolasstucki

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

@@ -0,0 +1,11 @@
+package dotty.tools.dotc.tastyreflect
+
+trait SettingsOpsImpl extends scala.tasty.reflect.SettingsOps with TastyCoreImpl {
+
+  def settings(implicit ctx: Context): Settings = ctx.settings
+
+  def SettingsDeco(settings: Settings): SettingsAPI = new SettingsAPI {
+    def color(implicit ctx: Context): Boolean = settings.color.value == "always"
+  }
+
+}

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

@@ -9,6 +9,8 @@ trait TastyCoreImpl extends scala.tasty.reflect.TastyCore {
 
   type Context = core.Contexts.Context
 
+  type Settings = config.ScalaSettings
+
   type TermOrTypeTree = tpd.Tree
 
   type Tree = tpd.Tree

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

@@ -14,6 +14,7 @@ class TastyImpl(val rootContext: Contexts.Context)
     with PatternOpsImpl
     with PositionOpsImpl
     with PrintersImpl
+    with SettingsOpsImpl
     with SignatureOpsImpl
     with StandardDefinitions
     with SymbolOpsImpl

library/src/scala/tasty/Tasty.scala

@@ -13,6 +13,7 @@ abstract class Tasty
     with PatternOps
     with PositionOps
     with Printers
+    with SettingsOps
     with SignatureOps
     with StandardDefinitions
     with SymbolOps

library/src/scala/tasty/reflect/SettingsOps.scala

@@ -0,0 +1,13 @@
+package scala.tasty.reflect
+
+trait SettingsOps extends TastyCore {
+
+  /** Compiler settings */
+  def settings(implicit ctx: Context): Settings
+
+  trait SettingsAPI {
+    def color(implicit ctx: Context): Boolean
+  }
+  implicit def SettingsDeco(settings: Settings): SettingsAPI
+
+}

library/src/scala/tasty/reflect/TastyCore.scala

@@ -114,6 +114,9 @@ trait TastyCore {
   /** Compilation context */
   type Context
 
+  /** Settings */
+  type Settings
+
   // TODO: When bootstrapped, remove and use `Term | TypeTree` type directly in other files
   /** Workaround missing `|` types in Scala 2 to represent `Term | TypeTree` */
   type TermOrTypeTree /* Term | TypeTree */

library/src/scala/tasty/util/ShowSourceCode.scala

@@ -6,6 +6,11 @@ import scala.annotation.switch
 class ShowSourceCode[T <: Tasty with Singleton](tasty0: T) extends Show[T](tasty0) {
   import tasty.{rootContext => _, _}
 
+  private[this] val color: Boolean = {
+    import tasty.rootContext
+    tasty.settings.color
+  }
+
   def showTree(tree: Tree)(implicit ctx: Context): String =
     (new Buffer).printTree(tree).result()