squeel 0.5.0

data/lib/squeel/visitors/order_visitor.rb

@@ -0,0 +1,107 @@
+require 'squeel/visitors/base'
+require 'squeel/contexts/join_dependency_context'
+
+module Squeel
+  module Visitors
+    class OrderVisitor < Base
+
+      def visit_Hash(o, parent)
+        o.map do |k, v|
+          if implies_context_change?(v)
+            visit_with_context_change(k, v, parent)
+          else
+            visit_without_context_change(k, v, parent)
+          end
+        end.flatten
+      end
+
+      def implies_context_change?(v)
+        Hash === v || can_accept?(v) ||
+        (Array === v && !v.empty? && v.all? {|val| can_accept?(val)})
+      end
+
+      def visit_with_context_change(k, v, parent)
+        parent = case k
+          when Nodes::KeyPath
+            traverse(k, parent, true)
+          else
+            find(k, parent)
+          end
+
+        if Array === v
+          v.map {|val| accept(val, parent || k)}
+        else
+          can_accept?(v) ? accept(v, parent || k) : v
+        end
+      end
+
+      def visit_without_context_change(k, v, parent)
+        v
+      end
+
+      def visit_Array(o, parent)
+        o.map { |v| can_accept?(v) ? accept(v, parent) : v }.flatten
+      end
+
+      def visit_Symbol(o, parent)
+        contextualize(parent)[o]
+      end
+
+      def visit_Squeel_Nodes_Stub(o, parent)
+        contextualize(parent)[o.symbol]
+      end
+
+      def visit_Squeel_Nodes_KeyPath(o, parent)
+        parent = traverse(o, parent)
+
+        accept(o.endpoint, parent)
+      end
+
+      def visit_Squeel_Nodes_Order(o, parent)
+        accept(o.expr, parent).send(o.descending? ? :desc : :asc)
+      end
+
+      def visit_Squeel_Nodes_Function(o, parent)
+        args = o.args.map do |arg|
+          case arg
+          when Nodes::Function, Nodes::KeyPath
+            accept(arg, parent)
+          when Symbol, Nodes::Stub
+            Arel.sql(arel_visitor.accept contextualize(parent)[arg.to_sym])
+          else
+            quoted?(arg) ? Arel.sql(arel_visitor.accept arg) : arg
+          end
+        end
+        Arel::Nodes::NamedFunction.new(o.name, args, o.alias)
+      end
+
+      def visit_Squeel_Nodes_Operation(o, parent)
+        args = o.args.map do |arg|
+          case arg
+          when Nodes::Function
+            accept(arg, parent)
+          when Symbol, Nodes::Stub
+            Arel.sql(arel_visitor.accept contextualize(parent)[arg.to_sym])
+          else
+            quoted?(arg) ? Arel.sql(arel_visitor.accept arg) : arg
+          end
+        end
+
+        op = case o.operator
+        when :+
+          Arel::Nodes::Addition.new(args[0], args[1])
+        when :-
+          Arel::Nodes::Subtraction.new(args[0], args[1])
+        when :*
+          Arel::Nodes::Multiplication.new(args[0], args[1])
+        when :/
+          Arel::Nodes::Division.new(args[0], args[1])
+        else
+          Arel.sql("#{arel_visitor.accept(args[0])} #{o.operator} #{arel_visitor.accept(args[1])}")
+        end
+        o.alias ? op.as(o.alias) : op
+      end
+
+    end
+  end
+end

data/lib/squeel/visitors/predicate_visitor.rb

@@ -0,0 +1,179 @@
+require 'squeel/visitors/base'
+require 'squeel/contexts/join_dependency_context'
+
+module Squeel
+  module Visitors
+    class PredicateVisitor < Base
+
+      def visit_Hash(o, parent)
+        predicates = o.map do |k, v|
+          if implies_context_change?(v)
+            visit_with_context_change(k, v, parent)
+          else
+            visit_without_context_change(k, v, parent)
+          end
+        end
+
+        predicates.flatten!
+
+        if predicates.size > 1
+          Arel::Nodes::Grouping.new(Arel::Nodes::And.new predicates)
+        else
+          predicates.first
+        end
+      end
+
+      def visit_Array(o, parent)
+        if o.first.is_a? String
+          sanitize_sql(o, parent)
+        else
+          o.map { |v| can_accept?(v) ? accept(v, parent) : v }.flatten
+        end
+      end
+
+      def visit_Squeel_Nodes_KeyPath(o, parent)
+        parent = traverse(o, parent)
+
+        accept(o.endpoint, parent)
+      end
+
+      def visit_Squeel_Nodes_Predicate(o, parent)
+        value = o.value
+        case value
+        when Nodes::Function
+          value = accept(value, parent)
+        when Nodes::KeyPath
+          value = can_accept?(value.endpoint) ? accept(value, parent) : contextualize(traverse(value, parent))[value.endpoint.to_sym]
+        end
+        if Nodes::Function === o.expr
+          accept(o.expr, parent).send(o.method_name, value)
+        else
+          contextualize(parent)[o.expr].send(o.method_name, value)
+        end
+      end
+
+      def visit_Squeel_Nodes_Function(o, parent)
+        args = o.args.map do |arg|
+          case arg
+          when Nodes::Function
+            accept(arg, parent)
+          when Nodes::KeyPath
+            can_accept?(arg.endpoint) ? accept(arg, parent) : contextualize(traverse(arg, parent))[arg.endpoint.to_sym]
+          when Symbol, Nodes::Stub
+            Arel.sql(arel_visitor.accept contextualize(parent)[arg.to_sym])
+          else
+            quoted?(arg) ? Arel.sql(arel_visitor.accept arg) : arg
+          end
+        end
+        Arel::Nodes::NamedFunction.new(o.name, args, o.alias)
+      end
+
+      def visit_Squeel_Nodes_Operation(o, parent)
+        args = o.args.map do |arg|
+          case arg
+          when Nodes::Function
+            accept(arg, parent)
+          when Nodes::KeyPath
+            can_accept?(arg.endpoint) ? accept(arg, parent) : contextualize(traverse(arg, parent))[arg.endpoint.to_sym]
+          when Symbol, Nodes::Stub
+            Arel.sql(arel_visitor.accept contextualize(parent)[arg.to_sym])
+          else
+            quoted?(arg) ? Arel.sql(arel_visitor.accept arg) : arg
+          end
+        end
+
+        op = case o.operator
+        when :+
+          Arel::Nodes::Addition.new(args[0], args[1])
+        when :-
+          Arel::Nodes::Subtraction.new(args[0], args[1])
+        when :*
+          Arel::Nodes::Multiplication.new(args[0], args[1])
+        when :/
+          Arel::Nodes::Division.new(args[0], args[1])
+        else
+          Arel::Nodes::InfixOperation(o.operator, args[0], args[1])
+        end
+        o.alias ? op.as(o.alias) : op
+      end
+
+      def visit_Squeel_Nodes_And(o, parent)
+        Arel::Nodes::Grouping.new(Arel::Nodes::And.new(accept(o.children, parent)))
+      end
+
+      def visit_Squeel_Nodes_Or(o, parent)
+        accept(o.left, parent).or(accept(o.right, parent))
+      end
+
+      def visit_Squeel_Nodes_Not(o, parent)
+        accept(o.expr, parent).not
+      end
+
+      def implies_context_change?(v)
+        case v
+        when Hash, Nodes::Predicate, Nodes::Unary, Nodes::Binary, Nodes::Nary
+          true
+        when Nodes::KeyPath
+          can_accept?(v.endpoint)
+        when Array
+          (!v.empty? && v.all? {|val| can_accept?(val)})
+        else
+          false
+        end
+      end
+
+      def visit_with_context_change(k, v, parent)
+        parent = case k
+          when Nodes::KeyPath
+            traverse(k, parent, true)
+          else
+            find(k, parent)
+          end
+
+        case v
+        when Hash, Nodes::KeyPath, Nodes::Predicate, Nodes::Unary, Nodes::Binary, Nodes::Nary
+          accept(v, parent || k)
+        when Array
+          v.map {|val| accept(val, parent || k)}
+        else
+          raise ArgumentError, <<-END
+          Hashes, Predicates, and arrays of visitables as values imply that their
+          corresponding keys are a parent. This didn't work out so well in the case
+          of key = #{k} and value = #{v}"
+          END
+        end
+      end
+
+      def visit_without_context_change(k, v, parent)
+        case v
+        when Nodes::Stub, Symbol
+          v = contextualize(parent)[v.to_sym]
+        when Nodes::KeyPath # If we could accept the endpoint, we wouldn't be here
+          v = contextualize(traverse(v, parent))[v.endpoint.to_sym]
+        end
+
+        case k
+        when Nodes::Predicate
+          accept(k % v, parent)
+        when Nodes::Function
+          arel_predicate_for(accept(k, parent), v, parent)
+        when Nodes::KeyPath
+          accept(k % v, parent)
+        else
+          attribute = contextualize(parent)[k.to_sym]
+          arel_predicate_for(attribute, v, parent)
+        end
+      end
+
+      def arel_predicate_for(attribute, value, parent)
+        if [Array, Range, Arel::SelectManager].include?(value.class)
+          attribute.in(value)
+        else
+          value = can_accept?(value) ? accept(value, parent) : value
+          attribute.eq(value)
+        end
+      end
+
+    end
+  end
+end

data/lib/squeel/visitors/select_visitor.rb

@@ -0,0 +1,103 @@
+require 'squeel/visitors/base'
+require 'squeel/contexts/join_dependency_context'
+
+module Squeel
+  module Visitors
+    class SelectVisitor < Base
+
+      def visit_Hash(o, parent)
+        o.map do |k, v|
+          if implies_context_change?(v)
+            visit_with_context_change(k, v, parent)
+          else
+            visit_without_context_change(k, v, parent)
+          end
+        end.flatten
+      end
+
+      def implies_context_change?(v)
+        Hash === v || can_accept?(v) ||
+        (Array === v && !v.empty? && v.all? {|val| can_accept?(val)})
+      end
+
+      def visit_with_context_change(k, v, parent)
+        parent = case k
+          when Nodes::KeyPath
+            traverse(k, parent, true)
+          else
+            find(k, parent)
+          end
+
+        if Array === v
+          v.map {|val| accept(val, parent || k)}
+        else
+          can_accept?(v) ? accept(v, parent || k) : v
+        end
+      end
+
+      def visit_without_context_change(k, v, parent)
+        v
+      end
+
+      def visit_Array(o, parent)
+        o.map { |v| can_accept?(v) ? accept(v, parent) : v }.flatten
+      end
+
+      def visit_Symbol(o, parent)
+        contextualize(parent)[o]
+      end
+
+      def visit_Squeel_Nodes_Stub(o, parent)
+        contextualize(parent)[o.symbol]
+      end
+
+      def visit_Squeel_Nodes_KeyPath(o, parent)
+        parent = traverse(o, parent)
+
+        accept(o.endpoint, parent)
+      end
+
+      def visit_Squeel_Nodes_Function(o, parent)
+        args = o.args.map do |arg|
+          case arg
+          when Nodes::Function, Nodes::KeyPath
+            accept(arg, parent)
+          when Symbol, Nodes::Stub
+            Arel.sql(arel_visitor.accept contextualize(parent)[arg.to_sym])
+          else
+            quoted?(arg) ? Arel.sql(arel_visitor.accept arg) : arg
+          end
+        end
+        Arel::Nodes::NamedFunction.new(o.name, args, o.alias)
+      end
+
+      def visit_Squeel_Nodes_Operation(o, parent)
+        args = o.args.map do |arg|
+          case arg
+          when Nodes::Function
+            accept(arg, parent)
+          when Symbol, Nodes::Stub
+            Arel.sql(arel_visitor.accept contextualize(parent)[arg.to_sym])
+          else
+            quoted?(arg) ? Arel.sql(arel_visitor.accept arg) : arg
+          end
+        end
+
+        op = case o.operator
+        when :+
+          Arel::Nodes::Addition.new(args[0], args[1])
+        when :-
+          Arel::Nodes::Subtraction.new(args[0], args[1])
+        when :*
+          Arel::Nodes::Multiplication.new(args[0], args[1])
+        when :/
+          Arel::Nodes::Division.new(args[0], args[1])
+        else
+          Arel.sql("#{arel_visitor.accept(args[0])} #{o.operator} #{arel_visitor.accept(args[1])}")
+        end
+        o.alias ? op.as(o.alias) : op
+      end
+
+    end
+  end
+end

data/spec/blueprints/articles.rb

@@ -0,0 +1,5 @@
+Article.blueprint do
+  person
+  title
+  body
+end

data/spec/blueprints/comments.rb

@@ -0,0 +1,5 @@
+Comment.blueprint do
+  article
+  person
+  body
+end

data/spec/blueprints/notes.rb

@@ -0,0 +1,3 @@
+Note.blueprint do
+  note
+end

data/spec/blueprints/people.rb

@@ -0,0 +1,4 @@
+Person.blueprint do
+  name
+  salary
+end

data/spec/blueprints/tags.rb

@@ -0,0 +1,3 @@
+Tag.blueprint do
+  name    { Sham.tag_name }
+end

data/spec/console.rb

@@ -0,0 +1,22 @@
+Bundler.setup
+require 'machinist/active_record'
+require 'sham'
+require 'faker'
+
+Dir[File.expand_path('../../spec/{helpers,support,blueprints}/*.rb', __FILE__)].each do |f|
+  require f
+end
+
+Sham.define do
+  name     { Faker::Name.name }
+  title    { Faker::Lorem.sentence }
+  body     { Faker::Lorem.paragraph }
+  salary   {|index| 30000 + (index * 1000)}
+  tag_name { Faker::Lorem.words(3).join(' ') }
+  note     { Faker::Lorem.words(7).join(' ') }
+end
+
+Schema.create
+
+require 'squeel'
+

data/spec/core_ext/symbol_spec.rb

@@ -0,0 +1,68 @@
+require 'spec_helper'
+
+describe Symbol do
+
+  Squeel::Constants::PREDICATES.each do |method_name|
+    it "creates #{method_name} predicates with no value" do
+      predicate = :attribute.send(method_name)
+      predicate.expr.should eq :attribute
+      predicate.method_name.should eq method_name
+      predicate.value?.should be_false
+    end
+
+    it "creates #{method_name} predicates with a value" do
+      predicate = :attribute.send(method_name, 'value')
+      predicate.expr.should eq :attribute
+      predicate.method_name.should eq method_name
+      predicate.value.should eq 'value'
+    end
+  end
+
+  Squeel::Constants::PREDICATE_ALIASES.each do |method_name, aliases|
+    aliases.each do |aliaz|
+      ['', '_any', '_all'].each do |suffix|
+        it "creates #{method_name.to_s + suffix} predicates with no value using the alias #{aliaz.to_s + suffix}" do
+          predicate = :attribute.send(aliaz.to_s + suffix)
+          predicate.expr.should eq :attribute
+          predicate.method_name.should eq "#{method_name}#{suffix}".to_sym
+          predicate.value?.should be_false
+        end
+
+        it "creates #{method_name.to_s + suffix} predicates with a value using the alias #{aliaz.to_s + suffix}" do
+          predicate = :attribute.send((aliaz.to_s + suffix), 'value')
+          predicate.expr.should eq :attribute
+          predicate.method_name.should eq "#{method_name}#{suffix}".to_sym
+          predicate.value.should eq 'value'
+        end
+      end
+    end
+  end
+
+  it 'creates ascending orders' do
+    order = :attribute.asc
+    order.should be_ascending
+  end
+
+  it 'creates descending orders' do
+    order = :attribute.desc
+    order.should be_descending
+  end
+
+  it 'creates functions' do
+    function = :function.func
+    function.should be_a Squeel::Nodes::Function
+  end
+
+  it 'creates inner joins' do
+    join = :join.inner
+    join.should be_a Squeel::Nodes::Join
+    join.type.should eq Arel::InnerJoin
+  end
+
+  it 'creates outer joins' do
+    join = :join.outer
+    join.should be_a Squeel::Nodes::Join
+    join.type.should eq Arel::OuterJoin
+  end
+
+end