sayso-meta_where 1.0.4.001

data/lib/meta_where/utility.rb

@@ -0,0 +1,51 @@
+module MetaWhere
+  module Utility
+    private
+
+    def array_of_activerecords(val)
+      val.is_a?(Array) && !val.empty? && val.all? {|v| v.is_a?(ActiveRecord::Base)}
+    end
+
+    def association_from_parent_and_column(parent, column)
+      parent.is_a?(Symbol) ? nil : @join_dependency.send(:find_join_association, column, parent)
+    end
+
+    def attribute_from_column_and_table(column, table)
+      case column
+      when String, Symbol
+        table[column]
+      when MetaWhere::Function
+        column.table = table
+        column.to_sqlliteral
+      else
+        nil
+      end
+    end
+
+    def args_for_predicate(value)
+      case value
+      when Array, ActiveRecord::Associations::AssociationCollection, ActiveRecord::Relation
+        value.to_a.map { |x|
+          x.respond_to?(:quoted_id) ? x.quoted_id : x
+        }
+      when ActiveRecord::Base
+        value.quoted_id
+      else
+        value
+      end
+    end
+
+    def method_from_value(value)
+      case value
+      when Array, Range, ActiveRecord::Associations::AssociationCollection, ActiveRecord::Relation, Arel::Relation
+        :in
+      else
+        :eq
+      end
+    end
+
+    def valid_comparison_method?(method)
+      MetaWhere::PREDICATES.map(&:to_s).include?(method.to_s)
+    end
+  end
+end

data/lib/meta_where/version.rb

@@ -0,0 +1,3 @@
+module MetaWhere
+  VERSION = "1.0.4.001"
+end

data/lib/meta_where/visitors/attribute.rb

@@ -0,0 +1,58 @@
+require 'meta_where/visitors/visitor'
+
+module MetaWhere
+  module Visitors
+    class Attribute < Visitor
+
+      def self.visitables
+        [Hash, Symbol, MetaWhere::Column]
+      end
+
+      def visit_Hash(o, parent)
+        parent = parent.name if parent.is_a? MetaWhere::JoinType
+        table = tables[parent]
+        built_attributes = o.map do |column, value|
+          if value.is_a?(Hash)
+            association = association_from_parent_and_column(parent, column)
+            accept(value, association || column)
+          elsif value.is_a?(Array) && value.all? {|v| can_accept?(v)}
+            association = association_from_parent_and_column(parent, column)
+            value.map {|val| self.accept(val, association || column)}
+          else
+            association = association_from_parent_and_column(parent, column)
+            can_accept?(value) ? self.accept(value, association || column) : value
+          end
+        end
+
+        built_attributes.flatten
+      end
+
+      def visit_Symbol(o, parent)
+        table = tables[parent]
+
+        unless attribute = table[o]
+          raise ::ActiveRecord::StatementInvalid, "No attribute named `#{o}` exists for table `#{table.name}`"
+        end
+
+        attribute
+      end
+
+      def visit_MetaWhere_Column(o, parent)
+        column_name = o.column.to_s
+        if column_name.include?('.')
+          table_name, column_name = column_name.split('.', 2)
+          table = Arel::Table.new(table_name, :engine => parent.arel_engine)
+        else
+          table = tables[parent]
+        end
+
+        unless attribute = table[column_name]
+          raise ::ActiveRecord::StatementInvalid, "No attribute named `#{column_name}` exists for table `#{table.name}`"
+        end
+
+        attribute.send(o.method)
+      end
+
+    end
+  end
+end

data/lib/meta_where/visitors/predicate.rb

@@ -0,0 +1,149 @@
+require 'meta_where/visitors/visitor'
+
+module MetaWhere
+  module Visitors
+    class Predicate < Visitor
+
+      def self.visitables
+        [Hash, Array, MetaWhere::Or, MetaWhere::And, MetaWhere::Not, MetaWhere::Condition, MetaWhere::Function]
+      end
+
+      def visit_Hash(o, parent)
+        parent ||= join_dependency.join_base
+        parent = parent.name if parent.is_a? MetaWhere::JoinType
+        table = tables[parent]
+        predicates = o.map do |column, value|
+          if value.is_a?(Hash)
+            association = association_from_parent_and_column(parent, column)
+            accept(value, association || column)
+          elsif [MetaWhere::Condition, MetaWhere::And, MetaWhere::Or, MetaWhere::Not].include?(value.class)
+            association = association_from_parent_and_column(parent, column)
+            accept(value, association || column)
+          elsif value.is_a?(Array) && !value.empty? && value.all? {|v| can_accept?(v)}
+            association = association_from_parent_and_column(parent, column)
+            value.map {|val| accept(val, association || column)}
+          elsif (value.is_a?(ActiveRecord::Base) || array_of_activerecords(value)) &&
+              reflection = parent.active_record.reflect_on_association(column.is_a?(MetaWhere::JoinType) ? column.name : column)
+            accept_activerecord_values(column, value, parent, reflection)
+          else
+            if column.is_a?(MetaWhere::Column)
+              method = column.method
+              column = column.column
+            else
+              method = method_from_value(value)
+            end
+
+            if [String, Symbol].include?(column.class) && column.to_s.include?('.')
+              table_name, column = column.to_s.split('.', 2)
+              table = Arel::Table.new(table_name, :engine => parent.arel_engine)
+            end
+
+            unless valid_comparison_method?(method)
+              raise ::ActiveRecord::StatementInvalid, "No comparison method named `#{method}` exists for column `#{column}`"
+            end
+
+            if attribute = attribute_from_column_and_table(column, table)
+              attribute.send(method, args_for_predicate(value))
+            else
+              raise ::ActiveRecord::StatementInvalid, "No attribute named `#{column}` exists for table `#{table.name}`"
+            end
+
+          end
+        end
+
+        predicates.flatten!
+
+        if predicates.size > 1
+          first = predicates.shift
+          Arel::Nodes::Grouping.new(predicates.inject(first) {|memo, expr| Arel::Nodes::And.new(memo, expr)})
+        else
+          predicates.first
+        end
+      end
+
+      def visit_Array(o, parent)
+        if o.first.is_a? String
+          join_dependency.join_base.send(:sanitize_sql, o)
+        else
+          o.map {|e| accept(e, parent)}.flatten
+        end
+      end
+
+      def visit_MetaWhere_Or(o, parent)
+        accept(o.condition1, parent).or(accept(o.condition2, parent))
+      end
+
+      def visit_MetaWhere_And(o, parent)
+        accept(o.condition1, parent).and(accept(o.condition2, parent))
+      end
+
+      def visit_MetaWhere_Not(o, parent)
+        accept(o.expr, parent).not
+      end
+
+      def visit_MetaWhere_Condition(o, parent)
+        table = tables[parent]
+
+        unless attribute = attribute_from_column_and_table(o.column, table)
+          raise ::ActiveRecord::StatementInvalid, "No attribute named `#{o.column}` exists for table `#{table.name}`"
+        end
+
+        unless valid_comparison_method?(o.method)
+          raise ::ActiveRecord::StatementInvalid, "No comparison method named `#{o.method}` exists for column `#{o.column}`"
+        end
+        attribute.send(o.method, args_for_predicate(o.value))
+      end
+
+      def visit_MetaWhere_Function(o, parent)
+        self.table = tables[parent]
+
+        o.to_sqlliteral
+      end
+
+      private
+
+      def accept_activerecord_values(column, value, parent, reflection)
+        groups = Array.wrap(value).group_by {|v| v.class.base_class}
+        unless reflection.options[:polymorphic] || groups.keys.all? {|k| reflection.klass == k}
+          raise ArgumentError, "An object you supplied to :#{reflection.name} is not a #{reflection.klass}!"
+        end
+        case reflection.macro
+        when :has_many, :has_one, :has_and_belongs_to_many
+          conditions = nil
+          groups.each do |klass, values|
+            condition = {
+              (reflection.options[:foreign_key] || reflection.klass.primary_key).to_sym => values.size == 1 ? values.first.id : values.map(&:id)
+            }
+            conditions = conditions ? conditions | condition : condition
+          end
+
+          accept(conditions, association_from_parent_and_column(parent, column) || column)
+        when :belongs_to
+          conditions = nil
+          groups.each do |klass, values|
+            condition = if reflection.options[:polymorphic]
+              {
+                (reflection.options[:foreign_key] || reflection.primary_key_name).to_sym => values.size == 1 ? values.first.id : values.map(&:id),
+                reflection.options[:foreign_type].to_sym => klass.name
+              }
+            else
+              {(reflection.options[:foreign_key] || reflection.primary_key_name).to_sym => values.size == 1 ? values.first.id : values.map(&:id)}
+            end
+            conditions = conditions ? conditions | condition : condition
+          end
+
+          accept(conditions, parent)
+        end
+      end
+
+      def sanitize_or_accept_reflection_conditions(reflection, parent, column)
+        if !can_accept?(reflection.options[:conditions])
+          reflection.sanitized_conditions
+        else
+          accept(reflection.options[:conditions], association_from_parent_and_column(parent, column) || column)
+        end
+      end
+
+    end
+  end
+end

data/lib/meta_where/visitors/visitor.rb

@@ -0,0 +1,47 @@
+require 'meta_where/utility'
+
+module MetaWhere
+  module Visitors
+    class Visitor
+      include MetaWhere::Utility
+
+      attr_reader :join_dependency, :tables
+
+      def initialize(join_dependency)
+        @join_dependency = join_dependency
+        jb = join_dependency.join_base
+        @engine = jb.arel_engine
+        @default_table = Arel::Table.new(jb.table_name, :as => jb.aliased_table_name, :engine => @engine)
+        @tables = Hash.new {|hash, key| hash[key] = get_table(key)}
+      end
+
+      def get_table(parent_or_table_name = nil)
+        if parent_or_table_name.is_a?(Symbol)
+          Arel::Table.new(parent_or_table_name, :engine => @engine)
+        elsif parent_or_table_name.respond_to?(:aliased_table_name)
+          Arel::Table.new(parent_or_table_name.table_name, :as => parent_or_table_name.aliased_table_name, :engine => @engine)
+        else
+          @default_table
+        end
+      end
+
+      def accept(object, parent = nil)
+        visit(object, parent)
+      end
+
+      def can_accept?(object)
+        respond_to? DISPATCH[object.class]
+      end
+
+      private
+
+      DISPATCH = Hash.new do |hash, klass|
+        hash[klass] = "visit_#{klass.name.gsub('::', '_')}"
+      end
+
+      def visit(object, parent)
+        send(DISPATCH[object.class], object, parent)
+      end
+    end
+  end
+end

data/lib/meta_where.rb

@@ -0,0 +1,51 @@
+module MetaWhere
+  METHOD_ALIASES = {
+    'ne' => :not_eq,
+    'like' => :matches,
+    'not_matches' => :does_not_match,
+    'nlike' => :does_not_match,
+    'lte' => :lteq,
+    'gte' => :gteq,
+    'nin' => :not_in
+  }
+
+  PREDICATES = [
+    :eq, :eq_any, :eq_all,
+    :not_eq, :not_eq_any, :not_eq_all,
+    :matches, :matches_any, :matches_all,
+    :does_not_match, :does_not_match_any, :does_not_match_all,
+    :lt, :lt_any, :lt_all,
+    :lteq, :lteq_any, :lteq_all,
+    :gt, :gt_any, :gt_all,
+    :gteq, :gteq_any, :gteq_all,
+    :in, :in_any, :in_all,
+    :not_in, :not_in_any, :not_in_all
+  ]
+
+  def self.operator_overload!
+    require 'core_ext/symbol_operators'
+  end
+end
+
+require 'arel'
+require 'active_record'
+require 'active_support'
+
+# TODO: Trim this down, as the individual files should be
+# handling their requires.
+require 'meta_where/column'
+require 'meta_where/condition'
+require 'meta_where/not'
+require 'meta_where/compound'
+require 'meta_where/function'
+require 'meta_where/join_type'
+require 'core_ext/symbol'
+require 'core_ext/hash'
+require 'meta_where/visitors/attribute'
+require 'meta_where/visitors/predicate'
+require 'meta_where/association_reflection'
+require 'meta_where/relation'
+require 'meta_where/join_dependency'
+ActiveRecord::Relation.send(:include, MetaWhere::Relation)
+ActiveRecord::Reflection::AssociationReflection.send(:include, MetaWhere::AssociationReflection)
+ActiveRecord::Associations::ClassMethods::JoinDependency.send(:include, MetaWhere::JoinDependency)

data/meta_where.gemspec

@@ -0,0 +1,48 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "meta_where/version"
+
+Gem::Specification.new do |s|
+  s.name        = "sayso-meta_where"
+  s.version     = MetaWhere::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["SaySo"]
+  s.email       = ["sayso@truvolabs.com"]
+  s.homepage    = "https://github.com/sayso/meta_where"
+  s.summary     = %q{ActiveRecord 3 query syntax on steroids - forked and gemified for sayso}
+  s.description = %q{
+      MetaWhere offers the ability to call any Arel predicate methods
+      (with a few convenient aliases) on your Model's attributes instead
+      of the ones normally offered by ActiveRecord's hash parameters. It also
+      adds convenient syntax for order clauses, smarter mapping of nested hash
+      conditions, and a debug_sql method to see the real SQL your code is
+      generating without running it against the database. If you like the new
+      AR 3.0 query interface, you'll love it with MetaWhere. - forked and gemified for sayso
+    }
+  s.post_install_message = %q{
+*** Thanks for installing MetaWhere! ***
+Be sure to check out http://metautonomo.us/projects/metawhere/ for a
+walkthrough of MetaWhere's features, and click the donate button if
+you're feeling especially appreciative. It'd help me justify this
+"open source" stuff to my lovely wife. :)
+
+}
+
+  s.extra_rdoc_files = [
+    "LICENSE",
+    "README.rdoc"
+  ]
+
+  s.rubyforge_project = "meta_where"
+
+  s.add_dependency 'activerecord', '~> 3.0.0'
+  s.add_dependency 'activesupport', '~> 3.0.0'
+  s.add_dependency 'arel', '~> 2.0.7'
+  s.add_development_dependency 'shoulda'
+  s.add_development_dependency 'sqlite3', '~> 1.3.3'
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+end

data/test/fixtures/companies.yml

@@ -0,0 +1,17 @@
+initech:
+  name      : Initech
+  id        : 1
+  created_at: 1999-02-19 08:00
+  updated_at: 1999-02-19 08:00
+
+aos:
+  name: Advanced Optical Solutions
+  id  : 2
+  created_at: 2004-02-01 08:00
+  updated_at: 2004-02-01 08:00
+
+mission_data:
+  name: Mission Data
+  id  : 3
+  created_at: 1996-09-21 08:00
+  updated_at: 1996-09-21 08:00

data/test/fixtures/company.rb

@@ -0,0 +1,7 @@
+class Company < ActiveRecord::Base
+  has_many :developers
+  has_many :developer_notes, :through => :developers, :source => :notes
+  has_many :slackers, :class_name => "Developer", :conditions => {:slacker => true}
+  has_many :notes, :as => :notable
+  has_many :data_types
+end

data/test/fixtures/data_type.rb

@@ -0,0 +1,3 @@
+class DataType < ActiveRecord::Base
+  belongs_to :company
+end

data/test/fixtures/data_types.yml

@@ -0,0 +1,15 @@
+<% 1.upto(9) do |n| %>
+dt_<%= n %>:
+  company_id: <%= n % 3 + 1 %>
+  str       : This string has <%= n %> exclamation points<%= '!' * n %>
+  txt       : <%= 'This is some text that may or may not repeat based on the value of n.' * n %>
+  int       : <%= n ** 3 %>
+  flt       : <%= n.to_f / 2.0 %>
+  dec       : <%= n.to_f ** (n + 0.1) %>
+  dtm       : <%= (Time.utc(2009, 12, 24) + 86400 * n).to_s(:db) %>
+  tms       : <%= (Time.utc(2009, 12, 24) + 86400 * n).to_s(:db) %>
+  tim       : <%= Time.utc(2000, 01, 01, n+8, n).to_s(:db) %>
+  dat       : <%= (Date.new(2009, 12, 24) + n).strftime("%Y-%m-%d") %>
+  bin       : <%= "BLOB#{n}" * n %>
+  bln       : <%= n % 2 > 0 ? true : false %>
+<% end %>

data/test/fixtures/developer.rb

@@ -0,0 +1,7 @@
+class Developer < ActiveRecord::Base
+  belongs_to :company
+  has_and_belongs_to_many :projects
+  has_many :notes, :as => :notable
+
+  scope :new_dev, where(:name => 'New Developer')
+end

data/test/fixtures/developers.yml

@@ -0,0 +1,55 @@
+peter:
+  id        : 1
+  company_id: 1
+  name      : Peter Gibbons
+  salary    : 100000
+  slacker   : true
+
+michael:
+  id        : 2
+  company_id: 1
+  name      : Michael Bolton
+  salary    : 70000
+  slacker   : false
+
+samir:
+  id        : 3
+  company_id: 1
+  name      : Samir Nagheenanajar
+  salary    : 65000
+  slacker   : false
+
+herb:
+  id        : 4
+  company_id: 2
+  name      : Herb Myers
+  salary    : 50000
+  slacker   : false
+
+dude:
+  id        : 5
+  company_id: 2
+  name      : Some Dude
+  salary    : 84000
+  slacker   : true
+
+ernie:
+  id        : 6
+  company_id: 3
+  name      : Ernie Miller
+  salary    : 45000
+  slacker   : true
+
+someone:
+  id        : 7
+  company_id: 3
+  name      : Someone Else
+  salary    : 70000
+  slacker   : true
+
+another:
+  id        : 8
+  company_id: 3
+  name      : Another Guy
+  salary    : 80000
+  slacker   : false

data/test/fixtures/developers_projects.yml

@@ -0,0 +1,25 @@
+<% 1.upto(3) do |d| %>
+y2k_<%= d %>:
+  developer_id: <%= d %>
+  project_id  : 1
+<% end %>
+
+virus:
+  developer_id: 2
+  project_id  : 2
+  
+<% 1.upto(8) do |d| %>
+awesome_<%= d %>:
+  developer_id: <%= d %>
+  project_id  : 3
+<% end %>
+
+metasearch:
+  developer_id: 6
+  project_id  : 4
+
+<% 4.upto(8) do |d| %>
+another_<%= d %>:
+  developer_id: <%= d %>
+  project_id  : 5
+<% end %>

data/test/fixtures/fixed_bid_project.rb

@@ -0,0 +1,2 @@
+class FixedBidProject < Project
+end

data/test/fixtures/invalid_company.rb

@@ -0,0 +1,4 @@
+class InvalidCompany < ActiveRecord::Base
+  set_table_name 'companies'
+  has_many :developers, :conditions => {:name.matches => '%Miller'}
+end

data/test/fixtures/invalid_developer.rb

@@ -0,0 +1,4 @@
+class InvalidDeveloper < ActiveRecord::Base
+  set_table_name 'developers'
+  has_many :notes, :as => :notable, :conditions => [:note.eq % 'GIPE']
+end

data/test/fixtures/note.rb

@@ -0,0 +1,3 @@
+class Note < ActiveRecord::Base
+  belongs_to :notable, :polymorphic => true, :conditions => '1=1'
+end

data/test/fixtures/notes.yml

@@ -0,0 +1,95 @@
+peter:
+  id          : 1
+  notable_type: Developer
+  notable_id  : 1
+  note        : A straight shooter with upper management written all over him.
+
+michael:
+  id          : 2
+  notable_type: Developer
+  notable_id  : 2
+  note        : Doesn't like the singer of the same name. The nerve!
+
+samir:
+  id          : 3
+  notable_type: Developer
+  notable_id  : 3
+  note        : Naga.... Naga..... Not gonna work here anymore anyway.
+
+herb:
+  id          : 4
+  notable_type: Developer
+  notable_id  : 4
+  note        : Will show you what he's doing.
+
+dude:
+  id          : 5
+  notable_type: Developer
+  notable_id  : 5
+  note        : Nothing of note.
+
+ernie:
+  id          : 6
+  notable_type: Developer
+  notable_id  : 6
+  note        : Complete slacker. Should probably be fired.
+
+someone:
+  id          : 7
+  notable_type: Developer
+  notable_id  : 7
+  note        : Just another developer.
+
+another:
+  id          : 8
+  notable_type: Developer
+  notable_id  : 8
+  note        : Placing a note in this guy's file for insubordination.
+
+initech:
+  id          : 9
+  notable_type: Company
+  notable_id  : 1
+  note        : Innovation + Technology!
+
+aos:
+  id          : 10
+  notable_type: Company
+  notable_id  : 2
+  note        : Advanced solutions of an optical nature.
+
+mission_data:
+  id          : 11
+  notable_type: Company
+  notable_id  : 3
+  note        : Best design + development shop in the 'ville.
+
+y2k:
+  id          : 12
+  notable_type: Project
+  notable_id  : 1
+  note        : It may have already passed but that's no excuse to be unprepared!
+
+virus:
+  id          : 13
+  notable_type: Project
+  notable_id  : 2
+  note        : It could bring the company to its knees.
+
+awesome:
+  id          : 14
+  notable_type: Project
+  notable_id  : 3
+  note        : This note is AWESOME!!!
+
+metasearch:
+  id          : 15
+  notable_type: Project
+  notable_id  : 4
+  note        : A complete waste of the developer's time.
+
+another:
+  id          : 16
+  notable_type: Project
+  notable_id  : 5
+  note        : This is another project note.

data/test/fixtures/people.yml

@@ -0,0 +1,14 @@
+grandpa:
+  name     : Abraham
+  id       : 1
+  parent_id: nil
+
+father:
+  name     : Isaac
+  id       : 2
+  parent_id: 1
+
+son:
+  name     : Jacob
+  id       : 3
+  parent_id: 2

data/test/fixtures/person.rb

@@ -0,0 +1,4 @@
+class Person < ActiveRecord::Base
+  belongs_to :parent, :class_name => 'Person', :foreign_key => :parent_id
+  has_many   :children, :class_name => 'Person', :foreign_key => :parent_id
+end