module RuboCop::Formatter::Colorizable

def colorize(string, *args)
  rainbow.wrap(string).color(*args)
end

def rainbow
  @rainbow ||= begin
    rainbow = Rainbow.new
    if options[:color]
      rainbow.enabled = true
    elsif options[:color] == false || !output.tty?
      rainbow.enabled = false
    end
    rainbow
  end
end