culturecode_stagehand 0.4.0 → 0.4.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: cd5f6596af77820e4b0fda648cf3a575cef70dd8
-  data.tar.gz: 73711aa60b690c1a9a7715d5ce26934d79e7727f
+  metadata.gz: 0e34f8b1144de550fdbf2d4cf1395b1ac0596d6a
+  data.tar.gz: 49ce2b899b0a202e7dae920ee831d7b0f43502a0
 SHA512:
-  metadata.gz: 9b685a7c6bd6f6c8638c86f8ff3a316ad2147942b7dcaa378daea28fbc31421cce4105097ac1a3a418d8a6639891e107c5663ee9a5177bdb25770f2d3cd555ac
-  data.tar.gz: 7f629766d1d1af995dc0870bf15bdc6a2c04c96105859cc5cdc4388cb104c4be11b7f0df4c17987f1b8e54a1fd6447e6847fd792196e346ee8f49da17a558cbe
+  metadata.gz: 4b5f80d47ab450ce86ff6a0d4e0e31c517486a7dd67a2cda007e2da64631f90d4e30113ecd2462a212716cbb992bfc7882de830bc239006b0a94afd7aaba64fc
+  data.tar.gz: a045c77ac6cdb29ff73c86f789bc0dafd43722c056911259af48351ee8d14f80f1a35c800af935349d2c51733efcca348e9643e626a1785b0177141a49e0fd3c

data/lib/stagehand/database.rb

@@ -12,12 +12,20 @@ module Stagehand
       current_connection_name == Configuration.staging_connection_name
     end
 
+    def production_connection
+      ProductionProbe.connection
+    end
+
     def staging_connection
       StagingProbe.connection
     end
 
-    def production_connection
-      ProductionProbe.connection
+    def production_database_name
+      database_name(Configuration.production_connection_name)
+    end
+
+    def staging_database_name
+      database_name(Configuration.staging_connection_name)
     end
 
     def with_connection(connection_name)
@@ -48,6 +56,9 @@ module Stagehand
       @@connection_name_stack.last
     end
 
+    def database_name(connection_name)
+      Rails.configuration.database_configuration[connection_name.to_s]['database']
+    end
 
     # CLASSES
 

data/lib/stagehand/staging/commit_entry.rb

@@ -50,10 +50,16 @@ module Stagehand
         return keys.present? ? where(sql.join(' OR '), *interpolates) : none
       end
 
-      def self.infer_class(table_name)
-        klass = ActiveRecord::Base.descendants.detect {|klass| klass.table_name == table_name && klass != Stagehand::Production::Record }
-        klass ||= table_name.classify.constantize # Try loading the class if it isn't loaded yet
+      def self.infer_class(table_name, record_id = nil)
+        classes = ActiveRecord::Base.descendants.select {|klass| klass.table_name == table_name }
+        classes.delete(Stagehand::Production::Record)
+        root_class = classes.first || table_name.classify.constantize # Try loading the class if it isn't loaded yet
 
+        if record_id && record = root_class.find_by_id(record_id)
+          klass = record.class
+        end
+
+        return klass || root_class
       rescue NameError
         raise(IndeterminateRecordClass, "Can't determine class from table name: #{table_name}")
       end
@@ -103,7 +109,7 @@ module Stagehand
       end
 
       def record_class
-        @record_class ||= self.class.infer_class(table_name)
+        @record_class ||= self.class.infer_class(table_name, record_id)
       end
 
       private

data/lib/stagehand/version.rb

@@ -1,3 +1,3 @@
 module Stagehand
-  VERSION = "0.4.0"
+  VERSION = "0.4.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: culturecode_stagehand
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.4.1
 platform: ruby
 authors:
 - Nicholas Jakobsen