1/*
2 * Copyright (c) 2022-2024 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16import * as ts from 'typescript';
17
18export function forEachNodeInSubtree(
19  node: ts.Node,
20  cb: (n: ts.Node) => void,
21  stopCond?: (n: ts.Node) => boolean
22): void {
23  cb(node);
24  if (stopCond?.(node)) {
25    return;
26  }
27
28  ts.forEachChild(node, (child) => {
29    forEachNodeInSubtree(child, cb, stopCond);
30  });
31}
32