module RuboCop::Formatter::Colorizable

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

def rainbow
  @rainbow ||= begin
    rainbow = Rainbow.new
    rainbow.enabled = false unless output.tty?
    rainbow
  end
end