idlize

1
/*
2
 * Copyright (c) 2022-2023 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

16
import * as ts from 'ohos-typescript'
17
import { AbstractVisitor } from './AbstractVisitor'
18

19
export class CallExpressionCollector extends AbstractVisitor {
20
    private readonly map = new Map<string, boolean>()
21

22
    constructor(sourceFile: ts.SourceFile, ctx: ts.TransformationContext, ...names: string[]) {
23
        super(sourceFile, ctx)
24
        for (const name of names) {
25
            this.map.set(name, false)
26
        }
27
    }
28

29
    isVisited(name: string): boolean {
30
        return this.map.get(name) === true
31
    }
32

33
    visitor(beforeChildren: ts.Node): ts.Node {
34
        const node = this.visitEachChild(beforeChildren)
35
        if (ts.isCallExpression(node)) {
36
            const expression = node.expression
37
            if (ts.isIdentifier(expression)) {
38
                const name = ts.idText(expression)
39
                const exist = this.map.get(name)
40
                if (!exist) this.map.set(name, true)
41
            }
42
        }
43
        return node
44
    }
45
}
46