provide an after_connect configuration option to run a Proc immadiately ...

README.md

@@ -74,6 +74,23 @@ You can configure some options with the usual rails mechanism, in
     config.sequel.load_database_tasks = false
 ```
 
+Enabling plugins
+================
+
+If you want to enable plugins for all your models, you should use the
+after_connect configuration option in `config/application.rb` (0.6.2+):
+
+```ruby
+    config.sequel.after_connect = proc do
+      Sequel::Model.plugin :timestamps, update_on_create: true
+    end
+```
+
+This will ensure that these plugins are loaded before any Sequel models are
+loaded. Loading plugins into `Sequel::Model` after subclasses are already
+created is not supported by Sequel. You can also load extensions in
+`after_connect` or perform any custom actions that you need.
+
 Available sequel specific rake tasks
 ====================================
 

lib/sequel_rails/configuration.rb

@@ -4,13 +4,18 @@ module SequelRails
 
   mattr_accessor :configuration
 
-  def self.setup(environment)
+  def self.setup(environment, app)
     config = configuration.environment_for(environment.to_s)
-    if config['url']
+    db = if config['url']
       ::Sequel.connect config['url'], config
     else
       ::Sequel.connect config
     end
+
+    callback = app.config.sequel.after_connect
+    callback.call if callback.respond_to?(:call)
+
+    db
   end
 
   class Configuration < ActiveSupport::OrderedOptions

lib/sequel_rails/railtie.rb

@@ -60,7 +60,7 @@ class Railtie < Rails::Railtie
     end
 
     initializer 'sequel.connect' do |app|
-      ::SequelRails.setup ::Rails.env
+      ::SequelRails.setup ::Rails.env, app
     end
 
     # Support overwriting crucial steps in subclasses

lib/sequel_rails/railties/database.rake

@@ -4,7 +4,7 @@ require 'sequel_rails/storage'
 namespace :db do
   def db_for_current_env
     @db_for_current_env ||= {}
-    @db_for_current_env[Rails.env] ||= ::SequelRails.setup(Rails.env)
+    @db_for_current_env[Rails.env] ||= ::SequelRails.setup(Rails.env, Rails.application)
   end
 
   # desc "Raises an error if there are pending migrations"

spec/lib/sequel_rails/configuration_spec.rb

@@ -45,7 +45,21 @@
       SequelRails.stub(:jruby?).and_return is_jruby
     end
 
-    subject { SequelRails.setup(environment) }
+    let(:sequel_config) { OpenStruct.new }
+    let(:app_config) { OpenStruct.new(sequel: sequel_config) }
+    let(:app) { OpenStruct.new(config: app_config) }
+    subject { SequelRails.setup(environment, app) }
+
+    context "after_connect hook" do
+      let(:environment) { 'development' }
+      let(:hook) { double }
+      let(:sequel_config) { OpenStruct.new(after_connect: hook) }
+
+      it "runs hook if provided" do
+        hook.should_receive(:call)
+        subject
+      end
+    end
 
     shared_examples "max_connections" do
       context "with max_connections config option" do