1/**
2 * Copyright (c) 2023-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
16#ifndef ES2PANDA_IR_ETS_TUPLE_H
17#define ES2PANDA_IR_ETS_TUPLE_H
18
19#include "checker/ETSchecker.h"
20#include "ir/typeNode.h"
21
22namespace ark::es2panda::ir {
23
24class ETSTuple : public TypeNode {
25public:
26    using TupleSizeType = uint32_t;
27
28    explicit ETSTuple(ArenaAllocator *const allocator)
29        : TypeNode(AstNodeType::ETS_TUPLE), typeAnnotationList_(allocator->Adapter())
30    {
31    }
32
33    explicit ETSTuple(ArenaAllocator *const allocator, const TupleSizeType size)
34        : TypeNode(AstNodeType::ETS_TUPLE), typeAnnotationList_(allocator->Adapter()), size_(size)
35    {
36    }
37    explicit ETSTuple(const ArenaVector<ir::TypeNode *> &typeList)
38        : TypeNode(AstNodeType::ETS_TUPLE),
39          typeAnnotationList_(typeList),
40          size_(static_cast<TupleSizeType>(typeList.size()))
41    {
42    }
43
44    [[nodiscard]] TupleSizeType GetTupleSize() const
45    {
46        return size_;
47    }
48
49    [[nodiscard]] ArenaVector<ir::TypeNode *> GetTupleTypeAnnotationsList() const
50    {
51        return typeAnnotationList_;
52    }
53
54    [[nodiscard]] bool HasSpreadType() const
55    {
56        return spreadType_ != nullptr;
57    }
58
59    void SetSpreadType(TypeNode *const newSpreadType)
60    {
61        spreadType_ = newSpreadType;
62    }
63
64    void SetTypeAnnotationsList(const ArenaVector<TypeNode *> &typeNodeList)
65    {
66        typeAnnotationList_ = typeNodeList;
67    }
68
69    void TransformChildren(const NodeTransformer &cb, std::string_view transformationName) override;
70    void Iterate(const NodeTraverser &cb) const override;
71    void Dump(ir::AstDumper *dumper) const override;
72    void Dump(ir::SrcDumper *dumper) const override;
73    void Compile([[maybe_unused]] compiler::PandaGen *pg) const override;
74    void Compile([[maybe_unused]] compiler::ETSGen *etsg) const override;
75    checker::Type *Check([[maybe_unused]] checker::TSChecker *checker) override;
76    checker::Type *Check([[maybe_unused]] checker::ETSChecker *checker) override;
77    checker::Type *GetType([[maybe_unused]] checker::ETSChecker *checker) override;
78
79    void Accept(ASTVisitorT *v) override
80    {
81        v->Accept(this);
82    }
83
84    // NOTE(vpukhov): hide in TypeCreation
85    static checker::Type *CalculateLUBForTuple(checker::ETSChecker *checker, ArenaVector<checker::Type *> &typeList,
86                                               checker::Type **spreadTypePtr);
87
88private:
89    ArenaVector<TypeNode *> typeAnnotationList_;
90    TypeNode *spreadType_ {};
91    TupleSizeType size_ {0};
92};
93
94}  // namespace ark::es2panda::ir
95
96#endif /* ES2PANDA_IR_ETS_TUPLE_H */
97