dm-core 0.9.2

data/lib/dm-core/hook.rb

@@ -0,0 +1,11 @@
+module DataMapper
+  module Hook
+    def self.included(model)
+      model.class_eval %{
+        include Extlib::Hook
+        register_instance_hooks :save, :create, :update, :destroy
+      }
+    end
+  end
+  DataMapper::Resource.append_inclusions Hook
+end # module DataMapper

data/lib/dm-core/identity_map.rb

@@ -0,0 +1,45 @@
+module DataMapper
+
+  # Tracks objects to help ensure that each object gets loaded only once.
+  # See: http://www.martinfowler.com/eaaCatalog/identityMap.html
+  class IdentityMap
+    # Get a resource from the IdentityMap
+    def get(key)
+      @cache[key]
+    end
+
+    alias [] get
+
+    # Add a resource to the IdentityMap
+    def set(key, resource)
+      @second_level_cache.set(key, resource) if @second_level_cache
+      @cache[key] = resource
+    end
+
+    alias []= set
+
+    # Remove a resource from the IdentityMap
+    def delete(key)
+      @second_level_cache.delete(key) if @second_level_cache
+      @cache.delete(key)
+    end
+
+    private
+
+    def initialize(second_level_cache = nil)
+      @cache = if @second_level_cache = second_level_cache
+        Hash.new { |h,key| h[key] = @second_level_cache.get(key) }
+      else
+        Hash.new
+      end
+    end
+
+    def cache
+      @cache
+    end
+
+    def method_missing(method, *args, &block)
+      cache.__send__(method, *args, &block)
+    end
+  end # class IdentityMap
+end # module DataMapper

data/lib/dm-core/is.rb

@@ -0,0 +1,16 @@
+module DataMapper
+  module Is
+
+    def is(plugin,*pars)
+      generator_method = "is_#{plugin}".to_sym
+
+      if self.respond_to?(generator_method)
+        self.send(generator_method,*pars)
+      else
+        raise PluginNotFoundError, "could not find plugin named #{plugin}"
+      end
+    end
+
+    Model.send(:include, self)
+  end # module Is
+end # module DataMapper

data/lib/dm-core/logger.rb

@@ -0,0 +1,233 @@
+require "time" # httpdate
+# ==== Public DataMapper Logger API
+#
+# Logger taken from Merb :)
+#
+# To replace an existing logger with a new one:
+#  DataMapper::Logger.set_log(log{String, IO},level{Symbol, String})
+#
+# Available logging levels are
+#   DataMapper::Logger::{ Fatal, Error, Warn, Info, Debug }
+#
+# Logging via:
+#   DataMapper.logger.fatal(message<String>)
+#   DataMapper.logger.error(message<String>)
+#   DataMapper.logger.warn(message<String>)
+#   DataMapper.logger.info(message<String>)
+#   DataMapper.logger.debug(message<String>)
+#
+# Flush the buffer to
+#   DataMapper.logger.flush
+#
+# Remove the current log object
+#   DataMapper.logger.close
+#
+# ==== Private DataMapper Logger API
+#
+# To initialize the logger you create a new object, proxies to set_log.
+#   DataMapper::Logger.new(log{String, IO},level{Symbol, String})
+#
+# Logger will not create the file until something is actually logged
+# This avoids file creation on DataMapper init when it creates the
+# default logger.
+module DataMapper
+
+  class << self #:nodoc:
+    attr_accessor :logger
+  end
+
+  class Logger
+
+    attr_accessor :aio
+    attr_accessor :delimiter
+    attr_reader   :level
+    attr_reader   :buffer
+    attr_reader   :log
+
+    # @note
+    #   Ruby (standard) logger levels:
+    #     off:   absolutely nothing
+    #     fatal: an unhandleable error that results in a program crash
+    #     error: a handleable error condition
+    #     warn:  a warning
+    #     info:  generic (useful) information about system operation
+    #     debug: low-level information for developers
+    #
+    #   DataMapper::Logger::LEVELS[:off, :fatal, :error, :warn, :info, :debug]
+    LEVELS =
+    {
+      :off   => 99999,
+      :fatal => 7,
+      :error => 6,
+      :warn  => 4,
+      :info  => 3,
+      :debug => 0
+    }
+
+    def level=(new_level)
+      @level = LEVELS[new_level.to_sym]
+      reset_methods(:close)
+    end
+
+    private
+
+    # The idea here is that instead of performing an 'if' conditional check on
+    # each logging we do it once when the log object is setup
+    def set_write_method
+      @log.instance_eval do
+
+        # Determine if asynchronous IO can be used
+        def aio?
+          @aio = !RUBY_PLATFORM.match(/java|mswin/) &&
+          !(@log == STDOUT) &&
+          @log.respond_to?(:write_nonblock)
+        end
+
+        # Define the write method based on if aio an be used
+        undef write_method if defined? write_method
+        if aio?
+          alias :write_method :write_nonblock
+        else
+          alias :write_method :write
+        end
+      end
+    end
+
+    def initialize_log(log)
+      close if @log # be sure that we don't leave open files laying around.
+      @log = log || "log/dm.log"
+    end
+
+    def reset_methods(o_or_c)
+      if o_or_c == :open
+        alias internal_push push_opened
+      elsif o_or_c == :close
+        alias internal_push push_closed
+      end
+    end
+
+    def push_opened(string)
+      message = Time.now.httpdate
+      message << delimiter
+      message << string
+      message << "\n" unless message[-1] == ?\n
+      @buffer << message
+      flush # Force a flush for now until we figure out where we want to use the buffering.
+    end
+
+    def push_closed(string)
+      unless @log.respond_to?(:write)
+        log = Pathname(@log)
+        log.dirname.mkpath
+        @log = log.open('a')
+        @log.sync = true
+      end
+      set_write_method
+      reset_methods(:open)
+      push(string)
+    end
+
+    alias internal_push push_closed
+
+    def prep_msg(message, level)
+      level << delimiter << message
+    end
+
+    public
+
+    # To initialize the logger you create a new object, proxies to set_log.
+    #   DataMapper::Logger.new(log{String, IO},level{Symbol, String})
+    #
+    # @param log<IO,String>        either an IO object or a name of a logfile.
+    # @param log_level<String>     the message string to be logged
+    # @param delimiter<String>     delimiter to use between message sections
+    # @param log_creation<Boolean> log that the file is being created
+    def initialize(*args)
+      set_log(*args)
+    end
+
+    # To replace an existing logger with a new one:
+    #  DataMapper::Logger.set_log(log{String, IO},level{Symbol, String})
+    #
+    #
+    # @param log<IO,String>        either an IO object or a name of a logfile.
+    # @param log_level<Symbol>     a symbol representing the log level from
+    #   {:off, :fatal, :error, :warn, :info, :debug}
+    # @param delimiter<String>     delimiter to use between message sections
+    # @param log_creation<Boolean> log that the file is being created
+    def set_log(log, log_level = :off, delimiter = " ~ ", log_creation = false)
+      delimiter    ||= " ~ "
+
+      if log_level && LEVELS[log_level.to_sym]
+        self.level = log_level.to_sym
+      else
+        self.level = :debug
+      end
+
+      @buffer    = []
+      @delimiter = delimiter
+
+      initialize_log(log)
+
+      DataMapper.logger = self
+
+      self.info("Logfile created") if log_creation
+    end
+
+    # Flush the entire buffer to the log object.
+    #   DataMapper.logger.flush
+    #
+    def flush
+      return unless @buffer.size > 0
+      @log.write_method(@buffer.slice!(0..-1).to_s)
+    end
+
+    # Close and remove the current log object.
+    #   DataMapper.logger.close
+    #
+    def close
+      flush
+      @log.close if @log.respond_to?(:close)
+      @log = nil
+    end
+
+    # Appends a string and log level to logger's buffer.
+
+    # @note
+    #   Note that the string is discarded if the string's log level less than the
+    #   logger's log level.
+    # @note
+    #   Note that if the logger is aio capable then the logger will use
+    #   non-blocking asynchronous writes.
+    #
+    # @param level<Fixnum>  the logging level as an integer
+    # @param string<String> the message string to be logged
+    def push(string)
+      internal_push(string)
+    end
+    alias << push
+
+    # Generate the following logging methods for DataMapper.logger as described
+    # in the API:
+    #  :fatal, :error, :warn, :info, :debug
+    #  :off only gets a off? method
+    LEVELS.each_pair do |name, number|
+      unless name.to_s == 'off'
+        class_eval <<-EOS
+        # DOC
+        def #{name}(message)
+          self.<<( prep_msg(message, "#{name}") ) if #{name}?
+        end
+        EOS
+      end
+
+      class_eval <<-EOS
+      # DOC
+      def #{name}?
+        #{number} >= level
+      end
+      EOS
+    end
+
+  end # class Logger
+end # module DataMapper

data/lib/dm-core/migrations/destructive_migrations.rb

@@ -0,0 +1,17 @@
+# TODO: move to dm-more/dm-migrations
+
+module DataMapper
+  module DestructiveMigrations
+    def self.included(model)
+      DestructiveMigrator.models << model
+    end
+  end # module DestructiveMigrations
+
+  class DestructiveMigrator < Migrator
+    def self.migrate(repository_name)
+      models.each do |model|
+        model.auto_migrate!
+      end
+    end
+  end # class DestructiveMigrator
+end # module DataMapper

data/lib/dm-core/migrator.rb

@@ -0,0 +1,29 @@
+# TODO: move to dm-more/dm-migrations
+
+module DataMapper
+  class Migrator
+    def self.subclasses
+      @@subclasses ||= []
+    end
+
+    def self.subclasses=(obj)
+      @@subclasses = obj
+    end
+
+    def self.inherited(klass)
+      subclasses << klass
+
+      class << klass
+        def models
+          @models ||= []
+        end
+      end
+    end
+
+    def self.migrate(repository_name)
+      subclasses.collect do |migrator|
+        migrator.migrate(repository_name)
+      end.flatten
+    end
+  end # class Migrator
+end # module DataMapper

data/lib/dm-core/model.rb

@@ -0,0 +1,399 @@
+module DataMapper
+  module Model
+    ##
+    #
+    # Extends the model with this module after DataMapper::Resource has been
+    # included.
+    #
+    # This is a useful way to extend DataMapper::Model while
+    # still retaining a self.extended method.
+    #
+    # @param [Module] extensions the module that is to be extend the model after
+    #   after DataMapper::Model
+    #
+    # @return [TrueClass, FalseClass] whether or not the inclusions have been
+    #   successfully appended to the list
+    #-
+    # @api public
+    #
+    # TODO: Move this do DataMapper::Model when DataMapper::Model is created
+    def self.append_extensions(*extensions)
+      extra_extensions.concat extensions
+      true
+    end
+    
+    def self.extra_extensions
+      @extra_extensions ||= []
+    end
+    
+    def self.extended(model)
+      model.instance_variable_set(:@storage_names, Hash.new { |h,k| h[k] = repository(k).adapter.resource_naming_convention.call(model.instance_eval { default_storage_name }) })
+      model.instance_variable_set(:@properties,    Hash.new { |h,k| h[k] = k == Repository.default_name ? PropertySet.new : h[Repository.default_name].dup })
+      extra_extensions.each { |extension| model.extend(extension) }
+    end
+
+    def inherited(target)
+      target.instance_variable_set(:@storage_names, @storage_names.dup)
+      target.instance_variable_set(:@properties, Hash.new { |h,k| h[k] = k == Repository.default_name ? self.properties(Repository.default_name).dup(target) : h[Repository.default_name].dup })
+
+      if @relationships
+        duped_relationships = {}; @relationships.each_pair{ |repos, rels| duped_relationships[repos] = rels.dup}
+        target.instance_variable_set(:@relationships, duped_relationships)
+      end
+    end
+
+    def self.new(storage_name, &block)
+      model = Class.new
+      model.send(:include, Resource)
+      model.class_eval <<-EOS, __FILE__, __LINE__
+        def self.default_storage_name
+          #{Extlib::Inflection.classify(storage_name).inspect}
+        end
+      EOS
+      model.instance_eval(&block) if block_given?
+      model
+    end
+
+    ##
+    # Get the repository with a given name, or the default one for the current
+    # context, or the default one for this class.
+    #
+    # @param name<Symbol>   the name of the repository wanted
+    # @param block<Block>   block to execute with the fetched repository as parameter
+    #
+    # @return <Object, DataMapper::Respository> whatever the block returns,
+    #   if given a block, otherwise the requested repository.
+    #-
+    # @api public
+    def repository(name = nil, &block)
+      #
+      # There has been a couple of different strategies here, but me (zond) and dkubb are at least
+      # united in the concept of explicitness over implicitness. That is - the explicit wish of the
+      # caller (+name+) should be given more priority than the implicit wish of the caller (Repository.context.last).
+      #
+      DataMapper.repository(*Array(name || (Repository.context.last ? nil : default_repository_name)), &block)
+    end
+
+    ##
+    # the name of the storage recepticle for this resource.  IE. table name, for database stores
+    #
+    # @return <String> the storage name (IE table name, for database stores) associated with this resource in the given repository
+    def storage_name(repository_name = default_repository_name)
+      @storage_names[repository_name]
+    end
+
+    ##
+    # the names of the storage recepticles for this resource across all repositories
+    #
+    # @return <Hash(Symbol => String)> All available names of storage recepticles
+    def storage_names
+      @storage_names
+    end
+
+    ##
+    # defines a property on the resource
+    #
+    # @param <Symbol> name the name for which to call this property
+    # @param <Type> type the type to define this property ass
+    # @param <Hash(Symbol => String)> options a hash of available options
+    # @see DataMapper::Property
+    def property(name, type, options = {})
+      property = Property.new(self, name, type, options)
+
+      create_property_getter(property)
+      create_property_setter(property)
+
+      @properties[repository.name] << property
+
+      # Add property to the other mappings as well if this is for the default
+      # repository.
+      if repository.name == default_repository_name
+        @properties.each_pair do |repository_name, properties|
+          next if repository_name == default_repository_name
+          properties << property
+        end
+      end
+
+      # Add the property to the lazy_loads set for this resources repository
+      # only.
+      # TODO Is this right or should we add the lazy contexts to all
+      # repositories?
+      if property.lazy?
+        context = options.fetch(:lazy, :default)
+        context = :default if context == true
+
+        Array(context).each do |item|
+          @properties[repository.name].lazy_context(item) << name
+        end
+      end
+
+      property
+    end
+
+    # TODO: make this a Set?
+    def repositories
+      [ repository ] + @properties.keys.collect { |repository_name| DataMapper.repository(repository_name) }
+    end
+
+    def properties(repository_name = default_repository_name)
+      @properties[repository_name]
+    end
+
+    def properties_with_subclasses(repository_name = default_repository_name)
+      #return properties if we're not interested in sti
+     if @properties[repository_name].inheritance_property.nil?
+       @properties[repository_name]
+     else
+        props = @properties[repository_name].dup
+        self.child_classes.each do |subclass|
+          subclass.properties(repository_name).each do |subprop|
+            props << subprop if not props.any? { |prop| prop.name == subprop.name }
+          end
+        end
+        props
+      end
+    end
+
+    def key(repository_name = default_repository_name)
+      @properties[repository_name].key
+    end
+
+    def inheritance_property(repository_name = default_repository_name)
+      @properties[repository_name].inheritance_property
+    end
+
+    def default_order
+      @default_order ||= key.map { |property| Query::Direction.new(property) }
+    end
+
+    def get(*key)
+      repository.identity_map(self).get(key) || first(to_query(repository, key))
+    end
+
+    def get!(*key)
+      get(*key) || raise(ObjectNotFoundError, "Could not find #{self.name} with key #{key.inspect}")
+    end
+
+    def all(query = {})
+      query = scoped_query(query)
+      query.repository.read_many(query)
+    end
+
+    def first(*args)
+      query = args.last.respond_to?(:merge) ? args.pop : {}
+      query = scoped_query(query.merge(:limit => args.first || 1))
+
+      if args.any?
+        query.repository.read_many(query)
+      else
+        query.repository.read_one(query)
+      end
+    end
+
+    def [](*key)
+      warn("#{name}[] is deprecated. Use #{name}.get! instead.")
+      get!(*key)
+    end
+
+    def first_or_create(query, attributes = {})
+      first(query) || begin
+        resource = allocate
+        query = query.dup
+
+        properties(repository.name).key.each do |property|
+          if value = query.delete(property.name)
+            resource.send("#{property.name}=", value)
+          end
+        end
+
+        resource.attributes = query.merge(attributes)
+        resource.save
+        resource
+      end
+    end
+
+    ##
+    # Create an instance of Resource with the given attributes
+    #
+    # @param <Hash(Symbol => Object)> attributes hash of attributes to set
+    def create(attributes = {})
+      resource = new(attributes)
+      resource.save
+      resource
+    end
+
+    ##
+    # Dangerous version of #create.  Raises if there is a failure
+    #
+    # @see DataMapper::Resource#create
+    # @param <Hash(Symbol => Object)> attributes hash of attributes to set
+    # @raise <PersistenceError> The resource could not be saved
+    def create!(attributes = {})
+      resource = create(attributes)
+      raise PersistenceError, "Resource not saved: :new_record => #{resource.new_record?}, :dirty_attributes => #{resource.dirty_attributes.inspect}" if resource.new_record?
+      resource
+    end
+
+    # TODO SPEC
+    def copy(source, destination, query = {})
+      repository(destination) do
+        repository(source).read_many(query).each do |resource|
+          self.create(resource)
+        end
+      end
+    end
+
+    # @api private
+    # TODO: spec this
+    def load(values, query)
+      repository = query.repository
+      model      = self
+
+      if inheritance_property_index = query.inheritance_property_index(repository)
+        model = values.at(inheritance_property_index) || model
+      end
+
+      if key_property_indexes = query.key_property_indexes(repository)
+        key_values   = values.values_at(*key_property_indexes)
+        identity_map = repository.identity_map(model)
+
+        if resource = identity_map.get(key_values)
+          return resource unless query.reload?
+        else
+          resource = model.allocate
+          resource.instance_variable_set(:@repository, repository)
+          identity_map.set(key_values, resource)
+        end
+      else
+        resource = model.allocate
+        resource.readonly!
+      end
+
+      resource.instance_variable_set(:@new_record, false)
+
+      query.fields.zip(values) do |property,value|
+        value = property.custom? ? property.type.load(value, property) : property.typecast(value)
+        property.set!(resource, value)
+
+        if track = property.track
+          case track
+            when :hash
+              resource.original_values[property.name] = value.dup.hash unless resource.original_values.has_key?(property.name) rescue value.hash
+            when :load
+               resource.original_values[property.name] = value unless resource.original_values.has_key?(property.name)
+          end
+        end
+      end
+
+      resource
+    end
+
+    # TODO: spec this
+    def to_query(repository, key, query = {})
+      conditions = Hash[ *self.key(repository.name).zip(key).flatten ]
+      Query.new(repository, self, query.merge(conditions))
+    end
+
+    private
+
+    def default_storage_name
+      self.name
+    end
+
+    def default_repository_name
+      Repository.default_name
+    end
+
+    def scoped_query(query = self.query)
+      assert_kind_of 'query', query, Query, Hash
+
+      return self.query if query == self.query
+
+      query = if query.kind_of?(Hash)
+        Query.new(query.has_key?(:repository) ? query.delete(:repository) : self.repository, self, query)
+      else
+        query
+      end
+
+      self.query ? self.query.merge(query) : query
+    end
+
+    # defines the getter for the property
+    def create_property_getter(property)
+      class_eval <<-EOS, __FILE__, __LINE__
+        #{property.reader_visibility}
+        def #{property.getter}
+          attribute_get(#{property.name.inspect})
+        end
+      EOS
+
+      if property.primitive == TrueClass && !property.model.instance_methods.include?(property.name.to_s)
+        class_eval <<-EOS, __FILE__, __LINE__
+          #{property.reader_visibility}
+          alias #{property.name} #{property.getter}
+        EOS
+      end
+    end
+
+    # defines the setter for the property
+    def create_property_setter(property)
+      unless instance_methods.include?(property.name.to_s + '=')
+        class_eval <<-EOS, __FILE__, __LINE__
+          #{property.writer_visibility}
+          def #{property.name}=(value)
+            attribute_set(#{property.name.inspect}, value)
+          end
+        EOS
+      end
+    end
+
+    def relationships(*args)
+      # DO NOT REMOVE!
+      # method_missing depends on these existing. Without this stub,
+      # a missing module can cause misleading recursive errors.
+      raise NotImplementedError.new
+    end
+    
+    def method_missing(method, *args, &block)      
+      if relationship = self.relationships(repository.name)[method]
+        klass = self == relationship.child_model ? relationship.parent_model : relationship.child_model
+        return DataMapper::Query::Path.new(repository, [ relationship ], klass)
+      end
+
+      if property = properties(repository.name)[method]
+        return property
+      end
+
+      super
+    end
+
+    # TODO: move to dm-more/dm-transactions
+    module Transaction
+      #
+      # Produce a new Transaction for this Resource class
+      #
+      # @return <DataMapper::Adapters::Transaction
+      #   a new DataMapper::Adapters::Transaction with all DataMapper::Repositories
+      #   of the class of this DataMapper::Resource added.
+      #-
+      # @api public
+      #
+      # TODO: move to dm-more/dm-transactions
+      def transaction(&block)
+        DataMapper::Transaction.new(self, &block)
+      end
+    end # module Transaction
+
+    include Transaction
+
+    # TODO: move to dm-more/dm-migrations
+    module Migration
+      # TODO: move to dm-more/dm-migrations
+      def storage_exists?(repository_name = default_repository_name)
+        repository(repository_name).storage_exists?(storage_name(repository_name))
+      end
+    end # module Migration
+
+    include Migration
+  end # module Model
+end # module DataMapper