lib/httpx/loggable.rb in httpx-0.0.3 vs lib/httpx/loggable.rb in httpx-0.0.4

@@ -1,11 +1,24 @@
 # frozen_string_literal: true
 
 module HTTPX
   module Loggable
-    def log(level = @options.debug_level, label = "", &msg)
+    COLORS = {
+      black:   30,
+      red:     31,
+      green:   32,
+      yellow:  33,
+      blue:    34,
+      magenta: 35,
+      cyan:    36,
+      white:   37,
+    }.freeze
+
+    def log(level: @options.debug_level, label: "", color: nil, &msg)
       return unless @options.debug
       return unless @options.debug_level >= level
-      @options.debug << (+label << msg.call << "\n")
+      message = (+label << msg.call << "\n")
+      message = "\e[#{COLORS[color]}m#{message}\e[0m" if color && @options.debug.isatty
+      @options.debug << message
     end
   end
 end