colorputs 0.2.1 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 32d29f8f142b80543ba387ce5cce70984630315d
-  data.tar.gz: eb7ea6d0841cf6c1589bbf97ff9dd5f9f8865993
+  metadata.gz: 9ae9a10c9171d187ce40cb99a92850bddc386d95
+  data.tar.gz: a30994927b620ea14c80948490dbebaf9786d53b
 SHA512:
-  metadata.gz: 0a4e4a6b334bd207aa9c20bc1bf6b16094b806945205bdc00f09302743d90aba83e7299a91e2cdc93e655aab0b79e743dccaec7837a94d3a30cb3abbd8babbaf
-  data.tar.gz: f8a147777e5d929f664e1e7832b211ae7dcf5445a7e59cf3c4137ffb240a38640462c2ccba63c602f4c52383b1f8bc2cbe07fe7a167de5310c1d668cd44efaee
+  metadata.gz: 5f9a1aa2d96ab7402f45eb9fdb6a28d9752f468ad0a513c0d3db797b533a4f4885ca3de5fe8f1380a6d4e0debb0332a1f0b3a182700327e8725f3aa2ba66f34d
+  data.tar.gz: f2b0d11d8db402a6efab3b07f595dd3775f0b2f474fb23a695ff425b6d369b62772ae8dd855ec9fc72e5d0b7047bf766b59466560d74b5493e99a01b5d8e85b8

data/lib/colorputs.rb

@@ -1,30 +1,34 @@
 module Kernel
-  __colors__ = {
-    red:     '[91m',
-    green:   '[92m',
-    yellow:  '[93m',
-    blue:    '[94m',
-    magenta: '[95m',
-    cyan:    '[96m',
-    white:   '[37m',
-    black:   '[30m'
-  }
-  
-  old_puts = method :puts
- 
-  aux_rb_body = proc do |str|
-    str.chars.inject("") {|acc,c| acc << "\e[5m\e" + __colors__.values[rand(__colors__.size - 1)] + c}.<< "\e[0m"
+
+  colors = Hash.new
+
+  std_coloring = proc do |color_code, str|
+    "\e#{color_code}#{str}\e[0m"
+  end.curry()
+
+  rainbow_coloring = proc do |str|
+    str.chars.inject("") { |acc,c| acc + colors.values[rand(colors.size - 2)].(c) }
   end
 
-  p_color_body = proc do |print_m_sym, obj, color = nil|
-    color_s = if color == :rainbow
-      aux_rb_body.(obj.send(print_m_sym))
-    elsif color && __colors__[color.to_sym]
-      "\e#{__colors__[color.to_sym]}#{obj.send(print_m_sym)}\e[0m"
-    else
-      obj
-    end
-    old_puts.(color_s)
+  colors.merge!({
+    red:     std_coloring.('[91m'),
+    green:   std_coloring.('[92m'),
+    yellow:  std_coloring.('[93m'),
+    blue:    std_coloring.('[94m'),
+    magenta: std_coloring.('[95m'),
+    cyan:    std_coloring.('[96m'),
+    white:   std_coloring.('[37m'),
+    black:   std_coloring.('[30m'),
+    rainbow: rainbow_coloring
+  })
+
+  old_puts = method :puts
+
+  p_color_body = proc do |print_m_sym, obj, *args|
+    maybe_color = args.last
+    fancy_str = colors[maybe_color].(obj.send(print_m_sym)) rescue obj
+    args.pop unless fancy_str == obj
+    old_puts.(fancy_str, args)
   end.curry()
 
   define_method :puts, p_color_body.(:to_s)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: colorputs
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.2
 platform: ruby
 authors:
 - Agustín Beamurguía, Lautaro De León