visibility_test.rb

require "bundler/inline"

gemfile do
  gem "graphql", "~>1.12.0"
end

class BaseField < GraphQL::Schema::Field
  def initialize(*args, federated: true, **kwargs, &block)
    @federated = federated
    super(*args, **kwargs, &block)
  end

  def visible?(context)
    if context[:federated] && @federated == false
      false
    else
      true
    end
  end
end


class BaseObject < GraphQL::Schema::Object
  def self.visible?(context)
    if context[:federated] && @federated == false
      false
    else
      true
    end
  end

  def self.federated(new_config = nil)
    if new_config.nil?
      @federated
    else
      @federated = new_config
    end
  end
end

class NonFederatedObject < BaseObject
  federated(false)

  field :int, Int, null: true
end


class Query < GraphQL::Schema::Object
  field_class BaseField

  field :shared_field, Int, null: true

  field :hidden_field, Int, null: true, federated: false

  field :non_federated_object, NonFederatedObject, null: true
end

class Schema < GraphQL::Schema
  query(Query)
end

puts "Whole schema:\n-----------------"
puts Schema.to_definition

puts "\n\nFederated schema:\n-----------------"
puts Schema.to_definition(context: { federated: true })

# $ ruby visibility_test.rb
# Whole schema:
# -----------------
# type NonFederatedObject {
#   int: Int
# }
#
# type Query {
#   hiddenField: Int
#   nonFederatedObject: NonFederatedObject
#   sharedField: Int
# }
#
#
# Federated schema:
# -----------------
# type Query {
#   sharedField: Int
# }