1 files changed, 77 insertions, 0 deletions
diff --git a/compiler/src/dotty/tools/dotc/printing/Highlighting.scala b/compiler/src/dotty/tools/dotc/printing/Highlighting.scala
new file mode 100644
index 000000000..3bda7fb7a
--- /dev/null
+++ b/compiler/src/dotty/tools/dotc/printing/Highlighting.scala
@@ -0,0 +1,77 @@
+package dotty.tools
+package dotc
+package printing
+
+import scala.collection.mutable
+import core.Contexts.Context
+
+object Highlighting {
+
+  implicit def highlightShow(h: Highlight)(implicit ctx: Context): String =
+    h.show
+
+  abstract class Highlight(private val highlight: String) {
+    def text: String
+
+    def show(implicit ctx: Context) =
+      if (ctx.settings.color.value == "never") text
+      else highlight + text + Console.RESET
+
+    override def toString =
+      highlight + text + Console.RESET
+
+    def +(other: Highlight)(implicit ctx: Context): HighlightBuffer =
+      new HighlightBuffer(this) + other
+
+    def +(other: String)(implicit ctx: Context): HighlightBuffer =
+      new HighlightBuffer(this) + other
+  }
+
+  abstract class Modifier(private val mod: String, text: String) extends Highlight(Console.RESET) {
+    override def show(implicit ctx: Context) =
+      if (ctx.settings.color.value == "never") ""
+      else mod + super.show
+  }
+
+  case class HighlightBuffer(hl: Highlight)(implicit ctx: Context) {
+    val buffer = new mutable.ListBuffer[String]
+
+    buffer += hl.show
+
+    def +(other: Highlight): HighlightBuffer = {
+      buffer += other.show
+      this
+    }
+
+    def +(other: String): HighlightBuffer = {
+      buffer += other
+      this
+    }
+
+    override def toString =
+      buffer.mkString
+  }
+
+  case class NoColor(text: String) extends Highlight(Console.RESET)
+
+  case class Red(text: String) extends Highlight(Console.RED)
+  case class Blue(text: String) extends Highlight(Console.BLUE)
+  case class Cyan(text: String) extends Highlight(Console.CYAN)
+  case class Black(text: String) extends Highlight(Console.BLACK)
+  case class Green(text: String) extends Highlight(Console.GREEN)
+  case class White(text: String) extends Highlight(Console.WHITE)
+  case class Yellow(text: String) extends Highlight(Console.YELLOW)
+  case class Magenta(text: String) extends Highlight(Console.MAGENTA)
+
+  case class RedB(text: String) extends Highlight(Console.RED_B)
+  case class BlueB(text: String) extends Highlight(Console.BLUE_B)
+  case class CyanB(text: String) extends Highlight(Console.CYAN_B)
+  case class BlackB(text: String) extends Highlight(Console.BLACK_B)
+  case class GreenB(text: String) extends Highlight(Console.GREEN_B)
+  case class WhiteB(text: String) extends Highlight(Console.WHITE_B)
+  case class YellowB(text: String) extends Highlight(Console.YELLOW_B)
+  case class MagentaB(text: String) extends Highlight(Console.MAGENTA_B)
+
+  case class Bold(text: String) extends Modifier(Console.BOLD, text)
+  case class Underlined(text: String) extends Modifier(Console.UNDERLINED, text)
+}