From 3bd8428322763df9e96d5b434aa6eba7b255bc53 Mon Sep 17 00:00:00 2001
From: Alinson S Xavier <axavier@anl.gov>
Date: Thu, 27 May 2021 18:01:32 -0500
Subject: [PATCH] Make logs more colorful

---
 src/log.jl       | 16 +++++++++++++++-
 test/runtests.jl |  3 +++
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/src/log.jl b/src/log.jl
index f90026e..d329c43 100644
--- a/src/log.jl
+++ b/src/log.jl
@@ -35,8 +35,17 @@ function handle_message(logger::TimeLogger,
                         kwargs...)
     elapsed_time = time() - logger.initial_time
     time_string = @sprintf("[%12.3f] ", elapsed_time)
+    
+    if level >= Logging.Error
+        color = :light_red
+    elseif level >= Logging.Warn
+        color = :light_yellow
+    else
+        color = :light_green
+    end
+    
     if level >= logger.screen_log_level
-        print(time_string)
+        printstyled(time_string, color=color)
         println(message)
     end
     if logger.file !== nothing && level >= logger.io_log_level
@@ -47,4 +56,9 @@ function handle_message(logger::TimeLogger,
     end
 end
 
+function setup_logger()
+    initial_time = time()
+    global_logger(TimeLogger(initial_time=initial_time))
+end
+
 export TimeLogger
\ No newline at end of file
diff --git a/test/runtests.jl b/test/runtests.jl
index 6a64a10..201452e 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -3,6 +3,9 @@
 # Released under the modified BSD license. See COPYING.md for more details.
 
 using Test
+using UnitCommitment
+
+UnitCommitment.setup_logger()
 
 @testset "UnitCommitment" begin
     include("instance_test.jl")