@apollo/federation-internals 2.7.7 → 2.8.0-connectors.0

package/src/error.ts

@@ -561,119 +561,6 @@ const INTERFACE_KEY_MISSING_IMPLEMENTATION_TYPE = makeCodeDefinition(
   { addedIn: '2.3.0' },
 )
 
-const SOURCE_FEDERATION_VERSION_REQUIRED = makeCodeDefinition(
-  'SOURCE_FEDERATION_VERSION_REQUIRED',
-  'Schemas using `@source{API,Type,Field}` directives must @link-import v2.7 or later of federation',
-  { addedIn: '2.7.1' },
-);
-
-const SOURCE_API_NAME_INVALID = makeCodeDefinition(
-  'SOURCE_API_NAME_INVALID',
-  'Each `@sourceAPI` directive must take a unique and valid name as an argument',
-  { addedIn: '2.7.0' },
-);
-
-const SOURCE_API_PROTOCOL_INVALID = makeCodeDefinition(
-  'SOURCE_API_PROTOCOL_INVALID',
-  'Each `@sourceAPI` directive must specify exactly one of the known protocols',
-  { addedIn: '2.7.0' },
-);
-
-const SOURCE_API_HTTP_BASE_URL_INVALID = makeCodeDefinition(
-  'SOURCE_API_HTTP_BASE_URL_INVALID',
-  'The `@sourceAPI` directive must specify a valid http.baseURL',
-  { addedIn: '2.7.0' },
-);
-
-const SOURCE_HTTP_HEADERS_INVALID = makeCodeDefinition(
-  'SOURCE_HTTP_HEADERS_INVALID',
-  'The `http.headers` argument of `@source*` directives must specify valid HTTP headers',
-  { addedIn: '2.7.0' },
-);
-
-const SOURCE_TYPE_API_ERROR = makeCodeDefinition(
-  'SOURCE_TYPE_API_ERROR',
-  'The `api` argument of the `@sourceType` directive must match a valid `@sourceAPI` name',
-  { addedIn: '2.7.0' },
-);
-
-const SOURCE_TYPE_PROTOCOL_INVALID = makeCodeDefinition(
-  'SOURCE_TYPE_PROTOCOL_INVALID',
-  'The `@sourceType` directive must specify the same protocol as its corresponding `@sourceAPI`',
-  { addedIn: '2.7.0' },
-);
-
-const SOURCE_TYPE_HTTP_METHOD_INVALID = makeCodeDefinition(
-  'SOURCE_TYPE_HTTP_METHOD_INVALID',
-  'The `@sourceType` directive must specify exactly one of `http.GET` or `http.POST`',
-  { addedIn: '2.7.0' },
-);
-
-const SOURCE_TYPE_HTTP_PATH_INVALID = makeCodeDefinition(
-  'SOURCE_TYPE_HTTP_PATH_INVALID',
-  'The `@sourceType` directive must specify a valid URL template for `http.GET` or `http.POST`',
-  { addedIn: '2.7.0' },
-);
-
-const SOURCE_TYPE_HTTP_BODY_INVALID = makeCodeDefinition(
-  'SOURCE_TYPE_HTTP_BODY_INVALID',
-  'If the `@sourceType` specifies `http.body`, it must be a valid `JSONSelection`',
-  { addedIn: '2.7.0' },
-);
-
-const SOURCE_TYPE_ON_NON_OBJECT_OR_NON_ENTITY = makeCodeDefinition(
-  'SOURCE_TYPE_ON_NON_OBJECT_OR_NON_ENTITY',
-  'The `@sourceType` directive must be applied to an object or interface type that also has `@key`',
-  { addedIn: '2.7.0' },
-);
-
-const SOURCE_TYPE_SELECTION_INVALID = makeCodeDefinition(
-  'SOURCE_TYPE_SELECTION_INVALID',
-  'The `selection` argument of the `@sourceType` directive must be a valid `JSONSelection` that outputs fields of the GraphQL type',
-);
-
-const SOURCE_FIELD_API_ERROR = makeCodeDefinition(
-  'SOURCE_FIELD_API_ERROR',
-  'The `api` argument of the `@sourceField` directive must match a valid `@sourceAPI` name',
-  { addedIn: '2.7.0' },
-);
-
-const SOURCE_FIELD_PROTOCOL_INVALID = makeCodeDefinition(
-  'SOURCE_FIELD_PROTOCOL_INVALID',
-  'If `@sourceField` specifies a protocol, it must match the corresponding `@sourceAPI` protocol',
-  { addedIn: '2.7.0' },
-);
-
-const SOURCE_FIELD_HTTP_METHOD_INVALID = makeCodeDefinition(
-  'SOURCE_FIELD_HTTP_METHOD_INVALID',
-  'The `@sourceField` directive must specify at most one of `http.{GET,POST,PUT,PATCH,DELETE}`',
-  { addedIn: '2.7.0' },
-);
-
-const SOURCE_FIELD_HTTP_PATH_INVALID = makeCodeDefinition(
-  'SOURCE_FIELD_HTTP_PATH_INVALID',
-  'The `@sourceField` directive must specify a valid URL template for `http.{GET,POST,PUT,PATCH,DELETE}`',
-  { addedIn: '2.7.0' },
-);
-
-const SOURCE_FIELD_HTTP_BODY_INVALID = makeCodeDefinition(
-  'SOURCE_FIELD_HTTP_BODY_INVALID',
-  'If `@sourceField` specifies http.body, it must be a valid `JSONSelection` matching available arguments and fields',
-  { addedIn: '2.7.0' },
-);
-
-const SOURCE_FIELD_SELECTION_INVALID = makeCodeDefinition(
-  'SOURCE_FIELD_SELECTION_INVALID',
-  'The `selection` argument of the `@sourceField` directive must be a valid `JSONSelection` that outputs fields of the GraphQL type',
-  { addedIn: '2.7.0' },
-);
-
-const SOURCE_FIELD_NOT_ON_ROOT_OR_ENTITY_FIELD = makeCodeDefinition(
-  'SOURCE_FIELD_NOT_ON_ROOT_OR_ENTITY_FIELD',
-  'The `@sourceField` directive must be applied to a field of the `Query` or `Mutation` types, or of an entity type',
-  { addedIn: '2.7.0' },
-);
-
 export const ERROR_CATEGORIES = {
   DIRECTIVE_FIELDS_MISSING_EXTERNAL,
   DIRECTIVE_UNSUPPORTED_ON_INTERFACE,
@@ -762,26 +649,6 @@ export const ERRORS = {
   INTERFACE_OBJECT_USAGE_ERROR,
   INTERFACE_KEY_NOT_ON_IMPLEMENTATION,
   INTERFACE_KEY_MISSING_IMPLEMENTATION_TYPE,
-  // Errors related to @sourceAPI, @sourceType, and/or @sourceField
-  SOURCE_FEDERATION_VERSION_REQUIRED,
-  SOURCE_API_NAME_INVALID,
-  SOURCE_API_PROTOCOL_INVALID,
-  SOURCE_API_HTTP_BASE_URL_INVALID,
-  SOURCE_HTTP_HEADERS_INVALID,
-  SOURCE_TYPE_API_ERROR,
-  SOURCE_TYPE_PROTOCOL_INVALID,
-  SOURCE_TYPE_HTTP_METHOD_INVALID,
-  SOURCE_TYPE_HTTP_PATH_INVALID,
-  SOURCE_TYPE_HTTP_BODY_INVALID,
-  SOURCE_TYPE_ON_NON_OBJECT_OR_NON_ENTITY,
-  SOURCE_TYPE_SELECTION_INVALID,
-  SOURCE_FIELD_API_ERROR,
-  SOURCE_FIELD_PROTOCOL_INVALID,
-  SOURCE_FIELD_HTTP_METHOD_INVALID,
-  SOURCE_FIELD_HTTP_PATH_INVALID,
-  SOURCE_FIELD_HTTP_BODY_INVALID,
-  SOURCE_FIELD_SELECTION_INVALID,
-  SOURCE_FIELD_NOT_ON_ROOT_OR_ENTITY_FIELD,
 };
 
 const codeDefByCode = Object.values(ERRORS).reduce((obj: {[code: string]: ErrorCodeDefinition}, codeDef: ErrorCodeDefinition) => { obj[codeDef.code] = codeDef; return obj; }, {});

package/src/federation.ts

@@ -39,15 +39,14 @@ import {
   KnownTypeNamesRule,
   PossibleTypeExtensionsRule,
   print as printAST,
-  Source,
   GraphQLErrorOptions,
   SchemaDefinitionNode,
   OperationTypeNode,
   OperationTypeDefinitionNode,
-  ConstDirectiveNode,
+  ConstDirectiveNode, parse,
 } from "graphql";
 import { KnownTypeNamesInFederationRule } from "./validation/KnownTypeNamesInFederationRule";
-import { buildSchema, buildSchemaFromAST } from "./buildSchema";
+import { buildSchemaFromAST } from "./buildSchema";
 import { parseSelectionSet, SelectionSet } from './operations';
 import { TAG_VERSIONS } from "./specs/tagSpec";
 import {
@@ -84,13 +83,9 @@ import {
 import { defaultPrintOptions, PrintOptions as PrintOptions, printSchema } from "./print";
 import { createObjectTypeSpecification, createScalarTypeSpecification, createUnionTypeSpecification } from "./directiveAndTypeSpecification";
 import { didYouMean, suggestionList } from "./suggestions";
-import { coreFeatureDefinitionIfKnown, validateKnownFeatures } from "./knownCoreFeatures";
+import { coreFeatureDefinitionIfKnown } from "./knownCoreFeatures";
 import { joinIdentity } from "./specs/joinSpec";
-import {
-  SourceAPIDirectiveArgs,
-  SourceFieldDirectiveArgs,
-  SourceTypeDirectiveArgs,
-} from "./specs/sourceSpec";
+import {validateSubgraphSchema} from "./wasm";
 
 const linkSpec = LINK_VERSIONS.latest();
 const tagSpec = TAG_VERSIONS.latest();
@@ -787,18 +782,6 @@ export class FederationMetadata {
     return this.getPost20FederationDirective(FederationDirectiveName.POLICY);
   }
 
-  sourceAPIDirective(): Post20FederationDirectiveDefinition<SourceAPIDirectiveArgs> {
-    return this.getPost20FederationDirective(FederationDirectiveName.SOURCE_API);
-  }
-
-  sourceTypeDirective(): Post20FederationDirectiveDefinition<SourceTypeDirectiveArgs> {
-    return this.getPost20FederationDirective(FederationDirectiveName.SOURCE_TYPE);
-  }
-
-  sourceFieldDirective(): Post20FederationDirectiveDefinition<SourceFieldDirectiveArgs> {
-    return this.getPost20FederationDirective(FederationDirectiveName.SOURCE_FIELD);
-  }
-
   allFederationDirectives(): DirectiveDefinition[] {
     const baseDirectives: DirectiveDefinition[] = [
       this.keyDirective(),
@@ -839,19 +822,6 @@ export class FederationMetadata {
       baseDirectives.push(policyDirective);
     }
 
-    const sourceAPIDirective = this.sourceAPIDirective();
-    if (isFederationDirectiveDefinedInSchema(sourceAPIDirective)) {
-      baseDirectives.push(sourceAPIDirective);
-    }
-    const sourceTypeDirective = this.sourceTypeDirective();
-    if (isFederationDirectiveDefinedInSchema(sourceTypeDirective)) {
-      baseDirectives.push(sourceTypeDirective);
-    }
-    const sourceFieldDirective = this.sourceFieldDirective();
-    if (isFederationDirectiveDefinedInSchema(sourceFieldDirective)) {
-      baseDirectives.push(sourceFieldDirective);
-    }
-
     return baseDirectives;
   }
 
@@ -1089,11 +1059,6 @@ export class FederationBlueprint extends SchemaBlueprint {
     validateKeyOnInterfacesAreAlsoOnAllImplementations(metadata, errorCollector);
     validateInterfaceObjectsAreOnEntities(metadata, errorCollector);
 
-    // FeatureDefinition objects passed to registerKnownFeature can register
-    // validation functions for subgraph schemas by overriding the
-    // validateSubgraphSchema method.
-    validateKnownFeatures(schema, errorCollector);
-
     // If tag is redefined by the user, make sure the definition is compatible with what we expect
     const tagDirective = metadata.tagDirective();
     if (tagDirective) {
@@ -1239,7 +1204,7 @@ export function setSchemaAsFed2Subgraph(schema: Schema, useLatest: boolean = fal
 
 // 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.7", import: ["@key", "@requires", "@provides", "@external", "@tag", "@extends", "@shareable", "@inaccessible", "@override", "@composeDirective", "@interfaceObject", "@authenticated", "@requiresScopes", "@policy", "@sourceAPI", "@sourceType", "@sourceField"])';
+export const FEDERATION2_LINK_WITH_FULL_IMPORTS = '@link(url: "https://specs.apollo.dev/federation/v2.7", import: ["@key", "@requires", "@provides", "@external", "@tag", "@extends", "@shareable", "@inaccessible", "@override", "@composeDirective", "@interfaceObject", "@authenticated", "@requiresScopes", "@policy"])';
 
 // This is the federation @link for tests that go through the asFed2SubgraphDocument function.
 export const FEDERATION2_LINK_WITH_AUTO_EXPANDED_IMPORTS = '@link(url: "https://specs.apollo.dev/federation/v2.7", import: ["@key", "@requires", "@provides", "@external", "@tag", "@extends", "@shareable", "@inaccessible", "@override", "@composeDirective", "@interfaceObject"])';
@@ -1376,9 +1341,9 @@ export function buildSubgraph(
   };
   let subgraph: Subgraph;
   try {
-    const schema = typeof source === 'string'
-      ? buildSchema(new Source(source, name), buildOptions)
-      : buildSchemaFromAST(source, buildOptions)
+    const parsed = typeof source === 'string' ? parse(source) : source;
+    const schema =  buildSchemaFromAST(parsed, buildOptions)
+    validateSubgraphSchema(schema, parsed)
     subgraph = new Subgraph(name, url, schema);
   } catch (e) {
     if (e instanceof GraphQLError && name !== FEDERATION_UNNAMED_SUBGRAPH_NAME) {

package/src/index.ts

@@ -23,4 +23,4 @@ export * from './argumentCompositionStrategies';
 export * from './specs/authenticatedSpec';
 export * from './specs/requiresScopesSpec';
 export * from './specs/policySpec';
-export * from './specs/sourceSpec';
+export * from './specs/connectSpec';

package/src/knownCoreFeatures.ts

@@ -1,5 +1,3 @@
-import { GraphQLError } from "graphql";
-import { Schema } from "./definitions";
 import { FeatureDefinition, FeatureDefinitions, FeatureUrl } from "./specs/coreSpec";
 
 const registeredFeatures = new Map<string, FeatureDefinitions>();
@@ -14,19 +12,6 @@ export function coreFeatureDefinitionIfKnown(url: FeatureUrl): FeatureDefinition
   return registeredFeatures.get(url.identity)?.find(url.version);
 }
 
-export function validateKnownFeatures(
-  schema: Schema,
-  errorCollector: GraphQLError[] = [],
-): GraphQLError[] {
-  registeredFeatures.forEach(definitions => {
-    const feature = definitions.latest();
-    if (feature.validateSubgraphSchema !== FeatureDefinition.prototype.validateSubgraphSchema) {
-      errorCollector.push(...feature.validateSubgraphSchema(schema));
-    }
-  });
-  return errorCollector;
-}
-
 /**
  * Removes a feature from the set of known features.
  *

package/src/specs/connectSpec.ts

@@ -0,0 +1,181 @@
+import {DirectiveLocation, GraphQLError} from 'graphql';
+import { CorePurpose, FeatureDefinition, FeatureDefinitions, FeatureUrl, FeatureVersion } from "./coreSpec";
+import {
+  Schema,
+  NonNullType,
+  InputObjectType,
+  InputFieldDefinition,
+  ListType,
+} from '../definitions';
+import { registerKnownFeature } from '../knownCoreFeatures';
+import { createDirectiveSpecification } from '../directiveAndTypeSpecification';
+
+export const connectIdentity = 'https://specs.apollo.dev/connect';
+
+const CONNECT = "connect";
+const SOURCE = "source";
+const URL_PATH_TEMPLATE = "URLPathTemplate";
+const JSON_SELECTION = "JSONSelection";
+const CONNECT_HTTP = "ConnectHTTP";
+const SOURCE_HTTP = "SourceHTTP";
+const HTTP_HEADER_MAPPING = "HTTPHeaderMapping";
+
+export class ConnectSpecDefinition extends FeatureDefinition {
+  constructor(version: FeatureVersion, readonly minimumFederationVersion: FeatureVersion) {
+    super(new FeatureUrl(connectIdentity, CONNECT, version), minimumFederationVersion);
+
+    this.registerDirective(createDirectiveSpecification({
+      name: CONNECT,
+      locations: [DirectiveLocation.FIELD_DEFINITION],
+      repeatable: true,
+      // We "compose" these directives using the  `@join__directive` mechanism,
+      // so they do not need to be composed in the way passing `composes: true`
+      // here implies.
+      composes: false,
+    }));
+
+    this.registerDirective(createDirectiveSpecification({
+      name: SOURCE,
+      locations: [DirectiveLocation.SCHEMA],
+      repeatable: true,
+      composes: false,
+    }));
+
+    this.registerType({ name: URL_PATH_TEMPLATE, checkOrAdd: () => [] });
+    this.registerType({ name: JSON_SELECTION, checkOrAdd: () => [] });
+    this.registerType({ name: CONNECT_HTTP, checkOrAdd: () => [] });
+    this.registerType({ name: SOURCE_HTTP, checkOrAdd: () => [] });
+    this.registerType({ name: HTTP_HEADER_MAPPING, checkOrAdd: () => [] });
+  }
+
+  addElementsToSchema(schema: Schema): GraphQLError[] {
+    /* scalar URLPathTemplate */
+    const URLPathTemplate = this.addScalarType(schema, URL_PATH_TEMPLATE);
+
+    /* scalar JSONSelection */
+    const JSONSelection = this.addScalarType(schema, JSON_SELECTION);
+
+    /*
+      directive @connect(
+        source: String
+        http: ConnectHTTP
+        selection: JSONSelection!
+      ) repeatable on FIELD_DEFINITION
+    */
+    const connect = this.addDirective(schema, CONNECT).addLocations(DirectiveLocation.FIELD_DEFINITION);
+    connect.repeatable = true;
+
+    connect.addArgument(SOURCE, schema.stringType());
+
+    /*
+      input HTTPHeaderMapping {
+        name: String!
+        as: String
+        value: String
+      }
+    */
+    const HTTPHeaderMapping = schema.addType(new InputObjectType(this.typeNameInSchema(schema, HTTP_HEADER_MAPPING)!));
+    HTTPHeaderMapping.addField(new InputFieldDefinition('name')).type =
+      new NonNullType(schema.stringType());
+    HTTPHeaderMapping.addField(new InputFieldDefinition('as')).type =
+      schema.stringType();
+    HTTPHeaderMapping.addField(new InputFieldDefinition('value')).type =
+      schema.stringType();
+
+    /*
+      input ConnectHTTP {
+        GET: URLPathTemplate
+        POST: URLPathTemplate
+        PUT: URLPathTemplate
+        PATCH: URLPathTemplate
+        DELETE: URLPathTemplate
+        body: JSONSelection
+        headers: [HTTPHeaderMapping!]
+      }
+    */
+    const ConnectHTTP = schema.addType(new InputObjectType(this.typeNameInSchema(schema, CONNECT_HTTP)!));
+    ConnectHTTP.addField(new InputFieldDefinition('GET')).type = URLPathTemplate;
+    ConnectHTTP.addField(new InputFieldDefinition('POST')).type = URLPathTemplate;
+    ConnectHTTP.addField(new InputFieldDefinition('PUT')).type = URLPathTemplate;
+    ConnectHTTP.addField(new InputFieldDefinition('PATCH')).type = URLPathTemplate;
+    ConnectHTTP.addField(new InputFieldDefinition('DELETE')).type = URLPathTemplate;
+    ConnectHTTP.addField(new InputFieldDefinition('body')).type = JSONSelection;
+    ConnectHTTP.addField(new InputFieldDefinition('headers')).type =
+      new ListType(new NonNullType(HTTPHeaderMapping));
+    connect.addArgument('http', new NonNullType(ConnectHTTP));
+
+    connect.addArgument('selection', JSONSelection);
+
+    /*
+      directive @source(
+        name: String!
+        http: ConnectHTTP
+      ) repeatable on SCHEMA
+    */
+    const source = this.addDirective(schema, SOURCE).addLocations(
+      DirectiveLocation.SCHEMA,
+    );
+    source.repeatable = true;
+    source.addArgument('name', new NonNullType(schema.stringType()));
+
+    /*
+      input SourceHTTP {
+        baseURL: String!
+        headers: [HTTPHeaderMapping!]
+      }
+    */
+    const SourceHTTP = schema.addType(new InputObjectType(this.typeNameInSchema(schema, SOURCE_HTTP)!));
+    SourceHTTP.addField(new InputFieldDefinition('baseURL')).type =
+      new NonNullType(schema.stringType());
+    SourceHTTP.addField(new InputFieldDefinition('headers')).type =
+      new ListType(new NonNullType(HTTPHeaderMapping));
+
+    source.addArgument('http', new NonNullType(SourceHTTP));
+
+    return [];
+  }
+
+  get defaultCorePurpose(): CorePurpose {
+    return 'EXECUTION';
+  }
+}
+
+export type SourceDirectiveArgs = {
+  name: string;
+  http: SourceDirectiveHTTP;
+};
+
+export type SourceDirectiveHTTP = {
+  baseURL: string;
+  headers?: HTTPHeaderMapping[];
+};
+
+type HTTPHeaderMapping = {
+  name: string;
+  as?: string;
+  value?: string;
+};
+
+type URLPathTemplate = string;
+type JSONSelection = string;
+
+export type ConnectDirectiveArgs = {
+  source: string;
+  http: ConnectDirectiveHTTP;
+  selection?: JSONSelection;
+};
+
+export type ConnectDirectiveHTTP = {
+  GET?: URLPathTemplate;
+  POST?: URLPathTemplate;
+  PUT?: URLPathTemplate;
+  PATCH?: URLPathTemplate;
+  DELETE?: URLPathTemplate;
+  body?: JSONSelection;
+  headers?: HTTPHeaderMapping[];
+};
+
+export const CONNECT_VERSIONS = new FeatureDefinitions<ConnectSpecDefinition>(connectIdentity)
+  .add(new ConnectSpecDefinition(new FeatureVersion(0, 1), new FeatureVersion(2, 8)));
+
+registerKnownFeature(CONNECT_VERSIONS);

package/src/specs/coreSpec.ts

@@ -117,11 +117,6 @@ export abstract class FeatureDefinition {
       .concat(this.typeSpecs().map((spec) => spec.name));
   }
 
-  // No-op implementation that can be overridden by subclasses.
-  validateSubgraphSchema(_schema: Schema): GraphQLError[] {
-    return [];
-  }
-
   protected nameInSchema(schema: Schema): string | undefined {
     const feature = this.featureInSchema(schema);
     return feature?.nameInSchema;

package/src/specs/federationSpec.ts

@@ -18,7 +18,6 @@ import { INACCESSIBLE_VERSIONS } from "./inaccessibleSpec";
 import { AUTHENTICATED_VERSIONS } from "./authenticatedSpec";
 import { REQUIRES_SCOPES_VERSIONS } from "./requiresScopesSpec";
 import { POLICY_VERSIONS } from './policySpec';
-import { SOURCE_VERSIONS } from './sourceSpec';
 
 export const federationIdentity = 'https://specs.apollo.dev/federation';
 
@@ -41,9 +40,6 @@ export enum FederationDirectiveName {
   AUTHENTICATED = 'authenticated',
   REQUIRES_SCOPES = 'requiresScopes',
   POLICY = 'policy',
-  SOURCE_API = 'sourceAPI',
-  SOURCE_TYPE = 'sourceType',
-  SOURCE_FIELD = 'sourceField',
 }
 
 const fieldSetTypeSpec = createScalarTypeSpecification({ name: FederationTypeName.FIELD_SET });
@@ -170,10 +166,6 @@ export class FederationSpecDefinition extends FeatureDefinition {
     if (version.gte(new FeatureVersion(2, 6))) {
       this.registerSubFeature(POLICY_VERSIONS.find(new FeatureVersion(0, 1))!);
     }
-
-    if (version.gte(new FeatureVersion(2, 7))) {
-      this.registerSubFeature(SOURCE_VERSIONS.find(new FeatureVersion(0, 1))!);
-    }
   }
 }
 

package/src/wasm.ts

@@ -0,0 +1,89 @@
+/**
+ * The TypeScript interface to the `wasm` module (implemented in Rust) with all the helpers needed to convert between
+ * the two.
+ */
+
+import {Schema} from "./definitions";
+import {
+    ConstDirectiveNode,
+    DefinitionNode,
+    DocumentNode,
+    GraphQLError,
+    print,
+} from "graphql/index";
+import {FeatureUrl, LinkDirectiveArgs} from "./specs/coreSpec";
+import {connectIdentity} from "./specs/connectSpec";
+import {aggregateError} from "./error";
+import {ErrorLocation, SourceDirective} from "../wasm/node";
+import {ASTNode, ConstValueNode, SchemaExtensionNode, StringValueNode} from "graphql";
+
+const {validate_connect_directives} =
+    require('../wasm/node') as typeof import('../wasm/node');
+
+/**
+ * @throws AggregateGraphQLError
+ */
+export function validateSubgraphSchema(schema: Schema, ast: DocumentNode) {
+    const errors = [];
+    if (hasConnectDirectives(schema)) {
+        errors.push(...validateConnectDirectives(ast));
+    }
+
+    if (errors.length > 0) {
+        throw aggregateError("FROM_WASM", "WASM validation detected errors", errors)
+    }
+}
+
+function hasConnectDirectives(schema: Schema): boolean {
+    return schema.schemaDefinition.appliedDirectivesOf<LinkDirectiveArgs>('link')
+        .some(linkDirective => {
+            const {url, import: imports} = linkDirective.arguments();
+            const featureUrl = FeatureUrl.maybeParse(url);
+            return imports && featureUrl && featureUrl.identity === connectIdentity && imports.some(nameOrRename => {
+                const originalName = typeof nameOrRename === 'string' ? nameOrRename : nameOrRename.name;
+                return originalName === '@source' || originalName === '@connect';
+            });
+        });
+}
+
+function validateConnectDirectives(ast: DocumentNode): GraphQLError[] {
+    // TODO: pass the _original_ source, not the one with added federation stuff (otherwise positions will be wrong)
+    // TODO: unless, of course, there's already infrastructure up above which does translation somehow
+    const source = print(ast)
+    return validate_connect_directives(source).map(
+        raw => new GraphQLError(raw.message, {
+            // source: new Source(source),
+            nodes: raw.location && findNode(ast, raw.location),
+            extensions: {code: raw.code},
+        }),
+    );
+}
+
+
+function findNode(ast: DocumentNode, location: ErrorLocation): ASTNode | undefined {
+    if (location.source) {
+        const sourceDirective = findSourceDirective(ast, location.source);
+        if (sourceDirective) {
+            return sourceDirective;
+        }
+    }
+    return undefined;
+}
+
+function findSourceDirective(ast: DocumentNode, sourceDirective: SourceDirective): ConstDirectiveNode | undefined {
+    const schema = ast.definitions.find(isSchemaExtension);
+    const sourceDirectives = schema?.directives?.
+        filter(directive => directive.name.value === "source");
+    return sourceDirectives?.find(directive => {
+        const nameArg = directive.arguments?.find(arg => arg.name.value === "name");
+        return nameArg && isStringValue(nameArg.value) && nameArg.value.value === sourceDirective.name;
+    });
+}
+
+function isSchemaExtension(node: DefinitionNode): node is SchemaExtensionNode {
+    return node.kind === "SchemaExtension";
+}
+
+function isStringValue(node: ConstValueNode): node is StringValueNode {
+    return node.kind === "StringValue";
+}

package/dist/specs/sourceSpec.d.ts

@@ -1,69 +0,0 @@
-import { GraphQLError } from 'graphql';
-import { FeatureDefinition, FeatureDefinitions, FeatureVersion } from "./coreSpec";
-import { Schema, DirectiveDefinition } from '../definitions';
-export declare const sourceIdentity = "https://specs.apollo.dev/source";
-export declare class SourceSpecDefinition extends FeatureDefinition {
-    readonly minimumFederationVersion: FeatureVersion;
-    constructor(version: FeatureVersion, minimumFederationVersion: FeatureVersion);
-    addElementsToSchema(schema: Schema): GraphQLError[];
-    allElementNames(): string[];
-    sourceAPIDirective(schema: Schema): DirectiveDefinition<SourceAPIDirectiveArgs>;
-    sourceTypeDirective(schema: Schema): DirectiveDefinition<SourceTypeDirectiveArgs>;
-    sourceFieldDirective(schema: Schema): DirectiveDefinition<SourceFieldDirectiveArgs>;
-    private getSourceDirectives;
-    validateSubgraphSchema(schema: Schema): GraphQLError[];
-    private validateSourceAPI;
-    private validateSourceType;
-    private validateSourceField;
-}
-export type SourceAPIDirectiveArgs = {
-    name: string;
-    http?: HTTPSourceAPI;
-};
-export type HTTPSourceAPI = {
-    baseURL: string;
-    headers?: HTTPHeaderMapping[];
-};
-export type HTTPHeaderMapping = {
-    name: string;
-    as?: string;
-    value?: string;
-};
-export type SourceTypeDirectiveArgs = {
-    api: string;
-    http?: HTTPSourceType;
-    selection: JSONSelection;
-    keyTypeMap?: KeyTypeMap;
-};
-export type HTTPSourceType = {
-    GET?: URLPathTemplate;
-    POST?: URLPathTemplate;
-    headers?: HTTPHeaderMapping[];
-    body?: JSONSelection;
-};
-type URLPathTemplate = string;
-type JSONSelection = string;
-type KeyTypeMap = {
-    key: string;
-    typeMap: {
-        [__typename: string]: string;
-    };
-};
-export type SourceFieldDirectiveArgs = {
-    api: string;
-    http?: HTTPSourceField;
-    selection?: JSONSelection;
-    keyTypeMap?: KeyTypeMap;
-};
-export type HTTPSourceField = {
-    GET?: URLPathTemplate;
-    POST?: URLPathTemplate;
-    PUT?: URLPathTemplate;
-    PATCH?: URLPathTemplate;
-    DELETE?: URLPathTemplate;
-    body?: JSONSelection;
-    headers?: HTTPHeaderMapping[];
-};
-export declare const SOURCE_VERSIONS: FeatureDefinitions<SourceSpecDefinition>;
-export {};
-//# sourceMappingURL=sourceSpec.d.ts.map

package/dist/specs/sourceSpec.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"sourceSpec.d.ts","sourceRoot":"","sources":["../../src/specs/sourceSpec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAqB,YAAY,EAAQ,MAAM,SAAS,CAAC;AAChE,OAAO,EAAE,iBAAiB,EAAE,kBAAkB,EAAc,cAAc,EAAqB,MAAM,YAAY,CAAC;AAClH,OAAO,EACL,MAAM,EAKN,mBAAmB,EAEpB,MAAM,gBAAgB,CAAC;AAKxB,eAAO,MAAM,cAAc,oCAAoC,CAAC;AAEhE,qBAAa,oBAAqB,SAAQ,iBAAiB;IACpB,QAAQ,CAAC,wBAAwB,EAAE,cAAc;gBAA1E,OAAO,EAAE,cAAc,EAAW,wBAAwB,EAAE,cAAc;IA4BtF,mBAAmB,CAAC,MAAM,EAAE,MAAM,GAAG,YAAY,EAAE;IA2EnD,eAAe,IAAI,MAAM,EAAE;IAgB3B,kBAAkB,CAAC,MAAM,EAAE,MAAM;IAIjC,mBAAmB,CAAC,MAAM,EAAE,MAAM;IAIlC,oBAAoB,CAAC,MAAM,EAAE,MAAM;IAInC,OAAO,CAAC,mBAAmB;IA+BlB,sBAAsB,CAAC,MAAM,EAAE,MAAM,GAAG,YAAY,EAAE;IA+B/D,OAAO,CAAC,iBAAiB;IAkEzB,OAAO,CAAC,kBAAkB;IAiG1B,OAAO,CAAC,mBAAmB;CAiH5B;AA2DD,MAAM,MAAM,sBAAsB,GAAG;IACnC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,aAAa,CAAC;CACtB,CAAC;AAEF,MAAM,MAAM,aAAa,GAAG;IAC1B,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,iBAAiB,EAAE,CAAC;CAC/B,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,MAAM,MAAM,uBAAuB,GAAG;IACpC,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,cAAc,CAAC;IACtB,SAAS,EAAE,aAAa,CAAC;IACzB,UAAU,CAAC,EAAE,UAAU,CAAC;CACzB,CAAC;AAEF,MAAM,MAAM,cAAc,GAAG;IAC3B,GAAG,CAAC,EAAE,eAAe,CAAC;IACtB,IAAI,CAAC,EAAE,eAAe,CAAC;IACvB,OAAO,CAAC,EAAE,iBAAiB,EAAE,CAAC;IAC9B,IAAI,CAAC,EAAE,aAAa,CAAC;CACtB,CAAC;AAEF,KAAK,eAAe,GAAG,MAAM,CAAC;AAC9B,KAAK,aAAa,GAAG,MAAM,CAAC;AAE5B,KAAK,UAAU,GAAG;IAChB,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,EAAE;QACP,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAAC;KAC9B,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IACrC,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,eAAe,CAAC;IACvB,SAAS,CAAC,EAAE,aAAa,CAAC;IAC1B,UAAU,CAAC,EAAE,UAAU,CAAC;CACzB,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG;IAC5B,GAAG,CAAC,EAAE,eAAe,CAAC;IACtB,IAAI,CAAC,EAAE,eAAe,CAAC;IACvB,GAAG,CAAC,EAAE,eAAe,CAAC;IACtB,KAAK,CAAC,EAAE,eAAe,CAAC;IACxB,MAAM,CAAC,EAAE,eAAe,CAAC;IACzB,IAAI,CAAC,EAAE,aAAa,CAAC;IACrB,OAAO,CAAC,EAAE,iBAAiB,EAAE,CAAC;CAC/B,CAAC;AAEF,eAAO,MAAM,eAAe,0CACwD,CAAC"}