module DtcRake
  module UI
    def error(msg)
      msg = Rainbow(msg).red if colorize?
      abort msg
    end
    module_function :error

    def announce(msg)
      msg = Rainbow(msg).aqua if colorize?
      puts msg
    end
    module_function :announce

    def success(msg)
      msg = Rainbow(msg).green if colorize?
      puts msg
    end
    module_function :success

    def warning(msg)
      msg = Rainbow(msg).yellow if colorize?
      puts msg
    end
    module_function :warning

    def info(msg)
      msg = Rainbow(msg).white if colorize?
      puts msg
    end
    module_function :info

    private

    def colorize?
      return @colorize if defined? @colorize

      colorize = Config.instance.colorize
      if colorize
        begin
          require "rainbow"
        rescue LoadError
          colorize = false
        end
      end
      @colorize = colorize
    end
  end
end