Merge pull request #199 from fazibear/fix-colorize

Use colorize dedicated method for color and modes
capistrano · Dec 16, 2014 · db4d1d3 · db4d1d3
2 parents 533c70c + 1905283
commit db4d1d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/sshkit/color.rb b/lib/sshkit/color.rb
@@ -1,7 +1,7 @@
 require 'colorize'
 
 module Color
-  String::COLORS.each_pair do |color, _|
+  String.colors.each do |color|
     instance_eval <<-RUBY, __FILE__, __LINE__
       def #{color}(string = '')
         string = yield if block_given?
@@ -10,7 +10,7 @@ def #{color}(string = '')
     RUBY
   end
 
-  String::MODES.each_pair do |mode, _|
+  String.modes.each do |mode|
     instance_eval <<-RUBY, __FILE__, __LINE__
       def #{mode}(string = '')
         string = yield if block_given?