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
import { Importer } from './Importer'
19
import { IssueTable, adaptorComponentName } from './utils'
20
import { id } from './ApiUtils'
21

22

23
export class NewTransformer extends AbstractVisitor {
24

25
    private static constructableEts = ['Path', 'Rect', 'Circle', 'Ellipse', 'Line', 'Polygon', 'Polyline', 'Shape']
26

27
    constructor(
28
        sourceFile: ts.SourceFile,
29
        ctx: ts.TransformationContext,
30
        private importer: Importer,
31
        private issueTable: IssueTable
32
    ) {
33
        super(sourceFile, ctx)
34
    }
35

36
    visitor(beforeChildren: ts.Node): ts.Node {
37
        const node = this.visitEachChild(beforeChildren)
38

39
        if (ts.isNewExpression(node)) {
40
            const constructible = node.expression
41
            if (ts.isIdentifier(constructible)) {
42
                const component = ts.idText(constructible)
43
                if (NewTransformer.constructableEts.includes(component)) {
44
                    const arkComponent = adaptorComponentName(component)
45
                    this.importer.addAdaptorImport(arkComponent)
46

47
                    return ts.factory.updateNewExpression(
48
                        node,
49
                        id(arkComponent),
50
                        node.typeArguments,
51
                        node.arguments,
52
                    )
53
                }
54
            }
55
        }
56

57
        return node
58
    }
59
}
60