1 files changed, 45 insertions, 1 deletions

diff --git a/common/log.cpp b/common/log.cpp
index 1d93404..5d35f40 100644
--- a/common/log.cpp
+++ b/common/log.cpp
@@ -33,6 +33,46 @@ private:
     Q_DISABLE_COPY(DebugStream)
 };
 
+    /*
+     * ANSI color codes:
+     * 0: reset colors/style
+     * 1: bold
+     * 4: underline
+     * 30 - 37: black, red, green, yellow, blue, magenta, cyan, and white text
+     * 40 - 47: black, red, green, yellow, blue, magenta, cyan, and white background
+     */
+enum ANSI_Colors {
+    DoNothing = -1,
+    Reset = 0,
+    Bold = 1,
+    Underline = 4,
+    Black = 30,
+    Red = 31,
+    Green = 32,
+    Yellow = 33,
+    Blue = 34
+};
+
+static QString colorCommand(int colorCode)
+{
+    return QString("\x1b[%1m").arg(colorCode);
+}
+
+static QString colorCommand(QList<int> colorCodes)
+{
+    colorCodes.removeAll(ANSI_Colors::DoNothing);
+    if (colorCodes.isEmpty()) {
+        return QString();
+    }
+    QString string("\x1b[");
+    for (int code : colorCodes) {
+        string += QString("%1;").arg(code);
+    }
+    string.chop(1);
+    string += "m";
+    return string;
+}
+
 QDebug debugStream(DebugLevel debugLevel, int line, const char* file, const char* function)
 {
     static DebugStream stream;
@@ -47,6 +87,7 @@ QDebug debugStream(DebugLevel debugLevel, int line, const char* file, const char
     }
 
     QString prefix;
+    int prefixColorCode = ANSI_Colors::DoNothing;
     switch (debugLevel) {
         case DebugLevel::Trace:
             prefix = "Trace:";
@@ -56,15 +97,18 @@ QDebug debugStream(DebugLevel debugLevel, int line, const char* file, const char
             break;
         case DebugLevel::Warning:
             prefix = "Warning:";
+            prefixColorCode = ANSI_Colors::Red;
             break;
         case DebugLevel::Error:
             prefix = "Error:";
+            prefixColorCode = ANSI_Colors::Red;
             break;
         default:
             break;
     };
 
-    debug << prefix + QString(" %1(%2) %3:").arg(QString::fromLatin1(programName)).arg(unsigned(getpid())).arg(function) /* << file << ":" << line */;
+    const QString resetColor = colorCommand(ANSI_Colors::Reset);
+    debug << colorCommand(QList<int>() << ANSI_Colors::Bold << prefixColorCode) + prefix + resetColor + QString(" %1(%2) %3:").arg(QString::fromLatin1(programName)).arg(unsigned(getpid())).arg(function) + resetColor/* << file << ":" << line */;
 
     return debug;
 }