Fix #6270: Allow change of color setting in TASTy Reflect

Author: nicolasstucki

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

@@ -32,6 +32,12 @@ class KernelImpl(val rootContext: core.Contexts.Context, val rootPosition: util.
 
   def Context_source(self: Context): java.nio.file.Path = self.compilationUnit.source.file.jpath
 
+  def Context_printColors(self: Context): Boolean = self.settings.color.value(self) == "always"
+
+  def Context_withColors(self: Context): Context = ctx.fresh.setSetting(ctx.settings.color, "always")
+
+  def Context_withoutColors(self: Context): Context = ctx.fresh.setSetting(ctx.settings.color, "never")
+
   //
   // REPORTING
   //

library/src-non-bootstrapped/scala/tasty/reflect/QuotedOps.scala

@@ -12,7 +12,9 @@ trait QuotedOps extends Core { self: Printers =>
     def cast[U: scala.quoted.Type](implicit ctx: Context): scala.quoted.Expr[U] =
       kernel.QuotedExpr_cast[U](expr)
 
-    /** Show a source code like representation of this expression */
+    /** Show a source code like representation of this expression.
+     *  Will print Ansi colors if ctx.printColors is enabled.
+     */
     def show(implicit ctx: Context): String =
       unseal.showCode
   }
@@ -22,7 +24,9 @@ trait QuotedOps extends Core { self: Printers =>
     def unseal(implicit ctx: Context): TypeTree =
       kernel.QuotedType_unseal(tpe)
 
-    /** Show a source code like representation of this type */
+    /** Show a source code like representation of this type
+     *  Will print Ansi colors if ctx.printColors is enabled.
+     */
     def show(implicit ctx: Context): String =
       unseal.showCode
   }

library/src/scala/tasty/Reflection.scala

@@ -16,7 +16,6 @@ class Reflection(val kernel: Kernel)
     with Printers
     with ReportingOps
     with RootPosition
-    with SettingsOps
     with SignatureOps
     with StandardDefinitions
     with SymbolOps

library/src/scala/tasty/reflect/ContextOps.scala

@@ -9,6 +9,15 @@ trait ContextOps extends Core {
 
     /** Returns the source file being compiled. The path is relative to the current working directory. */
     def source: java.nio.file.Path = kernel.Context_source(self)
+
+    /** Returns true if the generated strings are allowed to use colors */
+    def printColors: Boolean = kernel.Context_printColors(self)
+
+    /** Returns a new context where printColors is true */
+    def withColors: Context = kernel.Context_withColors(self)
+
+    /** Returns a new context where printColors is false */
+    def withoutColors: Context = kernel.Context_withoutColors(self)
   }
 
   /** Context of the macro expansion */

library/src/scala/tasty/reflect/Kernel.scala

@@ -139,6 +139,16 @@ trait Kernel {
   /** Returns the source file being compiled. The path is relative to the current working directory. */
   def Context_source(self: Context): java.nio.file.Path
 
+  /** Returns true if the generated strings are allowed to use colors */
+  def Context_printColors(self: Context): Boolean
+
+  /** Returns a new context where printColors is true */
+  def Context_withColors(self: Context): Context
+
+  /** Returns a new context where printColors is false */
+  def Context_withoutColors(self: Context): Context
+
+
   //
   // REPORTING
   //

library/src/scala/tasty/reflect/Printers.scala

@@ -14,7 +14,6 @@ trait Printers
   with ImportSelectorOps
   with PatternOps
   with PositionOps
-  with SettingsOps
   with SignatureOps
   with StandardDefinitions
   with SymbolOps
@@ -426,7 +425,7 @@ trait Printers
 
   class SourceCodePrinter extends Printer {
 
-    private[this] val color: Boolean = settings.color
+    private def color(implicit ctx: Context): Boolean = kernel.Context_printColors(ctx)
 
     def showTree(tree: Tree)(implicit ctx: Context): String =
       (new Buffer).printTree(tree).result()

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

@@ -0,0 +1,20 @@
+import scala.quoted._
+import scala.tasty._
+
+object api {
+  inline def (x: => String) reflect : String =
+    ${ reflImpl('x) }
+
+  private def reflImpl(x: Expr[String])(implicit refl: Reflection): Expr[String] = {
+    import refl._
+    x.show(the[Context].withoutColors).toExpr
+  }
+
+  inline def (x: => String) reflectColor : String =
+    ${ reflImplColor('x) }
+
+  private def reflImplColor(x: Expr[String])(implicit refl: Reflection): Expr[String] = {
+    import refl._
+    x.show(the[Context].withColors).toExpr
+  }
+}

tests/run-with-compiler/i6270/Macro_1.scala

@@ -0,0 +1,8 @@
+import api._
+
+object Test {
+  def main(args: Array[String]): Unit = {
+    assert(args(0).reflect == "args.apply(0)")
+    assert(args(0).reflectColor.contains("\u001b")) // Contains Ansi color
+  }
+}