Merge pull request #215 from devforce/params-context

Make context accessible when customizing relationship params

Author: richmolj

lib/graphiti/sideload.rb

@@ -197,7 +197,7 @@ def load(parents, query, graph_parent)
 
       with_error_handling Errors::SideloadParamsError do
         params = load_params(parents, query)
-        params_proc&.call(params, parents)
+        params_proc&.call(params, parents, context)
         return [] if blank_query?(params)
         opts = load_options(parents, query)
         opts[:sideload] = self
@@ -434,5 +434,9 @@ def evaluate_flag(flag)
         !!flag
       end
     end
+
+    def context
+      Graphiti.context[:object]
+    end
   end
 end

lib/graphiti/util/link.rb

@@ -63,7 +63,7 @@ def params
             params[:filter] = @sideload.base_filter([@model])
           end
 
-          @sideload.params_proc&.call(params, [@model])
+          @sideload.params_proc&.call(params, [@model], context)
         end
       end
 
@@ -75,6 +75,10 @@ def path
         end
         path
       end
+
+      def context
+        Graphiti.context[:object]
+      end
     end
   end
 end

spec/sideload_spec.rb

@@ -566,16 +566,16 @@ def self.name
 
   describe ".params" do
     before do
-      instance.class.params do |hash, parents, query|
+      instance.class.params do |hash, parents, context|
         hash[:parents] = parents
-        hash[:query] = query
+        hash[:context] = context
       end
     end
 
     it "sets params proc" do
-      hash, parents, query = {}, [double("parent")], double("query")
-      instance.params_proc.call(hash, parents, query)
-      expect(hash).to eq(parents: parents, query: query)
+      hash, parents, context = {}, [double("parent")], double("context")
+      instance.params_proc.call(hash, parents, context)
+      expect(hash).to eq(parents: parents, context: context)
     end
   end
 
@@ -655,19 +655,25 @@ def self.name
 
     context "when params customization" do
       before do
-        instance.class.params do |hash, parents|
+        instance.class.params do |hash, parents, context|
           hash[:a] = parents
+          hash[:b] = context.current_user
         end
       end
 
       it "is respected" do
+        current_user = double
         expected = {
           foo: "bar",
-          a: parents
+          a: parents,
+          b: current_user
         }
         expect(resource_class).to receive(:_all)
           .with(expected, anything, {type: :positions})
-        instance.load(parents, query, nil)
+
+        Graphiti.with_context(OpenStruct.new(current_user: current_user)) do
+          instance.load(parents, query, nil)
+        end
       end
     end