sequel-privacy 0.5 → 0.5.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8307cc7016667794361e143ba1a17735d2c189079af884a9928df3f67450fed1
-  data.tar.gz: f6ae84ecd2b0627deeb998d69cc7f38ec73313b5851b4e84b4367335ab6e7b06
+  metadata.gz: af7fc02c1fb57d47ba358babdebb3289c2d8eb6ec43d2fce25a2c3477a3f2ccc
+  data.tar.gz: a18b09c73b5b03540d2a74f985d32c99b6b2a090a714346f2c87ef5a2997252c
 SHA512:
-  metadata.gz: ff39c00cfd1df723a53bca2a75a3ab6fd3657e1475f9f53c48f369de686c3c81facea4af80325b6acfa23e409e6e5fe71b5794ed9c31fad7300f19727f25e7e7
-  data.tar.gz: fa3124720cb8a285a1033fc3923d3df5f61d04685585394d94614d9bc7331ada3f8863335e91c0198229bfc7238e827b004de2cbed2c88a06fd762328753f8b2
+  metadata.gz: 9f46a3f0f253061be75dcd2fb80cf797c1be229d5fd83b1c9f1c86860621dae4536ea60b06df4e38d2d6b1e66f0e90533f0d3a75dddc48be388898ac1d8517d4
+  data.tar.gz: f4aee413ab143a08366b223e2c34d62165c69f1e63871eee400d6f82af3c096266e0646f3b4d3d4649a761c913e2c07cef8f852586f0c535d0af3daf1d02f9a6

data/lib/sequel/plugins/privacy.rb

@@ -405,8 +405,10 @@ module Sequel
           case type
           when :many_to_one, :one_to_one
             _override_singular_association(name)
+            _override_association_dataset(name)
           when :one_to_many, :many_to_many
             _override_plural_association(name)
+            _override_association_dataset(name)
             # Check if there are already privacy policies defined for this association
             setup_association_privacy(name) if privacy_association_policies[name]
           end
@@ -436,6 +438,29 @@ module Sequel
 
         private
 
+        sig { params(name: Symbol).void }
+        def _override_association_dataset(name)
+          dataset_method = :"#{name}_dataset"
+          return unless method_defined?(dataset_method)
+
+          original = instance_method(dataset_method)
+          assoc_reflection = association_reflection(name)
+          assoc_class = T.let(nil, T.nilable(T.class_of(Sequel::Model)))
+
+          define_method(dataset_method) do |*args|
+            ds = original.bind(self).(*args)
+            vc = instance_variable_get(:@viewer_context)
+            return ds unless vc
+
+            assoc_class ||= assoc_reflection.associated_class
+            if assoc_class.respond_to?(:privacy_vc_key) && ds.respond_to?(:for_vc)
+              T.unsafe(ds).for_vc(vc)
+            else
+              ds
+            end
+          end
+        end
+
         sig { params(name: Symbol).void }
         def _override_singular_association(name)
           original = instance_method(name)

data/lib/sequel/privacy/version.rb

@@ -3,6 +3,6 @@
 
 module Sequel
   module Privacy
-    VERSION = '0.5'
+    VERSION = '0.5.1'
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sequel-privacy
 version: !ruby/object:Gem::Version
-  version: '0.5'
+  version: 0.5.1
 platform: ruby
 authors:
 - Austin Bales