OpenHarmony-v6.0-Release/s

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

#include "js_screen_listener.h"
#include "js_runtime_utils.h"
#include "window_manager_hilog.h"
#include "js_screen.h"

namespace OHOS {
namespace Rosen {
using namespace AbilityRuntime;
inline uint32_t SCREEN_DISCONNECT_TYPE = 0;
inline uint32_t SCREEN_CONNECT_TYPE = 1;

JsScreenListener::JsScreenListener(napi_env env) : env_(env), weakRef_(wptr<JsScreenListener> (this))
{
    TLOGI(WmsLogTag::DMS, "Constructor execution");
    napi_add_env_cleanup_hook(env_, CleanEnv, this);
}

JsScreenListener::~JsScreenListener()
{
    TLOGI(WmsLogTag::DMS, "Destructor execution");
    napi_remove_env_cleanup_hook(env_, CleanEnv, this);
    env_ = nullptr;
}

void JsScreenListener::CleanEnv(void* obj)
{
    JsScreenListener* thisObj = reinterpret_cast<JsScreenListener*>(obj);
    if (!thisObj) {
        TLOGE(WmsLogTag::DMS, "obj is nullptr");
        return;
    }
    TLOGI(WmsLogTag::DMS, "env_ is invalid, set to nullptr");
    thisObj->env_ = nullptr;
}

void JsScreenListener::AddCallback(const std::string& type, napi_value jsListenerObject)
{
    TLOGI(WmsLogTag::DMS, "called");
    std::lock_guard<std::mutex> lock(mtx_);
    std::unique_ptr<NativeReference> callbackRef;
    napi_ref result = nullptr;
    napi_create_reference(env_, jsListenerObject, 1, &result);
    callbackRef.reset(reinterpret_cast<NativeReference*>(result));
    jsCallBack_[type].emplace_back(std::move(callbackRef));
    TLOGI(WmsLogTag::DMS, "success jsCallBack_ size: %{public}u!",
        static_cast<uint32_t>(jsCallBack_[type].size()));
}

void JsScreenListener::RemoveAllCallback()
{
    std::lock_guard<std::mutex> lock(mtx_);
    jsCallBack_.clear();
}

void JsScreenListener::RemoveCallback(napi_env env, const std::string& type, napi_value jsListenerObject)
{
    std::lock_guard<std::mutex> lock(mtx_);
    auto it = jsCallBack_.find(type);
    if (it == jsCallBack_.end()) {
        TLOGE(WmsLogTag::DMS, "no callback to remove");
        return;
    }
    auto& listeners = it->second;
    for (auto iter = listeners.begin(); iter != listeners.end();) {
        bool isEquals = false;
        napi_strict_equals(env, jsListenerObject, (*iter)->GetNapiValue(), &isEquals);
        if (isEquals) {
            listeners.erase(iter);
        } else {
            iter++;
        }
    }
    TLOGI(WmsLogTag::DMS, "success jsCallBack_ size: %{public}u!",
        static_cast<uint32_t>(listeners.size()));
}

void JsScreenListener::CallJsMethod(const std::string& methodName, napi_value const * argv, size_t argc)
{
    if (methodName.empty()) {
        TLOGE(WmsLogTag::DMS, "empty method name str, call method failed");
        return;
    }
    TLOGD(WmsLogTag::DMS, "CallJsMethod methodName = %{public}s", methodName.c_str());
    if (env_ == nullptr) {
        TLOGE(WmsLogTag::DMS, "env_ nullptr");
        return;
    }
    for (auto& callback : jsCallBack_[methodName]) {
        napi_value method = callback->GetNapiValue();
        if (method == nullptr) {
            TLOGE(WmsLogTag::DMS, "Failed to get method callback from object");
            continue;
        }
        napi_call_function(env_, NapiGetUndefined(env_), method, argc, argv, nullptr);
    }
}

void JsScreenListener::OnConnect(ScreenId id)
{
    std::lock_guard<std::mutex> lock(mtx_);
    TLOGI(WmsLogTag::DMS, "called");
    if (jsCallBack_.empty()) {
        TLOGE(WmsLogTag::DMS, "not register!");
        return;
    }
    if (jsCallBack_.find(EVENT_CONNECT) == jsCallBack_.end()) {
        TLOGE(WmsLogTag::DMS, "not this event, return");
        return;
    }
    auto napiTask = [self = weakRef_, id, env = env_] () {
        HITRACE_METER_FMT(HITRACE_TAG_WINDOW_MANAGER, "JsScreenListener::OnConnect");
        auto thisListener = self.promote();
        if (thisListener == nullptr || env == nullptr) {
            TLOGNE(WmsLogTag::DMS, "[NAPI]this listener or env is nullptr");
            return;
        }
        napi_value argv[] = {CreateJsValue(env, static_cast<uint32_t>(id))};
        thisListener->CallJsMethod(EVENT_CONNECT, argv, ArraySize(argv));
    };

    if (env_ != nullptr) {
        napi_status ret = napi_send_event(env_, napiTask, napi_eprio_immediate, "OnConnect");
        if (ret != napi_status::napi_ok) {
            TLOGE(WmsLogTag::DMS, "Failed to SendEvent.");
        }
    } else {
        TLOGE(WmsLogTag::DMS, "env is nullptr");
    }
}

void JsScreenListener::OnDisconnect(ScreenId id)
{
    std::lock_guard<std::mutex> lock(mtx_);
    TLOGI(WmsLogTag::DMS, "called");
    if (jsCallBack_.empty()) {
        TLOGE(WmsLogTag::DMS, "not register!");
        return;
    }
    if (jsCallBack_.find(EVENT_DISCONNECT) == jsCallBack_.end()) {
        TLOGE(WmsLogTag::DMS, "not this event, return");
        return;
    }
    auto napiTask = [self = weakRef_, id, env = env_] () {
        HITRACE_METER_FMT(HITRACE_TAG_WINDOW_MANAGER, "JsScreenListener::OnDisconnect");
        auto thisListener = self.promote();
        if (thisListener == nullptr || env == nullptr) {
            TLOGNE(WmsLogTag::DMS, "[NAPI]this listener or env is nullptr");
            return;
        }
        napi_value argv[] = {CreateJsValue(env, static_cast<uint32_t>(id))};
        thisListener->CallJsMethod(EVENT_DISCONNECT, argv, ArraySize(argv));
    };

    if (env_ != nullptr) {
        napi_status ret = napi_send_event(env_, napiTask, napi_eprio_immediate, "OnDisconnect");
        if (ret != napi_status::napi_ok) {
            TLOGE(WmsLogTag::DMS, "Failed to SendEvent.");
        }
    } else {
        TLOGE(WmsLogTag::DMS, "env is nullptr");
    }
}

void JsScreenListener::OnChange(ScreenId id)
{
    std::lock_guard<std::mutex> lock(mtx_);
    TLOGI(WmsLogTag::DMS, "called");
    if (jsCallBack_.empty()) {
        TLOGE(WmsLogTag::DMS, "not register!");
        return;
    }
    if (jsCallBack_.find(EVENT_CHANGE) == jsCallBack_.end()) {
        TLOGE(WmsLogTag::DMS, "not this event, return");
        return;
    }
    auto napiTask = [self = weakRef_, id, env = env_] () {
        HITRACE_METER_FMT(HITRACE_TAG_WINDOW_MANAGER, "JsScreenListener::OnChange");
        auto thisListener = self.promote();
        if (thisListener == nullptr || env == nullptr) {
            TLOGNE(WmsLogTag::DMS, "[NAPI]this listener or env is nullptr");
            return;
        }
        napi_value argv[] = {CreateJsValue(env, static_cast<uint32_t>(id))};
        thisListener->CallJsMethod(EVENT_CHANGE, argv, ArraySize(argv));
    };

    if (env_ != nullptr) {
        napi_status ret = napi_send_event(env_, napiTask, napi_eprio_immediate, "OnChange");
        if (ret != napi_status::napi_ok) {
            TLOGE(WmsLogTag::DMS, "Failed to SendEvent.");
        }
    } else {
        TLOGE(WmsLogTag::DMS, "env is nullptr");
    }
}

napi_value JsScreenListener::CreateScreenIdArray(napi_env env, const std::vector<ScreenId>& data)
{
    napi_value arrayValue = nullptr;
    napi_create_array_with_length(env, data.size(), &arrayValue);
    if (arrayValue == nullptr) {
        TLOGE(WmsLogTag::DMS, "Failed to create screenid array");
        return NapiGetUndefined(env);
    }
    uint32_t index = 0;
    for (const auto& item : data) {
        napi_set_element(env, arrayValue, index++, CreateJsValue(env, static_cast<uint32_t>(item)));
    }
    return arrayValue;
}
} // namespace Rosen
} // namespace OHOS