mongo_mapper_ext 0.0.1

data/Rakefile

@@ -0,0 +1,68 @@
+require 'rake'
+require 'spec/rake/spectask'
+
+Dir.chdir File.dirname(__FILE__)	
+
+# Specs
+task :default => :spec
+
+Spec::Rake::SpecTask.new('spec') do |t|
+	t.spec_files = FileList["spec/**/*_spec.rb"].select{|f| f !~ /\/_/}
+  t.libs = ['lib'].collect{|f| "#{File.dirname __FILE__}/#{f}"}
+end
+
+# Gem
+require 'rake/clean'
+require 'rake/gempackagetask'
+require 'fileutils'
+
+spec = Gem::Specification.new do |s|
+	s.name = "mongo_mapper_ext"
+	s.version = "0.0.1" 
+	s.summary = "Extensions for MongoMapper"
+	s.description = "Extensions for MongoMapper"
+	s.author = "Alexey Petrushin"
+	#	s.email = ""
+	s.homepage = "http://github.com/alexeypetrushin/mongo_mapper_ext"
+  
+	s.platform = Gem::Platform::RUBY
+  s.has_rdoc = true
+  # s.extra_rdoc_files = ["README.rdoc"]
+	
+  # s.files = (%w{rakefile readme.md .gitignore} + Dir.glob("{app,lib,spec,.git}/**/*"))
+  s.files = (['Rakefile', 'readme.md'] + Dir.glob("{lib,spec}/**/*"))
+	
+	s.require_paths = ["lib"]
+
+  warn 'uncomment ruby_ext'
+  [
+    # 'ruby_ext',
+    'mongo_mapper',
+  ].each{|name| s.add_dependency(name)}
+
+end
+
+PACKAGE_DIR = "#{File.expand_path File.dirname(__FILE__)}/build"
+
+Rake::GemPackageTask.new(spec) do |p|
+	package_dir = PACKAGE_DIR
+#	FileUtils.mkdir package_dir unless File.exist? package_dir	
+	p.need_tar = true if RUBY_PLATFORM !~ /mswin/
+	p.need_zip = true
+	p.package_dir = package_dir
+end
+
+# CLEAN.include [ 'pkg', '*.gem']
+
+task :push do
+  dir = Dir.chdir PACKAGE_DIR do
+    gem_file = Dir.glob("mongo_mapper_ext*.gem").first
+    system "gem push #{gem_file}"
+  end
+end
+
+task :clean do
+  system "rm -r #{PACKAGE_DIR}"
+end
+
+task :release => [:gem, :push, :clean]

data/lib/mongo_mapper_ext/db_config.rb

@@ -0,0 +1,67 @@
+module MongoMapper
+  class ConnectionsPool < Hash
+    def [](database_alias)
+      database_alias = database_alias.to_s
+      unless connection = super(database_alias)            
+        MongoMapper.db_config.must.include database_alias
+        db_options = MongoMapper.db_config[database_alias]        
+        connection = Mongo::Connection.new(db_options['host'], db_options['port'], :logger => MongoMapper.logger)
+      
+        if defined?(PhusionPassenger)
+          PhusionPassenger.on_event(:starting_worker_process) do |forked|
+            connection.connect_to_master if forked
+          end
+        end
+        
+        self[database_alias] = connection
+      end            
+      return connection
+    end
+  end
+  
+  class DatabasesPool < Hash
+    def [](database_alias)
+      database_alias = database_alias.to_s
+      unless db = super(database_alias)
+        MongoMapper.db_config.must.include database_alias
+        db_options = MongoMapper.db_config[database_alias]
+        db = MongoMapper.connections[database_alias].db db_options['name'].must_be.a(String)
+        self[database_alias] = db
+      end
+      return db
+    end
+  end
+  
+  class << self
+    attr_accessor :db_config
+    
+    def db_config      
+      # unless @db_config
+      #   
+      #   if defined?(::Rails) and defined?(DATABASE)              
+      #     @db_config = {}
+      #     DATABASE["#{::Rails.env}!"].to_h.each do |db_alias, options|
+      #       @db_config[db_alias.to_s] = {
+      #         'name' => options.name!,
+      #         'host' => options.host(nil),
+      #         'port' => options.port(nil)
+      #       }
+      #     end
+      #   else
+      #     @db_config = {}
+      #     # raise "If you don't using Rails you must override this method and provide your own config!"
+      #   end          
+      # end
+      @db_config ||= {}
+    end
+    
+    def connections
+      @connections ||= ConnectionsPool.new
+    end
+    
+    def databases
+      @databases ||= DatabasesPool.new
+    end
+  end
+              
+end

data/lib/mongo_mapper_ext/hacks/fixes.rb

@@ -0,0 +1,20 @@
+# 
+# ObjectID
+# 
+# Mongo::ObjectID.class_eval do  
+#   def == other
+#     self.to_s == other.to_s
+#   end
+# 
+#   def to_yaml *args
+#     to_s.to_yaml *args
+#   end
+# end
+
+MongoMapper::Plugins::Associations::InArrayProxy.class_eval do
+  def delete(doc)
+    ids.delete(doc.id)
+    klass.delete(doc.id)
+    reset
+  end
+end

data/lib/mongo_mapper_ext/hacks/time_measuring.rb

@@ -0,0 +1,44 @@
+# Measures MongoDB requests time and adds it to log
+
+if defined?(Rails) and RAILS_ENV == 'development'
+  Mongo::Connection.class_eval do  
+    def send_message_with_time operation, message, log_message=nil
+      begin
+        logger = @logger
+        @logger = nil      
+        t = Time.now
+        send_message_without_time operation, message, log_message
+      ensure
+        logger.debug("  MONGODB (#{Time.now - t}) #{log_message || message}") if logger
+        @logger = logger
+      end
+    end
+    alias_method_chain :send_message, :time
+  
+    def send_message_with_safe_check_with_time operation, message, db_name, log_message=nil    
+      begin
+        logger = @logger
+        @logger = nil      
+        t = Time.now
+        send_message_with_safe_check_without_time operation, message, db_name, log_message
+      ensure
+        logger.debug("  MONGODB (#{Time.now - t}) #{log_message || message}") if logger
+        @logger = logger
+      end
+    end
+    alias_method_chain :send_message_with_safe_check, :time
+  
+    def receive_message_with_time operation, message, log_message=nil, socket=nil
+      begin
+        logger = @logger
+        @logger = nil      
+        t = Time.now
+        receive_message_without_time operation, message, log_message, socket
+      ensure
+        logger.debug("  MONGODB (#{Time.now - t}) #{log_message || message}") if logger
+        @logger = logger
+      end
+    end
+    alias_method_chain :receive_message, :time
+  end
+end

data/lib/mongo_mapper_ext/micelaneous.rb

@@ -0,0 +1,53 @@
+#
+# upsert
+# 
+Mongo::Collection.class_eval do
+  def upsert id, opt
+    opt.size.must == 1
+    opt.must_be.a Hash
+    opt.values.first.must_be.a Hash
+    
+    update({:_id => id}, opt, {:upsert => true, :safe => true})
+  end
+end
+
+
+# 
+# silence logging
+# 
+Mongo::Connection.class_eval do
+  attr_writer :logger
+end
+
+MongoMapper.class_eval do
+  def self.temporary_silince_logger &block
+    logger, nil_logger = self.logger, Logger.new(nil)
+    
+    begin  
+      MongoMapper.logger = nil_logger
+      connections.each do |name, connection|
+        connection.logger = nil_logger
+      end
+      block.call
+    ensure
+      MongoMapper.logger = logger
+      connections.each do |name, connection|
+        connection.logger = logger
+      end
+    end
+  end
+end
+
+
+# 
+# Silence index creation
+# 
+MongoMapper::Plugins::Indexes::ClassMethods.class_eval do
+  def ensure_index_with_silent_logger *args
+    MongoMapper.temporary_silince_logger do
+      ensure_index_without_silent_logger *args
+    end
+  end
+  
+  alias_method_chain :ensure_index, :silent_logger
+end

data/lib/mongo_mapper_ext/migrate.rake

@@ -0,0 +1,33 @@
+Rake::TaskManager.class_eval do
+  def remove_task(task_name)
+    @tasks.delete(task_name.to_s)
+  end
+end
+
+def remove_task(task_name)
+  Rake.application.remove_task(task_name)
+end
+
+namespace :db do
+  
+  remove_task 'db:migrate'
+  desc "Migrate Database"
+  task :migrate => :environment do
+    ::Migration = MongoMapper::Migration
+    Dir["#{Rails.root}/lib/db/**/*.rb"].each{|f| require f.sub(/\.rb$/, '')}
+    
+    database_alias = ENV['d'] || ENV['database']
+    database_alias = 'accounts' if database_alias.blank?
+    
+    version = ENV['v'] || ENV['version']
+    if version.blank?
+      size = MongoMapper::Migration.definitions[database_alias].size
+      highest_defined_version = size == 0 ? 0 : size - 1      
+      version = highest_defined_version
+    else
+      version = version.to_i
+    end
+    
+    MongoMapper::Migration.update database_alias, version
+  end
+end

data/lib/mongo_mapper_ext/migration.rb

@@ -0,0 +1,108 @@
+module MongoMapper::Migration
+  class MigrationDefinition
+    def upgrade &block
+      if block
+        @upgrade = block
+      else
+        @upgrade
+      end
+    end
+    alias_method :up, :upgrade
+    
+    def downgrade &block
+      if block
+        @downgrade = block
+      else
+        @downgrade
+      end
+    end
+    alias_method :down, :downgrade
+  end
+  
+  class Logger
+    attr_accessor :enabled
+    
+    def initialize enabled = true
+      self.enabled = enabled
+    end
+    
+    def info message
+      puts message if enabled
+    end
+  end
+  
+  class << self
+    def logger= logger
+      @logger = logger ? logger : Logger.new(false)
+    end
+    
+    def logger; @logger ||= Logger.new end
+    
+    def define database_alias, version, &block
+      database_alias = database_alias.to_s
+      
+      version.must_be.a Integer
+      definition = MigrationDefinition.new
+      block.call definition
+      definitions[database_alias][version] = definition
+    end
+            
+    def update database_alias, version  
+      database_alias = database_alias.to_s
+      db = MongoMapper.databases[database_alias]
+      
+      if metadata(db).version == version
+        logger.info "Database '#{database_alias}' already is of #{version} version, no migration needed."
+        return false
+      else
+        logger.info "Migration for '#{database_alias}' Database:"
+      end
+      
+      increase_db_version database_alias, db while metadata(db).version < version
+      decrease_db_version database_alias, db while metadata(db).version > version
+      true
+    end
+        
+    def metadata db
+      col = db.collection 'db_metadata'
+      metadata = (col.find_one || {:version => 0}).to_openobject
+    end
+    
+    def definitions
+      @definitions ||= Hash.new{|h, k| h[k] = []} 
+    end
+    
+    protected
+      def increase_db_version database_alias, db    
+        m = metadata(db)
+        migration = definitions[database_alias][m.version + 1]        
+        raise "No upgrade for version #{m.version + 1} of '#{database_alias}' Database!" unless migration and migration.up
+        
+        migration.up.call db
+        
+        m.version += 1        
+        update_metadata db, m
+        
+        logger.info "Database '#{database_alias}' upgraded to version #{m.version}."
+      end
+      
+      def decrease_db_version database_alias, db
+        m = metadata(db)
+        migration = definitions[database_alias][m.version]        
+        raise "No downgrade for version #{m.version} of '#{database_alias}' Database!" unless migration and migration.down
+        
+        migration.down.call db
+        
+        m.version -= 1
+        update_metadata db, m
+        
+        logger.info "Database '#{database_alias}' downgraded to version #{m.version}."
+      end
+
+      
+      def update_metadata db, metadata
+        col = db.collection 'db_metadata'
+        col.save metadata.to_hash
+      end
+  end
+end

data/lib/mongo_mapper_ext/mongo_mapper.rb

@@ -0,0 +1,25 @@
+MongoMapper.class_eval do
+  class << self
+    
+    def logger
+      # return @logger ||= Logger.new(nil) if defined?(Spec)
+      unless @logger
+        @logger = if "irb" == $0
+          Logger.new(STDOUT)
+        else          
+          if defined?(Rails)
+            Rails.test? ? Logger.new(nil) : Rails.logger
+          else
+            Logger.new(STDOUT)
+          end
+        end
+      end
+      @logger
+    end
+    
+    def logger= logger
+      @logger = logger
+    end
+    
+  end
+end

data/lib/mongo_mapper_ext/plugins/attributes_cache.rb

@@ -0,0 +1,22 @@
+module MongoMapper
+  module Plugins
+    module AttributesCache
+      
+      module InstanceMethods
+        def cache
+          @cache ||= {}
+        end
+
+        def clear_cache
+          @cache = {}
+        end
+
+        def reload
+          @cache.clear if @cache
+          super
+        end
+      end
+      
+    end
+  end
+end

data/lib/mongo_mapper_ext/plugins/db_config.rb

@@ -0,0 +1,47 @@
+# # 
+# # Connect to database_alias specified in config
+# #
+# module MongoMapper
+#   module Plugins
+#     module DbConfig
+#       
+#       module ClassMethods        
+#         inheritable_accessor :database_alias, nil
+#         def use_database database_alias
+#           self.database_alias = database_alias.to_s
+#         end
+#         
+#         def connection(mongo_connection=nil)
+#           assert_supported
+#           if mongo_connection.nil?
+#             @connection || connection_from_alias || MongoMapper.connection
+#           else
+#             @connection = mongo_connection
+#           end
+#         end
+# 
+#         def database_name
+#           assert_supported
+#           @database_name || database_name_from_alias
+#         end
+#         
+#         private
+#           def database_name_from_alias
+#             return unless database_alias
+#             
+#             MongoMapper.db_config.must.include database_alias
+#             MongoMapper.db_config[database_alias]['name']
+#           end
+#           
+#           def connection_from_alias
+#             return unless database_alias
+#             
+#             MongoMapper.db_config.must.include database_alias
+#             MongoMapper.connections[database_alias]
+#           end
+#         
+#       end
+#       
+#     end
+#   end
+# end

data/lib/mongo_mapper_ext/plugins/default_scope.rb

@@ -0,0 +1,80 @@
+module MongoMapper
+  module Plugins
+    module DefaultScope
+      
+      module ClassMethods
+        def query options = {}
+          super default_scope_options.merge(options)
+        end
+        
+        def find_one options = {}
+          super default_scope_options.merge(options)
+        end
+
+        def find_many options = {}
+          super default_scope_options.merge(options)
+        end
+        
+        def count options={}
+          super default_scope_options.merge(options)
+        end
+        
+        def with_exclusive_scope options = {}, &block
+          Thread.current['mm_with_exclusive_scope'].must_be.nil
+          
+          before = Thread.current['mm_with_exclusive_scope']
+          begin
+            Thread.current['mm_with_exclusive_scope'] = options
+            block.call if block
+          ensure
+            Thread.current['mm_with_exclusive_scope'] = before
+          end
+        end
+        
+        def with_scope options = {}, &block
+          Thread.current['mm_with_exclusive_scope'].must_be.nil
+          
+          before = Thread.current['mm_with_scope']
+          begin
+            options = before.merge options if before
+            Thread.current['mm_with_scope'] = options
+            block.call if block
+          ensure
+            Thread.current['mm_with_scope'] = before
+          end
+        end
+        
+        protected
+          def default_scope options = nil, &block          
+            options.must_be.a [NilClass, Hash]
+            self.write_inheritable_attribute(:default_scope, (options || block))
+          end
+                  
+        private
+          def default_scope_options
+            default_scope = self.read_inheritable_attribute(:default_scope)
+            
+            exclusive_options = Thread.current['mm_with_exclusive_scope']
+            return exclusive_options if exclusive_options
+            
+            options = if default_scope
+              if default_scope.respond_to? :call
+                default_scope.call self
+              else
+                default_scope
+              end
+            else
+              {}
+            end
+            
+            if scope_options = Thread.current['mm_with_scope']
+              options = options.merge scope_options
+            end
+            
+            options
+          end
+      end
+      
+    end
+  end
+end

data/lib/mongo_mapper_ext/plugins/micelaneous.rb

@@ -0,0 +1,97 @@
+module MongoMapper
+  # def self.deferred; @deferred ||= [] end
+  # def self.call_deferred
+  #   deferred.every.call
+  #   deferred.clear
+  # end
+  
+  module Plugins
+    module Micelaneous
+      
+      module InstanceMethods
+        def upsert *args
+          self.class.upsert id, *args
+        end
+      end
+      
+      module ClassMethods
+        # 
+        # Sequentiall :all for big collection
+        # 
+        def all_sequentially &block
+          page, per_page = 1, 5
+          begin
+            results = paginate(:page => page, :per_page => per_page, :order => '_id asc')
+            results.each{|o| block.call o}
+            page += 1
+          end until results.blank? or results.size < per_page
+        end
+        
+        
+        # 
+        # Deferred execution
+        # 
+        # def defer &block
+        #   MongoMapper.deferred << block
+        # end
+        
+        
+        # 
+        # Connect to database_alias specified in config
+        # 
+        def use_database database_alias
+          # defer do
+          database_alias = database_alias.to_s
+          MongoMapper.db_config.must.include database_alias
+      
+          self.connection MongoMapper.connections[database_alias]
+          set_database_name MongoMapper.db_config[database_alias]['name']
+          # end
+        end                
+        
+        
+        # 
+        # shortcut for upsert
+        # 
+        def upsert *args
+          collection.upsert *args          
+        end
+        
+        
+        # 
+        # CounterCache
+        # belongs_to :item, :counter_cashe => true
+        # 
+        def belongs_to association_id, options={}, &extension          
+          options.must_not.include :counter_cashe
+          if options.delete(:counter_cache) || options.delete('counter_cache')
+            association_id = association_id.to_s
+            association_key = "#{association_id}_id"
+            cache_attribute = "#{name.pluralize.underscore}_count"
+            cache_class = association_id.classify.constantize
+            cache_class.keys.must.include cache_attribute            
+            increase_method_name = "increase_#{cache_class.name.underscore}_#{name.pluralize.underscore}_counter"
+            decrease_method_name = "decrease_#{cache_class.name.underscore}_#{name.pluralize.underscore}_counter"
+            
+            define_method increase_method_name do
+              cache_class.upsert self.send(association_key), :$inc => {cache_attribute => 1}
+            end
+            protected increase_method_name            
+            
+            define_method decrease_method_name do
+              cache_class.upsert self.send(association_key), :$inc => {cache_attribute => -1}
+            end
+            protected decrease_method_name
+            
+            after_create increase_method_name
+            after_destroy decrease_method_name
+          end
+          
+          super association_id, options, &extension
+        end
+        
+      end
+      
+    end
+  end
+end

data/lib/mongo_mapper_ext/spec/helper.rb

@@ -0,0 +1,61 @@
+# 
+# Disabling indexes in test environment
+# 
+# if Object.const_defined?(:RAILS_ENV) and Object.const_get(:RAILS_ENV) != 'production'
+#   MongoMapper::Plugins::Indexes::ClassMethods.class_eval do
+#     def ensure_index(name_or_array, options={})
+#     end
+#   end
+# end
+
+if Object.const_defined?(:Crystal) and Crystal.respond_to?(:run)
+  MongoMapper::Plugins::Indexes::ClassMethods.class_eval do        
+    alias_method :original_ensure_index, :ensure_index
+    def ensure_index_stub *args; end
+  end
+  
+  crystal.after :config do |config|
+    if config.production?
+      MongoMapper::Plugins::Indexes::ClassMethods.send :alias_method, :ensure_index, :original_ensure_index
+    else
+      MongoMapper::Plugins::Indexes::ClassMethods.send :alias_method, :ensure_index, :ensure_index_stub
+    end
+  end  
+end
+
+
+# 
+# Clear db before each test
+# 
+Spec::Example::ExampleGroup.class_eval do
+  class << self
+    def with_mongo_mapper
+      before :each do
+        MongoMapper.db_config.each do |db_alias, opt|      
+          db = MongoMapper.databases[db_alias]
+          db.collection_names.each do |name|
+            next if name =~ /^system\./
+            db.collection(name).drop
+          end
+        end
+      end
+    end
+  end
+end
+# Spec::Runner.configure do |config|
+#   config.before(:each) do        
+#   end
+# end
+
+
+# 
+# Silence logger
+# 
+MongoMapper.logger = Logger.new(nil)
+if defined? Paperclip
+  Paperclip.class_eval do
+    def self.logger
+      @logger ||= Logger.new(nil)
+    end
+  end
+end

data/lib/mongo_mapper_ext/view_helpers.rb

@@ -0,0 +1,10 @@
+# 
+# HTML helpers
+# 
+[MongoMapper::EmbeddedDocument, MongoMapper::Document].each do |aclass|
+  aclass.class_eval do
+    def dom_id
+      new_record? ? "new_#{self.class.name.underscore}" : to_param
+    end
+  end
+end

data/lib/mongo_mapper_ext.rb

@@ -0,0 +1,32 @@
+gem 'mongo_mapper', '>=0.8'
+require 'mongo_mapper'
+
+
+[
+  'hacks/fixes',
+  'hacks/time_measuring',
+  'migration',
+  'mongo_mapper',
+  'view_helpers',
+  'db_config',
+  'micelaneous',
+  
+  'plugins/default_scope',
+  # 'plugins/db_config',  
+  'plugins/attributes_cache',
+  'plugins/micelaneous',  
+].each do |file|
+  require "mongo_mapper_ext/#{file}"
+end
+
+module CommonPluginsAddition
+  def self.included(model)
+    model.plugin MongoMapper::Plugins::DefaultScope
+    # model.plugin MongoMapper::Plugins::DbConfig
+    model.plugin MongoMapper::Plugins::AttributesCache
+    model.plugin MongoMapper::Plugins::Micelaneous
+    
+    model.attr_protected :id, :_id, :_type, :created_at, :updated_at
+  end
+end
+MongoMapper::Document.append_inclusions(CommonPluginsAddition)

data/readme.md

@@ -0,0 +1,3 @@
+# Extensions for MongoMapper.
+
+[add readme, for now go to specs]

data/spec/helper.rb

@@ -0,0 +1,8 @@
+dir = File.expand_path(File.dirname(__FILE__))
+lib_dir = File.expand_path("#{dir}/../../../lib")
+$LOAD_PATH << lib_dir unless $LOAD_PATH.include? lib_dir
+
+require 'spec_ext'
+gem 'mongo_mapper', '>=0.8'
+require "mongo_mapper"
+require 'ruby_ext'

data/spec/micelaneous_plugin_spec.rb

@@ -0,0 +1,56 @@
+require "#{File.expand_path(File.dirname(__FILE__))}/helper"
+
+require "mongo_mapper_ext/micelaneous"
+require "mongo_mapper_ext/plugins/micelaneous"
+
+describe "MongoMapper Default Scope" do
+  
+  before :all do
+    class ::Post
+      include MongoMapper::Document      
+      plugin MongoMapper::Plugins::Micelaneous
+      
+      key :comments_count, Integer, :default => 0
+      has_many :comments
+    end
+    
+    class ::Comment
+      include MongoMapper::Document
+      plugin MongoMapper::Plugins::Micelaneous
+      
+      key :post_id
+      belongs_to :post, :counter_cache => true
+    end
+    
+    MongoMapper.database = 'test'
+  end
+  
+  after :all do
+    %w{Post Comment}.each do |obj_name|
+      Object.send :remove_const, obj_name if Object.const_defined? obj_name
+    end
+  end
+  
+  before :each do 
+    [Post, Comment].every.delete_all
+  end
+    
+  it "should increase count of comments" do
+    post = Post.create!
+    comment = post.comments.create!
+    
+    post.reload
+    post.comments_count.should == 1
+  end
+  
+  it "should decrease count of comments" do
+    post = Post.create!
+    comment = post.comments.create!
+    post.reload
+    post.comments_count.should == 1
+    
+    comment.destroy    
+    post.reload
+    post.comments_count.should == 0
+  end
+end

data/spec/micelaneous_spec.rb

@@ -0,0 +1,53 @@
+require "#{File.expand_path(File.dirname(__FILE__))}/helper"
+
+require "mongo_mapper_ext/micelaneous"
+require "mongo_mapper_ext/plugins/micelaneous"
+
+describe "MongoMapper micelaneous" do
+  before :all do
+    @db = Mongo::Connection.new.db('test')
+    MongoMapper.database = 'test'
+  end
+  
+  before :each do
+    @db.collection('test').drop
+    @coll = @db.collection('test')
+  end
+    
+  it "upsert should update" do
+    id = @coll.save :count => 2
+    @coll.upsert id, :$inc => {:count => 1}
+    @coll.find(:_id => id).first['count'].should == 3
+  end
+  
+  it "upsert should set" do
+    id = @coll.save({})
+    @coll.upsert id, :$inc => {:count => 1}
+    @coll.find(:_id => id).first['count'].should == 1
+  end
+    
+  describe "handy upsert" do
+    class ::UpsertSample
+      include MongoMapper::Document
+      plugin MongoMapper::Plugins::Micelaneous
+      
+      key :counter, Integer, :default => 1
+    end  
+    
+    before :each do 
+      @model = UpsertSample.create!
+    end
+  
+    it "class upsert" do
+      UpsertSample.upsert @model.id, :$inc => {:counter => 1}
+      @model.reload
+      @model.counter.should == 2
+    end
+  
+    it "model upsert" do
+      @model.upsert :$inc => {:counter => 1}
+      @model.reload
+      @model.counter.should == 2
+    end
+  end
+end

data/spec/migration_spec.rb

@@ -0,0 +1,79 @@
+require "#{File.expand_path(File.dirname(__FILE__))}/helper"
+
+require "mongo_mapper_ext"
+require "mongo_mapper_ext/spec/helper"
+
+describe "MongoMapper Migration" do
+  with_mongo_mapper
+  
+  Migration = MongoMapper::Migration  
+  
+  before :all do
+    MongoMapper.logger = Logger.new(nil)
+    MongoMapper.db_config = {
+      'global' => {'name' => 'global_test'},
+      'accounts' => {'name' => "accounts_test"}
+    }
+    
+    class ::Sample
+      include MongoMapper::Document
+      use_database :global
+
+      key :name, String
+    end
+        
+    # MongoMapper.call_deferred
+    Migration.logger = nil        
+  end
+
+  after :all do
+    Object.send :remove_const, :Sample if Object.const_defined? :Sample    
+  end
+  
+  before :each do
+    Migration.definitions.clear
+  end
+  
+  it "Shouldn't update if versions are the same" do
+    Migration.update(:global, 0).should be_false
+  end
+  
+  it "migration should provide access to current database" do
+    Sample.count.should == 0    
+    Migration.define :global, 1 do |m|
+      m.up do |db|
+        Sample.create :name => 'name'
+        coll = db.collection 'samples'
+        coll.find.count.should == 1
+      end
+    end
+  
+    Migration.update(:global, 1).should be_true    
+    Sample.count.should == 1
+  end
+
+  it "increase_db_version" do
+    Sample.count.should == 0
+    Migration.define :global, 1 do |m|
+      m.up{Sample.create :name => 'name'}
+    end
+  
+    Migration.update(:global, 1).should be_true    
+    Sample.count.should == 1
+    Migration.metadata(MongoMapper.databases[:global]).version.should == 1
+  end
+  
+  it "decrease_db_version" do    
+    Migration.define :global, 1 do |m|
+      m.up{Sample.create :name => 'name'}
+      m.down{Sample.destroy_all}
+    end
+    Migration.update(:global, 1).should be_true
+    Sample.count.should == 1
+  
+    Migration.update(:global, 0).should be_true
+    Sample.count.should == 0
+    Migration.metadata(MongoMapper.databases[:global]).version.should == 0
+  end
+    
+end

data/spec/mm_spec.rb

@@ -0,0 +1,35 @@
+require 'mongo_mapper'
+
+describe "Special test to check some errors in the MongoMapper itself" do  
+  before :all do
+    MongoMapper.database = 'test'
+    
+    class ::AnObject
+      include MongoMapper::Document
+      
+      key :owner_name, String
+      key :viewers, Array
+      
+      def owner_name= owner_name
+        viewers.delete self.owner_name
+        viewers << owner_name        
+        # write_attribute :owner_name, owner_name
+        super owner_name
+        owner_name
+      end
+    end
+  end
+  
+  after :all do
+    Object.send :remove_const, :AnObject if Object.const_defined? :AnObject
+  end
+    
+  it do
+    o = AnObject.new
+    o.owner_name = 'user'
+    o.save!
+    
+    # o = AnObject.first # don't use reload, it willn't catch this error
+    o.viewers.should == %w{user}
+  end
+end

data/spec/scope_spec.rb

@@ -0,0 +1,124 @@
+require "#{File.expand_path(File.dirname(__FILE__))}/helper"
+
+require "mongo_mapper_ext/plugins/default_scope"
+
+describe "MongoMapper Default Scope" do
+  before :all do
+    MongoMapper.database = 'test'
+  end
+  
+  before :each do 
+    class ::ScopeSample
+      include MongoMapper::Document
+      plugin MongoMapper::Plugins::DefaultScope
+      
+      key :name, String
+      key :_type, String
+    end
+    
+    ScopeSample.delete_all
+    ScopeSample.create! :name => 'a'
+    ScopeSample.create! :name => 'b'
+  end
+  
+  after :each do 
+    Object.send :remove_const, :ScopeSample if Object.const_defined? :ScopeSample
+  end
+    
+  it "should not affect objects without default_scope" do
+    ScopeSample.count.should == 2
+    ScopeSample.all.count.should == 2
+  end
+  
+  it "default_scope" do
+    ScopeSample.send :default_scope, :name => 'a'
+    ScopeSample.count.should == 1
+    ScopeSample.all.count.should == 1
+    ScopeSample.first.name.should == 'a'
+  end
+  
+  it "default_scope as block" do
+    ScopeSample.send :default_scope do
+      {:name => 'a'}
+    end
+    ScopeSample.count.should == 1
+    ScopeSample.all.count.should == 1
+    ScopeSample.first.name.should == 'a'
+  end
+  
+  it "default_scope should be inherited" do    
+    ScopeSample.send :default_scope, :name => 'a'
+    class ScopeSampleAncestor < ScopeSample
+    end
+    
+    ScopeSampleAncestor.create! :name => 'a'
+    ScopeSampleAncestor.create! :name => 'b'
+
+    ScopeSampleAncestor.count.should == 1
+    ScopeSampleAncestor.all.count.should == 1
+    ScopeSampleAncestor.first.name.should == 'a'
+  end
+  
+  it "ancestors should be able to override default_scope" do
+    ScopeSample.send :default_scope, :name => 'a'
+    class ScopeSampleAncestor2 < ScopeSample
+      default_scope :name => 'b'
+    end
+    
+    ScopeSampleAncestor2.create! :name => 'a'
+    ScopeSampleAncestor2.create! :name => 'b'
+        
+    ScopeSampleAncestor2.count.should == 1
+    ScopeSampleAncestor2.all.count.should == 1
+    ScopeSampleAncestor2.first.name.should == 'b'
+  end
+  
+  it "shouldn't allow to nest with_exclusive_scope" do
+    lambda{
+      ScopeSample.with_exclusive_scope do
+        ScopeSample.with_exclusive_scope{}
+      end
+    }.should raise_error(AssertionError)
+    
+    lambda{
+      ScopeSample.with_exclusive_scope do
+        ScopeSample.with_scope{}
+      end
+    }.should raise_error(AssertionError)
+  end
+      
+  it "with_exclusive_scope should clear other scopes" do
+    ScopeSample.send :default_scope, :name => 'a'
+    
+    ScopeSample.with_scope :name => 'a' do
+      ScopeSample.with_exclusive_scope :name => 'b' do        
+        ScopeSample.count.should == 1
+        ScopeSample.first.name.should == 'b'
+      end
+    end
+  end
+  
+  it "with_scope" do
+    ScopeSample.with_scope :name => 'a' do
+      ScopeSample.count.should == 1
+      ScopeSample.first.name.should == 'a'
+    end
+  end
+  
+  it "with scope should merge scope" do
+    ScopeSample.class_eval do
+      key :name2, String
+      key :name3, String
+      default_scope :name => 'a'
+    end
+    ScopeSample.create! :name => 'a', :name2 => 'a2', :name3 => 'a3'
+    ScopeSample.create! :name => 'b', :name2 => 'b2', :name3 => 'b3'
+    
+    ScopeSample.with_scope :name2 => 'a2' do
+      ScopeSample.with_scope :name3 => 'a3' do
+        ScopeSample.count.should == 1
+        ScopeSample.first.name2.should == 'a2'
+      end
+    end
+  end
+end

data/spec/spec.opts

@@ -0,0 +1,4 @@
+--colour
+--format progress
+--loadby mtime
+--reverse

metadata

@@ -0,0 +1,102 @@
+--- !ruby/object:Gem::Specification 
+name: mongo_mapper_ext
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Alexey Petrushin
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-10-11 00:00:00 +04:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: mongo_mapper
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+description: Extensions for MongoMapper
+email: 
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- Rakefile
+- readme.md
+- lib/mongo_mapper_ext/db_config.rb
+- lib/mongo_mapper_ext/hacks/fixes.rb
+- lib/mongo_mapper_ext/hacks/time_measuring.rb
+- lib/mongo_mapper_ext/micelaneous.rb
+- lib/mongo_mapper_ext/migrate.rake
+- lib/mongo_mapper_ext/migration.rb
+- lib/mongo_mapper_ext/mongo_mapper.rb
+- lib/mongo_mapper_ext/plugins/attributes_cache.rb
+- lib/mongo_mapper_ext/plugins/db_config.rb
+- lib/mongo_mapper_ext/plugins/default_scope.rb
+- lib/mongo_mapper_ext/plugins/micelaneous.rb
+- lib/mongo_mapper_ext/spec/helper.rb
+- lib/mongo_mapper_ext/view_helpers.rb
+- lib/mongo_mapper_ext.rb
+- spec/helper.rb
+- spec/micelaneous_plugin_spec.rb
+- spec/micelaneous_spec.rb
+- spec/migration_spec.rb
+- spec/mm_spec.rb
+- spec/scope_spec.rb
+- spec/spec.opts
+has_rdoc: true
+homepage: http://github.com/alexeypetrushin/mongo_mapper_ext
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Extensions for MongoMapper
+test_files: []
+