Add callbacks on #create and #switch!

lib/apartment/adapters/abstract_adapter.rb

@@ -3,6 +3,9 @@
 module Apartment
   module Adapters
     class AbstractAdapter
+      include ActiveSupport::Callbacks
+      define_callbacks :create, :switch
+
       attr_writer :default_tenant
 
       #   @constructor
@@ -17,15 +20,17 @@ def initialize(config)
       #   @param {String} tenant Tenant name
       #
       def create(tenant)
-        create_tenant(tenant)
+        run_callbacks :create do
+          create_tenant(tenant)
 
-        switch(tenant) do
-          import_database_schema
+          switch(tenant) do
+            import_database_schema
 
-          # Seed data if appropriate
-          seed_data if Apartment.seed_after_create
+            # Seed data if appropriate
+            seed_data if Apartment.seed_after_create
 
-          yield if block_given?
+            yield if block_given?
+          end
         end
       end
 
@@ -80,10 +85,12 @@ def drop(tenant)
       #   @param {String} tenant name
       #
       def switch!(tenant = nil)
-        return reset if tenant.nil?
+        run_callbacks :switch do
+          return reset if tenant.nil?
 
-        connect_to_new(tenant).tap do
-          Apartment.connection.clear_query_cache
+          connect_to_new(tenant).tap do
+            Apartment.connection.clear_query_cache
+          end
         end
       end
 

lib/apartment/tenant.rb

@@ -9,7 +9,7 @@ module Tenant
     extend self
     extend Forwardable
 
-    def_delegators :adapter, :create, :drop, :switch, :switch!, :current, :each, :reset, :seed, :current_tenant, :default_tenant
+    def_delegators :adapter, :create, :drop, :switch, :switch!, :current, :each, :reset, :set_callback, :seed, :current_tenant, :default_tenant
 
     attr_writer :config