1/*
2 * Copyright (c) 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_GENERIC_BRIDGES_LOWERING_H
17#define ES2PANDA_GENERIC_BRIDGES_LOWERING_H
18
19#include "compiler/lowering/phase.h"
20
21namespace ark::es2panda::compiler {
22
23class GenericBridgesPhase : public Phase {
24public:
25    std::string_view Name() const override
26    {
27        return "CreateGenericBridges";
28    }
29
30    bool Perform(public_lib::Context *ctx, parser::Program *program) override;
31
32private:
33    struct Substitutions {
34        checker::Substitution *derivedSubstitutions = nullptr;
35        checker::Substitution *baseConstraints = nullptr;
36        checker::Substitution *derivedConstraints = nullptr;
37    };
38
39    ir::ClassDefinition *ProcessClassDefinition(ir::ClassDefinition *classDefinition) const;
40
41    void CreateGenericBridges(ir::ClassDefinition const *classDefinition, Substitutions &substitutions) const;
42
43    void MaybeAddGenericBridges(ir::ClassDefinition const *classDefinition, ir::MethodDefinition *baseMethod,
44                                ir::MethodDefinition *derivedMethod, Substitutions const &substitutions) const;
45
46    void ProcessScriptFunction(ir::ClassDefinition const *classDefinition, ir::ScriptFunction *baseFunction,
47                               ir::MethodDefinition *derivedMethod, Substitutions const &substitutions) const;
48
49    void AddGenericBridge(ir::ClassDefinition const *classDefinition, ir::MethodDefinition *methodDefinition,
50                          checker::Signature const *baseSignature, ir::ScriptFunction const *derivedFunction) const;
51
52    std::string CreateMethodDefinitionString(ir::ClassDefinition const *classDefinition,
53                                             checker::Signature const *baseSignature,
54                                             ir::ScriptFunction const *derivedFunction,
55                                             std::vector<ir::AstNode *> &typeNodes) const noexcept;
56
57    public_lib::Context *context_ = nullptr;
58};
59}  // namespace ark::es2panda::compiler
60
61#endif
62