1/*
2 * Copyright (c) 2021 - 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#ifndef ES2PANDA_CHECKER_TSANALYZER_H
16#define ES2PANDA_CHECKER_TSANALYZER_H
17
18#include "checker/SemanticAnalyzer.h"
19#include "checker/TSchecker.h"
20#include "util/helpers.h"
21
22namespace ark::es2panda::checker {
23
24class TSAnalyzer final : public SemanticAnalyzer {
25public:
26    explicit TSAnalyzer(Checker *checker) : SemanticAnalyzer(checker) {};
27
28    // NOLINTNEXTLINE(cppcoreguidelines-macro-usage)
29#define DECLARE_TSANALYZER_CHECK_METHOD(_, nodeType) checker::Type *Check(ir::nodeType *node) const override;
30    AST_NODE_MAPPING(DECLARE_TSANALYZER_CHECK_METHOD)
31#undef DECLARE_TSANALYZER_CHECK_METHOD
32
33// NOLINTNEXTLINE(cppcoreguidelines-macro-usage)
34#define DECLARE_TSANALYZER_CHECK_METHOD(_, __, nodeType, ___) \
35    virtual checker::Type *Check(ir::nodeType *node) const override;
36    AST_NODE_REINTERPRET_MAPPING(DECLARE_TSANALYZER_CHECK_METHOD)
37#undef DECLARE_TSANALYZER_CHECK_METHOD
38
39    checker::Type *CheckDeleteKeyword([[maybe_unused]] checker::TSChecker *checker, ir::UnaryExpression *expr) const;
40    checker::Type *CheckLiteral([[maybe_unused]] checker::TSChecker *checker, ir::UnaryExpression *expr) const;
41    checker::Type *CheckAssignmentExprOperatorType(ir::AssignmentExpression *expr, checker::Type *leftType,
42                                                   checker::Type *rightType) const;
43    checker::Type *CheckBinaryExprArithmLogical(ir::BinaryExpression *expr, ExpressionTypeInfo *leftRightType,
44                                                TSChecker *checker) const;
45    void CheckComputed(ir::MemberExpression *expr, checker::Type *indexType) const;
46    void CheckSpread(std::unordered_map<util::StringView, lexer::SourcePosition> &allPropertiesMap,
47                     checker::ObjectDescriptor *desc, ir::Expression *it) const;
48    checker::IndexInfo *CreateUnionTypeHelper(ArenaVector<checker::Type *> &computedPropTypes,
49                                              bool inConstContext) const;
50    void CheckNonComputed(checker::ObjectDescriptor *desc, ir::Expression *it,
51                          std::unordered_map<util::StringView, lexer::SourcePosition> &allPropertiesMap,
52                          bool inConstContext) const;
53
54private:
55    TSChecker *GetTSChecker() const;
56
57    varbinder::EnumMemberResult EvaluateBinaryExpression(checker::TSChecker *checker, varbinder::EnumVariable *enumVar,
58                                                         const ir::BinaryExpression *expr) const;
59    varbinder::EnumMemberResult EvaluateEnumMember(checker::TSChecker *checker, varbinder::EnumVariable *enumVar,
60                                                   const ir::AstNode *expr) const;
61    varbinder::EnumMemberResult EvaluateUnaryExpression(checker::TSChecker *checker, varbinder::EnumVariable *enumVar,
62                                                        const ir::UnaryExpression *expr) const;
63    void InferEnumVariableType(varbinder::EnumVariable *variable, double *value, bool *initNext, bool *isLiteralEnum,
64                               bool isConstEnum) const;
65    checker::Type *InferType(checker::TSChecker *checker, bool isConst, ir::TSEnumDeclaration *st) const;
66};
67
68}  // namespace ark::es2panda::checker
69
70#endif  // ES2PANDA_CHECKER_TSSANALYZER_H
71