dm-rails 1.0.0.rc1

data/Rakefile

@@ -0,0 +1,33 @@
+require 'rubygems'
+require 'rake'
+
+begin
+
+  require 'jeweler'
+
+  Jeweler::Tasks.new do |gem|
+
+    gem.name        = 'dm-rails'
+    gem.summary     = 'Use DataMapper with Rails 3'
+    gem.description = 'Integrate DataMapper with Rails 3'
+    gem.email       = 'gamsnjaga@gmail.com'
+    gem.homepage    = 'http://github.com/datamapper/dm-rails'
+    gem.authors     = [ 'Martin Gamsjaeger (snusnu)', 'Dan Kubb' ]
+
+    gem.add_dependency 'dm-core',           '~> 1.0.0.rc1'
+    gem.add_dependency 'dm-active_model',   '~> 0.4'
+
+    gem.add_dependency 'activesupport',     '~> 3.0.0.beta3'
+    gem.add_dependency 'actionpack',        '~> 3.0.0.beta3'
+    gem.add_dependency 'railties',          '~> 3.0.0.beta3'
+  end
+
+  Jeweler::GemcutterTasks.new
+
+  FileList['tasks/**/*.rake'].each { |task| import task }
+
+rescue LoadError
+  puts 'Jeweler (or a dependency) not available. Install it with: gem install jeweler'
+end
+
+task(:spec) {} # stub out the spec task for as long as we don't have any specs

data/VERSION

@@ -0,0 +1 @@
+1.0.0.rc1

data/dm-rails.gemspec

@@ -0,0 +1,85 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{dm-rails}
+  s.version = "1.0.0.rc1"
+
+  s.required_rubygems_version = Gem::Requirement.new("> 1.3.1") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Martin Gamsjaeger (snusnu)", "Dan Kubb"]
+  s.date = %q{2010-05-19}
+  s.description = %q{Integrate DataMapper with Rails 3}
+  s.email = %q{gamsnjaga@gmail.com}
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+     ".gitignore",
+     "Gemfile",
+     "LICENSE",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "dm-rails.gemspec",
+     "lib/dm-rails.rb",
+     "lib/dm-rails/configuration.rb",
+     "lib/dm-rails/middleware/identity_map.rb",
+     "lib/dm-rails/railtie.rb",
+     "lib/dm-rails/railties/benchmarking_mixin.rb",
+     "lib/dm-rails/railties/controller_runtime.rb",
+     "lib/dm-rails/railties/database.rake",
+     "lib/dm-rails/railties/i18n_support.rb",
+     "lib/dm-rails/railties/log_subscriber.rb",
+     "lib/dm-rails/session_store.rb",
+     "lib/dm-rails/setup.rb",
+     "lib/dm-rails/storage.rb",
+     "lib/generators/data_mapper.rb",
+     "lib/generators/data_mapper/migration/migration_generator.rb",
+     "lib/generators/data_mapper/migration/templates/migration.rb",
+     "lib/generators/data_mapper/model/model_generator.rb",
+     "lib/generators/data_mapper/model/templates/model.rb",
+     "lib/generators/data_mapper/observer/observer_generator.rb",
+     "lib/generators/data_mapper/observer/templates/observer.rb",
+     "tasks/ci.rake",
+     "tasks/clean.rake",
+     "tasks/local_gemfile.rake",
+     "tasks/metrics.rake",
+     "tasks/yard.rake",
+     "tasks/yardstick.rake"
+  ]
+  s.homepage = %q{http://github.com/datamapper/dm-rails}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.6}
+  s.summary = %q{Use DataMapper with Rails 3}
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<dm-core>, ["~> 1.0.0.rc1"])
+      s.add_runtime_dependency(%q<dm-active_model>, ["~> 0.4"])
+      s.add_runtime_dependency(%q<activesupport>, ["~> 3.0.0.beta3"])
+      s.add_runtime_dependency(%q<actionpack>, ["~> 3.0.0.beta3"])
+      s.add_runtime_dependency(%q<railties>, ["~> 3.0.0.beta3"])
+    else
+      s.add_dependency(%q<dm-core>, ["~> 1.0.0.rc1"])
+      s.add_dependency(%q<dm-active_model>, ["~> 0.4"])
+      s.add_dependency(%q<activesupport>, ["~> 3.0.0.beta3"])
+      s.add_dependency(%q<actionpack>, ["~> 3.0.0.beta3"])
+      s.add_dependency(%q<railties>, ["~> 3.0.0.beta3"])
+    end
+  else
+    s.add_dependency(%q<dm-core>, ["~> 1.0.0.rc1"])
+    s.add_dependency(%q<dm-active_model>, ["~> 0.4"])
+    s.add_dependency(%q<activesupport>, ["~> 3.0.0.beta3"])
+    s.add_dependency(%q<actionpack>, ["~> 3.0.0.beta3"])
+    s.add_dependency(%q<railties>, ["~> 3.0.0.beta3"])
+  end
+end
+

data/lib/dm-rails.rb

@@ -0,0 +1 @@
+require 'dm-rails/railtie'

data/lib/dm-rails/configuration.rb

@@ -0,0 +1,70 @@
+require 'active_support/core_ext/hash/except'
+require 'active_support/core_ext/class/attribute_accessors'
+
+module Rails
+  module DataMapper
+
+    mattr_accessor :configuration
+
+    class Configuration
+
+      def self.for(root, database_yml_hash)
+        Rails::DataMapper.configuration ||= new(root, database_yml_hash)
+      end
+
+      attr_reader :root, :raw
+
+      def environments
+        config.keys
+      end
+
+      def repositories
+        @repositories ||= @raw.reject { |k,v| k =~ /defaults/ }.inject({}) do |repositories, pair|
+          environment, config = pair.first, pair.last
+          repositories[environment] = begin
+            c = config['repositories'] || {}
+            c['default'] = config.except('repositories') if config.except('repositories')
+            normalize_repository_config(c)
+          end
+          repositories
+        end
+      end
+
+
+      def identity_map=(value)
+        @identity_map = value
+      end
+
+      def identity_map
+        @identity_map ||= true
+      end
+
+
+    private
+
+      def initialize(root, database_yml_hash)
+        @root, @raw = root, database_yml_hash
+      end
+
+      def normalize_repository_config(hash)
+        config = {}
+        hash.each do |key, value|
+          config[key] = if value.kind_of?(Hash)
+            normalize_repository_config(value)
+          elsif key == 'port'
+            value.to_i
+          elsif key == 'adapter' && value == 'postgresql'
+            'postgres'
+          elsif key == 'database' && hash['adapter'] == 'sqlite3'
+            value == ':memory:' ? value : File.expand_path(hash['database'], root)
+          else
+            value
+          end
+        end
+        config
+      end
+
+    end
+
+  end
+end

data/lib/dm-rails/middleware/identity_map.rb

@@ -0,0 +1,19 @@
+module Rails
+  module DataMapper
+    module Middleware
+
+      class IdentityMap
+        def initialize(app)
+          @app = app
+        end
+
+        def call(env)
+          ::DataMapper.repository do
+            @app.call(env)
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/dm-rails/railtie.rb

@@ -0,0 +1,103 @@
+require 'dm-core'
+require 'dm-active_model'
+
+require 'rails'
+require 'active_model/railtie'
+
+# Comment taken from active_record/railtie.rb
+#
+# For now, action_controller must always be present with
+# rails, so let's make sure that it gets required before
+# here. This is needed for correctly setting up the middleware.
+# In the future, this might become an optional require.
+require 'action_controller/railtie'
+
+require 'dm-rails/setup'
+require "dm-rails/railties/log_subscriber"
+require "dm-rails/railties/i18n_support"
+
+
+module Rails
+  module DataMapper
+
+    class Railtie < Rails::Railtie
+
+      log_subscriber :data_mapper, ::DataMapper::Railties::LogSubscriber.new
+
+      config.generators.orm :data_mapper, :migration => true
+
+
+      # Support overwriting crucial steps in subclasses
+
+      def configure_data_mapper(app)
+        app.config.data_mapper = Rails::DataMapper::Configuration.for(
+          Rails.root, app.config.database_configuration
+        )
+      end
+
+      def setup_i18n_support(app)
+        ::DataMapper::Model.append_inclusions(Rails::DataMapper::I18nSupport)
+      end
+
+      def setup_controller_runtime(app)
+        require "dm-rails/railties/controller_runtime"
+        ActionController::Base.send :include, Rails::DataMapper::Railties::ControllerRuntime
+      end
+
+      def setup_identity_map(app)
+        if app.config.data_mapper.identity_map
+          require 'dm-rails/middleware/identity_map'
+          app.config.middleware.use Rails::DataMapper::Middleware::IdentityMap
+        end
+      end
+
+      def setup_logger(app, logger)
+        Rails::DataMapper.setup_logger(logger)
+      end
+
+
+      initializer 'data_mapper.configuration' do |app|
+        configure_data_mapper(app)
+      end
+
+      initializer 'data_mapper.logger' do |app|
+        setup_logger(app, Rails.logger)
+      end
+
+      initializer 'data_mapper.i18n_support' do |app|
+        setup_i18n_support(app)
+      end
+
+      # Expose database runtime to controller for logging.
+      initializer "data_mapper.log_runtime" do |app|
+        setup_controller_runtime(app)
+      end
+
+      initializer 'data_mapper.setup_identity_map' do |app|
+        setup_identity_map(app)
+      end
+
+      # Preload all models once in production mode,
+      # and before every request in development mode
+      initializer "datamapper.add_to_prepare" do |app|
+        config.to_prepare { Rails::DataMapper.preload_models(app) }
+      end
+
+      # Run setup code once in after_initialize to make sure all initializers
+      # are in effect once we setup the connection. Also, this will make sure
+      # that the connection gets set up after all models have been loaded,
+      # because #after_initialize is guaranteed to run after #to_prepare.
+      # Both production and development environment will execute the setup
+      # code only once.
+      config.after_initialize do |app|
+        Rails::DataMapper.setup(Rails.env)
+      end
+
+      rake_tasks do
+        load 'dm-rails/railties/database.rake'
+      end
+
+    end
+
+  end
+end

data/lib/dm-rails/railties/benchmarking_mixin.rb

@@ -0,0 +1,23 @@
+module DataMapper
+  module Adapters
+    module Benchmarking
+
+      %w[ create read update delete ].each do |method|
+        class_eval <<-RUBY, __FILE__, __LINE__
+          def #{method}(*args, &block)
+            result = nil
+            @runtime ||= 0
+            @runtime += Benchmark.ms { result = super(*args, &block) }
+            result
+          end
+        RUBY
+      end
+
+      def reset_runtime
+        rt, @runtime = @runtime, 0
+        rt.to_f
+      end
+
+    end
+  end
+end

data/lib/dm-rails/railties/controller_runtime.rb

@@ -0,0 +1,45 @@
+require 'active_support/core_ext/module/attr_internal'
+
+module Rails
+  module DataMapper
+    module Railties
+
+      module ControllerRuntime
+
+        extend ActiveSupport::Concern
+
+        protected
+
+        attr_internal :db_runtime
+
+        def cleanup_view_runtime
+          # TODO add checks if DataMapper is connected to a repository.
+          # If it is, do this, if it isn't, just delegate to super
+          db_rt_before_render = ::DataMapper.repository.adapter.reset_runtime
+          runtime = super
+          db_rt_after_render = ::DataMapper.repository.adapter.reset_runtime
+          self.db_runtime = db_rt_before_render + db_rt_after_render
+          runtime - db_rt_after_render
+        end
+
+        def append_info_to_payload(payload)
+          super
+          payload[:db_runtime] = db_runtime
+        end
+
+
+        module ClassMethods
+
+          def log_process_action(payload)
+            messages, db_runtime = super, payload[:db_runtime]
+            messages << ("Models: %.1fms" % db_runtime.to_f) if db_runtime
+            messages
+          end
+
+        end
+
+      end
+
+    end
+  end
+end

data/lib/dm-rails/railties/database.rake

@@ -0,0 +1,124 @@
+require 'dm-rails/setup'
+require 'dm-rails/storage'
+
+namespace :db do
+
+  task :load_models => :environment do
+    FileList["app/models/**/*.rb"].each { |model| load model }
+  end
+
+  desc 'Create the database, load the schema, and initialize with the seed data'
+  task :setup => [ 'db:create', 'db:automigrate', 'db:seed' ]
+
+  namespace :test do
+    task :prepare => ['db:setup']
+  end
+
+  namespace :create do
+    desc 'Create all the local databases defined in config/database.yml'
+    task :all => :environment do
+      Rails::DataMapper.storage.create_all
+    end
+  end
+
+  desc "Create the database(s) defined in config/database.yml for the current Rails.env - also creates the test database(s) if Rails.env.development?"
+  task :create => :environment do
+    Rails::DataMapper.storage.create_environment(Rails::DataMapper.configuration.repositories[Rails.env])
+    if Rails.env.development? && Rails::DataMapper.configuration.repositories['test']
+      Rails::DataMapper.storage.create_environment(Rails::DataMapper.configuration.repositories['test'])
+    end
+  end
+
+  namespace :drop do
+    desc 'Drop all the local databases defined in config/database.yml'
+    task :all => :environment do
+      Rails::DataMapper.storage.drop_all
+    end
+  end
+
+  desc "Drops the database(s) for the current Rails.env - also drops the test database(s) if Rails.env.development?"
+  task :drop => :environment do
+    Rails::DataMapper.storage.drop_environment(Rails::DataMapper.configuration.repositories[Rails.env])
+    if Rails.env.development? && Rails::DataMapper.configuration.repositories['test']
+      Rails::DataMapper.storage.drop_environment(Rails::DataMapper.configuration.repositories['test'])
+    end
+  end
+
+
+  desc 'Perform destructive automigration of all repositories in the current Rails.env'
+  task :automigrate => :load_models do
+    require 'dm-migrations'
+    Rails::DataMapper.configuration.repositories[Rails.env].each do |repository, config|
+      ::DataMapper.auto_migrate!(repository.to_sym)
+      puts "[datamapper] Finished auto_migrate! for :#{repository} repository '#{config['database']}'"
+    end
+    if Rails.env.development? && Rails::DataMapper.configuration.repositories['test']
+      Rails::DataMapper.setup('test')
+      Rails::DataMapper.configuration.repositories['test'].each do |repository, config|
+        ::DataMapper.auto_migrate!(repository.to_sym)
+        puts "[datamapper] Finished auto_migrate! for :#{repository} repository '#{config['database']}'"
+      end
+    end
+  end
+
+  desc 'Perform non destructive automigration of all repositories in the current Rails.env'
+  task :autoupgrade => :load_models do
+    require 'dm-migrations'
+    Rails::DataMapper.configuration.repositories[Rails.env].each do |repository, config|
+      ::DataMapper.auto_upgrade!(repository.to_sym)
+      puts "[datamapper] Finished auto_upgrade! for :#{repository} repository '#{config['database']}'"
+    end
+    if Rails.env.development? && Rails::DataMapper.configuration.repositories['test']
+      Rails::DataMapper.setup('test')
+      Rails::DataMapper.configuration.repositories['test'].each do |repository, config|
+        ::DataMapper.auto_upgrade!(repository.to_sym)
+        puts "[datamapper] Finished auto_upgrade! for :#{repository} repository '#{config['database']}'"
+      end
+    end
+  end
+
+  desc 'Load the seed data from db/seeds.rb'
+  task :seed => :environment do
+    seed_file = File.join(Rails.root, 'db', 'seeds.rb')
+    load(seed_file) if File.exist?(seed_file)
+  end
+
+  namespace :migrate do
+    task :load => :environment do
+      require 'dm-migrations/migration_runner'
+      FileList['db/migrate/*.rb'].each do |migration|
+        load migration
+      end
+    end
+
+    desc 'Migrate up using migrations'
+    task :up, :version, :needs => :load do |t, args|
+      ::DataMapper::MigrationRunner.migrate_up!(args[:version])
+    end
+
+    desc 'Migrate down using migrations'
+    task :down, :version, :needs => :load do |t, args|
+      ::DataMapper::MigrationRunner.migrate_down!(args[:version])
+    end
+  end
+
+  desc 'Migrate the database to the latest version'
+  task :migrate => 'db:migrate:up'
+
+  namespace :sessions do
+    desc "Creates the sessions table for DataMapperStore"
+    task :create => :environment do
+      require 'dm-rails/session_store'
+      Rails::DataMapper::SessionStore::Session.auto_migrate!
+      puts "Created '#{Rails::DataMapper.configurations[Rails.env]['database']}.sessions'"
+    end
+
+    desc "Clear the sessions table for DataMapperStore"
+    task :clear => :environment do
+      require 'dm-rails/session_store'
+      Rails::DataMapper::SessionStore::Session.all.destroy!
+      puts "Deleted entries from '#{Rails::DataMapper.configurations[Rails.env]['database']}.sessions'"
+    end
+  end
+
+end