redistat/lib/redistat/finder.rb

module Redistat
  class Finder
    include Database
    
    attr_reader :options
    
    def initialize(options = {})
      @options = options
    end
    
    def valid_options?
      return true if !@options[:scope].blank? && !@options[:label].blank? && !@options[:from].blank? && !@options[:till].blank?
      false
    end
    
    def find(options = {})
      @options.merge!(options)
      raise InvalidOptions.new if !valid_options?
      if @options[:interval].nil? || !@options[:interval]
        find_by_magic
      else
        find_by_interval
      end
    end
    
    def find_by_interval(options = {})
      @options.merge!(options)
      raise InvalidOptions.new if !valid_options?
      key = build_key
      col = Collection.new(@options)
      col.total = Result.new(@options)
      build_date_sets.each do |set|
        set[:add].each do |date|
          result = Result.new
          result.date = Date.new(date).to_time
          db.hgetall("#{key.prefix}#{date}").each do |k, v|
            result[k] = v
            col.total.set_or_incr(k, v.to_i)
          end
          col << result
        end
      end
      col
    end
    
    def find_by_magic(options = {})
      @options.merge!(options)
      raise InvalidOptions.new if !valid_options?
      key = Key.new(@options[:scope], @options[:label])
      col = Collection.new(@options)
      col.total = Result.new(@options)
      col << col.total
      build_date_sets.each do |set|
        sum = Result.new
        sum = summarize_add_keys(set[:add], key, sum)
        sum = summarize_rem_keys(set[:rem], key, sum)
        sum.each do |k, v|
          col.total.set_or_incr(k, v.to_i)
        end
      end
      col
    end
    
    def build_date_sets
      Finder::DateSet.new(@options[:from], @options[:till], @options[:depth], @options[:interval])
    end
    
    def build_key
      Key.new(@options[:scope], @options[:label])
    end
    
    def summarize_add_keys(sets, key, sum)
      sets.each do |date|
        db.hgetall("#{key.prefix}#{date}").each do |k, v|
          sum.set_or_incr(k, v.to_i)
        end
      end
      sum
    end
    
    def summarize_rem_keys(sets, key, sum)
      sets.each do |date|
        db.hgetall("#{key.prefix}#{date}").each do |k, v|
          sum.set_or_incr(k, -v.to_i)
        end
      end
      sum
    end
    
    class << self
    
      def find(*args)
        new.find(*args)
      end
      
      def scope(scope)
        new.scope(scope)
      end
      
      def label(label)
        new.label(label)
      end
      
      def dates(from, till)
        new.dates(from, till)
      end
      alias :date :dates
      
      def from(date)
        new.from(date)
      end
      
      def till(date)
        new.till(date)
      end
      alias :untill :till
      
      def depth(unit)
        new.depth(unit)
      end
      
      def interval(unit)
        new.interval(unit)
      end
      
    end
    
    def scope(scope)
      @options[:scope] = scope
      self
    end
    
    def label(label)
      @options[:label] = label
      self
    end
    
    def dates(from, till)
      from(from).till(till)
    end
    alias :date :dates
    
    def from(date)
      @options[:from] = date
      self
    end
    
    def till(date)
      @options[:till] = date
      self
    end
    alias :until :till
    
    def depth(unit)
      @options[:depth] = unit
      self
    end
    
    def interval(unit)
      @options[:interval] = unit
      self
    end
    
  end
end