Function getAsyncAndAwaitOccurrences

internal/ls/documenthighlights.go:619–646

(node *ast.Node, sourceFile *ast.SourceFile)

}

func getAsyncAndAwaitOccurrences(node *ast.Node, sourceFile *ast.SourceFile) []*ast.Node {
	fun := ast.GetContainingFunction(node)
	if fun == nil {
		return nil
	}

	var keywords []*ast.Node

	for _, modifier := range fun.ModifierNodes() {
		if modifier.Kind == ast.KindAsyncKeyword {
			keywords = append(keywords, modifier)
		}
	}

	fun.ForEachChild(func(child *ast.Node) bool {
		traverseWithoutCrossingFunction(child, sourceFile, func(child *ast.Node) {
			if ast.IsAwaitExpression(child) {
				token := lsutil.GetFirstToken(child, sourceFile)
				if token.Kind == ast.KindAwaitKeyword {
					keywords = append(keywords, token)
				}
			}
		})
		return false // continue traversal
	})

	return keywords
}

func getYieldOccurrences(node *ast.Node, sourceFile *ast.SourceFile) []*ast.Node {
	parentFunc := ast.FindAncestor(node.Parent, ast.IsFunctionLike)