napi_utils/src/base_context.cpp

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

#include "event_manager.h"
#include "napi/native_api.h"
#include "napi/native_common.h"
#include "napi_utils.h"
#include "node_api.h"

namespace OHOS::NetStack {
BaseContext::BaseContext(napi_env env, EventManager *manager)
    : manager_(manager),
      env_(env),
      ref_(nullptr),
      parseOK_(false),
      requestOK_(false),
      errorCode_(0),
      callback_(nullptr),
      promiseRef_(nullptr),
      asyncWork_(nullptr),
      deferred_(nullptr),
      needPromise_(true),
      needThrowException_(false),
      permissionDenied_(false),
      noAllowedHost_(false)
{
}

BaseContext::BaseContext(napi_env env, const std::shared_ptr<EventManager> &sharedManager)
    : manager_(nullptr),
      env_(env),
      ref_(nullptr),
      parseOK_(false),
      requestOK_(false),
      errorCode_(0),
      callback_(nullptr),
      promiseRef_(nullptr),
      asyncWork_(nullptr),
      deferred_(nullptr),
      needPromise_(true),
      needThrowException_(false),
      permissionDenied_(false),
      noAllowedHost_(false),
      sharedManager_(sharedManager)
{
}

BaseContext::~BaseContext()
{
    DeleteCallback();
    DeletePromise();
    DeleteAsyncWork();
}

void BaseContext::SetParseOK(bool parseOK)
{
    parseOK_ = parseOK;
}

void BaseContext::SetExecOK(bool requestOK)
{
    requestOK_ = requestOK;
}

void BaseContext::SetErrorCode(int32_t errorCode)
{
    errorCode_ = errorCode;
}

void BaseContext::SetError(int32_t errorCode, const std::string &errorMessage)
{
    errorCode_ = errorCode;
    errorMessage_ = errorMessage;
}

napi_status BaseContext::SetCallback(napi_value callback)
{
    if (callback_ != nullptr) {
        (void)napi_delete_reference(env_, callback_);
    }
    auto status = napi_create_reference(env_, callback, 1, &callback_);
    callbackBak1_ = callback_;
    callbackBak2_ = callback_;
    callbackBak3_ = callback_;
    callbackBak4_ = callback_;
    return status;
}

void BaseContext::DeleteCallback()
{
    if (callback_ == nullptr || callback_ != callbackBak1_ || callback_ != callbackBak2_ ||
        callback_ != callbackBak3_ || callback_ != callbackBak4_) {
        return;
    }
    (void)napi_delete_reference(env_, callback_);
    callback_ = nullptr;
}

napi_async_work BaseContext::GetAsyncWork()
{
    return asyncWork_;
}

void BaseContext::CreateAsyncWork(const std::string &name, AsyncWorkExecutor executor, AsyncWorkCallback callback)
{
    napi_status ret = napi_create_async_work(env_, nullptr, NapiUtils::CreateStringUtf8(env_, name), executor, callback,
                                             this, &asyncWork_);
    asyncWorkBack1_ = asyncWork_;
    asyncWorkBack2_ = asyncWork_;
    asyncWorkBack3_ = asyncWork_;
    asyncWorkBack4_ = asyncWork_;
    if (ret != napi_ok) {
        return;
    }
    asyncWorkName_ = name;
    (void)napi_queue_async_work_with_qos(env_, asyncWork_, napi_qos_default);
}

void BaseContext::DeleteAsyncWork()
{
    if (asyncWork_ == nullptr) {
        return;
    }
    (void)napi_delete_async_work(env_, asyncWork_);
}

napi_value BaseContext::CreatePromise()
{
    napi_value result = nullptr;
    NAPI_CALL(env_, napi_create_promise(env_, &deferred_, &result));
    promiseRef_ = NapiUtils::CreateReference(env_, result);
    deferredBack1_ = deferred_;
    deferredBack2_ = deferred_;
    deferredBack3_ = deferred_;
    deferredBack4_ = deferred_;
    return result;
}

void BaseContext::DeletePromise()
{
    if (promiseRef_ == nullptr) {
        return;
    }
    (void)napi_delete_reference(env_, promiseRef_);
    promiseRef_ = nullptr;
}

bool BaseContext::IsParseOK() const
{
    return parseOK_;
}

bool BaseContext::IsExecOK() const
{
    return requestOK_;
}

napi_env BaseContext::GetEnv() const
{
    return env_;
}

int32_t BaseContext::GetErrorCode() const
{
    return errorCode_;
}

std::string BaseContext::GetErrorMessage() const
{
    return errorMessage_;
}

napi_value BaseContext::GetCallback() const
{
    if (callback_ == nullptr || callback_ != callbackBak1_ || callback_ != callbackBak2_ ||
        callback_ != callbackBak3_ || callback_ != callbackBak4_) {
        return nullptr;
    }
    napi_value callback = nullptr;
    NAPI_CALL(env_, napi_get_reference_value(env_, callback_, &callback));
    return callback;
}

napi_deferred BaseContext::GetDeferred() const
{
    return deferred_;
}

const std::string &BaseContext::GetAsyncWorkName() const
{
    return asyncWorkName_;
}

void BaseContext::EmitSharedManager(const std::string &type, const std::pair<napi_value, napi_value> &argv)
{
    if (sharedManager_ != nullptr) {
        sharedManager_->Emit(type, argv);
    }
}

void BaseContext::Emit(const std::string &type, const std::pair<napi_value, napi_value> &argv)
{
    if (manager_ != nullptr) {
        manager_->Emit(type, argv);
    }
}

void BaseContext::SetNeedPromise(bool needPromise)
{
    needPromise_ = needPromise;
}

bool BaseContext::IsNeedPromise() const
{
    return needPromise_;
}

EventManager *BaseContext::GetManager() const
{
    return manager_;
}

std::shared_ptr<EventManager> BaseContext::GetSharedManager() const
{
    return sharedManager_;
}

void BaseContext::SetSharedManager(const std::shared_ptr<EventManager> &sharedManager)
{
    sharedManager_ = sharedManager;
}

void BaseContext::ParseParams(napi_value *params, size_t paramsCount)
{
    if (paramsCount != 0 && paramsCount != 1) {
        return;
    }

    if (paramsCount == 1 && NapiUtils::GetValueType(env_, params[0]) != napi_function) {
        return;
    }

    if (paramsCount == 1) {
        SetParseOK(SetCallback(params[0]) == napi_ok);
        return;
    }
    SetParseOK(true);
}

void BaseContext::SetNeedThrowException(bool needThrowException)
{
    needThrowException_ = needThrowException;
}

bool BaseContext::IsNeedThrowException() const
{
    return needThrowException_;
}

void BaseContext::SetPermissionDenied(bool permissionDenied)
{
    permissionDenied_ = permissionDenied;
}

bool BaseContext::IsPermissionDenied() const
{
    return permissionDenied_;
}

void BaseContext::SetNoAllowedHost(bool noAllowed)
{
    noAllowedHost_ = noAllowed;
}

bool BaseContext::IsNoAllowedHost() const
{
    return noAllowedHost_;
}

void BaseContext::CreateReference(napi_value value)
{
    if (env_ != nullptr && value != nullptr) {
        ref_ = NapiUtils::CreateReference(env_, value);
    }
}

void BaseContext::DeleteReference()
{
    if (env_ != nullptr && ref_ != nullptr) {
        NapiUtils::DeleteReference(env_, ref_);
    }
}
} // namespace OHOS::NetStack