fix(federation): add reference resolver type to resolver map

subgraph-js/src/schema-helper/resolverMap.ts

@@ -1,22 +1,25 @@
 import { GraphQLFieldResolver, GraphQLScalarType, DocumentNode } from 'graphql';
+import type { GraphQLReferenceResolver } from '../schemaExtensions';
 
 export interface GraphQLSchemaModule {
   typeDefs: DocumentNode;
   resolvers?: GraphQLResolverMap<any>;
 }
 
 // eslint-disable-next-line @typescript-eslint/ban-types
-export interface GraphQLResolverMap<TContext = {}> {
+export interface GraphQLResolverMap<TContext = {}, TReference extends object = any> {
   [typeName: string]:
-    | {
+    | ({
         [fieldName: string]:
           | GraphQLFieldResolver<any, TContext>
           | {
               requires?: string;
               resolve?: GraphQLFieldResolver<any, TContext>;
               subscribe?: GraphQLFieldResolver<any, TContext>;
             };
-      }
+      } & {
+        __resolveReference?: GraphQLReferenceResolver<TContext, TReference>;
+      })
     | GraphQLScalarType
     | {
         [enumValue: string]: string | number;

subgraph-js/src/schemaExtensions.ts

@@ -5,8 +5,8 @@ import {
   GraphQLUnionTypeExtensions
 } from 'graphql';
 
-export type GraphQLReferenceResolver<TContext> = (
-  reference: object,
+export type GraphQLReferenceResolver<TContext, TReference extends object = any> = (
+  reference: TReference,
   context: TContext,
   info: GraphQLResolveInfo,
 ) => any;