mongo_mapper_ign 0.7.4

data/lib/mongo_mapper/plugins/logger.rb

@@ -0,0 +1,17 @@
+module MongoMapper
+  module Plugins
+    module Logger
+      module ClassMethods
+        def logger
+          MongoMapper.logger
+        end
+      end
+
+      module InstanceMethods
+        def logger
+          self.class.logger
+        end
+      end
+    end
+  end
+end

data/lib/mongo_mapper/plugins/modifiers.rb

@@ -0,0 +1,107 @@
+module MongoMapper
+  module Plugins
+    module Modifiers
+      module ClassMethods
+        def increment(*args)
+          modifier_update('$inc', args)
+        end
+
+        def decrement(*args)
+          criteria, keys = criteria_and_keys_from_args(args)
+          values, to_decrement = keys.values, {}
+          keys.keys.each_with_index { |k, i| to_decrement[k] = -values[i].abs }
+          collection.update(criteria, {'$inc' => to_decrement}, :multi => true)
+        end
+
+        def set(*args)
+          modifier_update('$set', args)
+        end
+
+        def unset(*args)
+          if args[0].is_a?(Hash)
+            criteria, keys = args.shift, args
+          else
+            keys, ids = args.partition { |arg| arg.is_a?(Symbol) }
+            criteria = {:id => ids}
+          end
+
+          criteria  = to_criteria(criteria)
+          modifiers = keys.inject({}) { |hash, key| hash[key] = 1; hash }
+          collection.update(criteria, {'$unset' => modifiers}, :multi => true)
+        end
+
+        def push(*args)
+          modifier_update('$push', args)
+        end
+
+        def push_all(*args)
+          modifier_update('$pushAll', args)
+        end
+
+        def add_to_set(*args)
+          modifier_update('$addToSet', args)
+        end
+        alias push_uniq add_to_set
+
+        def pull(*args)
+          modifier_update('$pull', args)
+        end
+
+        def pull_all(*args)
+          modifier_update('$pullAll', args)
+        end
+
+        def pop(*args)
+          modifier_update('$pop', args)
+        end
+
+        private
+          def modifier_update(modifier, args)
+            criteria, keys = criteria_and_keys_from_args(args)
+            collection.update(criteria, {modifier => keys}, :multi => true)
+          end
+
+          def criteria_and_keys_from_args(args)
+            keys     = args.pop
+            criteria = args[0].is_a?(Hash) ? args[0] : {:id => args}
+            [to_criteria(criteria), keys]
+          end
+      end
+
+      module InstanceMethods
+        def unset(*keys)
+          self.class.unset(id, *keys)
+        end
+
+        def increment(hash)
+          self.class.increment(id, hash)
+        end
+
+        def decrement(hash)
+          self.class.decrement(id, hash)
+        end
+
+        def set(hash)
+          self.class.set(id, hash)
+        end
+
+        def push(hash)
+          self.class.push(id, hash)
+        end
+
+        def pull(hash)
+          self.class.pull(id, hash)
+        end
+
+        def add_to_set(hash)
+          self.class.push_uniq(id, hash)
+        end
+        alias push_uniq add_to_set
+
+        def pop(hash)
+          self.class.pop(id, hash)
+        end
+      end
+    end
+  end
+end

data/lib/mongo_mapper/plugins/pagination.rb

@@ -0,0 +1,24 @@
+module MongoMapper
+  module Plugins
+    module Pagination
+      module ClassMethods
+        def per_page
+          25
+        end
+
+        def paginate(options)
+          per_page      = options.delete(:per_page) || self.per_page
+          page          = options.delete(:page)
+          total_entries = count(options)
+          pagination    = Pagination::Proxy.new(total_entries, page, per_page)
+
+          options.update(:limit => pagination.limit, :skip => pagination.skip)
+          pagination.subject = find_many(options)
+          pagination
+        end
+      end
+    end
+  end
+end
+
+require 'mongo_mapper/plugins/pagination/proxy'

data/lib/mongo_mapper/plugins/pagination/proxy.rb

@@ -0,0 +1,72 @@
+module MongoMapper
+  module Plugins
+    module Pagination
+      class Proxy
+        instance_methods.each { |m| undef_method m unless m =~ /(^__|^nil\?$|^send$|respond_to\?|proxy_|^object_id$)/ }
+                
+        attr_accessor :subject
+        attr_reader :total_entries, :per_page, :current_page
+        alias limit per_page
+
+        def initialize(total_entries, current_page, per_page=nil)
+          @total_entries    = total_entries.to_i
+          self.per_page     = per_page
+          self.current_page = current_page
+        end
+
+        def total_pages
+          (total_entries / per_page.to_f).ceil
+        end
+
+        def out_of_bounds?
+          current_page > total_pages
+        end
+
+        def previous_page
+          current_page > 1 ? (current_page - 1) : nil
+        end
+
+        def next_page
+          current_page < total_pages ? (current_page + 1) : nil
+        end
+
+        def skip
+          (current_page - 1) * per_page
+        end
+        alias offset skip # for will paginate support
+
+        def send(method, *args, &block)
+          if respond_to?(method)
+            super
+          else
+            subject.send(method, *args, &block)
+          end
+        end
+
+        def ===(other)
+          other === subject
+        end
+
+        def method_missing(name, *args, &block)
+          @subject.send(name, *args, &block)
+        end
+        
+        def respond_to?(name, *args, &block)
+          super || @subject.respond_to?(name, *args, &block)
+        end
+        
+        private
+          def per_page=(value)
+            value = 25 if value.blank?
+            @per_page = value.to_i
+          end
+
+          def current_page=(value)
+            value = value.to_i
+            value = 1 if value < 1
+            @current_page = value
+          end
+      end
+    end
+  end
+end

data/lib/mongo_mapper/plugins/persistence.rb

@@ -0,0 +1,68 @@
+module MongoMapper
+  module Plugins
+    module Persistence
+      module ClassMethods
+        class Unsupported < MongoMapperError; end
+
+        def connection(mongo_connection=nil)
+          not_supported_by_embedded
+          if mongo_connection.nil?
+            @connection ||= MongoMapper.connection
+          else
+            @connection = mongo_connection
+          end
+          @connection
+        end
+
+        def set_database_name(name)
+          not_supported_by_embedded
+          @database_name = name
+        end
+
+        def database_name
+          not_supported_by_embedded
+          @database_name
+        end
+
+        def database
+          not_supported_by_embedded
+          if database_name.nil?
+            MongoMapper.database
+          else
+            connection.db(database_name)
+          end
+        end
+
+        def set_collection_name(name)
+          not_supported_by_embedded
+          @collection_name = name
+        end
+
+        def collection_name
+          not_supported_by_embedded
+          @collection_name ||= self.to_s.tableize.gsub(/\//, '.')
+        end
+
+        def collection
+          not_supported_by_embedded
+          database.collection(collection_name)
+        end
+
+        private
+          def not_supported_by_embedded
+            raise Unsupported.new('This is not supported for embeddable documents at this time.') if embeddable?
+          end
+      end
+
+      module InstanceMethods
+        def collection
+          _root_document.class.collection
+        end
+
+        def database
+          _root_document.class.database
+        end
+      end
+    end
+  end
+end

data/lib/mongo_mapper/plugins/protected.rb

@@ -0,0 +1,45 @@
+module MongoMapper
+  module Plugins
+    module Protected
+      module ClassMethods
+        def attr_protected(*attrs)
+          self.write_inheritable_attribute(:attr_protected, Set.new(attrs) + (protected_attributes || []))
+        end
+
+        def protected_attributes
+          self.read_inheritable_attribute(:attr_protected)
+        end
+
+        def key(*args)
+          key = super
+          attr_protected key.name.to_sym if key.options[:protected]
+          key
+        end
+      end
+
+      module InstanceMethods
+        def assign(attrs={})
+          super(filter_protected_attrs(attrs))
+        end
+
+        def update_attributes(attrs={})
+          super(filter_protected_attrs(attrs))
+        end
+
+        def update_attributes!(attrs={})
+          super(filter_protected_attrs(attrs))
+        end
+
+        def protected_attributes
+          self.class.protected_attributes
+        end
+
+        protected
+          def filter_protected_attrs(attrs)
+            return attrs if protected_attributes.blank? || attrs.blank?
+            attrs.dup.delete_if { |key, val| protected_attributes.include?(key.to_sym) }
+          end
+      end
+    end
+  end
+end

data/lib/mongo_mapper/plugins/rails.rb

@@ -0,0 +1,57 @@
+module MongoMapper
+  module Plugins
+    module Rails
+      def self.configure(model)
+        model.extend ActiveModel::Naming if defined?(ActiveModel)
+      end
+
+      module InstanceMethods
+        def to_param
+          id.to_s if persisted?
+        end
+
+        def to_model
+          self
+        end
+
+        def to_key
+          [id] if persisted?
+        end
+
+        def new_record?
+          new?
+        end
+
+        def read_attribute(name)
+          self[name]
+        end
+
+        def read_attribute_before_typecast(name)
+          read_key_before_typecast(name)
+        end
+
+        def write_attribute(name, value)
+          self[name] = value
+        end
+      end
+
+      module ClassMethods
+        def has_one(*args)
+          one(*args)
+        end
+
+        def has_many(*args)
+          many(*args)
+        end
+
+        def column_names
+          keys.keys
+        end
+
+        def human_name
+          self.name.demodulize.titleize
+        end
+      end
+    end
+  end
+end

data/lib/mongo_mapper/plugins/serialization.rb

@@ -0,0 +1,91 @@
+require 'active_support/json'
+
+module MongoMapper
+  module Plugins
+    module Serialization
+      def self.configure(model)
+        model.class_eval { cattr_accessor :include_root_in_json, :instance_writer => true }
+      end
+      
+      module InstanceMethods
+        def to_xml(options = {}, &block)
+          options[:dasherize] = false unless options.has_key?(:dasherize)
+          serializer = Serialization::XmlSerializer.new(self, options)
+          block_given? ? serializer.to_s(&block) : serializer.to_s
+        end
+        
+        def as_json options={}
+          options ||= {}
+          unless options[:only]
+            methods = [options.delete(:methods)].flatten.compact
+            methods << :id
+            options[:methods] = methods.uniq
+          end
+
+          except = [options.delete(:except)].flatten.compact
+          except << :_id
+          options[:except] = except
+
+          # Direct rip from Rails 3 ActiveModel Serialization (#serializable_hash)
+          hash = begin
+            options[:only]   = Array.wrap(options[:only]).map { |n| n.to_s }
+            options[:except] = Array.wrap(options[:except]).map { |n| n.to_s }
+
+            attribute_names = attributes.keys.sort
+            if options[:only].any?
+              attribute_names &= options[:only]
+            elsif options[:except].any?
+              attribute_names -= options[:except]
+            end
+
+            method_names = Array.wrap(options[:methods]).inject([]) do |methods, name|
+              methods << name if respond_to?(name.to_s)
+              methods
+            end
+
+            (attribute_names + method_names).inject({}) { |hash, name|
+              hash[name] = send(name)
+              hash
+            }
+          end
+          # End rip
+          
+          options.delete(:only) if options[:only].nil? or options[:only].empty?
+
+          hash.each do |key, value|
+            if value.is_a?(Array)
+              hash[key] = value.map do |item|
+                item.respond_to?(:as_json) ? item.as_json(options) : item
+              end
+            elsif value.is_a? BSON::ObjectID
+              hash[key] = value.to_s
+            elsif value.respond_to?(:as_json)
+              hash[key] = value.as_json(options)
+            end
+          end
+          
+          # Replicate Rails 3 naming - and also bin anytihng after : for use in our dynamic classes from unit tests
+          hash = { ActiveSupport::Inflector.underscore(ActiveSupport::Inflector.demodulize(self)).gsub(/:.*/,'') => hash } if include_root_in_json
+          hash
+        end
+      end
+
+      module ClassMethods
+        def from_json(json)
+          self.new(ActiveSupport::JSON.decode(json))
+        end
+        
+        def from_xml(xml)
+          # instance = self.new
+          # instance.attributes = self.new
+          # instance
+          instance.attributes = self.new(Hash.from_xml(xml).values.first)
+        end        
+      end
+
+    end
+  end
+end
+
+require 'mongo_mapper/plugins/serialization/xml_serializer'
+require 'mongo_mapper/plugins/serialization/array'