@wundergraph/protographic 0.15.5 → 0.16.0

package/dist/src/required-fields-visitor.js

@@ -0,0 +1,513 @@
+import { getNamedType, isInterfaceType, isObjectType, isUnionType, Kind, parse, visit, } from 'graphql';
+import { FieldMapping } from '@wundergraph/cosmo-connect/dist/node/v1/node_pb';
+import { CompositeMessageKind } from './types.js';
+import { KEY_DIRECTIVE_NAME } from './string-constants.js';
+import { createEntityLookupRequestKeyMessageName, createRequestMessageName, createRequiredFieldsMethodName, createResponseMessageName, graphqlFieldToProtoField, formatKeyElements, } from './naming-conventions.js';
+import { getProtoTypeFromGraphQL } from './proto-utils.js';
+import { AbstractSelectionRewriter } from './abstract-selection-rewriter.js';
+/**
+ * Generates protobuf messages and RPC methods for @requires directive field sets.
+ *
+ * This visitor processes the field set defined in a @requires directive and generates:
+ * - RPC method definitions for fetching required fields
+ * - Request/response message definitions
+ * - Field mappings between GraphQL and protobuf
+ *
+ * The visitor handles each @key directive on the entity separately, creating
+ * distinct RPC methods for each key since required fields may need to be
+ * fetched using different entity keys.
+ *
+ * @example
+ * ```typescript
+ * const visitor = new RequiredFieldsVisitor(
+ *   schema,
+ *   ProductType,
+ *   weightField,
+ *   'dimensions { width height }'
+ * );
+ * visitor.visit();
+ *
+ * const messages = visitor.getMessageDefinitions();
+ * const rpcs = visitor.getRPCMethods();
+ * const mappings = visitor.getMapping();
+ * ```
+ */
+export class RequiredFieldsVisitor {
+    /**
+     * Creates a new RequiredFieldsVisitor.
+     *
+     * @param schema - The GraphQL schema containing type definitions
+     * @param objectType - The entity type that has the @requires directive
+     * @param requiredField - The field with the @requires directive
+     * @param fieldSet - The field set string from the @requires directive (e.g., "dimensions { width height }")
+     * @param options - Optional configuration options
+     * @throws Error if the object type is not an entity (has no @key directive)
+     */
+    constructor(schema, objectType, requiredField, fieldSet) {
+        this.schema = schema;
+        this.objectType = objectType;
+        this.requiredField = requiredField;
+        this.ancestors = [];
+        this.currentType = this.objectType;
+        this.keyDirectives = [];
+        this.currentKeyFieldsString = '';
+        /** Collected RPC methods for the required fields */
+        this.rpcMethods = [];
+        /** All generated protobuf message definitions */
+        this.messageDefinitions = [];
+        /** Stack for tracking nested message contexts */
+        this.stack = [];
+        this.inlineFragmentStack = [];
+        /** Mappings keyed by @key directive fields string */
+        this.mapping = {};
+        this.resolveKeyDirectives();
+        this.fieldSetDoc = parse(`{ ${fieldSet} }`);
+        this.normalizeSelectionSet();
+        this.visitor = this.createASTVisitor();
+        this.mapping = {};
+    }
+    /**
+     * Executes the visitor, processing the field set for each @key directive.
+     * Creates separate RPC methods and mappings for each entity key.
+     */
+    visit() {
+        for (const keyDirective of this.keyDirectives) {
+            const rawFieldSet = this.getKeyFieldsString(keyDirective);
+            if (rawFieldSet.length === 0) {
+                throw new Error(`Required field set is empty for key directive ${keyDirective.name.value}`);
+            }
+            this.currentKeyFieldsString = formatKeyElements(rawFieldSet).join(' ');
+            if (this.mapping[this.currentKeyFieldsString]) {
+                continue;
+            }
+            this.mapping[this.currentKeyFieldsString] = {
+                requiredFieldMapping: new FieldMapping({
+                    original: this.requiredField.name,
+                    mapped: graphqlFieldToProtoField(this.requiredField.name),
+                }),
+            };
+            visit(this.fieldSetDoc, this.visitor);
+        }
+    }
+    /**
+     * Normalizes the parsed field set operation by rewriting abstract selections.
+     * This ensures consistent handling of interface and union type selections.
+     */
+    normalizeSelectionSet() {
+        const visitor = new AbstractSelectionRewriter(this.fieldSetDoc, this.schema, this.objectType);
+        visitor.normalize();
+    }
+    /**
+     * Returns all generated protobuf message definitions.
+     *
+     * @returns Array of ProtoMessage definitions for request, response, and nested messages
+     */
+    getMessageDefinitions() {
+        return this.messageDefinitions;
+    }
+    /**
+     * Returns the generated RPC method definitions.
+     *
+     * @returns Array of RPCMethod definitions for fetching required fields
+     */
+    getRPCMethods() {
+        return this.rpcMethods;
+    }
+    /**
+     * Returns the field mappings keyed by @key directive fields string.
+     *
+     * @returns Record mapping key strings to their RequiredFieldMapping configurations
+     */
+    getMapping() {
+        return this.mapping;
+    }
+    /**
+     * Resolves all @key directives from the object type.
+     * Each key directive will result in a separate RPC method.
+     *
+     * @throws Error if the object type has no @key directives
+     */
+    resolveKeyDirectives() {
+        var _a, _b, _c;
+        this.keyDirectives = (_c = (_b = (_a = this.objectType.astNode) === null || _a === void 0 ? void 0 : _a.directives) === null || _b === void 0 ? void 0 : _b.filter((d) => d.name.value === KEY_DIRECTIVE_NAME)) !== null && _c !== void 0 ? _c : [];
+        if (this.keyDirectives.length === 0) {
+            throw new Error('Object type has to be an entity type to make use of the @requires directive');
+        }
+    }
+    /**
+     * Creates the AST visitor configuration for traversing the field set document.
+     *
+     * @returns An ASTVisitor with handlers for Document, Field, InlineFragment, and SelectionSet nodes
+     */
+    createASTVisitor() {
+        return {
+            Document: {
+                enter: (node) => {
+                    this.onEnterDocument(node);
+                },
+                leave: () => {
+                    this.onLeaveDocument();
+                },
+            },
+            Field: {
+                enter: (node, key, parent, path, ancestors) => {
+                    this.onEnterField({ node, key, parent, path, ancestors });
+                },
+            },
+            InlineFragment: {
+                enter: (node, key, parent, path, ancestors) => {
+                    this.onEnterInlineFragment({ node, key, parent, path, ancestors });
+                },
+                leave: (node, key, parent, path, ancestors) => {
+                    this.onLeaveInlineFragment({ node, key, parent, path, ancestors });
+                },
+            },
+            SelectionSet: {
+                enter: (node, key, parent, path, ancestors) => {
+                    this.onEnterSelectionSet({ node, key, parent, path, ancestors });
+                },
+                leave: (node, key, parent, path, ancestors) => {
+                    this.onLeaveSelectionSet({ node, key, parent, path, ancestors });
+                },
+            },
+        };
+    }
+    /**
+     * Handles leaving the document node.
+     * Finalizes the required field message and generates type field mappings.
+     */
+    onLeaveDocument() {
+        if (this.requiredFieldMessage) {
+            this.messageDefinitions.push(this.requiredFieldMessage);
+        }
+    }
+    /**
+     * Handles entering the document node.
+     * Creates the RPC method definition and all request/response message structures
+     * for fetching the required fields.
+     *
+     * @param node - The document node being entered
+     */
+    onEnterDocument(node) {
+        const requiredFieldsMethodName = createRequiredFieldsMethodName(this.objectType.name, this.requiredField.name, this.currentKeyFieldsString);
+        const requestMessageName = createRequestMessageName(requiredFieldsMethodName);
+        const responseMessageName = createResponseMessageName(requiredFieldsMethodName);
+        this.mapping[this.currentKeyFieldsString].rpc = {
+            name: requiredFieldsMethodName,
+            request: requestMessageName,
+            response: responseMessageName,
+        };
+        this.rpcMethods.push({
+            name: requiredFieldsMethodName,
+            request: requestMessageName,
+            response: responseMessageName,
+        });
+        // Request messages
+        const contextMessageName = `${requiredFieldsMethodName}Context`;
+        this.messageDefinitions.push({
+            messageName: requestMessageName,
+            fields: [
+                {
+                    fieldName: 'context',
+                    typeName: contextMessageName,
+                    fieldNumber: 1,
+                    isRepeated: true,
+                    description: `${contextMessageName} provides the context for the required fields method ${requiredFieldsMethodName}.`,
+                },
+            ],
+        });
+        const fieldsMessageName = `${requiredFieldsMethodName}Fields`;
+        const entityKeyRequestMessageName = createEntityLookupRequestKeyMessageName(this.objectType.name, this.currentKeyFieldsString);
+        this.messageDefinitions.push({
+            messageName: contextMessageName,
+            fields: [
+                {
+                    fieldName: 'key',
+                    typeName: entityKeyRequestMessageName,
+                    fieldNumber: 1,
+                },
+                {
+                    fieldName: 'fields',
+                    typeName: fieldsMessageName,
+                    fieldNumber: 2,
+                },
+            ],
+        });
+        // Define the prototype for the required fields message.
+        // This will be added to the message definitions when the document is left.
+        this.requiredFieldMessage = {
+            messageName: fieldsMessageName,
+            fields: [],
+        };
+        // Response messages
+        const resultMessageName = `${requiredFieldsMethodName}Result`;
+        this.messageDefinitions.push({
+            messageName: responseMessageName,
+            fields: [
+                {
+                    fieldName: 'result',
+                    typeName: resultMessageName,
+                    fieldNumber: 1,
+                    isRepeated: true,
+                    description: `${resultMessageName} provides the result for the required fields method ${requiredFieldsMethodName}.`,
+                },
+            ],
+        });
+        // Get the type name from the required field
+        const typeInfo = getProtoTypeFromGraphQL(false, this.requiredField.type);
+        this.messageDefinitions.push({
+            messageName: resultMessageName,
+            fields: [
+                {
+                    fieldName: graphqlFieldToProtoField(this.requiredField.name),
+                    typeName: typeInfo.typeName,
+                    fieldNumber: 1,
+                    isRepeated: typeInfo.isRepeated,
+                },
+            ],
+        });
+        this.stack.push(this.requiredFieldMessage);
+        this.current = this.requiredFieldMessage;
+    }
+    /**
+     * Handles entering a field node during traversal.
+     * Adds the field to the current proto message with appropriate type mapping.
+     *
+     * @param ctx - The visit context containing the field node and its ancestors
+     * @throws Error if the field definition is not found on the current type
+     */
+    onEnterField(ctx) {
+        var _a;
+        if (!this.current) {
+            return;
+        }
+        const fieldDefinition = this.fieldDefinition(ctx.node.name.value);
+        if (!fieldDefinition) {
+            throw new Error(`Field definition not found for field ${ctx.node.name.value}`);
+        }
+        let compositeType;
+        if (this.isCompositeType(fieldDefinition.type)) {
+            compositeType = this.handleCompositeType(fieldDefinition);
+        }
+        const protoFieldName = graphqlFieldToProtoField(fieldDefinition.name);
+        // Check if field already exists to avoid duplicates when reusing nested messages
+        if (this.current.fields.some((f) => f.fieldName === protoFieldName)) {
+            return;
+        }
+        const typeInfo = getProtoTypeFromGraphQL(false, fieldDefinition.type);
+        this.current.fields.push({
+            fieldName: protoFieldName,
+            typeName: typeInfo.typeName,
+            fieldNumber: ((_a = this.current) === null || _a === void 0 ? void 0 : _a.fields.length) + 1,
+            isRepeated: typeInfo.isRepeated,
+            graphqlName: fieldDefinition.name,
+            compositeType,
+        });
+    }
+    /**
+     * Handles entering an inline fragment node.
+     * Pushes the current inline fragment onto the stack for nested fragment handling.
+     *
+     * @param ctx - The visit context containing the inline fragment node
+     */
+    onEnterInlineFragment(ctx) {
+        if (this.currentInlineFragment) {
+            this.inlineFragmentStack.push(this.currentInlineFragment);
+        }
+        this.currentInlineFragment = ctx.node;
+    }
+    /**
+     * Handles leaving an inline fragment node.
+     * Records union member types when processing union type fragments.
+     *
+     * @param ctx - The visit context containing the inline fragment node
+     */
+    onLeaveInlineFragment(ctx) {
+        var _a;
+        this.currentInlineFragment = (_a = this.inlineFragmentStack.pop()) !== null && _a !== void 0 ? _a : undefined;
+    }
+    /**
+     * Handles entering a selection set node.
+     * Creates a new nested proto message for object type selections and updates
+     * the current type context for proper field resolution.
+     *
+     * @param ctx - The visit context containing the selection set node and its parent
+     */
+    onEnterSelectionSet(ctx) {
+        var _a, _b, _c, _d;
+        if (!ctx.parent || !this.current) {
+            return false;
+        }
+        let currentType;
+        if (this.isFieldNode(ctx.parent)) {
+            currentType = (_a = this.findObjectTypeForField(ctx.parent.name.value)) !== null && _a !== void 0 ? _a : undefined;
+        }
+        else if (this.isInlineFragmentNode(ctx.parent)) {
+            const typeName = (_b = ctx.parent.typeCondition) === null || _b === void 0 ? void 0 : _b.name.value;
+            if (typeName) {
+                currentType = this.findObjectType(typeName);
+            }
+        }
+        else {
+            return false;
+        }
+        if (!this.currentType || !currentType) {
+            return false;
+        }
+        this.ancestors.push(this.currentType);
+        this.currentType = currentType;
+        if (!this.current.nestedMessages) {
+            this.current.nestedMessages = [];
+        }
+        // Check if a nested message with the same name already exists
+        const existingNested = this.current.nestedMessages.find((msg) => { var _a; return msg.messageName === ((_a = this.currentType) === null || _a === void 0 ? void 0 : _a.name); });
+        let nested;
+        if (existingNested) {
+            // Reuse the existing nested message to avoid duplicates
+            nested = existingNested;
+        }
+        else {
+            // Create a new nested message for the current type
+            nested = {
+                messageName: (_d = (_c = this.currentType) === null || _c === void 0 ? void 0 : _c.name) !== null && _d !== void 0 ? _d : '',
+                fields: [],
+            };
+            this.current.nestedMessages.push(nested);
+        }
+        this.stack.push(this.current);
+        this.current = nested;
+    }
+    /**
+     * Handles leaving a selection set node.
+     * Restores the previous type and message context when ascending the tree,
+     * but only if onEnterSelectionSet actually pushed state.
+     *
+     * @param ctx - The visit context containing the selection set node
+     */
+    onLeaveSelectionSet(ctx) {
+        var _a;
+        this.currentType = (_a = this.ancestors.pop()) !== null && _a !== void 0 ? _a : this.currentType;
+        this.current = this.stack.pop();
+    }
+    /**
+     * Handles composite types (interfaces and unions) by setting up the
+     * appropriate composite type metadata on the current message.
+     *
+     * @param fieldDefinition - The field definition with a composite type
+     */
+    handleCompositeType(fieldDefinition) {
+        if (!this.current) {
+            return undefined;
+        }
+        const compositeType = getNamedType(fieldDefinition.type);
+        if (isInterfaceType(compositeType)) {
+            return {
+                kind: CompositeMessageKind.INTERFACE,
+                implementingTypes: this.schema.getImplementations(compositeType).objects.map((o) => o.name),
+                typeName: compositeType.name,
+            };
+        }
+        if (isUnionType(compositeType)) {
+            return {
+                kind: CompositeMessageKind.UNION,
+                memberTypes: this.schema.getPossibleTypes(compositeType).map((t) => t.name),
+                typeName: compositeType.name,
+            };
+        }
+        return undefined;
+    }
+    /**
+     * Type guard to check if a node is a FieldNode.
+     *
+     * @param node - The AST node or array of nodes to check
+     * @returns True if the node is a FieldNode
+     */
+    isFieldNode(node) {
+        if (Array.isArray(node)) {
+            return false;
+        }
+        return node.kind === Kind.FIELD;
+    }
+    /**
+     * Type guard to check if a node is an InlineFragmentNode.
+     *
+     * @param node - The AST node or array of nodes to check
+     * @returns True if the node is an InlineFragmentNode
+     */
+    isInlineFragmentNode(node) {
+        if (Array.isArray(node)) {
+            return false;
+        }
+        return node.kind === Kind.INLINE_FRAGMENT;
+    }
+    /**
+     * Finds the GraphQL object type for a field by looking up the field's return type.
+     *
+     * @param fieldName - The name of the field to look up
+     * @returns The GraphQL object type if the field returns an object type, undefined otherwise
+     */
+    findObjectTypeForField(fieldName) {
+        var _a, _b;
+        const fields = (_b = (_a = this.currentType) === null || _a === void 0 ? void 0 : _a.getFields()) !== null && _b !== void 0 ? _b : {};
+        const field = fields[fieldName];
+        if (!field) {
+            return undefined;
+        }
+        const namedType = getNamedType(field.type);
+        if (isObjectType(namedType)) {
+            return namedType;
+        }
+        return undefined;
+    }
+    /**
+     * Retrieves the field definition for a field name from the current type.
+     *
+     * @param fieldName - The name of the field to look up
+     * @returns The GraphQL field definition, or undefined if not found
+     */
+    fieldDefinition(fieldName) {
+        var _a;
+        return (_a = this.currentType) === null || _a === void 0 ? void 0 : _a.getFields()[fieldName];
+    }
+    /**
+     * Finds a GraphQL object type by name from the schema's type map.
+     *
+     * @param typeName - The name of the type to find
+     * @returns The GraphQL object type if found and is an object type, undefined otherwise
+     */
+    findObjectType(typeName) {
+        const type = this.schema.getTypeMap()[typeName];
+        if (!type) {
+            return undefined;
+        }
+        if (!isObjectType(type)) {
+            return undefined;
+        }
+        return type;
+    }
+    /**
+     * Extracts the fields string from a @key directive's fields argument.
+     *
+     * @param directive - The @key directive node
+     * @returns The fields string value, or empty string if not found
+     */
+    getKeyFieldsString(directive) {
+        var _a;
+        const fieldsArg = (_a = directive.arguments) === null || _a === void 0 ? void 0 : _a.find((arg) => arg.name.value === 'fields');
+        if (!fieldsArg) {
+            return '';
+        }
+        return fieldsArg.value.kind === Kind.STRING ? fieldsArg.value.value : '';
+    }
+    /**
+     * Checks if a GraphQL type is a composite type (interface or union).
+     *
+     * @param type - The GraphQL type to check
+     * @returns True if the type is an interface or union type
+     */
+    isCompositeType(type) {
+        const namedType = getNamedType(type);
+        return isInterfaceType(namedType) || isUnionType(namedType);
+    }
+}
+//# sourceMappingURL=required-fields-visitor.js.map

package/dist/src/required-fields-visitor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"required-fields-visitor.js","sourceRoot":"","sources":["../../src/required-fields-visitor.ts"],"names":[],"mappings":"AAAA,OAAO,EAML,YAAY,EAMZ,eAAe,EACf,YAAY,EACZ,WAAW,EACX,IAAI,EACJ,KAAK,EAEL,KAAK,GACN,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,YAAY,EAAE,MAAM,iDAAiD,CAAC;AAC/E,OAAO,EAA8B,oBAAoB,EAAyC,MAAM,YAAY,CAAC;AACrH,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EACL,uCAAuC,EACvC,wBAAwB,EACxB,8BAA8B,EAC9B,yBAAyB,EACzB,wBAAwB,EACxB,iBAAiB,GAClB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,uBAAuB,EAAE,MAAM,kBAAkB,CAAC;AAC3D,OAAO,EAAE,yBAAyB,EAAE,MAAM,kCAAkC,CAAC;AAmB7E;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,MAAM,OAAO,qBAAqB;IA0BhC;;;;;;;;;OASG;IACH,YACmB,MAAqB,EACrB,UAA6B,EAC7B,aAA0C,EAC3D,QAAgB;QAHC,WAAM,GAAN,MAAM,CAAe;QACrB,eAAU,GAAV,UAAU,CAAmB;QAC7B,kBAAa,GAAb,aAAa,CAA6B;QAnCrD,cAAS,GAAwB,EAAE,CAAC;QACpC,gBAAW,GAAkC,IAAI,CAAC,UAAU,CAAC;QAC7D,kBAAa,GAAoB,EAAE,CAAC;QACpC,2BAAsB,GAAG,EAAE,CAAC;QAEpC,oDAAoD;QAC5C,eAAU,GAAgB,EAAE,CAAC;QACrC,iDAAiD;QACzC,uBAAkB,GAAmB,EAAE,CAAC;QAKhD,iDAAiD;QACzC,UAAK,GAAmB,EAAE,CAAC;QAG3B,wBAAmB,GAAyB,EAAE,CAAC;QAEvD,qDAAqD;QAC7C,YAAO,GAA0B,EAAE,CAAC;QAkB1C,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC5B,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC,KAAK,QAAQ,IAAI,CAAC,CAAC;QAC5C,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC7B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACvC,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;IACpB,CAAC;IAED;;;OAGG;IACI,KAAK;QACV,KAAK,MAAM,YAAY,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YAC9C,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC;YAC1D,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,iDAAiD,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;YAC9F,CAAC;YAED,IAAI,CAAC,sBAAsB,GAAG,iBAAiB,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAEvE,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,sBAAsB,CAAC,EAAE,CAAC;gBAC9C,SAAS;YACX,CAAC;YAED,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,sBAAsB,CAAC,GAAG;gBAC1C,oBAAoB,EAAE,IAAI,YAAY,CAAC;oBACrC,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI;oBACjC,MAAM,EAAE,wBAAwB,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;iBAC1D,CAAC;aACH,CAAC;YACF,KAAK,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,qBAAqB;QAC3B,MAAM,OAAO,GAAG,IAAI,yBAAyB,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QAC9F,OAAO,CAAC,SAAS,EAAE,CAAC;IACtB,CAAC;IAED;;;;OAIG;IACI,qBAAqB;QAC1B,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACjC,CAAC;IAED;;;;OAIG;IACI,aAAa;QAClB,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED;;;;OAIG;IACI,UAAU;QACf,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED;;;;;OAKG;IACK,oBAAoB;;QAC1B,IAAI,CAAC,aAAa,GAAG,MAAA,MAAA,MAAA,IAAI,CAAC,UAAU,CAAC,OAAO,0CAAE,UAAU,0CAAE,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,KAAK,kBAAkB,CAAC,mCAAI,EAAE,CAAC;QACnH,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACpC,MAAM,IAAI,KAAK,CAAC,6EAA6E,CAAC,CAAC;QACjG,CAAC;IACH,CAAC;IAED;;;;OAIG;IACK,gBAAgB;QACtB,OAAO;YACL,QAAQ,EAAE;gBACR,KAAK,EAAE,CAAC,IAAI,EAAE,EAAE;oBACd,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;gBAC7B,CAAC;gBACD,KAAK,EAAE,GAAG,EAAE;oBACV,IAAI,CAAC,eAAe,EAAE,CAAC;gBACzB,CAAC;aACF;YACD,KAAK,EAAE;gBACL,KAAK,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;oBAC5C,IAAI,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;gBAC5D,CAAC;aACF;YACD,cAAc,EAAE;gBACd,KAAK,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;oBAC5C,IAAI,CAAC,qBAAqB,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;gBACrE,CAAC;gBACD,KAAK,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;oBAC5C,IAAI,CAAC,qBAAqB,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;gBACrE,CAAC;aACF;YACD,YAAY,EAAE;gBACZ,KAAK,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;oBAC5C,IAAI,CAAC,mBAAmB,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;gBACnE,CAAC;gBACD,KAAK,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;oBAC5C,IAAI,CAAC,mBAAmB,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;gBACnE,CAAC;aACF;SACF,CAAC;IACJ,CAAC;IAED;;;OAGG;IACK,eAAe;QACrB,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC9B,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACK,eAAe,CAAC,IAAkB;QACxC,MAAM,wBAAwB,GAAG,8BAA8B,CAC7D,IAAI,CAAC,UAAU,CAAC,IAAI,EACpB,IAAI,CAAC,aAAa,CAAC,IAAI,EACvB,IAAI,CAAC,sBAAsB,CAC5B,CAAC;QAEF,MAAM,kBAAkB,GAAG,wBAAwB,CAAC,wBAAwB,CAAC,CAAC;QAC9E,MAAM,mBAAmB,GAAG,yBAAyB,CAAC,wBAAwB,CAAC,CAAC;QAEhF,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC,GAAG,GAAG;YAC9C,IAAI,EAAE,wBAAwB;YAC9B,OAAO,EAAE,kBAAkB;YAC3B,QAAQ,EAAE,mBAAmB;SAC9B,CAAC;QAEF,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;YACnB,IAAI,EAAE,wBAAwB;YAC9B,OAAO,EAAE,kBAAkB;YAC3B,QAAQ,EAAE,mBAAmB;SAC9B,CAAC,CAAC;QAEH,mBAAmB;QACnB,MAAM,kBAAkB,GAAG,GAAG,wBAAwB,SAAS,CAAC;QAChE,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC;YAC3B,WAAW,EAAE,kBAAkB;YAC/B,MAAM,EAAE;gBACN;oBACE,SAAS,EAAE,SAAS;oBACpB,QAAQ,EAAE,kBAAkB;oBAC5B,WAAW,EAAE,CAAC;oBACd,UAAU,EAAE,IAAI;oBAChB,WAAW,EAAE,GAAG,kBAAkB,wDAAwD,wBAAwB,GAAG;iBACtH;aACF;SACF,CAAC,CAAC;QAEH,MAAM,iBAAiB,GAAG,GAAG,wBAAwB,QAAQ,CAAC;QAC9D,MAAM,2BAA2B,GAAG,uCAAuC,CACzE,IAAI,CAAC,UAAU,CAAC,IAAI,EACpB,IAAI,CAAC,sBAAsB,CAC5B,CAAC;QAEF,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC;YAC3B,WAAW,EAAE,kBAAkB;YAC/B,MAAM,EAAE;gBACN;oBACE,SAAS,EAAE,KAAK;oBAChB,QAAQ,EAAE,2BAA2B;oBACrC,WAAW,EAAE,CAAC;iBACf;gBACD;oBACE,SAAS,EAAE,QAAQ;oBACnB,QAAQ,EAAE,iBAAiB;oBAC3B,WAAW,EAAE,CAAC;iBACf;aACF;SACF,CAAC,CAAC;QAEH,wDAAwD;QACxD,2EAA2E;QAC3E,IAAI,CAAC,oBAAoB,GAAG;YAC1B,WAAW,EAAE,iBAAiB;YAC9B,MAAM,EAAE,EAAE;SACX,CAAC;QAEF,oBAAoB;QACpB,MAAM,iBAAiB,GAAG,GAAG,wBAAwB,QAAQ,CAAC;QAC9D,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC;YAC3B,WAAW,EAAE,mBAAmB;YAChC,MAAM,EAAE;gBACN;oBACE,SAAS,EAAE,QAAQ;oBACnB,QAAQ,EAAE,iBAAiB;oBAC3B,WAAW,EAAE,CAAC;oBACd,UAAU,EAAE,IAAI;oBAChB,WAAW,EAAE,GAAG,iBAAiB,uDAAuD,wBAAwB,GAAG;iBACpH;aACF;SACF,CAAC,CAAC;QAEH,4CAA4C;QAC5C,MAAM,QAAQ,GAAG,uBAAuB,CAAC,KAAK,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QAEzE,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC;YAC3B,WAAW,EAAE,iBAAiB;YAC9B,MAAM,EAAE;gBACN;oBACE,SAAS,EAAE,wBAAwB,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;oBAC5D,QAAQ,EAAE,QAAQ,CAAC,QAAQ;oBAC3B,WAAW,EAAE,CAAC;oBACd,UAAU,EAAE,QAAQ,CAAC,UAAU;iBAChC;aACF;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAC3C,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,oBAAoB,CAAC;IAC3C,CAAC;IAED;;;;;;OAMG;IACK,YAAY,CAAC,GAA4B;;QAC/C,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QAED,MAAM,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAClE,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,wCAAwC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;QACjF,CAAC;QAED,IAAI,aAAqD,CAAC;QAC1D,IAAI,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;YAC/C,aAAa,GAAG,IAAI,CAAC,mBAAmB,CAAC,eAAe,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,cAAc,GAAG,wBAAwB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAEtE,iFAAiF;QACjF,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,cAAc,CAAC,EAAE,CAAC;YACpE,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,uBAAuB,CAAC,KAAK,EAAE,eAAe,CAAC,IAAI,CAAC,CAAC;QACtE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC;YACvB,SAAS,EAAE,cAAc;YACzB,QAAQ,EAAE,QAAQ,CAAC,QAAQ;YAC3B,WAAW,EAAE,CAAA,MAAA,IAAI,CAAC,OAAO,0CAAE,MAAM,CAAC,MAAM,IAAG,CAAC;YAC5C,UAAU,EAAE,QAAQ,CAAC,UAAU;YAC/B,WAAW,EAAE,eAAe,CAAC,IAAI;YACjC,aAAa;SACd,CAAC,CAAC;IACL,CAAC;IAED;;;;;OAKG;IACK,qBAAqB,CAAC,GAAqC;QACjE,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC/B,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC5D,CAAC;QAED,IAAI,CAAC,qBAAqB,GAAG,GAAG,CAAC,IAAI,CAAC;IACxC,CAAC;IAED;;;;;OAKG;IACK,qBAAqB,CAAC,GAAqC;;QACjE,IAAI,CAAC,qBAAqB,GAAG,MAAA,IAAI,CAAC,mBAAmB,CAAC,GAAG,EAAE,mCAAI,SAAS,CAAC;IAC3E,CAAC;IAED;;;;;;OAMG;IACK,mBAAmB,CAAC,GAAmC;;QAC7D,IAAI,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACjC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,WAA0C,CAAC;QAC/C,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YACjC,WAAW,GAAG,MAAA,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,mCAAI,SAAS,CAAC;QAChF,CAAC;aAAM,IAAI,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YACjD,MAAM,QAAQ,GAAG,MAAA,GAAG,CAAC,MAAM,CAAC,aAAa,0CAAE,IAAI,CAAC,KAAK,CAAC;YACtD,IAAI,QAAQ,EAAE,CAAC;gBACb,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;aAAM,CAAC;YACN,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,WAAW,EAAE,CAAC;YACtC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACtC,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAE/B,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,CAAC,cAAc,GAAG,EAAE,CAAC;QACnC,CAAC;QAED,8DAA8D;QAC9D,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,WAAC,OAAA,GAAG,CAAC,WAAW,MAAK,MAAA,IAAI,CAAC,WAAW,0CAAE,IAAI,CAAA,CAAA,EAAA,CAAC,CAAC;QAE7G,IAAI,MAAoB,CAAC;QACzB,IAAI,cAAc,EAAE,CAAC;YACnB,wDAAwD;YACxD,MAAM,GAAG,cAAc,CAAC;QAC1B,CAAC;aAAM,CAAC;YACN,mDAAmD;YACnD,MAAM,GAAG;gBACP,WAAW,EAAE,MAAA,MAAA,IAAI,CAAC,WAAW,0CAAE,IAAI,mCAAI,EAAE;gBACzC,MAAM,EAAE,EAAE;aACX,CAAC;YACF,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAED;;;;;;OAMG;IACK,mBAAmB,CAAC,GAAmC;;QAC7D,IAAI,CAAC,WAAW,GAAG,MAAA,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,mCAAI,IAAI,CAAC,WAAW,CAAC;QAC5D,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;IAClC,CAAC;IAED;;;;;OAKG;IACK,mBAAmB,CAAC,eAA4C;QACtE,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,aAAa,GAAG,YAAY,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAEzD,IAAI,eAAe,CAAC,aAAa,CAAC,EAAE,CAAC;YACnC,OAAO;gBACL,IAAI,EAAE,oBAAoB,CAAC,SAAS;gBACpC,iBAAiB,EAAE,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;gBAC3F,QAAQ,EAAE,aAAa,CAAC,IAAI;aAC7B,CAAC;QACJ,CAAC;QAED,IAAI,WAAW,CAAC,aAAa,CAAC,EAAE,CAAC;YAC/B,OAAO;gBACL,IAAI,EAAE,oBAAoB,CAAC,KAAK;gBAChC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;gBAC3E,QAAQ,EAAE,aAAa,CAAC,IAAI;aAC7B,CAAC;QACJ,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;;;;OAKG;IACK,WAAW,CAAC,IAAsC;QACxD,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YACxB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAQ,IAAgB,CAAC,IAAI,KAAK,IAAI,CAAC,KAAK,CAAC;IAC/C,CAAC;IAED;;;;;OAKG;IACK,oBAAoB,CAAC,IAAsC;QACjE,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YACxB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAQ,IAAgB,CAAC,IAAI,KAAK,IAAI,CAAC,eAAe,CAAC;IACzD,CAAC;IAED;;;;;OAKG;IACK,sBAAsB,CAAC,SAAiB;;QAC9C,MAAM,MAAM,GAAG,MAAA,MAAA,IAAI,CAAC,WAAW,0CAAE,SAAS,EAAE,mCAAI,EAAE,CAAC;QACnD,MAAM,KAAK,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;QAChC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,SAAS,GAAG,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC3C,IAAI,YAAY,CAAC,SAAS,CAAC,EAAE,CAAC;YAC5B,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;;;;OAKG;IACK,eAAe,CAAC,SAAiB;;QACvC,OAAO,MAAA,IAAI,CAAC,WAAW,0CAAE,SAAS,GAAG,SAAS,CAAC,CAAC;IAClD,CAAC;IAED;;;;;OAKG;IACK,cAAc,CAAC,QAAgB;QACrC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,QAAQ,CAAC,CAAC;QAChD,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;YACxB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;OAKG;IACK,kBAAkB,CAAC,SAAwB;;QACjD,MAAM,SAAS,GAAG,MAAA,SAAS,CAAC,SAAS,0CAAE,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC;QAClF,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,OAAO,SAAS,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3E,CAAC;IAED;;;;;OAKG;IACK,eAAe,CAAC,IAAiB;QACvC,MAAM,SAAS,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;QACrC,OAAO,eAAe,CAAC,SAAS,CAAC,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;IAC9D,CAAC;CACF"}

package/dist/src/sdl-to-mapping-visitor.d.ts

@@ -12,7 +12,7 @@ import { GRPCMapping } from '@wundergraph/cosmo-connect/dist/node/v1/node_pb';
  * The generated mappings are used to translate between GraphQL and Protocol Buffer
  * representations in a consistent manner.
  */
-export declare class GraphQLToProtoVisitor {
+export declare class GraphQLToMappingVisitor {
     private readonly mapping;
     private readonly schema;
     /**
@@ -40,6 +40,14 @@ export declare class GraphQLToProtoVisitor {
      * lookup RPC methods and entity mappings.
      */
     private processEntityTypes;
+    private createRequiredFieldsMapping;
+    /**
+     * Extracts the required field set from a field's requires directive
+     * @param field - The GraphQL field to get the required field set from
+     * @returns The required field set string
+     * @throws Error if the required field set is not found or is not a string
+     */
+    private getRequiredFieldSet;
     /**
      * Extract all key directives from a GraphQL object type
      *