@graphql-codegen/visitor-plugin-common 5.6.1-alpha-20250105053549-084c750c69037c605fdcf600c20527dc5c74326e → 6.0.0-alpha-20250109122810-aa191e7ac94facb19d577e2fb0b27967d3590cbb

package/cjs/base-resolvers-visitor.js

@@ -440,6 +440,23 @@ class BaseResolversVisitor extends base_visitor_js_1.BaseVisitor {
             .map(([typeName, value]) => (0, utils_js_1.indent)(`${typeName}: ${value}${this.getPunctuation(declarationKind)}`))
             .join('\n')).string;
     }
+    buildFederationTypes() {
+        const federationMeta = this._federation.getMeta();
+        if (Object.keys(federationMeta).length === 0) {
+            return '';
+        }
+        const declarationKind = 'type';
+        return new utils_js_1.DeclarationBlock(this._declarationBlockConfig)
+            .export()
+            .asKind(declarationKind)
+            .withName(this.convertName('FederationTypes'))
+            .withComment('Mapping of federation types')
+            .withBlock(Object.keys(federationMeta)
+            .map(typeName => {
+            return (0, utils_js_1.indent)(`${typeName}: ${this.convertName(typeName)}${this.getPunctuation(declarationKind)}`);
+        })
+            .join('\n')).string;
+    }
     get schema() {
         return this._schema;
     }
@@ -646,6 +663,7 @@ class BaseResolversVisitor extends base_visitor_js_1.BaseVisitor {
                 fieldNode: original,
                 parentType,
                 parentTypeSignature: this.getParentTypeForSignature(node),
+                federationTypeSignature: 'FederationType',
             });
             const mappedTypeKey = isSubscriptionType ? `${mappedType}, "${node.name}"` : mappedType;
             const directiveMappings = node.directives
@@ -665,6 +683,7 @@ class BaseResolversVisitor extends base_visitor_js_1.BaseVisitor {
                     if (!federationDetails || federationDetails.resolvableKeyDirectives.length === 0) {
                         return '';
                     }
+                    signature.modifier = ''; // if a federation type has resolvable @key, then it should be required
                 }
                 this._federation.setMeta(parentType.name, { hasResolveReference: true });
                 signature.type = 'ReferenceResolver';
@@ -727,10 +746,18 @@ class BaseResolversVisitor extends base_visitor_js_1.BaseVisitor {
         if (!rootType) {
             fieldsContent.push((0, utils_js_1.indent)(`${this.config.internalResolversPrefix}isTypeOf?: IsTypeOfResolverFn<ParentType, ContextType>${this.getPunctuation(declarationKind)}`));
         }
+        const genericTypes = [
+            `ContextType = ${this.config.contextType.type}`,
+            this.transformParentGenericType(parentType),
+        ];
+        if (this._federation.getMeta()[typeName]) {
+            const typeRef = `${this.convertName('FederationTypes')}['${typeName}']`;
+            genericTypes.push(`FederationType extends ${typeRef} = ${typeRef}`);
+        }
         const block = new utils_js_1.DeclarationBlock(this._declarationBlockConfig)
             .export()
             .asKind(declarationKind)
-            .withName(name, `<ContextType = ${this.config.contextType.type}, ${this.transformParentGenericType(parentType)}>`)
+            .withName(name, `<${genericTypes.join(', ')}>`)
             .withBlock(fieldsContent.join('\n'));
         this._collectedResolvers[node.name] = {
             typename: name + '<ContextType>',

package/esm/base-resolvers-visitor.js

@@ -436,6 +436,23 @@ export class BaseResolversVisitor extends BaseVisitor {
             .map(([typeName, value]) => indent(`${typeName}: ${value}${this.getPunctuation(declarationKind)}`))
             .join('\n')).string;
     }
+    buildFederationTypes() {
+        const federationMeta = this._federation.getMeta();
+        if (Object.keys(federationMeta).length === 0) {
+            return '';
+        }
+        const declarationKind = 'type';
+        return new DeclarationBlock(this._declarationBlockConfig)
+            .export()
+            .asKind(declarationKind)
+            .withName(this.convertName('FederationTypes'))
+            .withComment('Mapping of federation types')
+            .withBlock(Object.keys(federationMeta)
+            .map(typeName => {
+            return indent(`${typeName}: ${this.convertName(typeName)}${this.getPunctuation(declarationKind)}`);
+        })
+            .join('\n')).string;
+    }
     get schema() {
         return this._schema;
     }
@@ -642,6 +659,7 @@ export class BaseResolversVisitor extends BaseVisitor {
                 fieldNode: original,
                 parentType,
                 parentTypeSignature: this.getParentTypeForSignature(node),
+                federationTypeSignature: 'FederationType',
             });
             const mappedTypeKey = isSubscriptionType ? `${mappedType}, "${node.name}"` : mappedType;
             const directiveMappings = node.directives
@@ -661,6 +679,7 @@ export class BaseResolversVisitor extends BaseVisitor {
                     if (!federationDetails || federationDetails.resolvableKeyDirectives.length === 0) {
                         return '';
                     }
+                    signature.modifier = ''; // if a federation type has resolvable @key, then it should be required
                 }
                 this._federation.setMeta(parentType.name, { hasResolveReference: true });
                 signature.type = 'ReferenceResolver';
@@ -723,10 +742,18 @@ export class BaseResolversVisitor extends BaseVisitor {
         if (!rootType) {
             fieldsContent.push(indent(`${this.config.internalResolversPrefix}isTypeOf?: IsTypeOfResolverFn<ParentType, ContextType>${this.getPunctuation(declarationKind)}`));
         }
+        const genericTypes = [
+            `ContextType = ${this.config.contextType.type}`,
+            this.transformParentGenericType(parentType),
+        ];
+        if (this._federation.getMeta()[typeName]) {
+            const typeRef = `${this.convertName('FederationTypes')}['${typeName}']`;
+            genericTypes.push(`FederationType extends ${typeRef} = ${typeRef}`);
+        }
         const block = new DeclarationBlock(this._declarationBlockConfig)
             .export()
             .asKind(declarationKind)
-            .withName(name, `<ContextType = ${this.config.contextType.type}, ${this.transformParentGenericType(parentType)}>`)
+            .withName(name, `<${genericTypes.join(', ')}>`)
             .withBlock(fieldsContent.join('\n'));
         this._collectedResolvers[node.name] = {
             typename: name + '<ContextType>',

package/package.json

@@ -1,12 +1,12 @@
 {
   "name": "@graphql-codegen/visitor-plugin-common",
-  "version": "5.6.1-alpha-20250105053549-084c750c69037c605fdcf600c20527dc5c74326e",
+  "version": "6.0.0-alpha-20250109122810-aa191e7ac94facb19d577e2fb0b27967d3590cbb",
   "peerDependencies": {
     "graphql": "^0.8.0 || ^0.9.0 || ^0.10.0 || ^0.11.0 || ^0.12.0 || ^0.13.0 || ^14.0.0 || ^15.0.0 || ^16.0.0"
   },
   "dependencies": {
     "@graphql-tools/optimize": "^2.0.0",
-    "@graphql-codegen/plugin-helpers": "^5.1.0",
+    "@graphql-codegen/plugin-helpers": "6.0.0-alpha-20250109122810-aa191e7ac94facb19d577e2fb0b27967d3590cbb",
     "@graphql-tools/relay-operation-optimizer": "^7.0.0",
     "@graphql-tools/utils": "^10.0.0",
     "auto-bind": "~4.0.0",

package/typings/base-resolvers-visitor.d.cts

@@ -693,6 +693,7 @@ export declare class BaseResolversVisitor<TRawConfig extends RawResolversConfig
     buildResolversParentTypes(): string;
     buildResolversUnionTypes(): string;
     buildResolversInterfaceTypes(): string;
+    buildFederationTypes(): string;
     get schema(): GraphQLSchema;
     get defaultMapperType(): string;
     get unusedMappers(): string[];

package/typings/base-resolvers-visitor.d.ts

@@ -693,6 +693,7 @@ export declare class BaseResolversVisitor<TRawConfig extends RawResolversConfig
     buildResolversParentTypes(): string;
     buildResolversUnionTypes(): string;
     buildResolversInterfaceTypes(): string;
+    buildFederationTypes(): string;
     get schema(): GraphQLSchema;
     get defaultMapperType(): string;
     get unusedMappers(): string[];