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 'typescript'
17
import { Rewrite } from './transformation-context';
18
import { AbstractVisitor } from './AbstractVisitor'
19
import { FunctionKind, isAnyMemoKind } from './util';
20

21
export class MemoArgumentDetector extends AbstractVisitor {
22
    constructor(
23
        public typechecker: ts.TypeChecker,
24
        public rewrite: Rewrite,
25
        public memoArgument: ts.Node,
26
        ctx: ts.TransformationContext,
27
    ) {
28
        super(ctx)
29
    }    
30
    hasMemoArgument : boolean = false
31
    
32

33
    visitor(node: ts.Node): ts.Node {
34

35
        if(ts.isCallExpression(node)){
36
            const callKind = this.rewrite.callTable.get(node) ?? FunctionKind.REGULAR
37
            if(isAnyMemoKind(callKind)) {
38
                this.hasMemoArgument = true
39
                this.memoArgument = node
40
            }
41
        }
42
        return this.visitEachChild(node)
43
    }
44

45
    usingMemoFunction(node: ts.Expression): boolean {
46
        this.hasMemoArgument = false
47
        this.visitor(node)
48
        return this.hasMemoArgument
49
    }    
50
}