import type { Maybe } from '../jsutils/Maybe.js';
import type { DocumentNode, OperationDefinitionNode } from '../language/ast.js';
import { Kind } from '../language/kinds.js';
export function getOperationAST(
documentAST: DocumentNode,
operationName?: Maybe<string>,
): Maybe<OperationDefinitionNode> {
let operation = null;
for (const definition of documentAST.definitions) {
if (definition.kind === Kind.OPERATION_DEFINITION) {
if (operationName == null) {
if (operation) {
return null;
}
operation = definition;
} else if (definition.name?.value === operationName) {
return definition;
}
}
}
return operation;
}