forceps 0.6.0 → 0.6.1

data/lib/forceps/client.rb

@@ -53,11 +53,24 @@ module Forceps
 
         include Forceps::ActsAsCopyableModel
 
-        def self.instantiate(record, column_types = {})
+        # Intercep intantiation of records to make the 'type' column point to the corresponding remote class
+
+        if Rails::VERSION::MAJOR >= 4
+          def self.instantiate(record, column_types = {})
+            __prepare_sti_column_for_forceps(record)
+            super
+          end
+        else
+          def self.instantiate(record)
+            __prepare_sti_column_for_forceps(record)
+            super
+          end
+        end
+
+        def self.__prepare_sti_column_for_forceps(record)
           if record[inheritance_column].present?
             record[inheritance_column] = "Forceps::Remote::#{record[inheritance_column]}"
           end
-          super
         end
 
         # We don't want to include STI condition automatically (the base class extends the original one)

data/lib/forceps/version.rb

@@ -1,3 +1,3 @@
 module Forceps
-  VERSION = "0.6.0"
+  VERSION = "0.6.1"
 end