arel-compat 0.4.0

data/lib/arel/algebra/relations.rb

@@ -0,0 +1,17 @@
+require 'arel/algebra/relations/relation'
+require 'arel/algebra/relations/utilities/compound'
+require 'arel/algebra/relations/utilities/nil'
+require 'arel/algebra/relations/utilities/externalization'
+require 'arel/algebra/relations/row'
+require 'arel/algebra/relations/writes'
+require 'arel/algebra/relations/operations/alias'
+require 'arel/algebra/relations/operations/from'
+require 'arel/algebra/relations/operations/group'
+require 'arel/algebra/relations/operations/having'
+require 'arel/algebra/relations/operations/join'
+require 'arel/algebra/relations/operations/order'
+require 'arel/algebra/relations/operations/project'
+require 'arel/algebra/relations/operations/where'
+require 'arel/algebra/relations/operations/skip'
+require 'arel/algebra/relations/operations/take'
+require 'arel/algebra/relations/operations/lock'

data/lib/arel/algebra/relations/operations/alias.rb

@@ -0,0 +1,7 @@
+module Arel
+  class Alias < Compound
+    attributes :relation
+    deriving :initialize
+    alias_method :==, :equal?
+  end
+end

data/lib/arel/algebra/relations/operations/from.rb

@@ -0,0 +1,6 @@
+module Arel
+  class From < Compound
+    attributes :relation, :sources
+    deriving :initialize, :==
+  end
+end

data/lib/arel/algebra/relations/operations/group.rb

@@ -0,0 +1,12 @@
+module Arel
+  class Group < Compound
+    attributes :relation, :groupings
+    deriving :==
+
+    def initialize(relation, *groupings, &block)
+      @relation = relation
+      @groupings = (groupings + arguments_from_block(relation, &block)) \
+        .collect { |g| g.bind(relation) }
+    end
+  end
+end

data/lib/arel/algebra/relations/operations/having.rb

@@ -0,0 +1,17 @@
+module Arel
+  class Having < Compound
+    attributes :relation, :predicates
+    deriving   :==
+    requires   :restricting
+
+    def initialize(relation, *predicates, &block)
+      predicates = [yield(relation)] + predicates if block_given?
+      @predicates = predicates.map { |p| p.bind(relation) }
+      @relation   = relation
+    end
+
+    def havings
+      @havings ||= relation.havings + predicates
+    end
+  end
+end

data/lib/arel/algebra/relations/operations/join.rb

@@ -0,0 +1,69 @@
+module Arel
+  class Join
+    include Relation
+
+    attributes :relation1, :relation2, :predicates
+    deriving :==
+    delegate :name, :to => :relation1
+
+    def initialize(relation1, relation2 = Nil.instance, *predicates)
+      @relation1, @relation2, @predicates = relation1, relation2, predicates
+    end
+
+    def hash
+      @hash ||= :relation1.hash
+    end
+
+    def eql?(other)
+      self == other
+    end
+
+    def attributes
+      @attributes ||= (relation1.externalize.attributes | relation2.externalize.attributes).bind(self)
+    end
+
+    def wheres
+      # TESTME bind to self?
+      relation1.externalize.wheres
+    end
+
+    def ons
+      @ons ||= @predicates.collect { |p| p.bind(self) }
+    end
+
+    # TESTME
+    def externalizable?
+      relation1.externalizable? or relation2.externalizable?
+    end
+
+    def join?
+      true
+    end
+
+    def engine
+      relation1.engine != relation2.engine ? Memory::Engine.new : relation1.engine
+    end
+  end
+
+  class InnerJoin  < Join; end
+  class OuterJoin  < Join; end
+  class StringJoin < Join
+    def externalizable?
+      relation1.externalizable?
+    end
+
+    def attributes
+      relation1.externalize.attributes
+    end
+
+    def engine
+      relation1.engine
+    end
+  end
+
+  module Relation
+    def join?
+      false
+    end
+  end
+end

data/lib/arel/algebra/relations/operations/lock.rb

@@ -0,0 +1,12 @@
+module Arel
+  class Lock < Compound
+    attributes :relation, :locked
+    deriving :initialize, :==
+
+    def initialize(relation, locked, &block)
+      @relation = relation
+      @locked   = locked.blank? ? " FOR UPDATE" : locked
+    end
+  end
+end
+

data/lib/arel/algebra/relations/operations/order.rb

@@ -0,0 +1,19 @@
+module Arel
+  class Order < Compound
+    attributes :relation, :orderings
+    deriving   :==
+    requires   :ordering
+
+    def initialize(relation, *orderings, &block)
+      @relation = relation
+      @orderings = (orderings + arguments_from_block(relation, &block)) \
+        .collect { |o| o.bind(relation) }
+    end
+
+    # TESTME
+    def orders
+      # QUESTION - do we still need relation.orders ?
+      (orderings + relation.orders).collect { |o| o.bind(self) }.collect { |o| o.to_ordering }
+    end
+  end
+end

data/lib/arel/algebra/relations/operations/project.rb

@@ -0,0 +1,20 @@
+module Arel
+  class Project < Compound
+    attributes :relation, :projections
+    deriving :==
+
+    def initialize(relation, *projections, &block)
+      @relation = relation
+      @projections = (projections + arguments_from_block(relation, &block)) \
+        .collect { |p| p.bind(relation) }
+    end
+
+    def attributes
+      @attributes ||= Header.new(projections).bind(self)
+    end
+
+    def externalizable?
+      attributes.any? { |a| a.respond_to?(:aggregation?) && a.aggregation? } || relation.externalizable?
+    end
+  end
+end

data/lib/arel/algebra/relations/operations/skip.rb

@@ -0,0 +1,7 @@
+module Arel
+  class Skip < Compound
+    attributes :relation, :skipped
+    deriving   :initialize, :==
+    requires   :skipping
+  end
+end

data/lib/arel/algebra/relations/operations/take.rb

@@ -0,0 +1,11 @@
+module Arel
+  class Take < Compound
+    attributes :relation, :taken
+    deriving   :initialize, :==
+    requires   :limiting
+
+    def externalizable?
+      true
+    end
+  end
+end

data/lib/arel/algebra/relations/operations/where.rb

@@ -0,0 +1,17 @@
+module Arel
+  class Where < Compound
+    attributes :relation, :predicates
+    deriving   :==
+    requires   :restricting
+
+    def initialize(relation, *predicates, &block)
+      predicates = [yield(relation)] + predicates if block_given?
+      @predicates = predicates.map { |p| p.bind(relation) }
+      @relation   = relation
+    end
+
+    def wheres
+      @wheres ||= relation.wheres + predicates
+    end
+  end
+end

data/lib/arel/algebra/relations/relation.rb

@@ -0,0 +1,136 @@
+module Arel
+  module Relation
+    attr_reader :count
+
+    def session
+      Session.new
+    end
+
+    def call
+      engine.read(self)
+    end
+
+    def bind(relation)
+      self
+    end
+
+    module Enumerable
+      include ::Enumerable
+
+      def each(&block)
+        session.read(self).each(&block)
+      end
+
+      def first
+        session.read(self).first
+      end
+    end
+    include Enumerable
+
+    module Operable
+      def join(other_relation = nil, join_class = InnerJoin)
+        case other_relation
+        when String
+          StringJoin.new(self, other_relation)
+        when Relation
+          JoinOperation.new(join_class, self, other_relation)
+        else
+          self
+        end
+      end
+
+      def outer_join(other_relation = nil)
+        join(other_relation, OuterJoin)
+      end
+
+      [:where, :project, :order, :take, :skip, :group, :from, :having].each do |operation_name|
+        class_eval <<-OPERATION, __FILE__, __LINE__
+          def #{operation_name}(*arguments, &block)
+            arguments.all?(&:blank?) && !block_given?? self : #{operation_name.to_s.classify}.new(self, *arguments, &block)
+          end
+        OPERATION
+      end
+
+      def lock(locking = nil)
+        Lock.new(self, locking)
+      end
+
+      def alias
+        Alias.new(self)
+      end
+
+      module Writable
+        def insert(record)
+          session.create Insert.new(self, record)
+        end
+
+        def update(assignments)
+          session.update Update.new(self, assignments)
+        end
+
+        def delete
+          session.delete Deletion.new(self)
+        end
+      end
+      include Writable
+
+      JoinOperation = Struct.new(:join_class, :relation1, :relation2) do
+        def on(*predicates)
+          join_class.new(relation1, relation2, *predicates)
+        end
+      end
+    end
+    include Operable
+
+    module AttributeAccessable
+      def [](index)
+        attributes[index]
+      end
+
+      def find_attribute_matching_name(name)
+        attributes.detect { |a| a.named?(name) } || Attribute.new(self, name)
+      end
+
+      def find_attribute_matching_attribute(attribute)
+        matching_attributes(attribute).max do |a1, a2|
+          (a1.original_attribute / attribute) <=> (a2.original_attribute / attribute)
+        end
+      end
+
+      def position_of(attribute)
+        (@position_of ||= Hash.new do |h, attribute|
+          h[attribute] = attributes.index(self[attribute])
+        end)[attribute]
+      end
+
+      private
+      def matching_attributes(attribute)
+        (@matching_attributes ||= attributes.inject({}) do |hash, a|
+          (hash[a.is_a?(Value) ? a.value : a.root] ||= []) << a
+          hash
+        end)[attribute.root] || []
+      end
+
+      def has_attribute?(attribute)
+        !matching_attributes(attribute).empty?
+      end
+    end
+    include AttributeAccessable
+
+    module DefaultOperations
+      def attributes;             Header.new  end
+      def projections;            []          end
+      def wheres;                 []          end
+      def orders;                 []          end
+      def inserts;                []          end
+      def groupings;              []          end
+      def havings;                []          end
+      def joins(formatter = nil); nil         end # FIXME
+      def taken;                  nil         end
+      def skipped;                nil         end
+      def sources;                []          end
+      def locked;                 []          end
+    end
+    include DefaultOperations
+  end
+end

data/lib/arel/algebra/relations/row.rb

@@ -0,0 +1,26 @@
+module Arel
+  class Row
+    attributes :relation, :tuple
+    deriving :==, :initialize
+
+    def [](attribute)
+      attribute.type_cast(tuple[relation.position_of(attribute)])
+    end
+
+    def slice(*attributes)
+      Row.new(relation, attributes.inject([]) do |cheese, attribute|
+        # FIXME TESTME method chaining
+        cheese << tuple[relation.relation.position_of(attribute)]
+        cheese
+      end)
+    end
+
+    def bind(relation)
+      Row.new(relation, tuple)
+    end
+
+    def combine(other, relation)
+      Row.new(relation, tuple + other.tuple)
+    end
+  end
+end

data/lib/arel/algebra/relations/utilities/compound.rb

@@ -0,0 +1,54 @@
+module Arel
+  class Compound
+    include Relation
+
+    attr_reader :relation
+    delegate :joins, :join?, :inserts, :taken, :skipped, :name, :externalizable?,
+             :column_for, :engine, :sources, :locked, :table_alias,
+             :to => :relation
+
+    def self.requires(feature = nil)
+      @requires ||= nil
+      @requires = feature if feature
+      @requires
+    end
+
+    [:wheres, :groupings, :orders, :havings, :projections].each do |operation_name|
+      class_eval <<-OPERATION, __FILE__, __LINE__
+        def #{operation_name}
+          @#{operation_name} ||= relation.#{operation_name}.collect { |o| o.bind(self) }
+        end
+      OPERATION
+    end
+
+    def attributes
+      @attributes ||= relation.attributes.bind(self)
+    end
+
+    def hash
+      @hash ||= :relation.hash
+    end
+
+    def eql?(other)
+      self == other
+    end
+
+    def engine
+      requires = self.class.requires
+      engine   = relation.engine
+
+      # Temporary check of whether or not the engine supports where.
+      if requires && engine.respond_to?(:supports) && !engine.supports(requires)
+        Memory::Engine.new
+      else
+        engine
+      end
+    end
+
+  private
+
+    def arguments_from_block(relation, &block)
+      block_given?? [yield(relation)] : []
+    end
+  end
+end