lib/uu/remote.rb in uu-0.1.1 vs lib/uu/remote.rb in uu-0.1.2

@@ -16,21 +16,33 @@
 
     def initialize(uri)
       @remote = DRbObject.new_with_uri(uri)
     end
 
+    def disable
+      @remote.disable
+    end
+
+    def enable
+      @remote.enable
+    end
+
     def eval_(code)
       @remote.eval_(code)
     end
 
     def klass(klass_)
       @remote.klass(klass_.to_s)
     end
 
     def run(&block)
-      @remote.disable
+      disable
       instance_eval(&block)
     ensure
-      @remote.enable
+      enable
+    end
+
+    def enable_at_exit
+      at_exit { enable }
     end
   end
 end