/*

 * Copyright (c) 2022-2024 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.

 */

#include "ecmascript/compiler/builtins/builtins_call_signature.h"

#include "ecmascript/compiler/builtins/builtins_stubs.h"

#include "ecmascript/compiler/call_signature.h"

#include "ecmascript/global_env_fields.h"



namespace panda::ecmascript::kungfu {



// Template factory for builtins stubs: takes an extra Gate::InvalidGateRef arg.

template <typename BuilderT>

static void *BuiltinStubBuilderFactory(void *env, void *data)

{

    return static_cast<void *>(

        new BuilderT(static_cast<CallSignature *>(data),

                     static_cast<Environment *>(env), Gate::InvalidGateRef));

}



CallSignature BuiltinsStubCSigns::callSigns_[BuiltinsStubCSigns::NUM_OF_BUILTINS_STUBS_EXTEND];

CallSignature BuiltinsStubCSigns::builtinsCSign_;

CallSignature BuiltinsStubCSigns::builtinsWithArgvCSign_;



void BuiltinsStubCSigns::Initialize()

{

#define COMMON_INIT(name)                                                                             \

    {                                                                                                 \

        callSigns_[ID_##name].SetID(ID_##name);                                                       \

        callSigns_[ID_##name].SetName(std::string("BuiltinStub_") + #name);                           \

        callSigns_[ID_##name].SetConstructor(                                                         \

            TargetConstructor(BuiltinStubBuilderFactory<name##StubBuilder>, &callSigns_[ID_##name])); \

    }



#define INIT_BUILTINS_METHOD(name)                                   \

    BuiltinsCallSignature::Initialize(&callSigns_[ID_##name]);       \

    COMMON_INIT(name)



#define INIT_BUILTINS_METHOD_DYN(name, type, ...)                    \

    BuiltinsCallSignature::Initialize(&callSigns_[ID_##type##name]); \

    COMMON_INIT(type##name)



#define INIT_BUILTINS_CONSTRUCTOR_METHOD(name)                         \

    BuiltinsWithArgvCallSignature::Initialize(&callSigns_[ID_##name]); \

    COMMON_INIT(name)



#define STW_COPY_INIT(name)                                          \

    callSigns_[ID_##name].SetStwCopyStub(true);                      \

    callSigns_[ID_##name].SetTargetKind(CallSignature::TargetKind::BUILTINS_STW_COPY_STUB);



#define INIT_BUILTINS_METHOD_STW_COPY(name)                          \

    INIT_BUILTINS_METHOD(name##StwCopy)                              \

    STW_COPY_INIT(name##StwCopy)



#define INIT_BUILTINS_METHOD_DYN_STW_COPY(name, type, ...)           \

    INIT_BUILTINS_METHOD_DYN(name##StwCopy, type)                    \

    STW_COPY_INIT(type##name##StwCopy)



#define INIT_BUILTINS_CONSTRUCTOR_METHOD_STW_COPY(name)              \

    INIT_BUILTINS_CONSTRUCTOR_METHOD(name##StwCopy)                  \

    STW_COPY_INIT(name##StwCopy)



    BUILTINS_STUB_LIST(INIT_BUILTINS_METHOD, INIT_BUILTINS_METHOD_DYN, INIT_BUILTINS_CONSTRUCTOR_METHOD)

    BUILTINS_STW_COPY_STUB_LIST(INIT_BUILTINS_METHOD_STW_COPY, INIT_BUILTINS_METHOD_DYN_STW_COPY, \

        INIT_BUILTINS_CONSTRUCTOR_METHOD_STW_COPY)



#undef INIT_BUILTINS_CONSTRUCTOR_METHOD_STW_COPY

#undef INIT_BUILTINS_METHOD_DYN_STW_COPY

#undef INIT_BUILTINS_METHOD_STW_COPY

#undef STW_COPY_INIT

#undef INIT_BUILTINS_CONSTRUCTOR_METHOD

#undef INIT_BUILTINS_METHOD_DYN

#undef INIT_BUILTINS_METHOD



#undef COMMON_INIT

    BuiltinsCallSignature::Initialize(&builtinsCSign_);

    BuiltinsWithArgvCallSignature::Initialize(&builtinsWithArgvCSign_);

}



void BuiltinsStubCSigns::GetCSigns(std::vector<const CallSignature*>& outCSigns)

{

    const size_t firstStubId = BUILTINS_STUB_ID(NONE) + 1;

    for (size_t i = firstStubId; i < NUM_OF_BUILTINS_STUBS_EXTEND; i++) {

        outCSigns.push_back(&callSigns_[i]);

    }

}



void BuiltinsStubCSigns::GetNormalCSigns(std::vector<const CallSignature*>& outCSigns)

{

    for (size_t i = CSID_NORMAL_START; i < CSID_NORMAL_END; i++) {

        outCSigns.push_back(&callSigns_[i]);

    }

}



void BuiltinsStubCSigns::GetStwCopyCSigns(std::vector<const CallSignature*>& outCSigns)

{

    for (size_t i = CSID_STW_COPY_START; i < CSID_STW_COPY_END; i++) {

        outCSigns.push_back(&callSigns_[i]);

    }

}



size_t BuiltinsStubCSigns::GetGlobalEnvIndex(ID builtinId)

{

    static const std::map<ID, GlobalEnvField> globalEnvIndex = {

        {ID::BooleanConstructor, GlobalEnvField::BOOLEAN_FUNCTION_INDEX},

        {ID::NumberConstructor, GlobalEnvField::NUMBER_FUNCTION_INDEX},

        {ID::ProxyConstructor, GlobalEnvField::PROXY_FUNCTION_INDEX},

        {ID::DateConstructor, GlobalEnvField::DATE_FUNCTION_INDEX},

        {ID::ArrayConstructor, GlobalEnvField::ARRAY_FUNCTION_INDEX},

        {ID::SetConstructor, GlobalEnvField::BUILTINS_SET_FUNCTION_INDEX},

        {ID::MapConstructor, GlobalEnvField::BUILTINS_MAP_FUNCTION_INDEX},

        {ID::BigIntConstructor, GlobalEnvField::BIGINT_FUNCTION_INDEX},

        {ID::ObjectConstructor, GlobalEnvField::OBJECT_FUNCTION_INDEX},

        {ID::ErrorConstructor, GlobalEnvField::ERROR_FUNCTION_INDEX},

        {ID::Int8ArrayConstructor, GlobalEnvField::INT8_ARRAY_FUNCTION_INDEX},

        {ID::Uint8ArrayConstructor, GlobalEnvField::UINT8_ARRAY_FUNCTION_INDEX},

        {ID::Uint8ClampedArrayConstructor, GlobalEnvField::UINT8_CLAMPED_ARRAY_FUNCTION_INDEX},

        {ID::Int16ArrayConstructor, GlobalEnvField::INT16_ARRAY_FUNCTION_INDEX},

        {ID::Uint16ArrayConstructor, GlobalEnvField::UINT16_ARRAY_FUNCTION_INDEX},

        {ID::Int32ArrayConstructor, GlobalEnvField::INT32_ARRAY_FUNCTION_INDEX},

        {ID::Uint32ArrayConstructor, GlobalEnvField::UINT32_ARRAY_FUNCTION_INDEX},

        {ID::Float32ArrayConstructor, GlobalEnvField::FLOAT32_ARRAY_FUNCTION_INDEX},

        {ID::Float64ArrayConstructor, GlobalEnvField::FLOAT64_ARRAY_FUNCTION_INDEX},

        {ID::BigInt64ArrayConstructor, GlobalEnvField::BIGINT64_ARRAY_FUNCTION_INDEX},

        {ID::BigUint64ArrayConstructor, GlobalEnvField::BIGUINT64_ARRAY_FUNCTION_INDEX},

        {ID::SharedArrayConstructor, GlobalEnvField::SHARED_ARRAY_FUNCTION_INDEX},

        {ID::SharedSetConstructor, GlobalEnvField::SHARED_BUILTIN_SET_FUNCTION_INDEX},

        {ID::SharedMapConstructor, GlobalEnvField::SHARED_BUILTIN_MAP_FUNCTION_INDEX},

    };

    if (globalEnvIndex.find(builtinId) != globalEnvIndex.end()) {

        return static_cast<size_t>(globalEnvIndex.at(builtinId));

    }

    LOG_COMPILER(FATAL) << "this branch is unreachable";

    UNREACHABLE();

}

}  // namespace panda::ecmascript::kungfu