@apollo/federation-internals 2.2.3 → 2.3.0-beta.1

package/src/federation.ts

@@ -28,7 +28,7 @@ import {
   sourceASTs,
   UnionType,
 } from "./definitions";
-import { assert, joinStrings, MultiMap, printHumanReadableList, OrderedMap, mapValues } from "./utils";
+import { assert, MultiMap, printHumanReadableList, OrderedMap, mapValues } from "./utils";
 import { SDLValidationRule } from "graphql/validation/ValidationContext";
 import { specifiedSDLRules } from "graphql/validation/specifiedRules";
 import {
@@ -40,8 +40,11 @@ import {
   PossibleTypeExtensionsRule,
   print as printAST,
   Source,
-  SchemaExtensionNode,
   GraphQLErrorOptions,
+  SchemaDefinitionNode,
+  OperationTypeNode,
+  OperationTypeDefinitionNode,
+  ConstDirectiveNode,
 } from "graphql";
 import { KnownTypeNamesInFederationRule } from "./validation/KnownTypeNamesInFederationRule";
 import { buildSchema, buildSchemaFromAST } from "./buildSchema";
@@ -302,6 +305,7 @@ function validateAllFieldSet<TParent extends SchemaElement<any, any>>({
   isOnParentType = false,
   allowOnNonExternalLeafFields = false,
   allowFieldsWithArguments = false,
+  allowOnInterface = false,
   onFields,
 }: {
   definition: DirectiveDefinition<{fields: any}>,
@@ -311,13 +315,14 @@ function validateAllFieldSet<TParent extends SchemaElement<any, any>>({
   isOnParentType?: boolean,
   allowOnNonExternalLeafFields?: boolean,
   allowFieldsWithArguments?: boolean,
+  allowOnInterface?: boolean,
   onFields?: (field: FieldDefinition<any>) => void,
 }): void {
   for (const application of definition.applications()) {
     const elt = application.parent as TParent;
     const type = targetTypeExtractor(elt);
     const parentType = isOnParentType ? type : (elt.parent as NamedType);
-    if (isInterfaceType(parentType)) {
+    if (isInterfaceType(parentType) && !allowOnInterface) {
       const code = ERROR_CATEGORIES.DIRECTIVE_UNSUPPORTED_ON_INTERFACE.get(definition.name);
       errorCollector.push(code.err(
         isOnParentType
@@ -438,43 +443,64 @@ function validateNoExternalOnInterfaceFields(metadata: FederationMetadata, error
   }
 }
 
-/**
- * Register errors when, for an interface field, some of the implementations of that field are @external
- * _and_ not all of those field implementation have the same type (which otherwise allowed because field
- * implementation types can be a subtype of the interface field they implement).
- * This is done because if that is the case, federation may later generate invalid query plans (see details
- * on https://github.com/apollographql/federation/issues/1257).
- * This "limitation" will be removed when we stop generating invalid query plans for it.
- */
-function validateInterfaceRuntimeImplementationFieldsTypes(
-  itf: InterfaceType,
-  metadata: FederationMetadata,
-  errorCollector: GraphQLError[],
-): void {
-  const requiresDirective = federationMetadata(itf.schema())?.requiresDirective();
-  assert(requiresDirective, 'Schema should be a federation subgraph, but @requires directive not found');
-  const runtimeTypes = itf.possibleRuntimeTypes();
-  for (const field of itf.fields()) {
-    const withExternalOrRequires: FieldDefinition<ObjectType>[] = [];
-    const typeToImplems: MultiMap<string, FieldDefinition<ObjectType>> = new MultiMap();
-    const nodes: ASTNode[] = [];
-    for (const type of runtimeTypes) {
-      const implemField = type.field(field.name);
-      if (!implemField) continue;
-      if (implemField.sourceAST) {
-        nodes.push(implemField.sourceAST);
+function validateKeyOnInterfacesAreAlsoOnAllImplementations(metadata: FederationMetadata, errorCollector: GraphQLError[]): void {
+  for (const itfType of metadata.schema.interfaceTypes()) {
+    const implementations = itfType.possibleRuntimeTypes();
+    for (const keyApplication of itfType.appliedDirectivesOf(metadata.keyDirective())) {
+      // Note that we will always have validated all @key fields at this point, so not bothering with extra validation
+      const fields = parseFieldSetArgument({parentType: itfType, directive: keyApplication, validate: false});
+      const isResolvable = !(keyApplication.arguments().resolvable === false);
+      const implementationsWithKeyButNotResolvable = new Array<ObjectType>();
+      const implementationsMissingKey = new Array<ObjectType>();
+      for (const type of implementations) {
+        const matchingApp = type.appliedDirectivesOf(metadata.keyDirective()).find((app) => {
+          const appFields = parseFieldSetArgument({parentType: type, directive: app, validate: false});
+          return fields.equals(appFields);
+        });
+        if (matchingApp) {
+          if (isResolvable && matchingApp.arguments().resolvable === false) {
+            implementationsWithKeyButNotResolvable.push(type);
+          }
+        } else {
+          implementationsMissingKey.push(type);
+        }
       }
-      if (metadata.isFieldExternal(implemField) || implemField.hasAppliedDirective(requiresDirective)) {
-        withExternalOrRequires.push(implemField);
+
+      if (implementationsMissingKey.length > 0) {
+        const typesString = printHumanReadableList(
+          implementationsMissingKey.map((i) => `"${i.coordinate}"`),
+          {
+            prefix: 'type',
+            prefixPlural: 'types',
+          }
+        );
+        errorCollector.push(ERRORS.INTERFACE_KEY_NOT_ON_IMPLEMENTATION.err(
+          `Key ${keyApplication} on interface type "${itfType.coordinate}" is missing on implementation ${typesString}.`,
+          { nodes: sourceASTs(...implementationsMissingKey) },
+        ));
+      } else if (implementationsWithKeyButNotResolvable.length > 0) {
+        const typesString = printHumanReadableList(
+          implementationsWithKeyButNotResolvable.map((i) => `"${i.coordinate}"`),
+          {
+            prefix: 'type',
+            prefixPlural: 'types',
+          }
+        );
+        errorCollector.push(ERRORS.INTERFACE_KEY_NOT_ON_IMPLEMENTATION.err(
+          `Key ${keyApplication} on interface type "${itfType.coordinate}" should be resolvable on all implementation types, but is declared with argument "@key(resolvable:)" set to false in ${typesString}.`,
+          { nodes: sourceASTs(...implementationsWithKeyButNotResolvable) },
+        ));
       }
-      const returnType = implemField.type!;
-      typeToImplems.add(returnType.toString(), implemField);
-    }
-    if (withExternalOrRequires.length > 0 && typeToImplems.size > 1) {
-      const typeToImplemsArray = [...typeToImplems.entries()];
-      errorCollector.push(ERRORS.INTERFACE_FIELD_IMPLEM_TYPE_MISMATCH.err(
-        `Some of the runtime implementations of interface field "${field.coordinate}" are marked @external or have a @require (${withExternalOrRequires.map(printFieldCoordinate)}) so all the implementations should use the same type (a current limitation of federation; see https://github.com/apollographql/federation/issues/1257), but ${formatFieldsToReturnType(typeToImplemsArray[0])} while ${joinStrings(typeToImplemsArray.slice(1).map(formatFieldsToReturnType), ' and ')}.`,
-        { nodes },
+    }
+  }
+}
+
+function validateInterfaceObjectsAreOnEntities(metadata: FederationMetadata, errorCollector: GraphQLError[]): void {
+  for (const application of metadata.interfaceObjectDirective().applications()) {
+    if (!isEntityType(application.parent)) {
+      errorCollector.push(ERRORS.INTERFACE_OBJECT_USAGE_ERROR.err(
+        `The @interfaceObject directive can only be applied to entity types but type "${application.parent.coordinate}" has no @key in this subgraph.`,
+        { nodes: application.parent.sourceAST }
       ));
     }
   }
@@ -521,13 +547,6 @@ function validateShareableNotRepeatedOnSameDeclaration(
   }
 }
 
-
-const printFieldCoordinate = (f: FieldDefinition<CompositeType>): string => `"${f.coordinate}"`;
-
-function formatFieldsToReturnType([type, implems]: [string, FieldDefinition<ObjectType>[]]) {
-  return `${joinStrings(implems.map(printFieldCoordinate))} ${implems.length == 1 ? 'has' : 'have'} type "${type}"`;
-}
-
 export class FederationMetadata {
   private _externalTester?: ExternalTester;
   private _sharingPredicate?: (field: FieldDefinition<CompositeType>) => boolean;
@@ -606,6 +625,11 @@ export class FederationMetadata {
     return this.sharingPredicate()(field);
   }
 
+  isInterfaceObjectType(type: NamedType): type is ObjectType {
+    return isObjectType(type)
+      && hasAppliedDirective(type, this.interfaceObjectDirective());
+  }
+
   federationDirectiveNameInSchema(name: string): string {
     if (this.isFed2Schema()) {
       const coreFeatures = this.schema.coreFeatures;
@@ -660,6 +684,15 @@ export class FederationMetadata {
     return this.schema.directive(this.federationDirectiveNameInSchema(name)) as DirectiveDefinition<TApplicationArgs> | undefined;
   }
 
+  private getPost20FederationDirective<TApplicationArgs extends {[key: string]: any}>(
+    name: FederationDirectiveName
+  ): Post20FederationDirectiveDefinition<TApplicationArgs> {
+    return this.getFederationDirective<TApplicationArgs>(name) ?? {
+      name,
+      applications: () => new Array<Directive<any, TApplicationArgs>>(),
+    };
+  }
+
   keyDirective(): DirectiveDefinition<{fields: any, resolvable?: boolean}> {
     return this.getLegacyFederationDirective(FederationDirectiveName.KEY);
   }
@@ -692,17 +725,18 @@ export class FederationMetadata {
     return this.getLegacyFederationDirective(FederationDirectiveName.TAG);
   }
 
-  composeDirective(): DirectiveDefinition<{name: string}> | FederationDirectiveNotDefinedInSchema<{name: string}> {
-    return this.getFederationDirective<{name: string}>(FederationDirectiveName.COMPOSE_DIRECTIVE) ?? {
-      name: FederationDirectiveName.COMPOSE_DIRECTIVE,
-      applications: () => new Array<Directive<any, {name: string}>>(),
-    };
+  composeDirective(): Post20FederationDirectiveDefinition<{name: string}> {
+    return this.getPost20FederationDirective(FederationDirectiveName.COMPOSE_DIRECTIVE);
   }
 
   inaccessibleDirective(): DirectiveDefinition<{}> {
     return this.getLegacyFederationDirective(FederationDirectiveName.INACCESSIBLE);
   }
 
+  interfaceObjectDirective(): Post20FederationDirectiveDefinition<{}> {
+    return this.getPost20FederationDirective(FederationDirectiveName.INTERFACE_OBJECT);
+  }
+
   allFederationDirectives(): DirectiveDefinition[] {
     const baseDirectives: DirectiveDefinition[] = [
       this.keyDirective(),
@@ -723,6 +757,11 @@ export class FederationMetadata {
     if (isFederationDirectiveDefinedInSchema(composeDirective)) {
       baseDirectives.push(composeDirective);
     }
+    const interfaceObjectDirective = this.interfaceObjectDirective();
+    if (isFederationDirectiveDefinedInSchema(interfaceObjectDirective)) {
+      baseDirectives.push(interfaceObjectDirective);
+    }
+
     return baseDirectives;
   }
 
@@ -762,12 +801,20 @@ export type FederationDirectiveNotDefinedInSchema<TApplicationArgs extends {[key
   applications: () => readonly Directive<any, TApplicationArgs>[],
 }
 
+export type Post20FederationDirectiveDefinition<TApplicationArgs extends {[key: string]: any}> =
+  DirectiveDefinition<TApplicationArgs>
+  | FederationDirectiveNotDefinedInSchema<TApplicationArgs>;
+
 export function isFederationDirectiveDefinedInSchema<TApplicationArgs extends {[key: string]: any}>(
-  definition: DirectiveDefinition<TApplicationArgs> | FederationDirectiveNotDefinedInSchema<TApplicationArgs>
+  definition: Post20FederationDirectiveDefinition<TApplicationArgs>
 ): definition is DirectiveDefinition<TApplicationArgs> {
   return definition instanceof DirectiveDefinition;
 }
 
+export function hasAppliedDirective(type: NamedType, definition: Post20FederationDirectiveDefinition<any>): boolean {
+  return isFederationDirectiveDefinedInSchema(definition) && type.hasAppliedDirective(definition);
+}
+
 export class FederationBlueprint extends SchemaBlueprint {
   constructor(private readonly withRootTypeRenaming: boolean) {
     super();
@@ -872,6 +919,7 @@ export class FederationBlueprint extends SchemaBlueprint {
       metadata,
       isOnParentType: true,
       allowOnNonExternalLeafFields: true,
+      allowOnInterface: metadata.federationFeature()!.url.version.compareTo(new FeatureVersion(2, 3)) >= 0,
       onFields: field => {
         const type = baseType(field.type!);
         if (isUnionType(type) || isInterfaceType(type)) {
@@ -925,6 +973,8 @@ export class FederationBlueprint extends SchemaBlueprint {
 
     validateNoExternalOnInterfaceFields(metadata, errorCollector);
     validateAllExternalFieldsUsed(metadata, errorCollector);
+    validateKeyOnInterfacesAreAlsoOnAllImplementations(metadata, errorCollector);
+    validateInterfaceObjectsAreOnEntities(metadata, errorCollector);
 
     // If tag is redefined by the user, make sure the definition is compatible with what we expect
     const tagDirective = metadata.tagDirective();
@@ -935,10 +985,6 @@ export class FederationBlueprint extends SchemaBlueprint {
       }
     }
 
-    for (const itf of schema.interfaceTypes()) {
-      validateInterfaceRuntimeImplementationFieldsTypes(itf, metadata, errorCollector);
-    }
-
     // While @shareable is "repeatable", this is only so one can use it on both a main
     // type definition _and_ possible other type extensions. But putting 2 @shareable
     // on the same type definition or field is both useless, and suggest some miscomprehension,
@@ -1070,15 +1116,22 @@ export function setSchemaAsFed2Subgraph(schema: Schema) {
 
 // This is the full @link declaration as added by `asFed2SubgraphDocument`. It's here primarily for uses by tests that print and match
 // subgraph schema to avoid having to update 20+ tests every time we use a new directive or the order of import changes ...
-export const FEDERATION2_LINK_WITH_FULL_IMPORTS = '@link(url: "https://specs.apollo.dev/federation/v2.2", import: ["@key", "@requires", "@provides", "@external", "@tag", "@extends", "@shareable", "@inaccessible", "@override", "@composeDirective"])';
-
-export function asFed2SubgraphDocument(document: DocumentNode): DocumentNode {
-  const fed2LinkExtension: SchemaExtensionNode = {
-    kind: Kind.SCHEMA_EXTENSION,
-    directives: [{
-      kind: Kind.DIRECTIVE,
-      name: { kind: Kind.NAME, value: linkDirectiveDefaultName },
-      arguments: [{
+export const FEDERATION2_LINK_WITH_FULL_IMPORTS = '@link(url: "https://specs.apollo.dev/federation/v2.3", import: ["@key", "@requires", "@provides", "@external", "@tag", "@extends", "@shareable", "@inaccessible", "@override", "@composeDirective", "@interfaceObject"])';
+
+/**
+ * Given a document that is assumed to _not_ be a fed2 schema (it does not have a `@link` to the federation spec),
+ * returns an equivalent document that `@link` to the last known federation spec.
+ *
+ * @param document - the document to "augment".
+ * @param options.addAsSchemaExtension - defines whethere the added `@link` is added as a schema extension (`extend schema`) or 
+ *   added to the schema definition. Defaults to `true` (added as an extension), as this mimics what we tends to write manually.
+ */
+export function asFed2SubgraphDocument(document: DocumentNode, options?: { addAsSchemaExtension: boolean }): DocumentNode {
+  const directiveToAdd: ConstDirectiveNode = ({
+    kind: Kind.DIRECTIVE,
+    name: { kind: Kind.NAME, value: linkDirectiveDefaultName },
+    arguments: [
+      {
         kind: Kind.ARGUMENT,
         name: { kind: Kind.NAME, value: 'url' },
         value: { kind: Kind.STRING, value: federationSpec.url.toString() }
@@ -1087,13 +1140,54 @@ export function asFed2SubgraphDocument(document: DocumentNode): DocumentNode {
         kind: Kind.ARGUMENT,
         name: { kind: Kind.NAME, value: 'import' },
         value: { kind: Kind.LIST, values: federationSpec.directiveSpecs().map((spec) => ({ kind: Kind.STRING, value: `@${spec.name}` })) }
-      }]
-    }]
-  };
-  return {
-    kind: Kind.DOCUMENT,
-    loc: document.loc,
-    definitions: document.definitions.concat(fed2LinkExtension)
+      }
+    ]
+  });
+  if (options?.addAsSchemaExtension ?? true) {
+    return {
+      kind: Kind.DOCUMENT,
+      loc: document.loc,
+      definitions: document.definitions.concat({
+        kind: Kind.SCHEMA_EXTENSION,
+        directives: [directiveToAdd]
+      }),
+    }
+  }
+
+  // We can't add a new schema definition if it already exists. If it doesn't we need to know if there is a mutation type or
+  // not.
+  const existingSchemaDefinition = document.definitions.find((d): d is SchemaDefinitionNode => d.kind == Kind.SCHEMA_DEFINITION);
+  if (existingSchemaDefinition) {
+    return {
+      kind: Kind.DOCUMENT,
+      loc: document.loc,
+      definitions: document.definitions.filter((d) => d !== existingSchemaDefinition).concat([{
+        ...existingSchemaDefinition,
+        directives: [directiveToAdd].concat(existingSchemaDefinition.directives ?? []),
+      }]),
+    }
+  } else {
+    const hasMutation = document.definitions.some((d) => d.kind === Kind.OBJECT_TYPE_DEFINITION && d.name.value === 'Mutation');
+    const makeOpType = (opType: OperationTypeNode, name: string): OperationTypeDefinitionNode => ({
+      kind: Kind.OPERATION_TYPE_DEFINITION,
+      operation: opType,
+      type: {
+        kind: Kind.NAMED_TYPE,
+        name: {
+          kind: Kind.NAME,
+          value: name,
+        }
+      },
+    });
+    return {
+      kind: Kind.DOCUMENT,
+      loc: document.loc,
+      definitions: document.definitions.concat({
+        kind: Kind.SCHEMA_DEFINITION,
+        directives: [directiveToAdd],
+        operationTypes: [ makeOpType(OperationTypeNode.QUERY, 'Query') ].concat(hasMutation ? makeOpType(OperationTypeNode.MUTATION, 'Mutation') : []),
+      }),
+    }
   }
 }
 
@@ -1123,13 +1217,21 @@ export function isFederationField(field: FieldDefinition<CompositeType>): boolea
 }
 
 export function isEntityType(type: NamedType): boolean {
-  if (type.kind !== "ObjectType") {
+  if (!isObjectType(type) && !isInterfaceType(type)) {
     return false;
   }
   const metadata = federationMetadata(type.schema());
   return !!metadata && type.hasAppliedDirective(metadata.keyDirective());
 }
 
+export function isInterfaceObjectType(type: NamedType): boolean {
+  if (!isObjectType(type)) {
+    return false;
+  }
+  const metadata = federationMetadata(type.schema());
+  return !!metadata && metadata.isInterfaceObjectType(type);
+}
+
 export function buildSubgraph(
   name: string,
   url: string,
@@ -1191,7 +1293,6 @@ function isFedSpecLinkDirective(directive: Directive<SchemaDefinition>): directi
 }
 
 function completeFed1SubgraphSchema(schema: Schema): GraphQLError[] {
-
   // We special case @key, @requires and @provides because we've seen existing user schema where those
   // have been defined in an invalid way, but in a way that fed1 wasn't rejecting. So for convenience,
   // if we detect one of those case, we just remove the definition and let the code afteward add the
@@ -1482,6 +1583,10 @@ export const serviceTypeSpec = createObjectTypeSpecification({
 export const entityTypeSpec = createUnionTypeSpecification({
   name: '_Entity',
   membersFct: (schema) => {
+    // Please note that `_Entity` cannot use "interface entities" since interface types cannot be in unions.
+    // It is ok in practice because _Entity is only use as return type for `_entities`, and even when interfaces
+    // are involve, the result of an `_entities` call will always be an object type anyway, and since we force
+    // all implementations of an interface entity to be entity themselves in a subgraph, we're fine.
     return schema.objectTypes().filter(isEntityType).map((t) => t.name);
   },
 });

package/src/federationSpec.ts

@@ -35,6 +35,7 @@ export enum FederationDirectiveName {
   TAG = 'tag',
   INACCESSIBLE = 'inaccessible',
   COMPOSE_DIRECTIVE = 'composeDirective',
+  INTERFACE_OBJECT = 'interfaceObject',
 }
 
 const fieldSetTypeSpec = createScalarTypeSpecification({ name: FederationTypeName.FIELD_SET });
@@ -93,7 +94,9 @@ const legacyFederationDirectives = [
   requiresDirectiveSpec,
   providesDirectiveSpec,
   externalDirectiveSpec,
-  TAG_VERSIONS.latest().tagDirectiveSpec,
+  // This should really be v0.1 instead of v0.2, but we can't change this to
+  // v0.1 without checking whether anyone relied on the v0.2 behavior.
+  TAG_VERSIONS.find(new FeatureVersion(0, 2))!.tagDirectiveSpec,
   extendsDirectiveSpec,
 ];
 
@@ -153,6 +156,16 @@ export class FederationSpecDefinition extends FeatureDefinition {
         }),
       }));
     }
+
+    if (version >= (new FeatureVersion(2, 3))) {
+      this.registerDirective(createDirectiveSpecification({
+        name: FederationDirectiveName.INTERFACE_OBJECT,
+        locations: [DirectiveLocation.OBJECT],
+      }));
+      this.registerDirective(
+        TAG_VERSIONS.find(new FeatureVersion(0, 3))!.tagDirectiveSpec
+      );
+    }
   }
 
   private registerDirective(spec: DirectiveSpecification) {
@@ -195,6 +208,7 @@ export class FederationSpecDefinition extends FeatureDefinition {
 export const FEDERATION_VERSIONS = new FeatureDefinitions<FederationSpecDefinition>(federationIdentity)
   .add(new FederationSpecDefinition(new FeatureVersion(2, 0)))
   .add(new FederationSpecDefinition(new FeatureVersion(2, 1)))
-  .add(new FederationSpecDefinition(new FeatureVersion(2, 2)));
+  .add(new FederationSpecDefinition(new FeatureVersion(2, 2)))
+  .add(new FederationSpecDefinition(new FeatureVersion(2, 3)));
 
 registerKnownFeature(FEDERATION_VERSIONS);

package/src/joinSpec.ts

@@ -64,11 +64,21 @@ export class JoinSpecDefinition extends FeatureDefinition {
     if (!this.isV01()) {
       joinType.addArgument('extension', new NonNullType(schema.booleanType()), false);
       joinType.addArgument('resolvable', new NonNullType(schema.booleanType()), true);
+
+      if (this.version >= (new FeatureVersion(0, 3))) {
+        joinType.addArgument('isInterfaceObject', new NonNullType(schema.booleanType()), false);
+      }
     }
 
     const joinField = this.addDirective(schema, 'field').addLocations(DirectiveLocation.FIELD_DEFINITION, DirectiveLocation.INPUT_FIELD_DEFINITION);
     joinField.repeatable = true;
-    joinField.addArgument('graph', new NonNullType(graphEnum));
+    // The `graph` argument used to be non-nullable, but @interfaceObject makes us add some field in
+    // the supergraph that don't "directly" come from any subgraph (they indirectly are inherited from
+    // an `@interfaceObject` type), and to indicate that, we use a `@join__field(graph: null)` annotation.
+    const graphArgType = this.version >= (new FeatureVersion(0, 3))
+      ? graphEnum
+      : new NonNullType(graphEnum);
+    joinField.addArgument('graph', graphArgType);
     joinField.addArgument('requires', joinFieldSet);
     joinField.addArgument('provides', joinFieldSet);
     if (!this.isV01()) {
@@ -87,6 +97,17 @@ export class JoinSpecDefinition extends FeatureDefinition {
       joinImplements.addArgument('interface', new NonNullType(schema.stringType()));
     }
 
+    if (this.version >= (new FeatureVersion(0, 3))) {
+      const joinUnionMember = this.addDirective(schema, 'unionMember').addLocations(DirectiveLocation.UNION);
+      joinUnionMember.repeatable = true;
+      joinUnionMember.addArgument('graph', new NonNullType(graphEnum));
+      joinUnionMember.addArgument('member', new NonNullType(schema.stringType()));
+
+      const joinEnumValue = this.addDirective(schema, 'enumValue').addLocations(DirectiveLocation.ENUM_VALUE);
+      joinEnumValue.repeatable = true;
+      joinEnumValue.addArgument('graph', new NonNullType(graphEnum));
+    }
+
     if (this.isV01()) {
       const joinOwner = this.addDirective(schema, 'owner').addLocations(DirectiveLocation.OBJECT);
       joinOwner.addArgument('graph', new NonNullType(graphEnum));
@@ -153,7 +174,7 @@ export class JoinSpecDefinition extends FeatureDefinition {
     return this.directive(schema, 'graph')!;
   }
 
-  typeDirective(schema: Schema): DirectiveDefinition<{graph: string, key?: string, extension?: boolean, resolvable?: boolean}> {
+  typeDirective(schema: Schema): DirectiveDefinition<{graph: string, key?: string, extension?: boolean, resolvable?: boolean, isInterfaceObject?: boolean}> {
     return this.directive(schema, 'type')!;
   }
 
@@ -162,7 +183,7 @@ export class JoinSpecDefinition extends FeatureDefinition {
   }
 
   fieldDirective(schema: Schema): DirectiveDefinition<{
-    graph: string,
+    graph?: string,
     requires?: string,
     provides?: string,
     override?: string,
@@ -173,6 +194,14 @@ export class JoinSpecDefinition extends FeatureDefinition {
     return this.directive(schema, 'field')!;
   }
 
+  unionMemberDirective(schema: Schema): DirectiveDefinition<{graph: string, member: string}> | undefined {
+    return this.directive(schema, 'unionMember');
+  }
+
+  enumValueDirective(schema: Schema): DirectiveDefinition<{graph: string}> | undefined {
+    return this.directive(schema, 'enumValue');
+  }
+
   ownerDirective(schema: Schema): DirectiveDefinition<{graph: string}> | undefined {
     return this.directive(schema, 'owner');
   }
@@ -182,15 +211,15 @@ export class JoinSpecDefinition extends FeatureDefinition {
   }
 }
 
-// Note: This declare a no-yet-agreed-upon join spec v0.2, that:
-//   1. allows a repeatable join__field (join-spec#15).
-//   2. allows the 'key' argument of join__type to be optional (join-spec#13)
-//   3. relax conditions on join__type in general so as to not relate to the notion of owner (join-spec#16).
-//   4. has join__implements (join-spec#13)
-// The changes from join-spec#17 and join-spec#18 are not yet implemented, but probably should be or we may have bugs
-// due to the query planner having an invalid understanding of the subgraph services API.
+// The versions are as follows:
+//  - 0.1: this is the version used by federation 1 composition. Federation 2 is still able to read supergraphs
+//    using that verison for backward compatibility, but never writes this spec version is not expressive enough
+//    for federation 2 in general.
+//  - 0.2: this is the original version released with federation 2.
+//  - 0.3: adds the `isInterfaceObject` argument to `@join__type`, and make the `graph` in `@join__field` skippable.
 export const JOIN_VERSIONS = new FeatureDefinitions<JoinSpecDefinition>(joinIdentity)
   .add(new JoinSpecDefinition(new FeatureVersion(0, 1)))
-  .add(new JoinSpecDefinition(new FeatureVersion(0, 2)));
+  .add(new JoinSpecDefinition(new FeatureVersion(0, 2)))
+  .add(new JoinSpecDefinition(new FeatureVersion(0, 3)));
 
 registerKnownFeature(JOIN_VERSIONS);

package/src/operations.ts

@@ -134,6 +134,15 @@ export class Field<TArgs extends {[key: string]: any} = {[key: string]: any}> ex
     return newField;
   }
 
+  withUpdatedAlias(newAlias: string | undefined): Field<TArgs> {
+    const newField = new Field<TArgs>(this.definition, this.args, this.variableDefinitions, newAlias);
+    for (const directive of this.appliedDirectives) {
+      newField.applyDirective(directive.definition!, directive.arguments());
+    }
+    this.copyAttachementsTo(newField);
+    return newField;
+  }
+
   appliesTo(type: ObjectType | InterfaceType): boolean {
     const definition = type.field(this.name);
     return !!definition && this.selects(definition);
@@ -141,7 +150,7 @@ export class Field<TArgs extends {[key: string]: any} = {[key: string]: any}> ex
 
   selects(definition: FieldDefinition<any>, assumeValid: boolean = false): boolean {
     // We've already validated that the field selects the definition on which it was built.
-    if (definition == this.definition) {
+    if (definition === this.definition) {
       return true;
     }
 
@@ -317,6 +326,10 @@ export class FragmentElement extends AbstractOperationElement<FragmentElement> {
     return this.withUpdatedTypes(newSourceType, this.typeCondition);
   }
 
+  withUpdatedCondition(newCondition: CompositeType | undefined): FragmentElement {
+    return this.withUpdatedTypes(this.sourceType, newCondition);
+  }
+
   withUpdatedTypes(newSourceType: CompositeType, newCondition: CompositeType | undefined): FragmentElement {
     // Note that we pass the type-condition name instead of the type itself, to ensure that if `newSourceType` was from a different
     // schema (typically, the supergraph) than `this.sourceType` (typically, a subgraph), then the new condition uses the
@@ -1019,6 +1032,22 @@ export class SelectionSet extends Freezable<SelectionSet> {
     return this._cachedSelections;
   }
 
+  fieldsInSet(): { path: string[], field: FieldSelection, directParent: SelectionSet }[] {
+    const fields = new Array<{ path: string[], field: FieldSelection, directParent: SelectionSet }>();
+    for (const selection of this.selections()) {
+      if (selection.kind === 'FieldSelection') {
+        fields.push({ path: [], field: selection, directParent: this });
+      } else {
+        const condition = selection.element().typeCondition;
+        const header = condition ? [`... on ${condition}`] : [];
+        for (const { path, field, directParent } of selection.selectionSet.fieldsInSet()) {
+          fields.push({ path: header.concat(path), field, directParent });
+        }
+      }
+    }
+    return fields;
+  }
+
   usedVariables(): Variables {
     let variables: Variables = [];
     for (const byResponseName of this._selections.values()) {
@@ -1200,6 +1229,23 @@ export class SelectionSet extends Freezable<SelectionSet> {
     return toAdd;
   }
 
+  /**
+   * If this selection contains a selection of a field with provided response name at top level, removes it.
+   *
+   * @return whether a selection was removed.
+   */
+  removeTopLevelField(responseName: string): boolean {
+    // It's a bug to try to remove from a frozen selection set
+    assert(!this.isFrozen(), () => `Cannot remove from frozen selection: ${this}`);
+
+    const wasRemoved = this._selections.delete(responseName);
+    if (wasRemoved) {
+      --this._selectionCount;
+      this._cachedSelections = undefined;
+    }
+    return wasRemoved;
+  }
+
   addPath(path: OperationPath, onPathEnd?: (finalSelectionSet: SelectionSet | undefined) => void) {
     let previousSelections: SelectionSet = this;
     let currentSelections: SelectionSet | undefined = this;
@@ -1524,6 +1570,10 @@ export class FieldSelection extends Freezable<FieldSelection> {
     this.selectionSet = isLeafType(type) ? undefined : (initialSelectionSet ? initialSelectionSet.cloneIfFrozen() : new SelectionSet(type as CompositeType));
   }
 
+  get parentType(): CompositeType {
+    return this.field.parentType;
+  }
+
   protected us(): FieldSelection {
     return this;
   }
@@ -1732,6 +1782,10 @@ export class FieldSelection extends Freezable<FieldSelection> {
     return new FieldSelection(this.field, newSubSelection);
   }
 
+  withUpdatedField(newField: Field<any>): FieldSelection {
+    return new FieldSelection(newField, this.selectionSet);
+  }
+
   equals(that: Selection): boolean {
     if (this === that) {
       return true;
@@ -1821,6 +1875,10 @@ export abstract class FragmentSelection extends Freezable<FragmentSelection> {
 
   abstract withUpdatedSubSelection(newSubSelection: SelectionSet | undefined): FragmentSelection;
 
+  get parentType(): CompositeType {
+    return this.element().parentType;
+  }
+
   protected us(): FragmentSelection {
     return this;
   }