/*
 * Copyright (c) 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 "async_lock.h"
#include "async_lock_manager.h"
#include "tools/log.h"

namespace Commonlibrary::Concurrent::LocksModule {
using namespace Commonlibrary::Concurrent::Common::Helper;

AsyncLock::AsyncLock(const std::string &lockName)
{
    lockName_ = lockName;
    anonymousLockId_ = 0;
}

AsyncLock::AsyncLock(uint32_t lockId)
{
    lockName_ = "";
    anonymousLockId_ = lockId;
}

napi_value AsyncLock::LockAsync(napi_env env, napi_ref cb, LockMode mode, const LockOptions &options)
{
    napi_value promise;
    napi_deferred deferred;
    napi_create_promise(env, &deferred, &promise);
    if (!NativeEngine::IsAlive(reinterpret_cast<NativeEngine *>(env))) {
        ErrorHelper::ThrowError(env, ErrorHelper::TYPE_ERROR, "env is not alive");
        return nullptr;
    }
    if (!CanAcquireLock(mode) && options.isAvailable) {
        napi_value err;
        NAPI_CALL(env, napi_create_string_utf8(env, "The lock is acquired", NAPI_AUTO_LENGTH, &err));
        napi_reject_deferred(env, deferred, err);
    } else {
        LockRequest *lockRequest =
            new LockRequest(this, AsyncLockManager::GetCurrentTid(env), env, cb, mode, options, deferred);
        std::unique_lock<std::mutex> lock(asyncLockMutex_);
        pendingList_.push_back(lockRequest);
        ProcessPendingLockRequestUnsafe(env, lockRequest);
    }
    return promise;
}

void AsyncLock::CleanUpLockRequestOnCompletion(LockRequest* lockRequest)
{
    std::unique_lock<std::mutex> lock(asyncLockMutex_);
    auto it = std::find(heldList_.begin(), heldList_.end(), lockRequest);
    if (it == heldList_.end()) {
        HILOG_FATAL("Lock is not found");
        return;
    }
    heldList_.erase(it);
    if (heldList_.empty()) {
        // There are may be other shared lock requests in the heldList_.
        // IF so, we mustn't change the status.
        lockStatus_ = LOCK_MODE_UNLOCK;
    }
    napi_env env = lockRequest->GetEnv();
    delete lockRequest;
    if (pendingList_.empty()) {
        if (refCount_ == 0 && heldList_.empty()) {
            lock.unlock();
            AsyncLockManager::CheckAndRemoveLock(this);
        }
        return;
    }
    ProcessPendingLockRequestUnsafe(env);
}

bool AsyncLock::CleanUpLockRequest(LockRequest *lockRequest)
{
    std::unique_lock<std::mutex> lock(asyncLockMutex_);
    auto it = std::find(pendingList_.begin(), pendingList_.end(), lockRequest);
    if (it == pendingList_.end()) {
        // the lock got held while we were waiting on the mutex, no-op
        return false;
    }
    // we won the race, need to remove the request from the queue and handle the time out event
    pendingList_.erase(it);
    return true;
}

template <bool isAsync>
void AsyncLock::ProcessLockRequest(napi_env env, LockRequest *lockRequest)
{
    heldList_.push_back(lockRequest);
    pendingList_.pop_front();
    asyncLockMutex_.unlock();
    if constexpr (isAsync) {
        lockRequest->CallCallbackAsync();
    } else {
        lockRequest->CallCallback();
    }
    asyncLockMutex_.lock();
}

void AsyncLock::ProcessPendingLockRequest(napi_env env, LockRequest* syncLockRequest)
{
    std::unique_lock<std::mutex> lock(asyncLockMutex_);
    if (pendingList_.empty()) {
        if (refCount_ == 0 && heldList_.empty()) {
            lock.unlock();
            AsyncLockManager::CheckAndRemoveLock(this);
        }
        return;
    }
    ProcessPendingLockRequestUnsafe(env, syncLockRequest);
}

void AsyncLock::ProcessPendingLockRequestUnsafe(napi_env env, LockRequest *syncLockRequest)
{
    LockRequest *lockRequest = pendingList_.front();
    if (!CanAcquireLockUnsafe(lockRequest->GetMode())) {
        return;
    }
    lockStatus_ = lockRequest->GetMode();
    if (lockStatus_ == LOCK_MODE_SHARED) {
        do {
            if (syncLockRequest == lockRequest) {
                ProcessLockRequest<false>(env, lockRequest);
            } else {
                ProcessLockRequest<true>(env, lockRequest);
            }
            if (pendingList_.empty()) {
                break;
            }
            lockRequest = pendingList_.front();
        } while (lockRequest->GetMode() == LOCK_MODE_SHARED);
    } else {
        ProcessLockRequest<true>(env, lockRequest);
    }
}

bool AsyncLock::CanAcquireLockUnsafe(LockMode mode)
{
    if (heldList_.empty()) {
        return true;
    }
    if (mode == LOCK_MODE_SHARED && lockStatus_ == LOCK_MODE_SHARED) {
        return true;
    }
    if (lockStatus_ == LOCK_MODE_UNLOCK) {
        return true;
    }
    return false;
}

bool AsyncLock::CanAcquireLock(LockMode mode)
{
    std::unique_lock<std::mutex> lock(asyncLockMutex_);
    return CanAcquireLockUnsafe(mode);
}

napi_status AsyncLock::FillLockState(napi_env env, napi_value held, napi_value pending)
{
    std::unique_lock<std::mutex> lock(asyncLockMutex_);
    uint32_t idx = 0;
    for (LockRequest *rq : heldList_) {
        napi_value info = CreateLockInfo(env, rq);
        bool pendingException = false;
        napi_is_exception_pending(env, &pendingException);
        if (pendingException) {
            return napi_pending_exception;
        }
        napi_value index;
        napi_create_int32(env, idx, &index);
        napi_status status = napi_set_property(env, held, index, info);
        if (status != napi_ok) {
            return status;
        }
        ++idx;
    }
    idx = 0;
    for (LockRequest *rq : pendingList_) {
        napi_value info = CreateLockInfo(env, rq);
        bool pendingException = false;
        napi_is_exception_pending(env, &pendingException);
        if (pendingException) {
            return napi_pending_exception;
        }
        napi_value index;
        napi_create_int32(env, idx, &index);
        napi_status status = napi_set_property(env, pending, index, info);
        if (status != napi_ok) {
            return status;
        }
        ++idx;
    }
    return napi_ok;
}

napi_value AsyncLock::CreateLockInfo(napi_env env, const LockRequest *rq)
{
    napi_value info;
    NAPI_CALL(env, napi_create_object(env, &info));
    napi_value name;
    NAPI_CALL(env, napi_create_string_utf8(env, lockName_.c_str(), NAPI_AUTO_LENGTH, &name));
    napi_value mode;
    NAPI_CALL(env, napi_create_int32(env, rq->GetMode(), &mode));
    napi_value tid;
    NAPI_CALL(env, napi_create_int32(env, rq->GetTid(), &tid));

    napi_property_descriptor properties[] = {
        DECLARE_NAPI_PROPERTY("name", name),
        DECLARE_NAPI_PROPERTY("mode", mode),
        DECLARE_NAPI_PROPERTY("contextId", tid),
    };
    NAPI_CALL(env, napi_define_properties(env, info, sizeof(properties) / sizeof(properties[0]), properties));
    return info;
}

bool AsyncLock::IsReadyForDeletion()
{
    std::unique_lock<std::mutex> lock(asyncLockMutex_);
    return refCount_ == 0 && pendingList_.empty() && heldList_.empty();
}

uint32_t AsyncLock::IncRefCount()
{
    std::unique_lock<std::mutex> lock(asyncLockMutex_);
    return ++refCount_;
}

uint32_t AsyncLock::DecRefCount()
{
    std::unique_lock<std::mutex> lock(asyncLockMutex_);
    return --refCount_;
}

std::vector<RequestCreationInfo> AsyncLock::GetSatisfiedRequestInfos()
{
    std::vector<RequestCreationInfo> result;
    std::unique_lock<std::mutex> lock(asyncLockMutex_);
    for (auto *request : heldList_) {
        result.push_back(RequestCreationInfo { request->GetTid(), request->GetCreationStacktrace() });
    }
    return result;
}

std::vector<RequestCreationInfo> AsyncLock::GetPendingRequestInfos()
{
    std::vector<RequestCreationInfo> result;
    std::unique_lock<std::mutex> lock(asyncLockMutex_);
    for (auto *request : pendingList_) {
        result.push_back(RequestCreationInfo { request->GetTid(), request->GetCreationStacktrace() });
    }
    return result;
}

} // namespace Commonlibrary::Concurrent::LocksModule