Function findNodes

packages/schematics/angular/utility/ast-utils.ts:130–183

(
  node: ts.Node,
  kindOrGuard: ts.SyntaxKind | ((node: ts.Node) => node is T),
  max = Infinity,
  recursive = false,
)

): T[];

export function findNodes<T extends ts.Node>(
  node: ts.Node,
  kindOrGuard: ts.SyntaxKind | ((node: ts.Node) => node is T),
  max = Infinity,
  recursive = false,
): T[] {
  if (!node || max == 0) {
    return [];
  }

  const test =
    typeof kindOrGuard === 'function'
      ? kindOrGuard
      : (node: ts.Node): node is T => node.kind === kindOrGuard;

  // Caching is only supported for the entire file
  if (ts.isSourceFile(node)) {
    const sourceFileCache = findNodesCache.get(node);
    if (sourceFileCache?.has(kindOrGuard)) {
      return sourceFileCache.get(kindOrGuard) as T[];
    }
  }

  const arr: T[] = [];
  if (test(node)) {
    arr.push(node);
    max--;
  }
  if (max > 0 && (recursive || !test(node))) {
    for (const child of node.getChildren()) {
      findNodes(child, test, max, recursive).forEach((node) => {
        if (max > 0) {
          arr.push(node);
        }
        max--;
      });

      if (max <= 0) {
        break;
      }
    }
  }

  if (ts.isSourceFile(node)) {
    let sourceFileCache = findNodesCache.get(node);
    if (!sourceFileCache) {
      sourceFileCache = new Map();
      findNodesCache.set(node, sourceFileCache);
    }
    sourceFileCache.set(kindOrGuard, arr);
  }

  return arr;
}

/**
 * Get all the nodes from a source.