active_record_disabler 0.0.2 → 0.0.3

data/README.md

@@ -14,9 +14,4 @@ ActiveRecord.enable
 
 User.first
 #=> #<User id: 1, name: "John Smith">
-
-ActiveRecord.disable "Can't touch this"
-
-User.first
-# ActiveRecordDisabledError: Can't touch this
 ```

data/lib/active_record_disabler.rb

@@ -4,15 +4,28 @@ class ActiveRecordDisabledError < StandardError; end
 
 module ActiveRecord
   class << self
+    def enabled?
+      @enabled
+    end
+
     def enable
-      Relation.send(:alias_method, :exec_queries, :ar_enabled)
+      @enabled = true
     end
 
-    def disable(message = "Active Record is disabled")
-      Relation.send(:alias_method, :ar_enabled, :exec_queries)
-      Relation.send(:define_method, :ar_disabled) { raise ActiveRecordDisabledError, message }
+    def disable
+      @enabled = false
+    end
+  end
+
+  class Relation
+    alias_method :_exec_queries, :exec_queries
 
-      Relation.send(:alias_method, :exec_queries, :ar_disabled)
+    def exec_queries(*args)
+      if ActiveRecord.enabled?
+        _exec_queries(*args)
+      else
+        raise ActiveRecordDisabledError, "Active Record is disabled"
+      end
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: active_record_disabler
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors: