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

#include "ecmascript/js_tagged_value-inl.h"
#include "ecmascript/global_env.h"

namespace panda::ecmascript {

void SnapshotEnv::AddGlobalConstToMap()
{
    auto globalConst = const_cast<GlobalEnvConstants *>(vm_->GetJSThread()->GlobalConstants());
    for (size_t index = 0; index < globalConst->GetConstantCount(); index++) {
        JSTaggedValue objectValue = globalConst->GetGlobalConstantObject(index);
        if (objectValue.IsHeapObject() && !objectValue.IsString()) {
            rootObjectMap_.emplace(objectValue.GetRawData(), index);
        }
    }
}

JSTaggedType SnapshotEnv::RelocateRootObjectAddr(uint32_t index)
{
    auto globalConst = const_cast<GlobalEnvConstants *>(vm_->GetJSThread()->GlobalConstants());
    size_t globalConstCount = globalConst->GetConstantCount();
    if (index < globalConstCount) {
        JSTaggedValue obj = globalConst->GetGlobalConstantObject(index);
        return obj.GetRawData();
    }
    JSHandle<JSTaggedValue> value = vm_->GetGlobalEnv()->GetNoLazyEnvObjectByIndex(index - globalConstCount);
    return value->GetRawData();
}

void SnapshotEnv::Iterate(const RootVisitor &v, VMRootVisitType type)
{
    if (multiThreadCheckValue_.exchange(JSThread::GetCurrentThreadId()) != 0) {
        LOG_ECMA(FATAL) << "SnapshotEnv push multi-thread check fail, thread id: " << multiThreadCheckValue_;
    }
    if (type == VMRootVisitType::UPDATE_ROOT) {
        // during update root, rootObjectMap_ key need update
        std::unordered_map<JSTaggedType, uint32_t> rootObjectMap(rootObjectMap_.size());
        for (auto &it : rootObjectMap_) {
            auto objectAddr = it.first;
            ObjectSlot slot(reinterpret_cast<uintptr_t>(&objectAddr));
            v(Root::ROOT_VM, slot);
            rootObjectMap.emplace(slot.GetTaggedType(), it.second);
        }
        std::swap(rootObjectMap_, rootObjectMap);
        rootObjectMap.clear();
    } else {
        for (auto &it : rootObjectMap_) {
            ObjectSlot slot(reinterpret_cast<uintptr_t>(&it));
            v(Root::ROOT_VM, slot);
        }
    }
    multiThreadCheckValue_ = 0;
}
}  // namespace panda::ecmascript