Contents

require 'gitter/drivers/abstract_driver'
module Gitter
  
  class ActiveRecordDriver < AbstractDriver
    
    delegate :count, to: :scope
    
    def order attr, desc = nil
      what = case desc
        when true, 'true' then "#{attr} DESC"
        when false, 'false' then attr
        when String then desc
        else attr 
      end
      new scope.except(:order).order(what.to_s)
    end
   
    def unordered
      new scope.except(:order)
    end

    def group arg
      new scope.group(arg)
    end

    def where attr_values, opts = {}
      exact = opts.fetch(:exact){true}
      ignore_case = opts.fetch(:ignore_case){false}
      find_format = opts.fetch(:find_format){nil}
      strip_blank = opts.fetch(:strip_blank){false}

      # has range?
      return new scope.where(attr_values) if Range === attr_values.values.first

      tokens = {}
      token_i = 0

      conditions = attr_values.map do |attr,value| 
        raise ArgumentError, "invalid range #{value} for #{attr}" if Range === value
        value = value.strip if strip_blank
        text = exact ? value : "%#{value}%"
        col, token = attr, ":q#{token_i}"
        col, token = upper(col), upper(token) if ignore_case
        col = find_format.call(col) if find_format
        tokens[:"q#{token_i}"] = text 
        token_i += 1
        "#{col} #{exact ? '=' : 'LIKE'} #{token}"
      end

      new scope.where("( #{conditions * ') OR ('} )", tokens)
    end

    def greater_or_equal attr, value
      new scope.where("#{attr} >= ?", value)
    end

    def less_or_equal attr, value
      new scope.where("#{attr} <= ?", value)
    end

    def each &block
      new scope.each(&block)
    end
    
    def named_scope name
      new scope.send(name)
    end
    
    def distinct_values attr
      attribute = attr.to_s.split(/\./).last || attr
      scope.select(attr).uniq.map(&:"#{attribute}").uniq
    end

    def to_s
      scope.to_sql
    end

    private 

    def upper(text)
      "upper(#{text})"
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
gitter-1.1.3	lib/gitter/drivers/active_record_driver.rb