module Riak
  class Node
    LAGER_LEVELS = [
                    :debug,
                    :info,
                    :notice,
                    :warning,
                    :error,
                    :critical,
                    :alert,
                    :emergency
                   ]

    def read_console_log(*levels)
      console_log = log + 'console.log'
      if console_log.exist?
        levels = levels.map { |level| expand_log_level(level) }.compact.flatten
        pattern = /(#{levels.map { |level| "\\[#{level}\\]" }.join("|")})/
        console_log.readlines.grep(pattern)
      end
    end

    def expand_log_level(level)
      case level
      when Range
        first = LAGER_LEVELS.index(level.begin) || 0
        last = LAGER_LEVELS.index(level.end) || -1
        LAGER_LEVELS[first..last]
      when Symbol
        level
      end
    end
  end
end