lookout-mongo_mapper 0.11.3

data/lib/mongo_mapper/extensions/date.rb

@@ -0,0 +1,25 @@
+# encoding: UTF-8
+module MongoMapper
+  module Extensions
+    module Date
+      def to_mongo(value)
+        if value.nil? || value == ''
+          nil
+        else
+          date = value.is_a?(::Date) || value.is_a?(::Time) ? value : ::Date.parse(value.to_s)
+          ::Time.utc(date.year, date.month, date.day)
+        end
+      rescue
+        nil
+      end
+
+      def from_mongo(value)
+        value.to_date if value.present?
+      end
+    end
+  end
+end
+
+class Date
+  extend MongoMapper::Extensions::Date
+end

data/lib/mongo_mapper/extensions/float.rb

@@ -0,0 +1,14 @@
+# encoding: UTF-8
+module MongoMapper
+  module Extensions
+    module Float
+      def to_mongo(value)
+        value.blank? ? nil : value.to_f
+      end
+    end
+  end
+end
+
+class Float
+  extend MongoMapper::Extensions::Float
+end

data/lib/mongo_mapper/extensions/hash.rb

@@ -0,0 +1,14 @@
+# encoding: UTF-8
+module MongoMapper
+  module Extensions
+    module Hash
+      def from_mongo(value)
+        HashWithIndifferentAccess.new(value || {})
+      end
+    end
+  end
+end
+
+class Hash
+  extend MongoMapper::Extensions::Hash
+end

data/lib/mongo_mapper/extensions/integer.rb

@@ -0,0 +1,19 @@
+# encoding: UTF-8
+module MongoMapper
+  module Extensions
+    module Integer
+      def to_mongo(value)
+        value_to_i = value.to_i
+        if value_to_i == 0 && value != value_to_i
+          value.to_s =~ /^(0x|0b)?0+/ ? 0 : nil
+        else
+          value_to_i
+        end
+      end
+    end
+  end
+end
+
+class Integer
+  extend MongoMapper::Extensions::Integer
+end

data/lib/mongo_mapper/extensions/kernel.rb

@@ -0,0 +1,9 @@
+# encoding: UTF-8
+module Kernel
+  # Returns the object's singleton class.
+  def singleton_class
+    class << self
+      self
+    end
+  end unless respond_to?(:singleton_class) # exists in 1.9.2
+end

data/lib/mongo_mapper/extensions/nil_class.rb

@@ -0,0 +1,18 @@
+# encoding: UTF-8
+module MongoMapper
+  module Extensions
+    module NilClass
+      def to_mongo(value)
+        value
+      end
+
+      def from_mongo(value)
+        value
+      end
+    end
+  end
+end
+
+class NilClass
+  include MongoMapper::Extensions::NilClass
+end

data/lib/mongo_mapper/extensions/object.rb

@@ -0,0 +1,26 @@
+# encoding: UTF-8
+module MongoMapper
+  module Extensions
+    module Object
+      extend ActiveSupport::Concern
+
+      module ClassMethods
+        def to_mongo(value)
+          value
+        end
+
+        def from_mongo(value)
+          value
+        end
+      end
+
+      def to_mongo
+        self.class.to_mongo(self)
+      end
+    end
+  end
+end
+
+class Object
+  include MongoMapper::Extensions::Object
+end

data/lib/mongo_mapper/extensions/object_id.rb

@@ -0,0 +1,32 @@
+# encoding: UTF-8
+module MongoMapper
+  module Extensions
+    module ObjectId
+      def to_mongo(value)
+        Plucky.to_object_id(value)
+      end
+
+      def from_mongo(value)
+        value
+      end
+    end
+  end
+end
+
+class ObjectId
+  extend MongoMapper::Extensions::ObjectId
+end
+
+class BSON::ObjectId
+  alias_method :original_to_json, :to_json
+
+  def as_json(options=nil)
+    to_s
+  end
+
+  def to_json(options = nil)
+    as_json.to_json
+  end
+
+  alias to_str to_s
+end

data/lib/mongo_mapper/extensions/set.rb

@@ -0,0 +1,20 @@
+# encoding: UTF-8
+require 'set'
+
+module MongoMapper
+  module Extensions
+    module Set
+      def to_mongo(value)
+        value.to_a
+      end
+
+      def from_mongo(value)
+        (value || []).to_set
+      end
+    end
+  end
+end
+
+class Set
+  extend MongoMapper::Extensions::Set
+end

data/lib/mongo_mapper/extensions/string.rb

@@ -0,0 +1,18 @@
+# encoding: UTF-8
+module MongoMapper
+  module Extensions
+    module String
+      def to_mongo(value)
+        value.nil? ? nil : value.to_s
+      end
+
+      def from_mongo(value)
+        value.nil? ? nil : value.to_s
+      end
+    end
+  end
+end
+
+class String
+  extend MongoMapper::Extensions::String
+end

data/lib/mongo_mapper/extensions/time.rb

@@ -0,0 +1,28 @@
+# encoding: UTF-8
+module MongoMapper
+  module Extensions
+    module Time
+      def to_mongo(value)
+        if value.nil? || value == ''
+          nil
+        else
+          time_class = ::Time.try(:zone).present? ? ::Time.zone : ::Time
+          time = value.is_a?(::Time) ? value : time_class.parse(value.to_s)
+          at(time.to_f).utc if time # ensure milliseconds are preserved with to_f (issue #308)
+        end
+      end
+
+      def from_mongo(value)
+        if ::Time.try(:zone).present? && value.present?
+          value.in_time_zone(::Time.zone)
+        else
+          value
+        end
+      end
+    end
+  end
+end
+
+class Time
+  extend MongoMapper::Extensions::Time
+end

data/lib/mongo_mapper/locale/en.yml

@@ -0,0 +1,5 @@
+en:
+  mongo_mapper:
+    errors:
+      messages:
+        taken: "has already been taken"

data/lib/mongo_mapper/middleware/identity_map.rb

@@ -0,0 +1,16 @@
+module MongoMapper
+  module Middleware
+    class IdentityMap
+      def initialize(app)
+        @app = app
+      end
+
+      def call(env)
+        MongoMapper::Plugins::IdentityMap.clear
+        @app.call(env)
+      ensure
+        MongoMapper::Plugins::IdentityMap.clear
+      end
+    end
+  end
+end

data/lib/mongo_mapper/plugins.rb

@@ -0,0 +1,22 @@
+# encoding: UTF-8
+module MongoMapper
+  module Plugins
+    include ActiveSupport::DescendantsTracker
+
+    def plugins
+      @plugins ||= []
+    end
+
+    def plugin(mod)
+      raise ArgumentError,  "Plugins must extend ActiveSupport::Concern" unless ActiveSupport::Concern === mod
+      include mod
+      direct_descendants.each {|model| model.send(:include, mod) }
+      plugins << mod
+    end
+
+    def included(base = nil, &block)
+      direct_descendants << base if base
+      super
+    end
+  end
+end

data/lib/mongo_mapper/plugins/accessible.rb

@@ -0,0 +1,52 @@
+module MongoMapper
+  module Plugins
+    module Accessible
+      extend ActiveSupport::Concern
+
+      included do
+        class_attribute :_accessible_attributes
+      end
+
+      module ClassMethods
+        def attr_accessible(*attrs)
+          raise AccessibleOrProtected.new(name) if try(:protected_attributes?)
+          self._accessible_attributes = Set.new(attrs) + (_accessible_attributes || [])
+        end
+
+        def accessible_attributes(*)
+          _accessible_attributes
+        end
+
+        def accessible_attributes?
+          _accessible_attributes?
+        end
+      end
+
+      def attributes=(attrs={})
+        super(filter_inaccessible_attrs(attrs))
+      end
+
+      def update_attributes(attrs={})
+        super(filter_inaccessible_attrs(attrs))
+      end
+
+      def update_attributes!(attrs={})
+        super(filter_inaccessible_attrs(attrs))
+      end
+
+      def accessible_attributes(*args)
+        self.class.accessible_attributes(*args)
+      end
+
+      def accessible_attributes?
+        self.class.accessible_attributes?
+      end
+
+      protected
+        def filter_inaccessible_attrs(attrs)
+          return attrs if !accessible_attributes? || attrs.blank?
+          attrs.dup.delete_if { |key, val| !accessible_attributes.include?(key.to_sym) }
+        end
+    end
+  end
+end

data/lib/mongo_mapper/plugins/active_model.rb

@@ -0,0 +1,18 @@
+# encoding: UTF-8
+module MongoMapper
+  module Plugins
+    module ActiveModel
+      extend ActiveSupport::Concern
+
+      include ::ActiveModel::Conversion
+      include ::ActiveModel::Serialization
+      include ::ActiveModel::Serializers::Xml
+      include ::ActiveModel::Serializers::JSON
+
+      included do
+        extend ::ActiveModel::Naming
+        extend ::ActiveModel::Translation
+      end
+    end
+  end
+end

data/lib/mongo_mapper/plugins/associations.rb

@@ -0,0 +1,90 @@
+# encoding: UTF-8
+module MongoMapper
+  module Plugins
+    module Associations
+      extend ActiveSupport::Concern
+
+      module ClassMethods
+        def inherited(subclass)
+          subclass.associations = associations.dup
+          super
+        end
+
+        def belongs_to(association_id, options={}, &extension)
+          create_association(BelongsToAssociation.new(association_id, options, &extension))
+        end
+
+        def many(association_id, options={}, &extension)
+          create_association(ManyAssociation.new(association_id, options, &extension))
+        end
+
+        def one(association_id, options={}, &extension)
+          create_association(OneAssociation.new(association_id, options, &extension))
+        end
+
+        def associations
+          @associations ||= {}
+        end
+
+        def associations=(hash)
+          @associations = hash
+        end
+
+        def associations_module_defined?
+          if method(:const_defined?).arity == 1 # Ruby 1.9 compat check
+            const_defined?('MongoMapperAssociations')
+          else
+            const_defined?('MongoMapperAssociations', false)
+          end
+        end
+
+        def associations_module
+          if associations_module_defined?
+            const_get 'MongoMapperAssociations'
+          else
+            Module.new.tap do |m|
+              const_set 'MongoMapperAssociations', m
+              include m
+            end
+          end
+        end
+
+        private
+          def create_association(association)
+            associations[association.name] = association
+            association.setup(self)
+          end
+      end
+
+      def associations
+        self.class.associations
+      end
+
+      def embedded_associations
+        associations.values.select { |assoc| assoc.embeddable? }
+      end
+
+      def build_proxy(association)
+        proxy = association.proxy_class.new(self, association)
+        self.instance_variable_set(association.ivar, proxy)
+
+        proxy
+      end
+
+      def get_proxy(association)
+        unless proxy = self.instance_variable_get(association.ivar)
+          proxy = build_proxy(association)
+        end
+        proxy
+      end
+
+      def save_to_collection(options={})
+        super if defined?(super)
+        associations.each do |association_name, association|
+          proxy = get_proxy(association)
+          proxy.save_to_collection(options) if proxy.proxy_respond_to?(:save_to_collection) && association.autosave?
+        end
+      end
+    end
+  end
+end