phi_attrs 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 3eacb70390f2672da6d28043638c4c50e54c5097
-  data.tar.gz: cc5f1b55c52300d1588d168f94168cef75219590
+  metadata.gz: 1b809e5c530692590d9ef986bace3be17d38dfff
+  data.tar.gz: 47f1af03b09d42744847c22a9052244c3b156cab
 SHA512:
-  metadata.gz: 0a3405b39cbe9006b92184db10ad4e31698512a5c742f43a201724042968e216ecde4977371f63d898a3143ca13e5bfcbbfe9cf7b47a53018d2103703d3ec1a4
-  data.tar.gz: b32c79d5304fe6c8bcecc94939b62b1d0d97cbae17db0aa3bc75dc5b47da01bd450f31e8a35113d18ec2542d4f047ca4e6e3aa73ad745b7f82f01450e92860b2
+  metadata.gz: 5e54820917cc4dd8579bc4d1a6707906423d2633d8bf13d50fa9bf7f956715f2c320a307b23ba666a13201d5f80b313b06c9277477c26f8a37619c52caecb70a
+  data.tar.gz: 4e018c7b2bc85907a70aa8759365d4d7cba4b00d1669da71f49522cbd8a198991b295f2382ef5fe007605775266ff1bf4863320b05049c35430b581d28934530

data/lib/phi_attrs/phi_record.rb

@@ -13,13 +13,15 @@ module PhiAttrs
     included do
       class_attribute :__phi_exclude_methods
       class_attribute :__phi_include_methods
-      class_attribute :__phi_extended_methods
+      class_attribute :__phi_extend_methods
       class_attribute :__phi_methods_wrapped
+      class_attribute :__phi_methods_extended
 
       after_initialize :wrap_phi
 
+      # These have to default to an empty array
       self.__phi_methods_wrapped = []
-      self.__phi_extended_methods = []
+      self.__phi_methods_extended = []
     end
 
     class_methods do
@@ -49,15 +51,15 @@ module PhiAttrs
       # is allowed. The methods that are extended should return ActiveRecord
       # models that also extend PhiAttrs.
       #
-      # If they do not, this is essentially an alias for PhiRecord#include_in_phi
-      #
       # @param [Array<Symbol>] *methods Any number of methods to extend access to
       #
       # @example
+      #   has_one :foo
+      #   has_one :bar
       #   extend_phi_access :foo, :bar
       #
       def extend_phi_access(*methods)
-        self.__phi_extended_methods = methods.map(&:to_s)
+        self.__phi_extend_methods = methods.map(&:to_s)
       end
 
       # Enable PHI access for any instance of this class.
@@ -100,11 +102,18 @@ module PhiAttrs
     # @return [Array<String>] the method names to be wrapped with PHI access logging
     #
     def __phi_wrapped_methods
-      extended_methods = self.class.__phi_extended_methods.to_a
       excluded_methods = self.class.__phi_exclude_methods.to_a
       included_methods = self.class.__phi_include_methods.to_a
 
-      extended_methods + attribute_names - excluded_methods + included_methods - [self.class.primary_key]
+      attribute_names - excluded_methods + included_methods - [self.class.primary_key]
+    end
+
+    # Get all method names to be wrapped with PHI access extension
+    #
+    # @return [Array<String>] the method names to be wrapped with PHI access extension
+    #
+    def __phi_extended_methods
+      self.class.__phi_extend_methods.to_a
     end
 
     # Enable PHI access for a single instance of this class.
@@ -167,7 +176,8 @@ module PhiAttrs
       @__phi_access_logged = false
 
       # Wrap attributes with PHI Logger and Access Control
-      __phi_wrapped_methods.each { |attr| phi_wrap_method(attr) }
+      __phi_wrapped_methods.each { |m| phi_wrap_method(m) }
+      __phi_extended_methods.each { |m| phi_extend_access(m) }
     end
 
     # Log Key for an instance of this class. If the instance is persisted in the
@@ -255,26 +265,46 @@ module PhiAttrs
             @__phi_access_logged = true
           end
 
-          # extend PHI access to relationships if needed
-          if self.class.__phi_extended_methods.include? method_name
+          send(unwrapped_method, *args, &block)
+        end
+      end
 
-            # get the unwrapped relation
-            relation = send(unwrapped_method, *args, &block)
+      # method_name => wrapped_method => unwrapped_method
+      self.class.send(:alias_method, unwrapped_method, method_name)
+      self.class.send(:alias_method, method_name, wrapped_method)
 
-            if relation.class.included_modules.include?(PhiRecord)
-              relation.allow_phi!(phi_allowed_by, phi_access_reason) unless relation.phi_allowed?
-            end
-          end
+      self.class.__phi_methods_wrapped << method_name
+    end
 
-          send(unwrapped_method, *args, &block)
+    # Core logic for wrapping methods in PHI access extensions. Almost
+    # functionally equivalent to the phi_wrap_method call above,
+    # this method doesn't add any logging or access restriction, but
+    # simply proxies the PhiRecord#allow_phi! call.
+    #
+    # @private
+    #
+    def phi_extend_access(method_name)
+      return if self.class.__phi_methods_extended.include? method_name
+
+      wrapped_method = :"__#{method_name}_phi_access_extended"
+      unwrapped_method = :"__#{method_name}_phi_access_original"
+
+      self.class.send(:define_method, wrapped_method) do |*args, &block|
+        # get the unwrapped relation
+        relation = send(unwrapped_method, *args, &block)
+
+        if phi_allowed? && relation.class.included_modules.include?(PhiRecord)
+          relation.allow_phi!(phi_allowed_by, phi_access_reason) unless relation.phi_allowed?
         end
+
+        relation
       end
 
       # method_name => wrapped_method => unwrapped_method
       self.class.send(:alias_method, unwrapped_method, method_name)
       self.class.send(:alias_method, method_name, wrapped_method)
 
-      self.class.__phi_methods_wrapped << method_name
+      self.class.__phi_methods_extended << method_name
     end
   end
 end

data/lib/phi_attrs/version.rb

@@ -1,3 +1,3 @@
 module PhiAttrs
-  VERSION = '0.1.3'.freeze
+  VERSION = '0.1.4'.freeze
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: phi_attrs
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Wyatt Kirby