import type { ObjMap } from '../jsutils/ObjMap.js';
import type {
DocumentNode,
OperationDefinitionNode,
SelectionSetNode,
} from '../language/ast.js';
import { Kind } from '../language/kinds.js';
import { visit } from '../language/visitor.js';
export function separateOperations(
documentAST: DocumentNode,
): ObjMap<DocumentNode> {
const operations: Array<OperationDefinitionNode> = [];
const depGraph: DepGraph = Object.create(null);
for (const definitionNode of documentAST.definitions) {
switch (definitionNode.kind) {
case Kind.OPERATION_DEFINITION:
operations.push(definitionNode);
break;
case Kind.FRAGMENT_DEFINITION:
depGraph[definitionNode.name.value] = collectDependencies(
definitionNode.selectionSet,
);
break;
default:
}
}
const separatedDocumentASTs = Object.create(null);
for (const operation of operations) {
const dependencies = new Set<string>();
for (const fragmentName of collectDependencies(operation.selectionSet)) {
collectTransitiveDependencies(dependencies, depGraph, fragmentName);
}
const operationName = operation.name ? operation.name.value : '';
separatedDocumentASTs[operationName] = {
kind: Kind.DOCUMENT,
definitions: documentAST.definitions.filter(
(node) =>
node === operation ||
(node.kind === Kind.FRAGMENT_DEFINITION &&
dependencies.has(node.name.value)),
),
};
}
return separatedDocumentASTs;
}
type DepGraph = ObjMap<ReadonlyArray<string>>;
function collectTransitiveDependencies(
collected: Set<string>,
depGraph: DepGraph,
fromName: string,
): void {
if (!collected.has(fromName)) {
collected.add(fromName);
const immediateDeps = depGraph[fromName];
if (immediateDeps !== undefined) {
for (const toName of immediateDeps) {
collectTransitiveDependencies(collected, depGraph, toName);
}
}
}
}
function collectDependencies(selectionSet: SelectionSetNode): Array<string> {
const dependencies: Array<string> = [];
visit(selectionSet, {
FragmentSpread(node) {
dependencies.push(node.name.value);
},
});
return dependencies;
}