module AmazingPrint::Colorize

def colorize(str, type)
  str = CGI.escapeHTML(str) if options[:html]
  if options[:plain] || !options[:color][type] || !inspector.colorize?
    str
  #
  # Check if the string color method is defined by amazing_print and accepts
  # html parameter or it has been overriden by some gem such as colorize.
  #
  elsif str.method(options[:color][type]).arity == -1 # Accepts html parameter.
    str.send(options[:color][type], options[:html])
  else
    if options[:html]
      str = %(<kbd style="color:#{options[:color][type]}">#{str}</kbd>)
    end
    str.send(options[:color][type])
  end
end