mongo_mapper-unstable 2009.10.11

data/.gitignore

@@ -0,0 +1,8 @@
+*.sw?
+.DS_Store
+coverage
+rdoc
+pkg
+*~
+*.gem
+tmp

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 John Nunemaker
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,50 @@
+= MongoMapper
+
+Awesome gem for modeling your domain and storing it in mongo.
+
+Releases are tagged on github and released on gemcutter. Master is pushed to whenever I add a patch or a new feature, but I do not release a new gem version each time I push.
+
+== Note on Patches/Pull Requests
+
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a future version unintentionally.
+* Commit, do not mess with rakefile, version, or history. (if you want to have your own version, that is fine but bump version in a commit by itself in another branch so I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Install
+
+MongoMapper is only released on gemcutter. To install, you can setup gemcutter as your default gem source.
+
+ $ gem install gemcutter
+ $ gem tumble
+ 
+Then you can install MM:
+ 
+ $ gem install mongo_mapper
+
+You can also just declare the source:
+
+ $ gem install mongo_mapper -s http://gemcutter.org
+
+== Dependencies
+
+* ActiveSupport (typically the latest version)
+* Mongo Ruby Driver (mongo)
+* My fork of the validatable gem (jnunemaker-validatable)
+
+== Documentation
+
+http://rdoc.info/projects/jnunemaker/mongomapper
+
+== More Info
+
+You can learn more about mongo here:
+http://www.mongodb.org/
+
+You can learn more about the mongo ruby driver here:
+http://github.com/mongodb/mongo-ruby-driver/tree/master
+
+== Copyright
+
+Copyright (c) 2009 John Nunemaker. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,87 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "mongo_mapper"
+    gem.summary = %Q{Awesome gem for modeling your domain and storing it in mongo}
+    gem.email = "nunemaker@gmail.com"
+    gem.homepage = "http://github.com/jnunemaker/mongomapper"
+    gem.authors = ["John Nunemaker"]
+    gem.rubyforge_project = "mongomapper"
+    
+    gem.add_dependency('activesupport')
+    gem.add_dependency('mongo', '0.15.1')
+    gem.add_dependency('jnunemaker-validatable', '1.7.4')
+    
+    gem.add_development_dependency('mocha', '0.9.4')
+    gem.add_development_dependency('jnunemaker-matchy', '0.4.0')
+  end
+  
+  Jeweler::RubyforgeTasks.new do |rubyforge|
+    rubyforge.doc_task = "rdoc"
+  end
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+namespace :test do
+  Rake::TestTask.new(:units) do |test|
+    test.libs << 'lib' << 'test'
+    test.pattern = 'test/unit/**/test_*.rb'
+    test.verbose = true
+  end
+  
+  Rake::TestTask.new(:functionals) do |test|
+    test.libs << 'lib' << 'test'
+    test.pattern = 'test/functional/**/test_*.rb'
+    test.verbose = true
+  end
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/test_*.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+
+begin
+  require 'cucumber/rake/task'
+  Cucumber::Rake::Task.new(:features)
+rescue LoadError
+  task :features do
+    abort "Cucumber is not available. In order to run features, you must: sudo gem install cucumber"
+  end
+end
+
+task :default => :test
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  if File.exist?('VERSION.yml')
+    config = YAML.load(File.read('VERSION.yml'))
+    version = "#{config[:major]}.#{config[:minor]}.#{config[:patch]}"
+  else
+    version = ""
+  end
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "MongoMapper #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+2009.10.11

data/bin/mmconsole

@@ -0,0 +1,55 @@
+#!/usr/bin/env ruby
+$:.unshift File.dirname(__FILE__)+"/../lib"
+
+require 'mongo_mapper'
+require 'irb'
+
+IRB.setup(nil)
+irb = IRB::Irb.new
+
+IRB.conf[:MAIN_CONTEXT] = irb.context
+
+irb.context.evaluate("require 'irb/completion'", 0)
+irb.context.evaluate(%@
+include MongoMapper
+
+MongoMapper.database = "mmtest"
+$db = MongoMapper.database
+
+@, 0)
+
+puts %@
+Welcome to the MongoMapper Console!
+
+Example 1:
+  things = $db.collection("things")
+  things.insert("name" => "Raw Thing")
+  things.insert("name" => "Another Thing", "date" => Time.now)
+
+  cursor = things.find("name" => "Raw Thing")
+  puts cursor.next_object.inspect
+
+Example 2:
+  class Thing
+    include MongoMapper::Document
+    key :name, String, :required => true
+    key :date, Time
+  end
+
+  thing = Thing.new
+  thing.name = "My thing"
+  thing.date = Time.now
+  thing.save
+
+  all_things = Thing.all
+  puts all_things.map { |object| object.name }.inspect
+
+@
+
+trap("SIGINT") do
+  irb.signal_handle
+end
+catch(:IRB_EXIT) do
+  irb.eval_input
+end
+

data/lib/mongo_mapper/associations/base.rb

@@ -0,0 +1,83 @@
+module MongoMapper
+  module Associations
+    class Base
+      attr_reader :type, :name, :options
+
+      def initialize(type, name, options = {})
+        @type, @name, @options = type, name, options
+      end
+
+      def class_name
+        @class_name ||= begin
+          if cn = options[:class_name]
+            cn
+          elsif many?
+            name.to_s.singularize.camelize
+          else
+            name.to_s.camelize
+          end
+        end
+      end
+
+      def klass
+        @klass ||= class_name.constantize
+      end
+
+      def many?
+        @many_type ||= @type == :many
+      end
+
+      def belongs_to?
+        @belongs_to_type ||= @type == :belongs_to
+      end
+
+      def polymorphic?
+        !!@options[:polymorphic]
+      end
+
+      def as?
+        !!@options[:as]
+      end
+
+      def type_key_name
+        @type_key_name ||= many? ? '_type' : "#{as}_type"
+      end
+
+      def as
+        @options[:as] || self.name
+      end
+
+      def foreign_key
+        @options[:foreign_key] || "#{name}_id"
+      end
+
+      def ivar
+        @ivar ||= "@_#{name}"
+      end
+
+      def embeddable?
+        many? && klass.embeddable?
+      end
+
+      def proxy_class
+        @proxy_class ||= begin
+          if many?
+            if self.klass.embeddable?
+              polymorphic? ? ManyEmbeddedPolymorphicProxy : ManyEmbeddedProxy
+            else
+              if polymorphic?
+                ManyPolymorphicProxy
+              elsif as?
+                ManyDocumentsAsProxy
+              else
+                ManyProxy
+              end
+            end
+          else
+            polymorphic? ? BelongsToPolymorphicProxy : BelongsToProxy
+          end
+        end # end begin
+      end # end proxy_class
+    end
+  end
+end

data/lib/mongo_mapper/associations/belongs_to_polymorphic_proxy.rb

@@ -0,0 +1,34 @@
+module MongoMapper
+  module Associations
+    class BelongsToPolymorphicProxy < Proxy
+      def replace(doc)
+        if doc
+          doc.save if doc.new?
+          id, type = doc.id, doc.class.name
+        end
+        
+        @owner.send("#{@association.foreign_key}=", id)
+        @owner.send("#{@association.type_key_name}=", type)
+        reset
+      end
+      
+      protected
+        def find_target
+          if proxy_id && proxy_class
+            proxy_class.find_by_id(proxy_id)
+          end
+        end
+        
+        def proxy_id
+          @proxy_id ||= @owner.send(@association.foreign_key)
+        end
+        
+        def proxy_class
+          @proxy_class ||= begin
+            klass = @owner.send(@association.type_key_name)
+            klass && klass.constantize
+          end
+        end
+    end
+  end
+end

data/lib/mongo_mapper/associations/belongs_to_proxy.rb

@@ -0,0 +1,22 @@
+module MongoMapper
+  module Associations
+    class BelongsToProxy < Proxy
+      def replace(doc)
+        if doc
+          doc.save if doc.new?
+          id = doc.id
+        end
+
+        @owner.send("#{@association.foreign_key}=", id)
+        reset
+      end
+
+      protected
+        def find_target
+          if association_id = @owner.send(@association.foreign_key)
+            @association.klass.find_by_id(association_id)
+          end
+        end
+    end
+  end
+end

data/lib/mongo_mapper/associations/many_documents_as_proxy.rb

@@ -0,0 +1,27 @@
+module MongoMapper
+  module Associations
+    class ManyDocumentsAsProxy < ManyDocumentsProxy
+      protected
+        def scoped_conditions
+          {as_type_name => @owner.class.name, as_id_name => @owner.id}
+        end
+
+        def apply_scope(doc)
+          ensure_owner_saved
+
+          doc.send("#{as_type_name}=", @owner.class.name)
+          doc.send("#{as_id_name}=", @owner.id)
+
+          doc
+        end
+
+        def as_type_name
+          @as_type_name ||= @association.options[:as].to_s + "_type"
+        end
+
+        def as_id_name
+          @as_id_name ||= @association.options[:as].to_s + "_id"
+        end
+    end
+  end
+end

data/lib/mongo_mapper/associations/many_documents_proxy.rb

@@ -0,0 +1,116 @@
+module MongoMapper
+  module Associations
+    class ManyDocumentsProxy < Proxy
+      delegate :klass, :to => :@association
+      delegate :collection, :to => :klass
+      
+      include MongoMapper::Finders
+      
+      def find(*args)
+        options = args.extract_options!
+        klass.find(*args << scoped_options(options))
+      end
+
+      def paginate(options)
+        klass.paginate(scoped_options(options))
+      end
+
+      def all(options={})
+        find(:all, scoped_options(options))
+      end
+
+      def first(options={})
+        find(:first, scoped_options(options))
+      end
+
+      def last(options={})
+        find(:last, scoped_options(options))
+      end
+
+      def count(conditions={})
+        klass.count(conditions.deep_merge(scoped_conditions))
+      end
+
+      def replace(docs)
+        @target.map(&:destroy) if load_target
+        docs.each { |doc| apply_scope(doc).save }
+        reset
+      end
+
+      def <<(*docs)
+        ensure_owner_saved
+        flatten_deeper(docs).each { |doc| apply_scope(doc).save }
+        reset
+      end
+      alias_method :push, :<<
+      alias_method :concat, :<<
+
+      def build(attrs={})
+        doc = klass.new(attrs)
+        apply_scope(doc)
+        doc
+      end
+
+      def create(attrs={})
+        doc = klass.new(attrs)
+        apply_scope(doc).save
+        doc
+      end
+
+      def destroy_all(conditions={})
+        all(:conditions => conditions).map(&:destroy)
+        reset
+      end
+
+      def delete_all(conditions={})
+        klass.delete_all(conditions.deep_merge(scoped_conditions))
+        reset
+      end
+      
+      def nullify
+        criteria = FinderOptions.to_mongo_criteria(scoped_conditions)
+        all(criteria).each do |doc|
+          doc.update_attributes self.foreign_key => nil
+        end
+        reset
+      end
+      
+      def method_missing(method, *args)
+        finder = DynamicFinder.new(method)
+        
+        if finder.found?
+          dynamic_find(finder, args)
+        else
+          super
+        end
+      end
+      
+      protected
+        def scoped_conditions
+          {self.foreign_key => @owner.id}
+        end
+
+        def scoped_options(options)
+          options.deep_merge({:conditions => scoped_conditions})
+        end
+
+        def find_target
+          find(:all)
+        end
+
+        def ensure_owner_saved
+          @owner.save if @owner.new?
+        end
+
+        def apply_scope(doc)
+          ensure_owner_saved
+          doc.send("#{self.foreign_key}=", @owner.id)
+          doc
+        end
+
+        def foreign_key
+          @association.options[:foreign_key] || @owner.class.name.underscore.gsub("/", "_") + "_id"
+        end
+    end
+  end
+end

data/lib/mongo_mapper/associations/many_embedded_polymorphic_proxy.rb

@@ -0,0 +1,33 @@
+module MongoMapper
+  module Associations
+    class ManyEmbeddedPolymorphicProxy < Proxy      
+      def replace(v)
+        @_values = v.map do |doc_or_hash|
+          if doc_or_hash.kind_of?(EmbeddedDocument)
+            doc = doc_or_hash
+            {@association.type_key_name => doc.class.name}.merge(doc.attributes)
+          else
+            doc_or_hash
+          end
+        end
+        
+        reset
+      end
+
+      protected
+        def find_target
+          (@_values || []).map do |hash|
+            polymorphic_class(hash).new(hash)
+          end
+        end
+        
+        def polymorphic_class(doc)
+          if class_name = doc[@association.type_key_name]
+            class_name.constantize
+          else
+            @association.klass
+          end
+        end
+    end
+  end
+end

data/lib/mongo_mapper/associations/many_embedded_proxy.rb

@@ -0,0 +1,67 @@
+module MongoMapper
+  module Associations
+    class ManyEmbeddedProxy < Proxy
+      def replace(v)
+        @_values = v.map { |e| e.kind_of?(EmbeddedDocument) ? e.attributes : e }
+        reset
+      end
+      
+      def build(opts={})
+        owner = @owner
+        child = @association.klass.new(opts)
+        assign_parent_reference(child)
+        child._root_document = owner
+        self << child
+        child
+      end
+      
+      def find(opts)
+        case opts
+        when :all
+          self
+        when String
+          if load_target
+            child = @target.detect {|item| item.id == opts}
+            assign_parent_reference(child)
+            child
+          end
+        end
+      end
+
+      def <<(*docs)
+        if load_target
+          root = @owner._root_document || @owner
+          docs.each do |doc|
+            doc._root_document = root
+            @target << doc
+          end
+        end
+      end
+      alias_method :push, :<<
+      alias_method :concat, :<<
+
+      protected
+        def find_target
+          (@_values || []).map do |e|
+            child = @association.klass.new(e)
+            assign_parent_reference(child)
+            child
+          end
+        end
+        
+      private
+      
+        def assign_parent_reference(child)
+          return unless child && @owner
+          return if @owner.class.name.blank?
+          owner = @owner
+          child.class_eval do
+            define_method(owner.class.name.underscore) do
+              owner
+            end
+          end
+        end
+      
+    end
+  end
+end

data/lib/mongo_mapper/associations/many_polymorphic_proxy.rb

@@ -0,0 +1,11 @@
+module MongoMapper
+  module Associations
+    class ManyPolymorphicProxy < ManyDocumentsProxy      
+      private
+        def apply_scope(doc)
+          doc.send("#{@association.type_key_name}=", doc.class.name)
+          super
+        end
+    end
+  end
+end

data/lib/mongo_mapper/associations/many_proxy.rb

@@ -0,0 +1,6 @@
+module MongoMapper
+  module Associations
+    class ManyProxy < ManyDocumentsProxy
+    end
+  end
+end