Function bindPrecompilation

ark/schema/scope.ts:182–211

(
	references: readonly BaseNode[],
	precompiler: CompiledFunction<() => PrecompiledReferences>,
	owningScope?: BaseScope
)

	bindPrecompilation(references, precompileReferences(references))

const bindPrecompilation = (
	references: readonly BaseNode[],
	precompiler: CompiledFunction<() => PrecompiledReferences>,
	owningScope?: BaseScope
): void => {
	const precompilation = precompiler.write(rootScopeFnName, 4)
	const compiledTraversals = precompiler.compile()()

	for (const node of references) {
		if (node.precompilation && (!owningScope || node.$ !== owningScope)) {
			// if node has already been bound to another scope or anonymous type, don't rebind it
			continue
		}
		node.traverseAllows =
			compiledTraversals[`${node.id}Allows`].bind(compiledTraversals)
		if (node.isRoot() && !node.allowsRequiresContext) {
			// if the reference doesn't require context, we can assign over
			// it directly to avoid having to initialize it
			node.allows = node.traverseAllows as never
		}
		node.traverseApply =
			compiledTraversals[`${node.id}Apply`].bind(compiledTraversals)

		if (compiledTraversals[`${node.id}Optimistic`]) {
			;(node as UnionNode).traverseOptimistic =
				compiledTraversals[`${node.id}Optimistic`].bind(compiledTraversals)
		}
		node.precompilation = precompilation
	}
}

export type PrecompiledReferences = {
	[k: `${string}Allows`]: TraverseAllows