Simple rails logging output colorize

colorize.rb

#!/usr/bin/env ruby -w

# SETUP:
#     (assumes ~/bin/ is in your path)
#     cp colorize.rb ~/bin/colorize
#     chmod +x ~/bin/colorize
# USAGE:
#     tail -f log/development.log | colorize

class String
  COLORS = {
    black: 30,
    blue:  34,
    yellow: 33,
    cyan: 36,
    green: 32,
    magenta: 35,
    red: 31,
    white: 37
  }

  COLORS.each do |color, code|
    define_method(color) do
      "\e[#{code}m#{self.gsub("\n", "")}\033[0m"
    end
  end
end

STDIN.each do |line|
  puts case line
  when /\[ERROR/i, /\[FATAL/i
    line.red
  when /\[WARNING/i
    line.orange
  when /\[INFO/i
    line.yellow
  else
    line
  end
end