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 
16 #ifndef ES2PANDA_IR_MODULE_EXPORT_DECLARATION_H
17 #define ES2PANDA_IR_MODULE_EXPORT_DECLARATION_H
18 
19 #include "ir/statement.h"
20 
21 namespace ark::es2panda::ir {
22 class StringLiteral;
23 class ExportSpecifier;
24 
25 class ExportNamedDeclaration : public Statement {
26 public:
ExportNamedDeclaration(ArenaAllocator *allocator, StringLiteral *source, ArenaVector<ExportSpecifier *> &&specifiers)27     explicit ExportNamedDeclaration(ArenaAllocator *allocator, StringLiteral *source,
28                                     ArenaVector<ExportSpecifier *> &&specifiers)
29         : Statement(AstNodeType::EXPORT_NAMED_DECLARATION),
30           decorators_(allocator->Adapter()),
31           source_(source),
32           specifiers_(std::move(specifiers))
33     {
34     }
35 
ExportNamedDeclaration(ArenaAllocator *allocator, AstNode *decl, ArenaVector<ExportSpecifier *> &&specifiers)36     explicit ExportNamedDeclaration(ArenaAllocator *allocator, AstNode *decl,
37                                     ArenaVector<ExportSpecifier *> &&specifiers)
38         : Statement(AstNodeType::EXPORT_NAMED_DECLARATION),
39           decorators_(allocator->Adapter()),
40           decl_(decl),
41           specifiers_(std::move(specifiers))
42     {
43     }
44 
ExportNamedDeclaration(ArenaAllocator *allocator, AstNode *decl)45     explicit ExportNamedDeclaration(ArenaAllocator *allocator, AstNode *decl)
46         : Statement(AstNodeType::EXPORT_NAMED_DECLARATION),
47           decorators_(allocator->Adapter()),
48           decl_(decl),
49           specifiers_(allocator->Adapter())
50     {
51     }
52 
Decl() const53     const AstNode *Decl() const
54     {
55         return decl_;
56     }
57 
Source() const58     const StringLiteral *Source() const
59     {
60         return source_;
61     }
62 
Specifiers() const63     const ArenaVector<ExportSpecifier *> &Specifiers() const
64     {
65         return specifiers_;
66     }
67 
68     void AddDecorators([[maybe_unused]] ArenaVector<ir::Decorator *> &&decorators) override
69     {
70         decorators_ = std::move(decorators);
71     }
72 
73     bool CanHaveDecorator([[maybe_unused]] bool inTs) const override
74     {
75         return !inTs && (source_ == nullptr);
76     }
77 
78     void TransformChildren(const NodeTransformer &cb, std::string_view transformationName) override;
79     void Iterate(const NodeTraverser &cb) const override;
80     void Dump(ir::AstDumper *dumper) const override;
81     void Dump(ir::SrcDumper *dumper) const override;
82     void Compile(compiler::PandaGen *pg) const override;
83     void Compile(compiler::ETSGen *etsg) const override;
84     checker::Type *Check(checker::TSChecker *checker) override;
85     checker::Type *Check(checker::ETSChecker *checker) override;
86 
87     void Accept(ASTVisitorT *v) override
88     {
89         v->Accept(this);
90     }
91 
92 private:
93     ArenaVector<Decorator *> decorators_;
94     StringLiteral *source_ {};
95     AstNode *decl_ {};
96     ArenaVector<ExportSpecifier *> specifiers_;
97 };
98 }  // namespace ark::es2panda::ir
99 
100 #endif
101