/*
 * 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 "uc_observer_mgr.h"

#include "app_mgr_client.h"
#include "hiview_logger.h"
#include "iservice_registry.h"
#include "process_status.h"
#include "system_ability_definition.h"

namespace OHOS {
namespace HiviewDFX {
DEFINE_LOG_TAG("UnifiedCollector");
using namespace OHOS::HiviewDFX::UCollectUtil;

void UcObserverManager::RegisterObservers()
{
    RegisterSysAbilityListener();
}

void UcObserverManager::UnregisterObservers()
{
    UnregisterSysAbilityListener();
}

void UcObserverManager::RegisterSysAbilityListener()
{
    auto samgr = SystemAbilityManagerClient::GetInstance().GetSystemAbilityManager();
    if (samgr == nullptr) {
        HIVIEW_LOGE("failed to get samgr");
        return;
    }
    sysAbilityListener_ = new(std::nothrow) UcSystemAbilityListener();
    if (sysAbilityListener_ == nullptr) {
        HIVIEW_LOGE("new listener is null");
        return;
    }
    if (auto ret = samgr->SubscribeSystemAbility(APP_MGR_SERVICE_ID, sysAbilityListener_); ret != ERR_OK) {
        HIVIEW_LOGE("failed to subscribe app mgr service status, ret=%{public}d", ret);
        return;
    }
    HIVIEW_LOGI("succ to subscribe app mgr service status");
}

void UcObserverManager::UnregisterSysAbilityListener()
{
    if (sysAbilityListener_ == nullptr) {
        return;
    }
    auto samgr = SystemAbilityManagerClient::GetInstance().GetSystemAbilityManager();
    if (samgr == nullptr) {
        HIVIEW_LOGE("failed to get samgr");
        return;
    }
    if (auto ret = samgr->UnSubscribeSystemAbility(APP_MGR_SERVICE_ID, sysAbilityListener_); ret != ERR_OK) {
        HIVIEW_LOGE("failed to unSubscribe app mgr service status, ret=%{public}d", ret);
        return;
    }
    sysAbilityListener_ = nullptr;
    HIVIEW_LOGI("succ to unsubscribe app mgr service status");
}

void UcObserverManager::RegisterAppObserver()
{
    appStateObserver_ = new(std::nothrow) UcAppStateObserver();
    if (appStateObserver_ == nullptr) {
        HIVIEW_LOGE("observer is null");
        return;
    }
    auto res = DelayedSingleton<AppExecFwk::AppMgrClient>::GetInstance()->
        RegisterApplicationStateObserver(appStateObserver_);
    if (res != ERR_OK) {
        HIVIEW_LOGE("failed to register observer, res=%{public}d", res);
        return;
    }
    HIVIEW_LOGI("succ to register observer");
}

void UcObserverManager::RegisterRenderObserver()
{
    renderStateObserver_ = new(std::nothrow) UcRenderStateObserver();
    if (renderStateObserver_ == nullptr) {
        HIVIEW_LOGE("observer is null");
        return;
    }

    int32_t res = -1; // -1: default value
    constexpr uint32_t maxTryTimes = 3; // 3: max three times
    uint32_t curTryTime = 1; // 1: first time
    do {
        res = DelayedSingleton<AppExecFwk::AppMgrClient>::GetInstance()->
            RegisterRenderStateObserver(renderStateObserver_);
        HIVIEW_LOGI("register observer, res=%{public}d, curTryTime=%{public}u", res, curTryTime);
        if (res == ERR_OK) {
            HIVIEW_LOGI("succ to register observer");
            return;
        }
        ++curTryTime;
        sleep(1); // sleep 1s
    } while (res != ERR_OK && curTryTime <= maxTryTimes);
}

void UcObserverManager::UnregisterAppObserver()
{
    if (appStateObserver_ == nullptr) {
        return;
    }
    auto res = DelayedSingleton<AppExecFwk::AppMgrClient>::GetInstance()->
        UnregisterApplicationStateObserver(appStateObserver_);
    if (res != ERR_OK) {
        HIVIEW_LOGE("failed to unregister observer, res=%{public}d", res);
        return;
    }
    appStateObserver_ = nullptr;
    HIVIEW_LOGI("succ to unregister observer");
}

void UcObserverManager::UnregisterRenderObserver()
{
    if (renderStateObserver_ == nullptr) {
        return;
    }
    auto res = DelayedSingleton<AppExecFwk::AppMgrClient>::GetInstance()->
        UnregisterRenderStateObserver(renderStateObserver_);
    if (res != ERR_OK) {
        HIVIEW_LOGE("failed to unregister observer, res=%{public}d", res);
        return;
    }
    renderStateObserver_ = nullptr;
    HIVIEW_LOGI("succ to unregister observer");
}
}  // namespace HiviewDFX
}  // namespace OHOS