| import { Kind } from '../language/kinds.mjs'; |
| /** |
| * Returns an operation AST given a document AST and optionally an operation |
| * name. If a name is not provided, an operation is only returned if only one is |
| * provided in the document. |
| */ |
| |
| export function getOperationAST(documentAST, operationName) { |
| let operation = null; |
| |
| for (const definition of documentAST.definitions) { |
| if (definition.kind === Kind.OPERATION_DEFINITION) { |
| var _definition$name; |
| |
| if (operationName == null) { |
| // If no operation name was provided, only return an Operation if there |
| // is one defined in the document. Upon encountering the second, return |
| // null. |
| if (operation) { |
| return null; |
| } |
| |
| operation = definition; |
| } else if ( |
| ((_definition$name = definition.name) === null || |
| _definition$name === void 0 |
| ? void 0 |
| : _definition$name.value) === operationName |
| ) { |
| return definition; |
| } |
| } |
| } |
| |
| return operation; |
| } |