maksar-meta_where 1.0.4

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.foreign_key).to_sym => values.size == 1 ? values.first.id : values.map(&:id),
+                reflection.foreign_type.to_sym => klass.name
+              }
+            else
+              {(reflection.options[:foreign_key] || reflection.foreign_key).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,52 @@
+require 'meta_where/utility'
+
+module MetaWhere
+  module Visitors
+    class Visitor
+      include MetaWhere::Utility
+
+      attr_reader :tables
+      attr_reader :join_dependency
+
+      def initialize
+        @tables = Hash.new {|hash, key| hash[key] = get_table(key)}
+      end
+
+      def join_dependency=(jd)
+        @join_dependency = jd
+        jb = jd.join_base
+        @engine = jb.arel_engine
+        @default_table = Arel::Table.new(jb.table_name, :as => jb.aliased_table_name, :engine => @engine)
+        self
+      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/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        = "maksar-meta_where"
+  s.version     = MetaWhere::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Ernie Miller"]
+  s.email       = ["ernie@metautonomo.us"]
+  s.homepage    = "http://metautonomo.us/projects/metawhere"
+  s.summary     = %q{ActiveRecord 3 query syntax on steroids.}
+  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.
+    }
+
+  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.1.0.rc1'
+  s.add_dependency 'activesupport', '~> 3.1.0.rc1'
+  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,5 @@
+class Developer < ActiveRecord::Base
+  belongs_to :company
+  has_and_belongs_to_many :projects
+  has_many :notes, :as => :notable
+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.