eslint-plugin-unicorn/rules/utils/contains-suspension-point.js

More than 300 powerful ESLint rules

import {isFunction} from '../ast/index.js';
import getVisitorChildNodes from './get-visitor-child-nodes.js';

/**
Check whether `node` or any of its descendants, excluding nested functions, is a suspension point (`await`, `for await…of`, or `yield`).

Nested functions are not descended into, since their suspension points belong to a different function.

@param {import('estree').Node} node
@param {import('eslint').SourceCode['visitorKeys']} visitorKeys
@returns {boolean}
*/
export default function containsSuspensionPoint(node, visitorKeys) {
	if (
		node.type === 'AwaitExpression'
		|| node.type === 'YieldExpression'
		|| (node.type === 'ForOfStatement' && node.await)
	) {
		return true;
	}

	if (isFunction(node)) {
		return false;
	}

	return getVisitorChildNodes(node, visitorKeys).some(child => containsSuspensionPoint(child, visitorKeys));
}