graphql 2.5.23 → 2.5.24

data/lib/graphql/execution/finalize.rb

@@ -0,0 +1,217 @@
+# frozen_string_literal: true
+module GraphQL
+  module Execution
+    class Finalize
+      def initialize(query, data, runner)
+        @query = query
+        @data = data
+        @static_type_at = runner.static_type_at
+        @runner = runner
+        @current_exec_path = query.path.dup
+        @current_result_path = query.path.dup
+        @finalizers = runner.finalizers ? runner.finalizers[query] : {}.compare_by_identity
+        @finalizers_count = 0
+        @finalizers.each do |key, values|
+          values.each do |key2, values2|
+            case values2
+            when Array
+              @finalizers_count += values2.size
+            else
+              @finalizers_count += 1
+            end
+          end
+        end
+
+        query.context.errors.each do |err|
+          err_path = err.path - @current_exec_path
+          key = err_path.pop
+          targets = [data]
+          while (part = err_path.shift)
+            targets.map! { |t| t[part] }
+            targets.flatten!
+          end
+
+          targets.each_with_index do |target, idx|
+            if target.is_a?(Hash)
+              if target[key].equal?(err)
+                tf = @finalizers[target] ||= {}.compare_by_identity
+                tf[key] = err
+                @finalizers_count += 1
+              elsif (arr = target[key]).is_a?(Array)
+                arr.each_with_index do |el, idx|
+                  if el.equal?(err)
+                    tf = @finalizers[arr] ||= {}.compare_by_identity
+                    tf[idx] = err
+                    @finalizers_count += 1
+                  end
+                end
+              end
+            end
+          end
+        end
+      end
+
+      def run
+        if (selected_operation = @query.selected_operation) && @data
+          if @data.is_a?(Hash)
+            check_object_result(@data, @query.root_type, selected_operation.selections)
+          elsif @data.is_a?(Array)
+            check_list_result(@data, @query.root_type, selected_operation.selections)
+          elsif @data.is_a?(Finalizer)
+            dummy_data = {}
+            dummy_key = "__dummy"
+            @data.path = @query.path
+            @data.finalize_graphql_result(@query, dummy_data, dummy_key)
+            dummy_data[dummy_key]
+          else
+            raise ArgumentError, "Unexpected @data: #{@data.inspect}"
+          end
+        else
+          @data
+        end
+      end
+
+      private
+
+      def run_finalizers(result_path, finalizer_or_finalizers, result_data, result_key)
+        if finalizer_or_finalizers.is_a?(Array)
+          finalizer_or_finalizers.each { |f|
+            f.path = result_path
+            f.finalize_graphql_result(@query, result_data, result_key)
+          }
+          @finalizers_count -= finalizer_or_finalizers.size
+        else
+          f = finalizer_or_finalizers
+          f.path = result_path
+          f.finalize_graphql_result(@query, result_data, result_key)
+          @finalizers_count -= 1
+        end
+      end
+
+      def finalizers(result_value, key)
+        finalizers_for_value = @finalizers[result_value]
+        finalizers_for_value && finalizers_for_value[key]
+      end
+
+      def check_object_result(result_h, parent_type, ast_selections)
+        if (f = finalizers(result_h, nil))
+          run_finalizers(@current_result_path.dup, f, result_h, nil)
+          return result_h if @finalizers_count == 0
+        end
+
+        if parent_type.kind.abstract?
+          parent_type = @runner.runtime_type_at[result_h]
+        end
+
+        ast_selections.each do |ast_selection|
+          case ast_selection
+          when Language::Nodes::Field
+            key = ast_selection.alias || ast_selection.name
+            if (f = finalizers(result_h, key))
+              result_value = result_h[key]
+              run_finalizers(@current_result_path.dup << key, f, result_h, key)
+              new_result_value = result_h.key?(key) ? result_h[key] : :unassigned
+            end
+            next if !(f || result_h.key?(key))
+            begin
+              @current_exec_path << key
+              @current_result_path << key
+
+              field_defn = @query.context.types.field(parent_type, ast_selection.name) || raise("Invariant: No field found for #{static_type.to_type_signature}.#{ast_selection.name}")
+              result_type = field_defn.type
+              if (result_type_non_null = result_type.non_null?)
+                result_type = result_type.of_type
+              end
+
+              if !f
+                result_value = result_h[key]
+                new_result_value = if result_type.list? && result_value
+                  check_list_result(result_value, result_type.of_type, ast_selection.selections)
+                elsif !result_type.kind.leaf? && result_value
+                  check_object_result(result_value, result_type, ast_selection.selections)
+                else
+                  result_value
+                end
+              end
+
+              if new_result_value.nil? && result_type_non_null
+                return nil
+              elsif :unassigned.equal?(new_result_value)
+                # Do nothing
+                break if @finalizers_count == 0
+              elsif !new_result_value.equal?(result_value)
+                result_h[key] = new_result_value
+                break if @finalizers_count == 0
+              end
+            ensure
+              @current_exec_path.pop
+              @current_result_path.pop
+            end
+          when Language::Nodes::InlineFragment
+            static_type_at_result = @static_type_at[result_h]
+            if static_type_at_result && (
+                (t = ast_selection.type).nil? ||
+                @runner.type_condition_applies?(@query.context, static_type_at_result, t.name)
+              )
+              result_h = check_object_result(result_h, parent_type, ast_selection.selections)
+            end
+          when Language::Nodes::FragmentSpread
+            fragment_defn = @query.document.definitions.find { |defn| defn.is_a?(Language::Nodes::FragmentDefinition) && defn.name == ast_selection.name }
+            static_type_at_result = @static_type_at[result_h]
+            if static_type_at_result && @runner.type_condition_applies?(@query.context, static_type_at_result, fragment_defn.type.name)
+              result_h = check_object_result(result_h, parent_type, fragment_defn.selections)
+            end
+          end
+        end
+
+        result_h
+      end
+
+      def check_list_result(result_arr, inner_type, ast_selections)
+        inner_type_non_null = false
+        if inner_type.non_null?
+          inner_type_non_null = true
+          inner_type = inner_type.of_type
+        end
+
+        new_invalid_null = false
+
+        if (f = finalizers(result_arr, nil))
+          run_finalizers(@current_result_path.dup, f, result_arr, nil)
+          return result_arr if @finalizers_count == 0
+        end
+
+        result_arr.each_with_index do |result_item, idx|
+          @current_result_path << idx
+          new_result = if (f = finalizers(result_arr, idx))
+            run_finalizers(@current_result_path.dup, f, result_arr, idx)
+            result_arr[idx]
+          elsif inner_type.list? && result_item
+            check_list_result(result_item, inner_type.of_type, ast_selections)
+          elsif !inner_type.kind.leaf? && result_item
+            check_object_result(result_item, inner_type, ast_selections)
+          else
+            result_item
+          end
+
+          if new_result.nil? && inner_type_non_null
+            new_invalid_null = true
+            result_arr[idx] = nil
+            break if @finalizers_count == 0
+          elsif !new_result.equal?(result_item)
+            result_arr[idx] = new_result
+            break if @finalizers_count == 0
+          end
+        ensure
+          @current_result_path.pop
+        end
+
+        if new_invalid_null
+          nil
+        else
+          result_arr
+        end
+      end
+    end
+  end
+end

data/lib/graphql/execution/input_values.rb

@@ -0,0 +1,261 @@
+# frozen_string_literal: true
+module GraphQL
+  module Execution
+    class InputValues
+      def initialize(query, runner)
+        @query = query
+        @runner = runner
+        @variable_values = nil
+      end
+
+      def variable_values
+        @variable_values ||= begin
+          variable_nodes = @query.selected_operation.variables
+          if variable_nodes.empty?
+            EmptyObjects::EMPTY_HASH
+          else
+            raw_values = @query.provided_variables
+            values = {}
+            variable_nodes.each do |var_node|
+              var_ast_value = if raw_values.key?(var_node.name)
+                raw_values[var_node.name]
+              elsif raw_values.key?(sym_name = var_node.name.to_sym)
+                raw_values[sym_name]
+              elsif !var_node.default_value.nil?
+                var_node.default_value
+              else
+                next
+              end
+
+              var_type = @runner.schema.type_from_ast(var_node.type, context: @query.context)
+              values[var_node.name] = variable_value(var_ast_value, var_type)
+            end
+            values
+          end
+        end
+      end
+
+      def argument_values(owner_defn, argument_nodes, field_resolve_step)
+        arg_defns = @query.types.arguments(owner_defn)
+        argument_values = {}
+
+        arg_defns.each do |argument_definition|
+          arg_ruby_key = argument_definition.keyword
+          arg_graphql_key = argument_definition.graphql_name
+          arg_node = argument_nodes.find { |a| a.name == arg_graphql_key }
+          if arg_node.nil? || (arg_node.value.is_a?(Language::Nodes::VariableIdentifier) && !variable_values.key?(arg_node.value.name))
+            if argument_definition.default_value?
+              arg_value = value_from_ast(argument_definition.default_value, argument_definition.type)
+              argument_value(argument_values, arg_ruby_key, argument_definition, arg_value, nil, field_resolve_step)
+            end
+          else
+            arg_value = value_from_ast(arg_node.value, argument_definition.type)
+            argument_value(argument_values, arg_ruby_key, argument_definition, arg_value, nil, field_resolve_step)
+          end
+        end
+
+        argument_values
+      rescue GraphQL::ExecutionError => exec_err
+        exec_err
+      end
+
+      private
+
+      def variable_value(value, type)
+        if type.non_null?
+          type = type.of_type
+        end
+
+        if value.is_a?(Language::Nodes::Enum)
+          value = value.name
+        end
+
+        if value.nil?
+          nil
+        elsif type.list?
+          inner_type = type.of_type
+          if value.is_a?(Array)
+            value.map { |v| variable_value(v, inner_type) }.freeze
+          else
+            [variable_value(value, inner_type)].freeze
+          end
+        elsif type.kind.input_object?
+          coerced_obj = {}
+
+          @query.types.arguments(type).each do |arg|
+            arg_key = arg.keyword
+            if value.key?(arg.graphql_name)
+              arg_value = value[arg.graphql_name]
+            elsif value.key?(sym_name = arg.graphql_name.to_sym)
+              arg_value = value[sym_name]
+            elsif arg.default_value?
+              coerced_obj[arg_key] = arg.default_value
+              next
+            else
+              next
+            end
+
+            coerced_obj[arg_key] = variable_value(arg_value, arg.type)
+          end
+
+          coerced_obj
+        elsif type.kind.leaf?
+          type.coerce_input(value, @query.context)
+        else
+          raise GraphQL::Error, "Unexpected input type: #{type.graphql_name}."
+        end
+      end
+
+      def argument_value(argument_values, argument_key, argument_definition, arg_value, override_type, field_resolve_step)
+        treat_as_type = override_type || argument_definition.type
+        if treat_as_type.non_null?
+          treat_as_type = treat_as_type.of_type
+        end
+
+        if treat_as_type.kind.list? && !arg_value.nil?
+          inner_t = treat_as_type.unwrap
+          arg_value = if arg_value.is_a?(Array)
+            values = Array.new(arg_value.size)
+            arg_value.each_with_index { |inner_v, idx| argument_value(values, idx, argument_definition, inner_v, inner_t, field_resolve_step)}
+            values
+          else
+            values = [nil]
+            argument_value(values, 0, argument_definition, arg_value, inner_t, field_resolve_step)
+            values
+          end
+        end
+
+        if arg_value && treat_as_type.kind.input_object?
+          arg_defns = @query.types.arguments(treat_as_type)
+          new_arg_value = {}
+          arg_defns.each do |inner_arg_defn|
+            inner_arg_key = inner_arg_defn.keyword
+            if arg_value.is_a?(Hash)
+              if arg_value.key?(inner_arg_key)
+                inner_arg_value = arg_value[inner_arg_key]
+                argument_value(new_arg_value, inner_arg_key, inner_arg_defn, inner_arg_value, nil, field_resolve_step)
+              end
+            else
+              inner_arg_name = inner_arg_defn.graphql_name
+              inner_arg_value = arg_value.arguments.find { |a| a.name == inner_arg_name } # rubocop:disable Development/ContextIsPassedCop
+              if inner_arg_value
+                argument_value(new_arg_value, inner_arg_key, inner_arg_defn, inner_arg_value, nil, field_resolve_step)
+              end
+            end
+          end
+          arg_value = treat_as_type.new(nil, ruby_kwargs: new_arg_value, context: @query.context, defaults_used: nil)
+        end
+
+        if override_type.nil? # only on root arguments, not list elements
+          arg_value = begin
+            argument_definition.prepare_value(nil, arg_value, context: @query.context)
+          rescue StandardError => err
+            @runner.schema.handle_or_reraise(@query.context, err)
+          end
+        end
+
+        if field_resolve_step && arg_value && override_type.nil? && argument_definition.loads
+          field_defn = field_resolve_step.field_definition
+          load_receiver = if (r = field_defn.resolver)
+            r.new(field: field_defn, context: @query.context, object: nil)
+          else
+            field_defn
+          end
+          ps = field_resolve_step.pending_steps ||= []
+
+          if argument_definition.type.list?
+            results = Array.new(arg_value.size, nil)
+            argument_values[argument_key] = results
+            arg_value.each_with_index do |inner_v, idx|
+              loads_step = LoadArgumentStep.new(
+                field_resolve_step: field_resolve_step,
+                load_receiver: load_receiver,
+                argument_value: inner_v,
+                argument_definition: argument_definition,
+                arguments: results,
+                argument_key: idx,
+              )
+              ps.push(loads_step)
+              @runner.add_step(loads_step)
+            end
+          else
+            loads_step = LoadArgumentStep.new(
+              field_resolve_step: field_resolve_step,
+              load_receiver: load_receiver,
+              argument_value: arg_value,
+              argument_definition: argument_definition,
+              arguments: argument_values,
+              argument_key: argument_key,
+            )
+            ps.push(loads_step)
+            @runner.add_step(loads_step)
+          end
+        else
+          argument_values[argument_key] = arg_value
+        end
+        nil
+      end
+
+      def value_from_ast(value_node, type)
+        if type.non_null?
+          type = type.of_type
+        end
+
+        if value_node.nil?
+          nil
+        elsif value_node.is_a?(GraphQL::Language::Nodes::VariableIdentifier)
+          variable_values[value_node.name]
+        elsif value_node.is_a?(GraphQL::Language::Nodes::NullValue)
+          nil
+        elsif type.list?
+          inner_type = type.of_type
+          if value_node.is_a?(Array)
+            coerced_items = value_node.map do |inner_value_node|
+              value_from_ast(inner_value_node, inner_type)
+            end
+            coerced_items.freeze
+          else
+            item_value = value_from_ast(value_node, inner_type)
+            [item_value].freeze
+          end
+
+        elsif type.kind.input_object?
+          coerced_obj = {}
+          arg_nodes_by_name = value_node.arguments.each_with_object({}) do |arg_node, acc| # rubocop:disable Development/ContextIsPassedCop
+            acc[arg_node.name] = arg_node
+          end
+
+          @query.types.arguments(type).each do |arg|
+            arg_node = arg_nodes_by_name[arg.graphql_name]
+            arg_key = arg.keyword
+            if arg_node.nil? || (arg_node.value.is_a?(Language::Nodes::VariableIdentifier) && !variable_values.key?(arg_node.value.name))
+              if arg.default_value?
+                coerced_obj[arg_key] = arg.default_value
+              end
+              next
+            end
+
+            arg_value = value_from_ast(arg_node.value, arg.type)
+            coerced_obj[arg_key] = arg_value
+          end
+
+          coerced_obj
+        elsif type.kind.leaf?
+          if type.kind.enum?
+            if value_node.is_a?(GraphQL::Language::Nodes::Enum)
+              value_node = value_node.name
+            end
+          end
+
+          begin
+            type.coerce_input(value_node, @query.context)
+          rescue GraphQL::UnauthorizedEnumValueError => enum_err
+            @runner.schema.unauthorized_object(enum_err)
+          end
+        else
+          raise "Unexpected input type: #{type.to_type_signature}."
+        end
+      end
+    end
+  end
+end

data/lib/graphql/execution/interpreter/handles_raw_value.rb

@@ -5,6 +5,12 @@ module GraphQL
     class Interpreter
       # Wrapper for raw values
       class RawValue
+        include GraphQL::Execution::Finalizer
+
+        def finalize_graphql_result(query, result_data, result_key)
+          result_data[result_key] = @object
+        end
+
         def initialize(obj = nil)
           @object = obj
         end

data/lib/graphql/execution/load_argument_step.rb

@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+module GraphQL
+  module Execution
+    class LoadArgumentStep
+      def initialize(field_resolve_step:, arguments:, load_receiver:, argument_value:, argument_definition:, argument_key:)
+        @field_resolve_step = field_resolve_step
+        @load_receiver = load_receiver
+        @arguments = arguments
+        @argument_value = argument_value
+        @argument_definition = argument_definition
+        @argument_key = argument_key
+        @loaded_value = nil
+      end
+
+      def value
+        @loaded_value = @field_resolve_step.sync(@loaded_value)
+        assign_value
+      end
+
+      def call
+        context = @field_resolve_step.selections_step.query.context
+        @loaded_value = begin
+          @load_receiver.load_and_authorize_application_object(@argument_definition, @argument_value, context)
+        rescue GraphQL::UnauthorizedError => auth_err
+          context.schema.unauthorized_object(auth_err)
+        end
+        if (runner = @field_resolve_step.runner).resolves_lazies && runner.lazy?(@loaded_value)
+          runner.dataloader.lazy_at_depth(@field_resolve_step.path.size, self)
+        else
+          assign_value
+        end
+      rescue GraphQL::RuntimeError => err
+        @loaded_value = err
+        assign_value
+      rescue StandardError => stderr
+        @loaded_value = begin
+          context.query.handle_or_reraise(stderr)
+        rescue GraphQL::ExecutionError => ex_err
+          ex_err
+        end
+        assign_value
+      end
+
+      private
+
+      def assign_value
+        if @loaded_value.is_a?(GraphQL::RuntimeError)
+          @loaded_value.path = @field_resolve_step.path
+          @field_resolve_step.arguments = @loaded_value
+        else
+          query = @field_resolve_step.selections_step.query
+          query.current_trace.object_loaded(@argument_definition, @loaded_value, query.context)
+          @arguments[@argument_key] = @loaded_value
+        end
+
+        field_pending_steps = @field_resolve_step.pending_steps
+        field_pending_steps.delete(self)
+        if @field_resolve_step.arguments && field_pending_steps.size == 0 # rubocop:disable Development/ContextIsPassedCop
+          @field_resolve_step.runner.add_step(@field_resolve_step)
+        end
+      end
+    end
+  end
+end

data/lib/graphql/execution/next.rb

@@ -1,11 +1,29 @@
 # frozen_string_literal: true
-require "graphql/execution/next/prepare_object_step"
-require "graphql/execution/next/field_resolve_step"
-require "graphql/execution/next/load_argument_step"
-require "graphql/execution/next/runner"
-require "graphql/execution/next/selections_step"
+require "graphql/execution/prepare_object_step"
+require "graphql/execution/input_values"
+require "graphql/execution/field_resolve_step"
+require "graphql/execution/finalize"
+require "graphql/execution/load_argument_step"
+require "graphql/execution/runner"
+require "graphql/execution/selections_step"
 module GraphQL
   module Execution
+    module Finalizer
+      attr_accessor :path
+      def finalize_graphql_result(query, result_data, result_key)
+        raise RequiredImplementationMissingError
+      end
+    end
+
+    module HaltExecution
+    end
+
+    module PostProcessor
+      def after_resolve(field_results)
+        raise RequiredImplementationMissingError, "#{self.class}#after_resolve should handle `field_results` and return a new value to use"
+      end
+    end
+
     module Next
       module SchemaExtension
         def execute_next(query_str = nil, context: nil, document: nil, operation_name: nil, variables: nil, root_value: nil, validate: true, visibility_profile: nil)