Add highlighter string interpolator

1 files changed, 16 insertions, 0 deletions

diff --git a/src/dotty/tools/dotc/printing/SyntaxHighlighting.scala b/src/dotty/tools/dotc/printing/SyntaxHighlighting.scala
index 83c428976..932b980db 100644
--- a/src/dotty/tools/dotc/printing/SyntaxHighlighting.scala
+++ b/src/dotty/tools/dotc/printing/SyntaxHighlighting.scala
@@ -5,9 +5,25 @@ package printing
 import parsing.Tokens._
 import scala.annotation.switch
 import scala.collection.mutable.StringBuilder
+import core.Contexts.Context
 
 /** This object provides functions for syntax highlighting in the REPL */
 object SyntaxHighlighting {
+
+  implicit class SyntaxFormatting(val sc: StringContext) extends AnyVal {
+    def hl(args: Any*)(implicit ctx: Context): String =
+      new SyntaxFormatter(sc).assemble(args)
+  }
+
+  private class SyntaxFormatter(sc: StringContext) extends Formatting.StringFormatter(sc) {
+    override def assemble(args: Seq[Any])(implicit ctx: Context): String = super.assemble {
+      args.map {
+        case str: String => new String(apply(str).toArray)
+        case x => x
+      }
+    }
+  }
+
   val NoColor         = Console.RESET
   val CommentColor    = Console.GREEN
   val KeywordColor    = Console.CYAN