graphql 2.5.23 → 2.6.3

data/lib/graphql/execution/finalize.rb

@@ -0,0 +1,230 @@
+# 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 #{parent_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)
+              return nil if result_h.nil?
+            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)
+              return nil if result_h.nil?
+            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
+
+        effective_idx = -1
+        result_arr.each_with_index do |result_item, before_idx|
+          effective_idx += 1
+          @current_result_path << before_idx
+          new_result = if (f = finalizers(result_arr, before_idx))
+            before_size = result_arr.size
+            run_finalizers(@current_result_path.dup, f, result_arr, effective_idx)
+            after_size = result_arr.size
+            if after_size < before_size
+              effective_idx -= 1
+              :unassigned
+            else
+              result_arr[effective_idx]
+            end
+          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[effective_idx] = nil
+            break if @finalizers_count == 0
+          elsif :unassigned.equal?(new_result)
+            break if @finalizers_count == 0
+          elsif !new_result.equal?(result_item)
+            result_arr[effective_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,333 @@
+# 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 = {}
+        errors = nil
+
+        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
+        rescue GraphQL::RuntimeError => exec_err
+          errors ||= []
+          errors << exec_err
+        end
+
+        return argument_values, errors
+      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 = {}
+
+          if value.is_a?(Hash)
+            @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
+
+              if arg_value.nil? && arg.replace_null_with_default?
+                arg_value = arg.default_value
+              end
+
+              coerced_obj[arg_key] = variable_value(arg_value, arg.type)
+            end
+          else
+            @query.types.arguments(type).each do |arg|
+              arg_key = arg.keyword
+              arg_name = arg.graphql_name
+              if (v_node = value.arguments.find { |a| a.name == arg_name }) # rubocop:disable Development/ContextIsPassedCop
+                arg_value = v_node.value
+                coerced_obj[arg_key] = if arg_value.nil? && arg.replace_null_with_default?
+                  arg.default_value
+                else
+                  variable_value(arg_value, arg.type)
+                end
+              elsif arg.default_value?
+                coerced_obj[arg_key] = arg.default_value
+              else
+                # Nothing
+              end
+            end
+          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?
+          if arg_value.nil?
+            treat_as_type.coerce_input(arg_value, @query.context)
+          end
+          treat_as_type = treat_as_type.of_type
+        end
+
+        if arg_value.nil? && argument_definition.replace_null_with_default?
+          arg_value = argument_definition.default_value
+        end
+
+        if treat_as_type.kind.list? && !arg_value.nil?
+          inner_t = treat_as_type.of_type
+          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, object: nil, arguments: argument_values, field: field_resolve_step&.field_definition)
+          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 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
+          elsif value_node.is_a?(Language::Nodes::NullValue)
+            nil
+          else
+            item_value = value_from_ast(value_node, inner_type)
+            [item_value].freeze
+          end
+        elsif type.kind.input_object?
+          coerced_obj = {}
+          # TODO manually handle NullValue here?
+          if value_node.is_a?(Hash)
+            @query.types.arguments(type).each do |arg|
+              arg_value = value_node[arg.keyword]
+              arg_key = arg.keyword
+              if arg_value.nil?
+                if arg.default_value?
+                  coerced_obj[arg_key] = arg.default_value
+                end
+                next
+              end
+
+              coerced_obj[arg_key] = value_from_ast(arg_value, arg.type)
+            end
+          else
+            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
+          end
+
+
+          coerced_obj
+        elsif type.kind.leaf?
+          if value_node.is_a?(Language::Nodes::AbstractNode) || value_node.is_a?(Array)
+            value_node = coerce_untyped_input(value_node)
+          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
+
+      private
+
+      def coerce_untyped_input(input_value)
+        case input_value
+        when Language::Nodes::AbstractNode
+          case input_value
+          when Language::Nodes::NullValue
+            nil
+          when Language::Nodes::Enum
+            input_value.name
+          when Language::Nodes::InputObject
+            value_h = {}
+            input_value.arguments.each do |arg| # rubocop:disable Development/ContextIsPassedCop
+              value_h[arg.name] = coerce_untyped_input(arg.value)
+            end
+            value_h
+          else
+            raise "Unhandled untyped input AST node: #{input_value.class}"
+          end
+        when Array
+          input_value.map { |v| coerce_untyped_input(v) }
+        else
+          input_value
+        end
+      end
+    end
+  end
+end

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

@@ -30,7 +30,10 @@ module GraphQL
               @storage[argument_owner][parent_object][ast_node] = resolved_args
             end
           end
+        end
 
+        def cached_arguments_for(ast_node, argument_owner)
+          @storage[argument_owner][nil][ast_node]
         end
 
         # @yield [Interpreter::Arguments, Lazy<Interpreter::Arguments>] The finally-loaded arguments

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