lib/graphql/schema/union.rb



# frozen_string_literal: true
module GraphQL
  class Schema
    class Union < GraphQL::Schema::Member
      extend GraphQL::Schema::Member::HasUnresolvedTypeError

      class << self
        def inherited(child_class)
          add_unresolved_type_error(child_class)
          super
        end

        def possible_types(*types, context: GraphQL::Query::NullContext, **options)
          if types.any?
            types.each do |t|
              assert_valid_union_member(t)
              type_memberships << type_membership_class.new(self, t, **options)
            end
          else
            visible_types = []
            warden = Warden.from_context(context)
            type_memberships.each do |type_membership|
              if warden.visible_type_membership?(type_membership, context)
                visible_types << type_membership.object_type
              end
            end
            visible_types
          end
        end

        def all_possible_types
          type_memberships.map(&:object_type)
        end

        def type_membership_class(membership_class = nil)
          if membership_class
            @type_membership_class = membership_class
          else
            @type_membership_class || find_inherited_value(:type_membership_class, GraphQL::Schema::TypeMembership)
          end
        end

        def kind
          GraphQL::TypeKinds::UNION
        end

        def type_memberships
          @type_memberships ||= []
        end

        # Update a type membership whose `.object_type` is a string or late-bound type
        # so that the type membership's `.object_type` is the given `object_type`.
        # (This is used for updating the union after the schema as lazily loaded the union member.)
        # @api private
        def assign_type_membership_object_type(object_type)
          assert_valid_union_member(object_type)
          type_memberships.each { |tm|
            possible_type = tm.object_type
            if possible_type.is_a?(String) && (possible_type == object_type.name)
              # This is a match of Ruby class names, not graphql names,
              # since strings are used to refer to constants.
              tm.object_type = object_type
            elsif possible_type.is_a?(LateBoundType) && possible_type.graphql_name == object_type.graphql_name
              tm.object_type = object_type
            end
          }
          nil
        end

        private

        def assert_valid_union_member(type_defn)
          case type_defn
          when Class
            if !type_defn.kind.object?
              raise ArgumentError, "Union possible_types can only be object types (not #{type_defn.kind.name}, #{type_defn.inspect})"
            end
          when Module
            # it's an interface type, defined as a module
            raise ArgumentError, "Union possible_types can only be object types (not interface types), remove #{type_defn.graphql_name} (#{type_defn.inspect})"
          when String, GraphQL::Schema::LateBoundType
            # Ok - assume it will get checked later
          else
            raise ArgumentError, "Union possible_types can only be class-based GraphQL types (not #{type_defn.inspect} (#{type_defn.class.name}))."
          end
        end
      end
    end
  end
end