Contents

# frozen_string_literal: true

module GraphQL
  module Client
    module Query
      module HasSelectionSet
        ID_FIELD_NAME = 'id'
        INVALID_FIELD = Class.new(StandardError)
        UNDEFINED_FRAGMENT = Class.new(StandardError)

        attr_accessor :selection_set

        def add_connection(connection_name, as: nil, **arguments)
          node_field = nil

          add_field(connection_name, as: as, **arguments) do |connection|
            connection.add_field('edges') do |edges|
              edges.add_field('cursor')
              edges.add_field('node') do |node|
                node_field = node
                yield node
              end
            end

            connection.add_field('pageInfo') do |page_info|
              page_info.add_field('hasPreviousPage')
              page_info.add_field('hasNextPage')
            end
          end

          node_field
        end

        def add_field(field_name, as: nil, **arguments)
          field_defn = resolve(field_name)
          field = Field.new(field_defn, arguments: arguments, as: as, document: document)
          selection_set.add_field(field)

          field.add_field(ID_FIELD_NAME) if field.node?

          if block_given?
            yield field
          else
            field
          end
        end

        def add_fields(*field_names)
          field_names.each do |field_name|
            add_field(field_name)
          end
        end

        def add_fragment(fragment_name)
          fragment = document.fragments.fetch(fragment_name) do
            raise UNDEFINED_FRAGMENT, "a fragment named #{fragment_name} has not been defined"
          end

          selection_set.add_fragment(fragment)
        end

        private

        def resolve(field_name)
          resolver_type.field(field_name) do
            raise INVALID_FIELD, "#{field_name} is not a valid field for #{resolver_type}"
          end
        end
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
graphql_client-0.4.1	lib/graphql_client/query/has_selection_set.rb
graphql_client-0.3.3	lib/graphql_client/query/has_selection_set.rb