Support terminal colors for logger.

CMakeLists.txt

@@ -9,6 +9,7 @@ endif()
 project(webcc)
 
 option(WEBCC_ENABLE_SOAP "Enable SOAP support (need pugixml)?" ON)
+option(WEBCC_ENABLE_TEST "Build test?" ON)
 option(WEBCC_ENABLE_UNITTEST "Build unit test?" ON)
 option(WEBCC_ENABLE_EXAMPLES "Build examples?" ON)
 
@@ -140,6 +141,10 @@ if(WEBCC_ENABLE_EXAMPLES)
     add_subdirectory(examples)
 endif()
 
+if(WEBCC_ENABLE_TEST)
+    add_subdirectory(test)
+endif()
+
 if(WEBCC_ENABLE_UNITTEST)
     add_subdirectory(${THIRD_PARTY_DIR}/src/gtest)
     add_subdirectory(unittest)

test/CMakeLists.txt

@@ -0,0 +1,19 @@
+# Tests
+
+# Common libraries to link.
+set(TEST_COMMON_LIBS webcc ${Boost_LIBRARIES} ${OPENSSL_LIBRARIES}
+	"${CMAKE_THREAD_LIBS_INIT}")
+if(WIN32)
+	set(TEST_COMMON_LIBS ${TEST_COMMON_LIBS} zlibstatic crypt32)
+else()
+	set(TEST_COMMON_LIBS ${TEST_COMMON_LIBS} ${ZLIB_LIBRARIES})
+endif()
+
+if(UNIX)
+	# Add `-ldl` for Linux to avoid "undefined reference to `dlopen'".
+	set(TEST_COMMON_LIBS ${TEST_COMMON_LIBS} ${CMAKE_DL_LIBS})
+endif()
+
+add_executable(test_logger test_logger.cc)
+target_link_libraries(test_logger ${TEST_COMMON_LIBS})
+

test/test_logger.cc

@@ -0,0 +1,15 @@
+#include "webcc/logger.h"
+
+int main() {
+  WEBCC_LOG_INIT("", webcc::LOG_CONSOLE);
+
+  LOG_INFO("info");
+  LOG_WARN("warn");
+  LOG_ERRO("erro");
+  LOG_VERB("verb");
+  LOG_FATA("fata");
+  LOG_INFO("info");
+
+  return 0;
+}
+

webcc/logger.cc

@@ -24,6 +24,16 @@
 
 namespace webcc {
 
+// -----------------------------------------------------------------------------
+
+static std::string g_main_thread_id;
+
+static const char* kLevelNames[] = {
+  "VERB", "INFO", "WARN", "ERRO", "FATA"
+};
+
+// -----------------------------------------------------------------------------
+
 struct Logger {
   Logger() : file(nullptr), modes(0) {
   }
@@ -56,11 +66,96 @@ struct Logger {
 // Global logger.
 static Logger g_logger;
 
-static std::string g_main_thread_id;
+// -----------------------------------------------------------------------------
 
-static const char* kLevelNames[] = {
-  "VERB", "INFO", "WARN", "ERRO", "FATA"
-};
+// The colors related code was adapted from Loguru. See:
+//   https://github.com/emilk/loguru/blob/master/loguru.cpp
+// Thanks to Loguru!
+
+bool g_colorlogtostderr = true;
+
+static const bool g_terminal_has_color = []() {
+#if (defined(WIN32) || defined(_WIN64))
+#ifndef ENABLE_VIRTUAL_TERMINAL_PROCESSING
+#define ENABLE_VIRTUAL_TERMINAL_PROCESSING 0x0004
+#endif
+  HANDLE houtput = GetStdHandle(STD_OUTPUT_HANDLE);
+  if (houtput != INVALID_HANDLE_VALUE) {
+    DWORD mode = 0;
+    GetConsoleMode(houtput, &mode);
+    mode |= ENABLE_VIRTUAL_TERMINAL_PROCESSING;
+    return SetConsoleMode(houtput, mode) != 0;
+  }
+  return false;
+#else
+  const char* term = getenv("TERM");
+  if (term == nullptr) {
+    return false;
+  }
+
+  return strcmp(term, "cygwin") == 0 || strcmp(term, "linux") == 0 ||
+         strcmp(term, "rxvt-unicode-256color") == 0 ||
+         strcmp(term, "screen") == 0 || strcmp(term, "screen-256color") == 0 ||
+         strcmp(term, "tmux-256color") == 0 || strcmp(term, "xterm") == 0 ||
+         strcmp(term, "xterm-256color") == 0 ||
+         strcmp(term, "xterm-termite") == 0 || strcmp(term, "xterm-color") == 0;
+#endif
+}();
+
+// Colors
+
+#ifdef _WIN32
+#define VTSEQ(ID) ("\x1b[1;" #ID "m")
+#else
+#define VTSEQ(ID) ("\x1b[" #ID "m")
+#endif
+
+const char* TerminalBlack() {
+  return g_terminal_has_color ? VTSEQ(30) : "";
+}
+const char* TerminalRed() {
+  return g_terminal_has_color ? VTSEQ(31) : "";
+}
+const char* TerminalGreen() {
+  return g_terminal_has_color ? VTSEQ(32) : "";
+}
+const char* TerminalYellow() {
+  return g_terminal_has_color ? VTSEQ(33) : "";
+}
+const char* TerminalBlue() {
+  return g_terminal_has_color ? VTSEQ(34) : "";
+}
+const char* TerminalPurple() {
+  return g_terminal_has_color ? VTSEQ(35) : "";
+}
+const char* TerminalCyan() {
+  return g_terminal_has_color ? VTSEQ(36) : "";
+}
+const char* TerminalLightGray() {
+  return g_terminal_has_color ? VTSEQ(37) : "";
+}
+const char* TerminalWhite() {
+  return g_terminal_has_color ? VTSEQ(37) : "";
+}
+const char* TerminalLightRed() {
+  return g_terminal_has_color ? VTSEQ(91) : "";
+}
+const char* TerminalDim() {
+  return g_terminal_has_color ? VTSEQ(2) : "";
+}
+
+// Formating
+const char* TerminalBold() {
+  return g_terminal_has_color ? VTSEQ(1) : "";
+}
+const char* TerminalUnderline() {
+  return g_terminal_has_color ? VTSEQ(4) : "";
+}
+
+// You should end each line with this!
+const char* TerminalReset() {
+  return g_terminal_has_color ? VTSEQ(0) : "";
+}
 
 namespace bfs = boost::filesystem;
 
@@ -168,6 +263,25 @@ void LogWrite(int level, const char* file, int line, const char* format, ...) {
     va_list args;
     va_start(args, format);
 
+    if (g_colorlogtostderr && g_terminal_has_color) {
+      if (level < WEBCC_WARN) {
+        fprintf(stderr, "%s%s%s, %s, %7s, %25s, %4d, %s",
+                TerminalReset(), TerminalDim(),
+                timestamp.c_str(), kLevelNames[level], thread_id.c_str(),
+                file, line,
+                level == WEBCC_INFO ? TerminalReset() : "");  // un-dim for INFO
+      } else {
+        fprintf(stderr, "%s%s%s, %s, %7s, %25s, %4d, ",
+                TerminalReset(),
+                level == WEBCC_WARN ? TerminalYellow() : TerminalRed(),
+                timestamp.c_str(), kLevelNames[level], thread_id.c_str(),
+                file, line);
+      }
+
+      vfprintf(stderr, format, args);
+
+      fprintf(stderr, "%s\n", TerminalReset());
+    } else {
       fprintf(stderr, "%s, %s, %7s, %25s, %4d, ",
               timestamp.c_str(), kLevelNames[level], thread_id.c_str(),
               file, line);
@@ -175,6 +289,7 @@ void LogWrite(int level, const char* file, int line, const char* format, ...) {
       vfprintf(stderr, format, args);
 
       fprintf(stderr, "\n");
+    }
 
     if ((g_logger.modes & LOG_FLUSH) != 0) {
       fflush(stderr);