dont 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8df12a0318bc20065d835a2140936de4a303b0b1
-  data.tar.gz: 4ce2cae8395f12d253c9de429316a1ba60daeb2b
+  metadata.gz: 79e390ca714c283c11ab6c6ed2df51bc9623dcfe
+  data.tar.gz: 0e65bfb6b8d054d1cfe0311bc951b13f2edf7996
 SHA512:
-  metadata.gz: 24da3afc49cf4a355be86e2b716165d413164597a38650de4fdd4befb9684a608875bff42b6f9de5e44ffc46c50b18b0bacb91c5818e0fde74908551c34c25cb
-  data.tar.gz: c2deede14c9f51997d34021a5b3885566f59a7e475e6944e14c474542e12b2ce342c43cecaf2a7a5ea9f944191a17623062b0b569d91cf71bb453630137cbe8a
+  metadata.gz: 5673e14de0efd99161fd5e474d1809c8312c3e9e17e2048710a94f2ba3659fd5c71b46edb6e8068692ff18ddea013f3d6fc9b4a69a3f4cfeb67fa537181cbe8b
+  data.tar.gz: ebd77b1c6e42494ec0988fcc4c2477468702fa66edba988645e0b168cdcb5567a1ae1ab3d037fbca21db9854b8c7ea512b5ec6e0cdbc4d471e59db30a021daed

data/lib/dont.rb

@@ -62,10 +62,28 @@ class Dont < Module
   def initialize(key)
     handler = Dont.fetch_handler(key)
     @implementation = ->(method) {
-      old_method = instance_method(method)
+      # The moment `dont_use` is called in ActiveRecord is before AR defines
+      # the attributes in the model. So you get an error when calling
+      # instance_method with the regular implementation.
+      #
+      # This hack determines if it's an ActiveRecord attribute or not, and
+      # adapts the code.
+      is_ar_attribute = defined?(ActiveRecord::Base) &&
+        ancestors.include?(ActiveRecord::Base) &&
+        !method_defined?(method)
+
+      old_method = instance_method(method) unless is_ar_attribute
       define_method(method) do |*args|
         handler.call(self, method)
-        old_method.bind(self).call(*args)
+        if is_ar_attribute
+          if method =~ /=$/
+            self[method] = args.first
+          else
+            self[method]
+          end
+        else
+          old_method.bind(self).call(*args)
+        end
       end
     }
   end

data/lib/dont/version.rb

@@ -1,3 +1,3 @@
 class Dont < Module
-  VERSION = "0.1.0".freeze
+  VERSION = "0.1.1".freeze
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: dont
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Maarten Claes