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_COMPILER_LOWERING_LOCAL_CLASS_LOWERING_H
17 #define ES2PANDA_COMPILER_LOWERING_LOCAL_CLASS_LOWERING_H
18 
19 #include "compiler/lowering/phase.h"
20 
21 namespace ark::es2panda::compiler {
22 
23 class LocalClassConstructionPhase : public Phase {
24 public:
25     std::string_view Name() const override;
26     bool Perform(public_lib::Context *ctx, parser::Program *program) override;
27 
28 protected:
29     void CreateClassPropertiesForCapturedVariables(public_lib::Context *ctx, ir::ClassDefinition *classDef,
30                                                    ArenaSet<varbinder::Variable *> const &capturedVars,
31                                                    ArenaMap<varbinder::Variable *, varbinder::Variable *> &variableMap,
32                                                    ArenaMap<varbinder::Variable *, ir::ClassProperty *> &propertyMap);
33 
34     void ModifyConstructorParameters(public_lib::Context *ctx, ir::ClassDefinition *classDef,
35                                      ArenaSet<varbinder::Variable *> const &capturedVars,
36                                      ArenaMap<varbinder::Variable *, varbinder::Variable *> &variableMap,
37                                      ArenaMap<varbinder::Variable *, varbinder::Variable *> &parameterMap);
38 
39     void RemapReferencesFromCapturedVariablesToClassProperties(
40         ir::ClassDefinition *classDef, ArenaMap<varbinder::Variable *, varbinder::Variable *> &variableMap);
41 
42     ir::ETSParameterExpression *CreateParam(checker::ETSChecker *const checker, varbinder::FunctionParamScope *scope,
43                                             util::StringView name, checker::Type *type);
44 };
45 
46 }  // namespace ark::es2panda::compiler
47 
48 #endif
49