/*
 * Copyright (c) 2021-2025 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef ES2PANDA_COMPILER_CORE_ETS_EMITTER_H
#define ES2PANDA_COMPILER_CORE_ETS_EMITTER_H

#include "annotation.h"
#include "assembly-literals.h"
#include "emitter.h"

namespace ark::es2panda::parser {
class Program;
}  // namespace ark::es2panda::parser

namespace ark::es2panda::varbinder {
class RecordTable;
}  // namespace ark::es2panda::varbinder

namespace ark::es2panda::ir {
class ClassDefinition;
}  // namespace ark::es2panda::ir

namespace ark::es2panda::checker {
class ETSObjectType;
class ETSArrayType;
class Signature;
}  // namespace ark::es2panda::checker

namespace ark::pandasm {
struct Field;
struct Record;
class ItemMetadata;
class AnnotationData;
}  // namespace ark::pandasm

namespace ark::es2panda::compiler {

using LiteralArrayPair = std::pair<std::string, std::vector<pandasm::LiteralArray::Literal>>;
using LiteralArrayVector = std::vector<LiteralArrayPair>;

class ETSFunctionEmitter : public FunctionEmitter {
public:
    ETSFunctionEmitter(const CodeGen *cg, ProgramElement *programElement) : FunctionEmitter(cg, programElement) {}
    ~ETSFunctionEmitter() override = default;
    NO_COPY_SEMANTIC(ETSFunctionEmitter);
    NO_MOVE_SEMANTIC(ETSFunctionEmitter);

protected:
    const ETSGen *Etsg() const
    {
        return reinterpret_cast<const ETSGen *>(Cg());
    }

    pandasm::Function *GenFunctionSignature() override;

    void GenFunctionAnnotations(pandasm::Function *func) override;
    void GenVariableSignature(pandasm::debuginfo::LocalVariable &variableDebug,
                              varbinder::LocalVariable *variable) const override;
    void GenSourceFileDebugInfo(pandasm::Function *func) override;
};

class ETSEmitter : public Emitter {
public:
    explicit ETSEmitter(const public_lib::Context *context) : Emitter(context) {}
    ~ETSEmitter() override = default;
    NO_COPY_SEMANTIC(ETSEmitter);
    NO_MOVE_SEMANTIC(ETSEmitter);

    void GenAnnotation() override;
    std::vector<pandasm::AnnotationData> GenCustomAnnotations(
        const ArenaVector<ir::AnnotationUsage *> &annotationUsages, const std::string &baseName);

private:
    using DynamicCallNamesMap = ArenaMap<const ArenaVector<util::StringView>, uint32_t>;

    void GenExternalRecord(varbinder::RecordTable *recordTable, const parser::Program *extProg);
    void GenGlobalArrayRecord(const checker::ETSArrayType *arrayType, checker::Signature *signature);
    std::vector<pandasm::AnnotationData> GenAnnotations(const ir::ClassDefinition *classDef);
    void GenClassRecord(const ir::ClassDefinition *classDef, bool external);
    pandasm::AnnotationElement ProcessArrayType(const ir::ClassProperty *prop, std::string &baseName,
                                                const ir::Expression *init);
    pandasm::AnnotationElement ProcessETSEnumType(std::string &baseName, const ir::Expression *init,
                                                  const checker::Type *type);
    pandasm::AnnotationElement GenCustomAnnotationElement(const ir::ClassProperty *prop, std::string &baseName);
    pandasm::AnnotationData GenCustomAnnotation(ir::AnnotationUsage *anno, std::string &baseName);
    void CreateEnumProp(const ir::ClassProperty *prop, pandasm::Field &field);
    void ProcessArrayElement(const ir::Expression *elem, std::vector<pandasm::LiteralArray::Literal> &literals,
                             std::string &baseName, LiteralArrayVector &result);
    LiteralArrayVector CreateLiteralArray(std::string &baseName, const ir::Expression *array);
    void CreateLiteralArrayProp(const ir::ClassProperty *prop, std::string &baseName, pandasm::Field &field);
    void GenCustomAnnotationProp(const ir::ClassProperty *prop, std::string &baseName, pandasm::Record &record,
                                 bool external);
    void GenCustomAnnotationRecord(const ir::AnnotationDeclaration *annoDecl, std::string &baseName, bool external);
    void GenEnumRecord(const ir::TSEnumDeclaration *enumDecl, bool external);
    void GenAnnotationRecord(std::string_view recordNameView, bool isRuntime = false, bool isType = false);
    void GenInterfaceRecord(const ir::TSInterfaceDeclaration *interfaceDecl, bool external);
    void EmitDefaultFieldValue(pandasm::Field &classField, const ir::Expression *init);
    void GenClassField(const ir::ClassProperty *prop, pandasm::Record &classRecord, bool external);

    void GenInterfaceMethodDefinition(const ir::MethodDefinition *methodDef, bool external);
    void GenClassInheritedFields(const checker::ETSObjectType *baseType, pandasm::Record &classRecord);
    pandasm::AnnotationData GenAnnotationSignature(const ir::ClassDefinition *classDef);
    pandasm::AnnotationData GenAnnotationModule(const ir::ClassDefinition *classDef);
    pandasm::AnnotationData GenAnnotationEnclosingClass(std::string_view className);
    pandasm::AnnotationData GenAnnotationEnclosingMethod(const ir::MethodDefinition *methodDef);
    pandasm::AnnotationData GenAnnotationFunctionalReference(const ir::ClassDefinition *classDef);
    pandasm::AnnotationData GenAnnotationInnerClass(const ir::ClassDefinition *classDef, const ir::AstNode *parent);
    pandasm::AnnotationData GenAnnotationAsync(ir::ScriptFunction *scriptFunc);
    pandasm::AnnotationData GenAnnotationDynamicCall(DynamicCallNamesMap &callNames);
    ir::MethodDefinition *FindAsyncImpl(ir::ScriptFunction *asyncFunc);
    void ProcessArrayExpression(std::string &baseName, LiteralArrayVector &result,
                                std::vector<pandasm::LiteralArray::Literal> &literals, const ir::Expression *elem);
};
}  // namespace ark::es2panda::compiler

#endif