dm-rails 1.0.0.rc1

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

@@ -0,0 +1,12 @@
+module Rails
+  module DataMapper
+
+    module I18nSupport
+      # Set the i18n scope to overwrite ActiveModel.
+      def i18n_scope #:nodoc:
+        :data_mapper
+      end
+    end
+
+  end
+end

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

@@ -0,0 +1,31 @@
+module DataMapper
+  module Railties
+
+    class LogSubscriber < Rails::LogSubscriber
+
+      def sql(event)
+        name = '%s (%.1fms)' % [event.payload[:name], event.duration]
+        sql  = event.payload[:sql].squeeze(' ')
+
+        if odd?
+          name = color(name, :cyan, true)
+          sql  = color(sql, nil, true)
+        else
+          name = color(name, :magenta, true)
+        end
+
+        debug "  #{name}  #{sql}"
+      end
+
+      def odd?
+        @odd_or_even = !@odd_or_even
+      end
+
+      def logger
+        ::DataMapper.logger
+      end
+
+    end
+
+  end
+end

data/lib/dm-rails/session_store.rb

@@ -0,0 +1,61 @@
+require 'dm-core'
+
+# Implements DataMapper-specific session store.
+
+module Rails
+  module DataMapper
+
+    class SessionStore < ActionDispatch::Session::AbstractStore
+
+      class Session
+
+        include ::DataMapper::Resource
+
+        property :id,         Serial
+        property :session_id, String,   :required => true, :unique => true, :unique_index => true
+        property :data,       Object,   :required => true, :default => ActiveSupport::Base64.encode64(Marshal.dump({}))
+        property :updated_at, DateTime, :required => false, :index => true
+
+        def self.name
+          'session'
+        end
+
+      end
+
+      SESSION_RECORD_KEY = 'rack.session.record'.freeze
+
+      cattr_accessor :session_class
+      self.session_class = Session
+
+      private
+
+      def get_session(env, sid)
+        sid ||= generate_sid
+        session = find_session(sid)
+        env[SESSION_RECORD_KEY] = session
+        [ sid, session.data ]
+      end
+
+      def set_session(env, sid, session_data)
+        session            = get_session_resource(env, sid)
+        session.data       = session_data
+        session.updated_at = Time.now if session.dirty?
+        session.save ? sid : false
+      end
+
+      def get_session_resource(env, sid)
+        if env[ENV_SESSION_OPTIONS_KEY][:id].nil?
+          env[SESSION_RECORD_KEY] = find_session(sid)
+        else
+          env[SESSION_RECORD_KEY] ||= find_session(sid)
+        end
+      end
+
+      def find_session(sid)
+        self.class.session_class.first_or_new(:session_id => sid)
+      end
+
+    end
+
+  end
+end

data/lib/dm-rails/setup.rb

@@ -0,0 +1,41 @@
+require 'active_support/core_ext/hash/except'
+
+require 'dm-rails/configuration'
+require 'dm-rails/railties/benchmarking_mixin'
+
+module Rails
+  module DataMapper
+
+    def self.setup(environment)
+      puts "[datamapper] Setting up the #{environment.inspect} environment:"
+      configuration.repositories[environment].each do |name, config|
+        setup_with_instrumentation(name.to_sym, config)
+      end
+      initialize_foreign_keys
+    end
+
+    def self.setup_logger(logger)
+      ::DataMapper.logger = logger
+    end
+
+    def self.setup_with_instrumentation(name, options)
+      puts "[datamapper] Setting up #{name.inspect} repository: '#{options['database']}' on #{options['adapter']}"
+      adapter = ::DataMapper.setup(name, options)
+      adapter.extend ::DataMapper::Adapters::Benchmarking
+    end
+
+    def self.initialize_foreign_keys
+      ::DataMapper::Model.descendants.each do |model|
+        model.relationships.each_value { |r| r.child_key }
+      end
+    end
+
+    def self.preload_models(app)
+      app.config.paths.app.models.each do |path|
+        Dir.glob("#{path}/**/*.rb").sort.each { |file| require_dependency file }
+      end
+      initialize_foreign_keys
+    end
+
+  end
+end

data/lib/dm-rails/storage.rb

@@ -0,0 +1,165 @@
+module Rails
+  module DataMapper
+
+    def self.storage
+      Storage
+    end
+
+    class Storage
+      attr_reader :name, :config
+
+      def self.create_all
+        with_local_repositories { |config| create_environment(config) }
+      end
+
+      def self.drop_all
+        with_local_repositories { |config| drop_environment(config) }
+      end
+
+      def self.create_environment(config)
+        config.each { |repo_name, repo_config| new(repo_name, repo_config).create }
+      end
+
+      def self.drop_environment(config)
+        config.each { |repo_name, repo_config| new(repo_name, repo_config).drop }
+      end
+
+      def self.new(name, config)
+        klass = lookup_class(config['adapter'])
+        if klass.equal?(self)
+          super(name, config)
+        else
+          klass.new(name, config)
+        end
+      end
+
+      class << self
+      private
+
+        def with_local_repositories
+          Rails::DataMapper.configuration.repositories.each_value do |config|
+            if config['host'].blank? || %w[ 127.0.0.1 localhost ].include?(config['host'])
+              yield(config)
+            else
+              puts "This task only modifies local databases. #{config['database']} is on a remote host."
+            end
+          end
+        end
+
+        def lookup_class(adapter)
+          klass_name = adapter.camelize.to_sym
+
+          unless Storage.const_defined?(klass_name)
+            raise "Adapter #{adapter} not supported (#{klass_name.inspect})"
+          end
+
+          const_get(klass_name)
+        end
+
+      end
+
+      def initialize(name, config)
+        @name, @config = name.to_sym, config
+      end
+
+      def create
+        _create
+        puts "[datamapper] Created database '#{database}'"
+      end
+
+      def drop
+        _drop
+        puts "[datamapper] Dropped database '#{database}'"
+      end
+
+      def database
+        @database ||= config['database'] || config['path']
+      end
+
+      def username
+        @username ||= config['username'] || ''
+      end
+
+      def password
+        @password ||= config['password'] || ''
+      end
+
+      def charset
+        @charset ||= config['charset'] || ENV['CHARSET'] || 'utf8'
+      end
+
+      class Sqlite3 < Storage
+        def _create
+          return if in_memory?
+          ::DataMapper.setup(name, config.merge('database' => path))
+        end
+
+        def _drop
+          return if in_memory?
+          path.unlink if path.file?
+        end
+
+      private
+
+        def in_memory?
+          database == ':memory:'
+        end
+
+        def path
+          @path ||= Pathname(File.expand_path(database, Rails.root))
+        end
+
+      end
+
+      class Mysql < Storage
+        def _create
+          execute("CREATE DATABASE `#{database}` DEFAULT CHARACTER SET #{charset} DEFAULT COLLATE #{collation}")
+        end
+
+        def _drop
+          execute("DROP DATABASE IF EXISTS `#{database}`")
+        end
+
+      private
+
+        def execute(statement)
+          system(
+            'mysql',
+            (username.blank? ? '' : "--user=#{username}"),
+            (password.blank? ? '' : "--password=#{password}"),
+            '-e',
+            statement
+          )
+        end
+
+        def collation
+          @collation ||= config['collation'] || ENV['COLLATION'] || 'utf8_unicode_ci'
+        end
+
+      end
+
+      class Postgres < Storage
+        def _create
+          system(
+            'createdb',
+            '-E',
+            charset,
+            '-U',
+            username,
+            database
+          )
+        end
+
+        def _drop
+          system(
+            'dropdb',
+            '-U',
+            username,
+            database
+          )
+        end
+
+      end
+    end
+  end
+end

data/lib/generators/data_mapper.rb

@@ -0,0 +1,82 @@
+require 'rails/generators/named_base'
+require 'rails/generators/migration'
+require 'rails/generators/active_model'
+
+module DataMapper
+  module Generators
+
+    class Base < ::Rails::Generators::NamedBase #:nodoc:
+
+      include ::Rails::Generators::Migration
+
+      def self.source_root
+        @_datamapper_source_root ||=
+        File.expand_path("../#{base_name}/#{generator_name}/templates", __FILE__)
+      end
+
+      protected
+
+      # Datamapper does not care if migrations have the same name as long as
+      # they have different ids.
+      #
+      def migration_exists?(dirname, file_name) #:nodoc:
+        false
+      end
+
+      # Implement the required interface for Rails::Generators::Migration.
+      #
+      def self.next_migration_number(dirname) #:nodoc:
+        "%.3d" % (current_migration_number(dirname) + 1)
+      end
+
+    end
+
+    class ActiveModel < ::Rails::Generators::ActiveModel #:nodoc:
+      def self.all(klass)
+        "#{klass}.all"
+      end
+
+      def self.find(klass, params=nil)
+        "#{klass}.get(#{params})"
+      end
+
+      def self.build(klass, params=nil)
+        if params
+          "#{klass}.new(#{params})"
+        else
+          "#{klass}.new"
+        end
+      end
+
+      def save
+        "#{name}.save"
+      end
+
+      def update_attributes(params=nil)
+        "#{name}.update(#{params})"
+      end
+
+      def errors
+        "#{name}.errors"
+      end
+
+      def destroy
+        "#{name}.destroy"
+      end
+    end
+
+  end
+end
+
+module Rails
+
+  module Generators
+    class GeneratedAttribute #:nodoc:
+      def type_class
+        return 'DateTime' if type.to_s == 'datetime'
+        return type.to_s.camelcase
+      end
+    end
+  end
+
+end

data/lib/generators/data_mapper/migration/migration_generator.rb

@@ -0,0 +1,30 @@
+require 'generators/data_mapper'
+
+module DataMapper
+  module Generators
+
+    class MigrationGenerator < Base
+
+      argument :attributes, :type => :array, :default => [], :banner => "field:type field:type"
+      class_option :id, :type => :numeric, :desc => "The id to be used in this migration"
+
+      def create_migration_file
+        set_local_assigns!
+        migration_template "migration.rb", "db/migrate/#{file_name}.rb"
+      end
+
+      protected
+
+      attr_reader :migration_action
+
+      def set_local_assigns!
+        if file_name =~ /^(add|remove|drop)_.*_(?:to|from)_(.*)/
+          @migration_action = $1 == 'add' ? 'add' : 'drop'
+          @table_name       = $2.pluralize
+        end
+      end
+
+    end
+
+  end
+end

data/lib/generators/data_mapper/migration/templates/migration.rb

@@ -0,0 +1,23 @@
+migration <%= migration_number.to_i %>, :<%= migration_file_name %> do
+
+  up do
+<% unless attributes.empty? -%>
+    modify_table :<%= table_name %> do
+<% attributes.each do |attribute| -%>
+      <%= migration_action %>_column :<%= attribute.name %><% if migration_action == 'add' %>, :<%= attribute.type_class %><% end -%>
+<% end -%>
+    end
+<% end -%>
+  end
+
+  down do
+<% unless attributes.empty? -%>
+    modify_table :<%= table_name %> do
+<% attributes.reverse.each do |attribute| -%>
+      <%= migration_action == 'add' ? 'drop' : 'add' %>_column :<%= attribute.name %><% if migration_action == 'drop' %>, :<%= attribute.type_class %><% end -%>
+<% end -%>
+    end
+<% end -%>
+  end
+
+end