@apollo/federation-internals 2.7.1 → 2.7.3-testing.0

package/dist/operations.js

@@ -536,25 +536,26 @@ function computeFragmentsToKeep(selectionSet, fragments, minUsagesToOptimize) {
     return toExpand.size === 0 ? fragments : fragments.filter((f) => !toExpand.has(f.name));
 }
 class Operation {
-    constructor(schema, rootKind, selectionSet, variableDefinitions, fragments, name) {
+    constructor(schema, rootKind, selectionSet, variableDefinitions, fragments, name, directives) {
         this.schema = schema;
         this.rootKind = rootKind;
         this.selectionSet = selectionSet;
         this.variableDefinitions = variableDefinitions;
         this.fragments = fragments;
         this.name = name;
+        this.directives = directives;
     }
     withUpdatedSelectionSet(newSelectionSet) {
         if (this.selectionSet === newSelectionSet) {
             return this;
         }
-        return new Operation(this.schema, this.rootKind, newSelectionSet, this.variableDefinitions, this.fragments, this.name);
+        return new Operation(this.schema, this.rootKind, newSelectionSet, this.variableDefinitions, this.fragments, this.name, this.directives);
     }
     withUpdatedSelectionSetAndFragments(newSelectionSet, newFragments) {
         if (this.selectionSet === newSelectionSet && newFragments === this.fragments) {
             return this;
         }
-        return new Operation(this.schema, this.rootKind, newSelectionSet, this.variableDefinitions, newFragments, this.name);
+        return new Operation(this.schema, this.rootKind, newSelectionSet, this.variableDefinitions, newFragments, this.name, this.directives);
     }
     optimize(fragments, minUsagesToOptimize = 2) {
         (0, utils_1.assert)(minUsagesToOptimize >= 1, `Expected 'minUsagesToOptimize' to be at least 1, but got ${minUsagesToOptimize}`);
@@ -582,6 +583,10 @@ class Operation {
         }
         return this.withUpdatedSelectionSetAndFragments(optimizedSelection, finalFragments !== null && finalFragments !== void 0 ? finalFragments : undefined);
     }
+    generateQueryFragments() {
+        const [minimizedSelectionSet, fragments] = this.selectionSet.minimizeSelectionSet();
+        return new Operation(this.schema, this.rootKind, minimizedSelectionSet, this.variableDefinitions, fragments, this.name, this.directives);
+    }
     expandAllFragments() {
         const expanded = this.selectionSet.expandFragments();
         return this.withUpdatedSelectionSetAndFragments(expanded.normalize({ parentType: expanded.parentType }), undefined);
@@ -616,7 +621,7 @@ class Operation {
         return defaultedVariableValues;
     }
     toString(expandFragments = false, prettyPrint = true) {
-        return this.selectionSet.toOperationString(this.rootKind, this.variableDefinitions, this.fragments, this.name, expandFragments, prettyPrint);
+        return this.selectionSet.toOperationString(this.rootKind, this.variableDefinitions, this.fragments, this.name, this.directives, expandFragments, prettyPrint);
     }
 }
 exports.Operation = Operation;
@@ -703,6 +708,10 @@ class NamedFragmentDefinition extends definitions_1.DirectiveTargetElement {
     computeExpandedSelectionSetAtType(type) {
         const expandedSelectionSet = this.expandedSelectionSet();
         const selectionSet = expandedSelectionSet.normalize({ parentType: type });
+        if (!(0, definitions_1.isObjectType)(this.typeCondition)) {
+            const validator = FieldsConflictValidator.build(expandedSelectionSet);
+            return { selectionSet, validator };
+        }
         const trimmed = expandedSelectionSet.minus(selectionSet);
         const validator = trimmed.isEmpty() ? undefined : FieldsConflictValidator.build(trimmed);
         return { selectionSet, validator };
@@ -815,17 +824,6 @@ class NamedFragments {
             return fragment.withUpdatedSelectionSet(reoptimizedSelectionSet);
         });
     }
-    selectionSetIsWorthUsing(selectionSet) {
-        const selections = selectionSet.selections();
-        if (selections.length === 0) {
-            return false;
-        }
-        if (selections.length === 1) {
-            const s = selections[0];
-            return !(s.kind === 'FieldSelection' && s.element.isLeafField());
-        }
-        return true;
-    }
     rebaseOn(schema) {
         return this.mapInDependencyOrder((fragment, newFragments) => {
             const rebasedType = schema.type(fragment.selectionSet.parentType.name);
@@ -834,7 +832,7 @@ class NamedFragments {
             }
             let rebasedSelection = fragment.selectionSet.rebaseOn({ parentType: rebasedType, fragments: newFragments, errorIfCannotRebase: false });
             rebasedSelection = rebasedSelection.normalize({ parentType: rebasedType });
-            return this.selectionSetIsWorthUsing(rebasedSelection)
+            return rebasedSelection.isWorthUsing()
                 ? new NamedFragmentDefinition(schema, fragment.name, rebasedType).setSelectionSet(rebasedSelection)
                 : undefined;
         });
@@ -923,6 +921,39 @@ class SelectionSet {
         this._keyedSelections = keyedSelections;
         this._selections = (0, utils_1.mapValues)(keyedSelections);
     }
+    minimizeSelectionSet(namedFragments = new NamedFragments(), seenSelections = new Map()) {
+        const minimizedSelectionSet = this.lazyMap((selection) => {
+            var _a;
+            if (selection.kind === 'FragmentSelection' && selection.element.typeCondition && selection.element.appliedDirectives.length === 0
+                && selection.selectionSet && selection.selectionSet.isWorthUsing()) {
+                const mockHashCode = `on${selection.element.typeCondition}` + selection.selectionSet.selections().length;
+                const equivalentSelectionSetCandidates = seenSelections.get(mockHashCode);
+                if (equivalentSelectionSetCandidates) {
+                    const match = equivalentSelectionSetCandidates.find(([candidateSet]) => candidateSet.equals(selection.selectionSet));
+                    if (match) {
+                        return new FragmentSpreadSelection(this.parentType, namedFragments, match[1], []);
+                    }
+                }
+                const [minimizedSelectionSet] = selection.selectionSet.minimizeSelectionSet(namedFragments, seenSelections);
+                const fragmentDefinition = new NamedFragmentDefinition(this.parentType.schema(), `_generated_${mockHashCode}_${(_a = equivalentSelectionSetCandidates === null || equivalentSelectionSetCandidates === void 0 ? void 0 : equivalentSelectionSetCandidates.length) !== null && _a !== void 0 ? _a : 0}`, selection.element.typeCondition).setSelectionSet(minimizedSelectionSet);
+                if (!equivalentSelectionSetCandidates) {
+                    seenSelections.set(mockHashCode, [[selection.selectionSet, fragmentDefinition]]);
+                    namedFragments.add(fragmentDefinition);
+                }
+                else {
+                    equivalentSelectionSetCandidates.push([selection.selectionSet, fragmentDefinition]);
+                }
+                return new FragmentSpreadSelection(this.parentType, namedFragments, fragmentDefinition, []);
+            }
+            else if (selection.kind === 'FieldSelection') {
+                if (selection.selectionSet) {
+                    selection = selection.withUpdatedSelectionSet(selection.selectionSet.minimizeSelectionSet(namedFragments, seenSelections)[0]);
+                }
+            }
+            return selection;
+        });
+        return [minimizedSelectionSet, namedFragments];
+    }
     selectionsInReverseOrder() {
         const length = this._selections.length;
         const reversed = new Array(length);
@@ -1219,7 +1250,7 @@ class SelectionSet {
         }
         return false;
     }
-    toOperationString(rootKind, variableDefinitions, fragments, operationName, expandFragments = false, prettyPrint = true) {
+    toOperationString(rootKind, variableDefinitions, fragments, operationName, directives, expandFragments = false, prettyPrint = true) {
         const indent = prettyPrint ? '' : undefined;
         const fragmentsDefinitions = !expandFragments && fragments && !fragments.isEmpty()
             ? fragments.toString(indent) + "\n\n"
@@ -1230,7 +1261,8 @@ class SelectionSet {
         const nameAndVariables = operationName
             ? " " + (operationName + (variableDefinitions.isEmpty() ? "" : variableDefinitions.toString()))
             : (variableDefinitions.isEmpty() ? "" : " " + variableDefinitions.toString());
-        return fragmentsDefinitions + rootKind + nameAndVariables + " " + this.toString(expandFragments, true, indent);
+        const directives_str = (0, definitions_1.directivesToString)(directives);
+        return fragmentsDefinitions + rootKind + nameAndVariables + directives_str + " " + this.toString(expandFragments, true, indent);
     }
     toString(expandFragments = true, includeExternalBrackets = true, indent) {
         if (this.isEmpty()) {
@@ -1248,6 +1280,17 @@ class SelectionSet {
                 : selectionsToString;
         }
     }
+    isWorthUsing() {
+        const selections = this.selections();
+        if (selections.length === 0) {
+            return false;
+        }
+        if (selections.length === 1) {
+            const s = selections[0];
+            return !(s.kind === 'FieldSelection' && s.element.isLeafField());
+        }
+        return true;
+    }
 }
 exports.SelectionSet = SelectionSet;
 class SelectionSetUpdates {
@@ -2163,7 +2206,7 @@ class FragmentSpreadSelection extends FragmentSelection {
     }
     key() {
         if (!this.computedKey) {
-            this.computedKey = '...' + this.namedFragment.name + (this.spreadDirectives.length === 0 ? '' : ' ' + this.spreadDirectives.join(' '));
+            this.computedKey = '...' + this.namedFragment.name + (0, definitions_1.directivesToString)(this.spreadDirectives);
         }
         return this.computedKey;
     }
@@ -2179,18 +2222,7 @@ class FragmentSpreadSelection extends FragmentSelection {
         validate((0, definitions_1.runtimeTypesIntersects)(this.parentType, this.namedFragment.typeCondition), () => `Fragment "${this.namedFragment.name}" cannot be spread inside type ${this.parentType} as the runtime types do not intersect ${this.namedFragment.typeCondition}`);
     }
     toSelectionNode() {
-        const directiveNodes = this.spreadDirectives.length === 0
-            ? undefined
-            : this.spreadDirectives.map(directive => {
-                return {
-                    kind: graphql_1.Kind.DIRECTIVE,
-                    name: {
-                        kind: graphql_1.Kind.NAME,
-                        value: directive.name,
-                    },
-                    arguments: directive.argumentsToAST()
-                };
-            });
+        const directiveNodes = (0, definitions_1.directivesToDirectiveNodes)(this.spreadDirectives);
         return {
             kind: graphql_1.Kind.FRAGMENT_SPREAD,
             name: { kind: graphql_1.Kind.NAME, value: this.namedFragment.name },
@@ -2266,9 +2298,7 @@ class FragmentSpreadSelection extends FragmentSelection {
             return (indent !== null && indent !== void 0 ? indent : '') + this.element + ' ' + this.selectionSet.toString(true, true, indent);
         }
         else {
-            const directives = this.spreadDirectives;
-            const directiveString = directives.length == 0 ? '' : ' ' + directives.join(' ');
-            return (indent !== null && indent !== void 0 ? indent : '') + '...' + this.namedFragment.name + directiveString;
+            return (indent !== null && indent !== void 0 ? indent : '') + '...' + this.namedFragment.name + (0, definitions_1.directivesToString)(this.spreadDirectives);
         }
     }
 }
@@ -2321,6 +2351,7 @@ function selectionOfNode(parentType, node, variableDefinitions, fragments, field
 }
 function operationFromDocument(schema, document, options) {
     let operation;
+    let operation_directives;
     const operationName = options === null || options === void 0 ? void 0 : options.operationName;
     const fragments = new NamedFragments();
     document.definitions.forEach(definition => {
@@ -2329,6 +2360,7 @@ function operationFromDocument(schema, document, options) {
                 validate(!operation || operationName, () => 'Must provide operation name if query contains multiple operations.');
                 if (!operationName || (definition.name && definition.name.value === operationName)) {
                     operation = definition;
+                    operation_directives = directivesOfNodes(schema, definition.directives);
                 }
                 break;
             case graphql_1.Kind.FRAGMENT_DEFINITION:
@@ -2358,10 +2390,10 @@ function operationFromDocument(schema, document, options) {
         }
     });
     fragments.validate(variableDefinitions);
-    return operationFromAST({ schema, operation, variableDefinitions, fragments, validateInput: options === null || options === void 0 ? void 0 : options.validate });
+    return operationFromAST({ schema, operation, operation_directives, variableDefinitions, fragments, validateInput: options === null || options === void 0 ? void 0 : options.validate });
 }
 exports.operationFromDocument = operationFromDocument;
-function operationFromAST({ schema, operation, variableDefinitions, fragments, validateInput, }) {
+function operationFromAST({ schema, operation, operation_directives, variableDefinitions, fragments, validateInput, }) {
     var _a;
     const rootType = schema.schemaDefinition.root(operation.operation);
     validate(rootType, () => `The schema has no "${operation.operation}" root type defined`);
@@ -2372,7 +2404,7 @@ function operationFromAST({ schema, operation, variableDefinitions, fragments, v
         variableDefinitions,
         fragments: fragmentsIfAny,
         validate: validateInput,
-    }), variableDefinitions, fragmentsIfAny, (_a = operation.name) === null || _a === void 0 ? void 0 : _a.value);
+    }), variableDefinitions, fragmentsIfAny, (_a = operation.name) === null || _a === void 0 ? void 0 : _a.value, operation_directives);
 }
 function parseOperation(schema, operation, options) {
     return operationFromDocument(schema, (0, graphql_1.parse)(operation), options);
@@ -2403,6 +2435,7 @@ function operationToDocument(operation) {
         name: operation.name ? { kind: graphql_1.Kind.NAME, value: operation.name } : undefined,
         selectionSet: operation.selectionSet.toSelectionSetNode(),
         variableDefinitions: operation.variableDefinitions.toVariableDefinitionNodes(),
+        directives: (0, definitions_1.directivesToDirectiveNodes)(operation.directives),
     };
     const fragmentASTs = operation.fragments
         ? (_a = operation.fragments) === null || _a === void 0 ? void 0 : _a.toFragmentDefinitionNodes()