benhoskings-ambitious-activerecord 0.1.3

data/Manifest

@@ -0,0 +1,22 @@
+lib/ambition/adapters/active_record/base.rb
+lib/ambition/adapters/active_record/query.rb
+lib/ambition/adapters/active_record/select.rb
+lib/ambition/adapters/active_record/slice.rb
+lib/ambition/adapters/active_record/sort.rb
+lib/ambition/adapters/active_record/statements.rb
+lib/ambition/adapters/active_record/context.rb
+lib/ambition/adapters/active_record.rb
+test/benchmark.rb
+test/chaining_test.rb
+test/count_test.rb
+test/enumerable_test.rb
+test/helper.rb
+test/join_test.rb
+test/profiler.rb
+test/ruby_test.rb
+test/select_test.rb
+test/slice_test.rb
+test/sort_test.rb
+test/source_test.rb
+test/types_test.rb
+Manifest

data/ambitious-activerecord.gemspec

@@ -0,0 +1,72 @@
+
+# Gem::Specification for Ambitious-activerecord-0.1.3
+# Originally generated by Echoe
+
+Gem::Specification.new do |s|
+  s.name = %q{ambitious-activerecord}
+  s.version = "0.1.3"
+
+  s.specification_version = 2 if s.respond_to? :specification_version=
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Chris Wanstrath"]
+  s.date = %q{2008-04-26}
+  s.description = %q{An ambitious adapter for ActiveRecord}
+  s.email = %q{chris@ozmm.org}
+  s.extra_rdoc_files = ["lib/ambition/adapters/active_record/base.rb", "lib/ambition/adapters/active_record/query.rb", "lib/ambition/adapters/active_record/select.rb", "lib/ambition/adapters/active_record/slice.rb", "lib/ambition/adapters/active_record/sort.rb", "lib/ambition/adapters/active_record/statements.rb", "lib/ambition/adapters/active_record.rb"]
+  s.files = ["lib/ambition/adapters/active_record/base.rb", "lib/ambition/adapters/active_record/query.rb", "lib/ambition/adapters/active_record/select.rb", "lib/ambition/adapters/active_record/slice.rb", "lib/ambition/adapters/active_record/sort.rb", "lib/ambition/adapters/active_record/statements.rb", "lib/ambition/adapters/active_record.rb", "test/benchmark.rb", "test/chaining_test.rb", "test/count_test.rb", "test/enumerable_test.rb", "test/helper.rb", "test/join_test.rb", "test/profiler.rb", "test/ruby_test.rb", "test/select_test.rb", "test/slice_test.rb", "test/sort_test.rb", "test/source_test.rb", "test/types_test.rb", "Manifest", "ambitious-activerecord.gemspec"]
+  s.has_rdoc = true
+  s.homepage = %q{http://ambition.rubyforge.org/}
+  s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "Ambitious-activerecord", "--main", "README"]
+  s.require_paths = ["lib"]
+  s.rubyforge_project = %q{ambition}
+  s.rubygems_version = %q{1.3.1}
+  s.summary = %q{An ambitious adapter for ActiveRecord}
+  s.test_files = ["test/chaining_test.rb", "test/count_test.rb", "test/enumerable_test.rb", "test/join_test.rb", "test/ruby_test.rb", "test/select_test.rb", "test/slice_test.rb", "test/sort_test.rb", "test/source_test.rb", "test/types_test.rb"]
+
+  s.add_dependency(%q<activerecord>, [">= 1.15.6"])
+  s.add_dependency(%q<ambition>, [">= 0.5.3"])
+end
+
+
+# # Original Rakefile source (requires the Echoe gem):
+# 
+# require 'rake'
+# 
+# Version = '0.1.3'
+# 
+# begin
+#   require 'rubygems'
+#   gem 'echoe', '>=2.7'
+#   ENV['RUBY_FLAGS'] = ""
+#   require 'echoe'
+# 
+#   Echoe.new('ambitious-activerecord') do |p|
+#     p.dependencies  << 'activerecord >=1.15.6'
+#     p.summary        = "An ambitious adapter for ActiveRecord"
+#     p.author         = 'Chris Wanstrath'
+#     p.email          = "chris@ozmm.org"
+# 
+#     p.project        = 'ambition'
+#     p.url            = "http://ambition.rubyforge.org/"
+#     p.test_pattern   = 'test/*_test.rb'
+#     p.version        = Version
+#     p.dependencies  << 'ambition >=0.5.3'
+#   end
+# 
+# rescue LoadError 
+#   puts "Not doing any of the Echoe gemmy stuff, because you don't have the specified gem versions"
+# 
+#   require 'rake/testtask'
+#   Rake::TestTask.new do |t|
+#     t.pattern = "test/*_test.rb"
+#   end
+# end
+# 
+# desc 'Install as a gem'
+# task :install_gem do
+#   puts `rake manifest package && gem install pkg/ambitious-activerecord-#{Version}.gem`
+# end
+# 
+# task :default => :test 
+# 

data/lib/ambition/adapters/active_record.rb

@@ -0,0 +1,12 @@
+require 'ambition'
+require 'active_record'
+require 'ambition/adapters/active_record/query'
+require 'ambition/adapters/active_record/base'
+require 'ambition/adapters/active_record/select'
+require 'ambition/adapters/active_record/sort'
+require 'ambition/adapters/active_record/slice'
+require 'ambition/adapters/active_record/statements'
+require 'ambition/adapters/active_record/context'
+
+ActiveRecord::Base.extend Ambition::API
+ActiveRecord::Base.ambition_adapter = Ambition::Adapters::ActiveRecord

data/lib/ambition/adapters/active_record/base.rb

@@ -0,0 +1,53 @@
+require 'active_record/connection_adapters/abstract/quoting'
+
+module Ambition
+  module Adapters
+    module ActiveRecord
+      class Base
+        include ::ActiveRecord::ConnectionAdapters::Quoting
+
+        def sanitize(value)
+          if value.is_a? Array
+            return value.map { |v| sanitize(v) }.join(', ')
+          end
+
+          case value
+          when true,  'true'  
+            '1'
+          when false, 'false' 
+            '0'
+          when Regexp
+            "'#{value.source}'"
+          else 
+            if owner.connected?
+              ::ActiveRecord::Base.connection.quote(value) 
+            else
+              quote(value)
+            end
+          end
+        rescue
+          "'#{value}'"
+        end
+
+        def quote_column_name(value)
+          if owner.connected?
+            ::ActiveRecord::Base.connection.quote_column_name(value) 
+          else
+            value.to_s
+          end
+        end
+
+        def dbadapter_name
+          ::ActiveRecord::Base.connection.adapter_name
+        rescue ::ActiveRecord::ConnectionNotEstablished
+          'Abstract'
+        end
+
+        def statement(*args)
+          @statement_instance ||= DatabaseStatements.const_get(dbadapter_name).new
+          @statement_instance.send(*args)
+        end
+      end
+    end
+  end
+end

data/lib/ambition/adapters/active_record/query.rb

@@ -0,0 +1,75 @@
+module Ambition
+  module Adapters
+    module ActiveRecord
+      class Query
+        @@select = 'SELECT * FROM %s %s'
+
+        def kick
+          owner.find(:all, to_hash)
+        end
+
+        def size
+          owner.count(to_hash)
+        end
+        alias_method :length, :size
+
+        def to_hash
+          hash = {}
+
+          unless (joins = stash[:joins]).blank?
+            hash[:joins] = joins
+            stash[:include] -= stash[:joins] unless stash[:include].blank?
+          end
+
+          unless (where = clauses[:select]).blank?
+            hash[:conditions] = Array(where)
+            hash[:conditions] *= ' AND '
+          end
+
+          unless (order = clauses[:sort]).blank?
+            hash[:order] = order.join(', ')
+          end
+
+          if Array(clauses[:slice]).last =~ /LIMIT (\d+)/
+            hash[:limit] = $1.to_i
+          end
+
+          if Array(clauses[:slice]).last =~ /OFFSET (\d+)/
+            hash[:offset] = $1.to_i
+          end
+
+          hash[:include] = stash[:include] if stash[:include]
+
+          hash
+        end
+
+        def to_s
+          hash = to_hash
+
+          raise "Sorry, I can't construct SQL with complex joins (yet)" unless hash[:include].blank?
+
+          sql = []
+          sql << joins_sql(hash[:joins])      unless hash[:joins].blank?
+          sql << "WHERE #{hash[:conditions]}" unless hash[:conditions].blank?
+          sql << "ORDER BY #{hash[:order]}"   unless hash[:order].blank?
+          sql << clauses[:slice].last         unless hash[:slice].blank?
+
+          @@select % [ owner.table_name, sql.join(' ') ]
+        end
+        alias_method :to_sql, :to_s
+
+        private
+
+        def joins_sql(joins)
+          joins.map {|join|
+            ::ActiveRecord::Associations::ClassMethods::InnerJoinDependency.
+              new(owner, join, nil).
+              join_associations.
+              map(&:association_join).
+              join(' ')
+          }
+        end
+      end
+    end
+  end
+end

data/lib/ambition/adapters/active_record/select.rb

@@ -0,0 +1,99 @@
+module Ambition
+  module Adapters
+    module ActiveRecord
+      class Select < Base
+        def call(method)
+          "#{owner.table_name}.#{quote_column_name method}"
+        end
+
+        def chained_call(*methods)
+          if reflection = owner.reflections[methods.first]
+            stash[:include] ||= []
+            stash[:include] << methods.first
+            "#{reflection.table_name}.#{quote_column_name methods.last}"
+          elsif respond_to? methods[1]
+            send(methods[1], methods.first)
+          else
+            raise "I don't understand: #{methods.inspect}"
+          end
+        end
+
+        def both(left, right)
+          "(#{left} AND #{right})"
+        end
+
+        def either(left, right)
+          "(#{left} OR #{right})"
+        end
+
+        def ==(left, right)
+          if right.nil?
+            "#{left} IS NULL"
+          else
+            "#{left} = #{sanitize right}"
+          end
+        end
+
+        # !=
+        def not_equal(left, right)
+          if right.nil?
+            "#{left} IS NOT NULL"
+          else
+            "#{left} <> #{sanitize right}"
+          end
+        end
+
+        def =~(left, right)
+          if right.is_a? Regexp
+            "#{left} #{statement(:regexp, right)} #{sanitize right}"
+          else
+            "#{left} LIKE #{sanitize right}"
+          end
+        end
+
+        # !~
+        def not_regexp(left, right)
+          if right.is_a? Regexp
+            "#{left} #{statement(:not_regexp, right)} #{sanitize right}"
+          else
+            "#{left} NOT LIKE #{sanitize right}"
+          end
+        end
+
+        def <(left, right)
+          "#{left} < #{sanitize right}"
+        end
+
+        def >(left, right)
+          "#{left} > #{sanitize right}"
+        end
+
+        def >=(left, right)
+          "#{left} >= #{sanitize right}"
+        end
+
+        def <=(left, right)
+          "#{left} <= #{sanitize right}"
+        end
+
+        def include?(left, right)
+          left = left.map { |element| sanitize element }.join(', ')
+          "#{right} IN (#{left})"
+        end
+
+        def nil?(column)
+          left = "#{owner.table_name}.#{quote_column_name column}"
+          negated? ? not_equal(left, nil) : self.==(left, nil)
+        end
+
+        def downcase(column)
+          "LOWER(#{owner.table_name}.#{quote_column_name column})"
+        end
+
+        def upcase(column)
+          "UPPER(#{owner.table_name}.#{quote_column_name column})"
+        end
+      end
+    end
+  end
+end

data/lib/ambition/adapters/active_record/slice.rb

@@ -0,0 +1,19 @@
+module Ambition
+  module Adapters
+    module ActiveRecord
+      class Slice < Base
+        def slice(start, length=nil)
+          if start.is_a? Range
+            length  = start.end
+            length -= 1 if start.exclude_end?
+            start = start.first - 1
+            length -= start
+          end
+          out  = "LIMIT #{length} "
+          out << "OFFSET #{start}" if start.to_i.nonzero?
+          out
+        end
+      end
+    end
+  end
+end

data/lib/ambition/adapters/active_record/sort.rb

@@ -0,0 +1,43 @@
+module Ambition
+  module Adapters
+    module ActiveRecord
+      class Sort < Base
+        def sort_by(method)
+          "#{owner.table_name}.#{quote_column_name method}" 
+        end
+
+        def reverse_sort_by(method)
+          "#{owner.table_name}.#{quote_column_name method} DESC" 
+        end
+
+        def chained_sort_by(receiver, method)
+          if reflection = owner.reflections[receiver]
+            stash[:include] ||= []
+            stash[:include] << receiver
+            "#{reflection.table_name}.#{quote_column_name method}"
+          else 
+            raise [ receiver, method ].inspect
+          end
+        end
+
+        def chained_reverse_sort_by(receiver, method)
+          if reflection = owner.reflections[receiver]
+            stash[:include] ||= []
+            stash[:include] << receiver
+            "#{reflection.table_name}.#{quote_column_name method} DESC"
+          else 
+            raise [ receiver, method ].inspect
+          end
+        end
+
+        def to_proc(symbol)
+          "#{owner.table_name}.#{symbol}"
+        end
+
+        def rand
+          'RAND()'
+        end
+      end
+    end
+  end
+end

data/lib/ambition/adapters/active_record/statements.rb

@@ -0,0 +1,39 @@
+module Ambition
+  module Adapters
+    module ActiveRecord
+      module DatabaseStatements
+        def self.const_missing(*args)
+          Abstract
+        end
+
+        class Abstract
+          def regexp(regexp)
+            'REGEXP'
+          end
+
+          def not_regexp(regexp)
+            'NOT REGEXP'
+          end
+        end
+
+        class PostgreSQL < Abstract
+          def regexp(regexp)
+            if regexp.options == 1
+              '~*'
+            else
+              '~'
+            end
+          end
+
+          def not_regexp(regexp)
+            if regexp.options == 1
+              '!~*'
+            else
+              '!~'
+            end
+          end
+        end
+      end
+    end
+  end
+end