From 5f106855dfe3a93c84cac6fd036d6a91d6dcbfb6 Mon Sep 17 00:00:00 2001
From: Mateusz Pietras <mateusz.pietras@leancode.pl>
Date: Thu, 7 Nov 2024 13:31:24 +0100
Subject: [PATCH] Implement ColoredTextPrintStrategy

---
 lib/src/print_strategy.dart | 40 +++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/lib/src/print_strategy.dart b/lib/src/print_strategy.dart
index 3dc8b14..9f94b42 100644
--- a/lib/src/print_strategy.dart
+++ b/lib/src/print_strategy.dart
@@ -46,3 +46,43 @@ class PlainTextPrintStrategy extends PrintStrategy {
     return log.toString();
   }
 }
+
+/// Instructs [LoggingBugfenderListener] to print colored text.
+/// The color is based on the log level.
+/// The colored text works only if the output terminal supports ANSI escape
+/// sequences. In most cases you can check if ANSI escapes are supported, using
+/// `kIsWeb || io.stdout.supportsAnsiEscapes`.
+class ColoredTextPrintStrategy extends PlainTextPrintStrategy {
+  /// Creates a strategy that prints colored plain text.
+  const ColoredTextPrintStrategy();
+
+  static final _levelColors = <Level, String?>{
+    Level.FINEST: '8',
+    Level.FINER: '8',
+    Level.FINE: '8',
+    Level.CONFIG: null,
+    Level.INFO: '12',
+    Level.WARNING: '208',
+    Level.SEVERE: '196',
+    Level.SHOUT: '199',
+  };
+
+  /// ANSI Control Sequence Introducer, signals the terminal for new settings.
+  static const ansiEsc = '\x1B[';
+
+  /// Reset all colors and options to terminal defaults.
+  static const ansiDefault = '${ansiEsc}0m';
+
+  @override
+  String print(LogRecord record) {
+    final log = super.print(record);
+
+    final color = _levelColors[record.level];
+
+    if (color != null) {
+      return '${ansiEsc}38;5;${color}m$log$ansiDefault';
+    } else {
+      return log;
+    }
+  }
+}