/*
 * Copyright (c) 2021 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/js_method.h"

#include "ecmascript/jspandafile/js_pandafile.h"
#include "libpandafile/method_data_accessor-inl.h"

namespace panda::ecmascript {
JSMethod::JSMethod(const JSPandaFile *jsPandaFile, panda_file::File::EntityId fileId,
                   panda_file::File::EntityId codeId, uint32_t accessFlags,
                   uint32_t numArgs)
    : Method(nullptr, jsPandaFile != nullptr ? jsPandaFile->GetPandaFile() : nullptr,
             fileId, codeId, accessFlags, numArgs, nullptr)
{
    bytecodeArray_ = GetInstructions();
    bytecodeArraySize_ = GetCodeSize();
    jsPandaFile_ = jsPandaFile;
}

// It's not allowed '#' token appear in ECMA function(method) name, which discriminates same names in panda methods.
CString JSMethod::ParseFunctionName() const
{
    CString methodName(utf::Mutf8AsCString(GetName().data));
    if (LIKELY(methodName[0] != '#')) {
        return methodName;
    }
    size_t index = methodName.find_last_of('#');
    return CString(methodName.substr(index + 1));
}

void JSMethod::InitializeCallField()
{
    uint32_t callType = UINT32_MAX;  // UINT32_MAX means not found
    const panda_file::File *pandaFile = GetPandaFile();
    panda_file::File::EntityId fieldId = GetFileId();
    panda_file::MethodDataAccessor mda(*pandaFile, fieldId);
    mda.EnumerateAnnotations([&](panda_file::File::EntityId annotation_id) {
        panda_file::AnnotationDataAccessor ada(*pandaFile, annotation_id);
        auto *annotation_name = reinterpret_cast<const char *>(pandaFile->GetStringData(ada.GetClassId()).data);
        if (::strcmp("L_ESCallTypeAnnotation;", annotation_name) == 0) {
            uint32_t elem_count = ada.GetCount();
            for (uint32_t i = 0; i < elem_count; i++) {
                panda_file::AnnotationDataAccessor::Elem adae = ada.GetElement(i);
                auto *elem_name = reinterpret_cast<const char *>(pandaFile->GetStringData(adae.GetNameId()).data);
                if (::strcmp("callType", elem_name) == 0) {
                    callType = adae.GetScalarValue().GetValue();
                }
            }
        }
    });
    // Needed info for call can be got by loading callField only once
    callField_ = (callType & CALL_TYPE_MASK) |
                 NumVregsBits::Encode(GetNumVregs()) |
                 NumArgsBits::Encode(GetNumArgs() - HaveFuncBit::Decode(callType)  // exclude func
                                                  - HaveNewTargetBit::Decode(callType)  // exclude new target
                                                  - HaveThisBit::Decode(callType)) |  // exclude this
                 IsNativeBit::Encode(IsNative());
}
} // namespace panda::ecmascript