phi_attrs 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8f28f8b165404acd488de741070670268bb5e1a3
-  data.tar.gz: 8610013fc23d93100ba5829327a85c32516a2f38
+  metadata.gz: 07c9497b2d8b7c17ad211cd9bd9ed40fdf669e78
+  data.tar.gz: 3e42b4499d9b45d1f46a52d325869f171becb27d
 SHA512:
-  metadata.gz: 4b181df2aacce039f31c31a97051434f99a5dc96f975f2668eba97a7d7cff0a357f6aed8eca75a03a75945b1b4489083e489318e4ba680a446968b93d3142285
-  data.tar.gz: 1bfa9fac24405baa8a8e3e91eb8f5db4f7602413254143d9b8526d61f45f396250351bb333ca99da35a49a293d303521466974c79ffa0a83ecf9ce8e391463a7
+  metadata.gz: 7f4e4acdcc682ba9b0cca43644f39a248ea3878e2138d8c6105246c74d9df6c1ad1c304cec13a790398c7430b70b140f56d60e701e973c374a9dfc250ef65a3c
+  data.tar.gz: 67737de996af40fa8ceea1b68a69f2ce516e7035c02d0e70183e8b56058e2a17ea2e6d5fb2030cf9df25a0e688072e1cc871c0b41ae365544f50132cf6a26bd0

data/lib/phi_attrs/phi_record.rb

@@ -13,6 +13,7 @@ module PhiAttrs
       after_initialize :wrap_phi
 
       self.__phi_methods_wrapped = []
+      self.__phi_extended_methods = []
     end
 
     class_methods do
@@ -59,11 +60,14 @@ module PhiAttrs
     end
 
     def __phi_wrapped_methods
-      attribute_names - self.class.__phi_exclude_methods.to_a + self.class.__phi_include_methods.to_a - [self.class.primary_key]
+      associations = self.class.reflect_on_all_associations.map(&:name).map(&:to_s)
+      excluded_methods = self.class.__phi_exclude_methods.to_a
+      included_methods = self.class.__phi_include_methods.to_a
+      associations + attribute_names - excluded_methods + included_methods - [self.class.primary_key]
     end
 
     def allow_phi!(user_id, reason)
-      PhiAttrs::Logger.tagged( *phi_log_keys ) do
+      PhiAttrs::Logger.tagged(*phi_log_keys) do
         @__phi_access_allowed = true
         @__phi_user_id = user_id
         @__phi_access_reason = reason
@@ -112,10 +116,21 @@ module PhiAttrs
           raise PhiAttrs::Exceptions::PhiAccessException, "Attempted PHI access for #{self.class.name} #{@__phi_user_id}" unless phi_allowed?
 
           unless @__phi_access_logged
-            PhiAttrs::Logger.info("'#{phi_allowed_by}' accessing #{self.class.name}.\n\t access logging triggered by method: #{method_name}")
+            PhiAttrs::Logger.info("'#{phi_allowed_by}' accessing #{self.class.name}. Triggered by method: #{method_name}")
             @__phi_access_logged = true
           end
 
+          # extend PHI access to relationships if needed
+          if self.class.__phi_extended_methods.include? method_name
+
+            # get the unwrapped relation
+            relation = send(unwrapped_method, *args, &block)
+
+            if relation.class.included_modules.include?(PhiRecord)
+              relation.allow_phi!(phi_allowed_by, phi_access_reason) unless relation.phi_allowed?
+            end
+          end
+
           send(unwrapped_method, *args, &block)
         end
       end

data/lib/phi_attrs/version.rb

@@ -1,3 +1,3 @@
 module PhiAttrs
-  VERSION = '0.1.1'.freeze
+  VERSION = '0.1.2'.freeze
 end

metadata

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