Contents

# encoding: UTF-8
module MongoMapper
  module Plugins
    module Keys
      class Key
        attr_accessor :name, :type, :options, :default_value

        def initialize(*args)
          options = args.extract_options!
          @name, @type = args.shift.to_s, args.shift
          self.options = (options || {}).symbolize_keys
          self.default_value = self.options[:default]
        end

        def ==(other)
          @name == other.name && @type == other.type
        end

        def embeddable?
          return false unless type.respond_to?(:embeddable?)
          type.embeddable?
        end

        def can_default_id?
          type && (type == ObjectId || type == BSON::ObjectId || type == String)
        end

        def number?
          type == Integer || type == Float
        end

        def get(value)
          if value.nil? && !default_value.nil?
            if default_value.respond_to?(:call)
              return default_value.call
            else
              # Using Marshal is easiest way to get a copy of mutable objects
              # without getting an error on immutable objects
              return Marshal.load(Marshal.dump(default_value))
            end
          end

          if options[:typecast].present?
            type.from_mongo(value).map! { |v| typecast_class.from_mongo(v) }
          else
            type.from_mongo(value)
          end
        end

        def set(value)
          type.to_mongo(value).tap do |values|
            if options[:typecast].present?
              values.map! { |v| typecast_class.to_mongo(v) }
            end
          end
        end

        private
          def typecast_class
            @typecast_class ||= options[:typecast].constantize
          end
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
mongo_mapper-0.9.2	lib/mongo_mapper/plugins/keys/key.rb
mongo_mapper-0.9.1	lib/mongo_mapper/plugins/keys/key.rb
mongo_mapper-0.9.0	lib/mongo_mapper/plugins/keys/key.rb