lib/memento/action/destroy.rb in memento-0.3.7 vs lib/memento/action/destroy.rb in memento-0.4.0

@@ -1,31 +1,32 @@
-class Memento::Action::Destroy < Memento::Action::Base
-  
-  def fetch
-    record.attributes_for_memento
-  end
-  
-  def undo
-    rebuild_object do |object|
-      begin
-        object.save!
-      rescue
-        object.id = nil
-        object.save!
+module Memento
+  class Action::Destroy < Memento::Action::Base
+
+    def fetch
+      record.attributes_for_memento
+    end
+
+    def undo
+      rebuild_object do |object|
+        begin
+          object.save!
+        rescue
+          object.id = nil
+          object.save!
+        end
+        state.record = object
+        state.save
       end
-      state.record = object
-      state.save
     end
-  end
-  
-  private
-  
-  def rebuild_object
-    new_object do |object|
-      state.record_data.each do |attribute, value|
-        object.send(:"#{attribute}=", value)
+
+    private
+
+    def rebuild_object
+      new_object do |object|
+        state.record_data.each do |attribute, value|
+          object.send(:"#{attribute}=", value)
+        end
+        yield(object) if block_given?
       end
-      yield(object) if block_given?
     end
   end
-  
 end
\ No newline at end of file