/*
 * 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 "interfaces/inner_api/ui_session/ui_session_manager.h"

#include "adapter/ohos/entrance/ui_session/include/ui_service_hilog.h"
namespace OHOS::Ace {
std::mutex UiSessionManager::mutex_;
constexpr int32_t ONCE_IPC_SEND_DATA_MAX_SIZE = 4096;
UiSessionManager& UiSessionManager::GetInstance()
{
    static UiSessionManager instance_;
    return instance_;
}

void UiSessionManager::ReportClickEvent(const std::string& data)
{
    for (auto pair : reportObjectMap_) {
        auto reportService = iface_cast<ReportService>(pair.second);
        if (reportService != nullptr) {
            reportService->ReportClickEvent(data);
        } else {
            LOGW("report click event failed,process id:%{public}d", pair.first);
        }
    }
}

void UiSessionManager::ReportSearchEvent(const std::string& data)
{
    for (auto pair : reportObjectMap_) {
        auto reportService = iface_cast<ReportService>(pair.second);
        if (reportService != nullptr) {
            reportService->ReportSearchEvent(data);
        } else {
            LOGW("report search event failed,process id:%{public}d", pair.first);
        }
    }
}

void UiSessionManager::ReportRouterChangeEvent(const std::string& data)
{
    for (auto pair : reportObjectMap_) {
        auto reportService = iface_cast<ReportService>(pair.second);
        if (reportService != nullptr) {
            reportService->ReportRouterChangeEvent(data);
        } else {
            LOGW("report switch event failed,process id:%{public}d", pair.first);
        }
    }
}

void UiSessionManager::ReportComponentChangeEvent(const std::string& key, const std::string& value)
{
    for (auto pair : reportObjectMap_) {
        auto reportService = iface_cast<ReportService>(pair.second);
        if (reportService != nullptr && GetComponentChangeEventRegistered()) {
            auto data = InspectorJsonUtil::Create();
            data->Put(key.data(), value.data());
            reportService->ReportComponentChangeEvent(data->ToString());
        } else {
            LOGW("report component event failed,process id:%{public}d", pair.first);
        }
    }
}

void UiSessionManager::ReportWebUnfocusEvent(int64_t accessibilityId, const std::string& data)
{
    auto jsonValue = InspectorJsonUtil::Create(true);
    jsonValue->Put("id", accessibilityId);
    jsonValue->Put("$type", "web");
    jsonValue->Put("text", data.c_str());
    for (auto pair : reportObjectMap_) {
        auto reportService = iface_cast<ReportService>(pair.second);
        if (reportService != nullptr) {
            reportService->ReportWebUnfocusEvent(accessibilityId, jsonValue->ToString());
        } else {
            LOGW("report web unfocus event failed,process id:%{public}d", pair.first);
        }
    }
}

void UiSessionManager::SaveReportStub(sptr<IRemoteObject> reportStub, int32_t processId)
{
    reportObjectMap_[processId] = reportStub;
}

void UiSessionManager::SetClickEventRegistered(bool status)
{
    std::lock_guard<std::mutex> lock(mutex_);
    if (status) {
        clickEventRegisterProcesses_++;
    } else {
        clickEventRegisterProcesses_--;
    }
}

void UiSessionManager::SetSearchEventRegistered(bool status)
{
    std::lock_guard<std::mutex> lock(mutex_);
    if (status) {
        searchEventRegisterProcesses_++;
    } else {
        searchEventRegisterProcesses_--;
    }
}

void UiSessionManager::SetRouterChangeEventRegistered(bool status)
{
    std::lock_guard<std::mutex> lock(mutex_);
    if (status) {
        routerChangeEventRegisterProcesses_++;
    } else {
        routerChangeEventRegisterProcesses_--;
    }
}

void UiSessionManager::SetComponentChangeEventRegistered(bool status)
{
    std::lock_guard<std::mutex> lock(mutex_);
    if (status) {
        componentChangeEventRegisterProcesses_++;
    } else {
        componentChangeEventRegisterProcesses_--;
    }
}

bool UiSessionManager::GetClickEventRegistered()
{
    return clickEventRegisterProcesses_ > 0 ? true : false;
}

bool UiSessionManager::GetSearchEventRegistered()
{
    return searchEventRegisterProcesses_ > 0 ? true : false;
}

bool UiSessionManager::GetRouterChangeEventRegistered()
{
    return routerChangeEventRegisterProcesses_ > 0 ? true : false;
}

bool UiSessionManager::GetComponentChangeEventRegistered()
{
    return componentChangeEventRegisterProcesses_ > 0 ? true : false;
}

void UiSessionManager::GetInspectorTree()
{
    jsonValue_ = InspectorJsonUtil::Create(true);
    webTaskNums = 0;
    WebTaskNumsChange(1);
    inspectorFunction_();
}

void UiSessionManager::SaveInspectorTreeFunction(InspectorFunction&& function)
{
    inspectorFunction_ = std::move(function);
}

void UiSessionManager::AddValueForTree(int32_t id, const std::string& value)
{
    std::string key = std::to_string(id);
    if (jsonValue_->Contains(key)) {
        jsonValue_->Replace(key.c_str(), value.c_str());
    } else {
        jsonValue_->Put(key.c_str(), value.c_str());
    }
}

void UiSessionManager::WebTaskNumsChange(int32_t num)
{
    webTaskNums += num;
    if (webTaskNums == 0) {
        std::string data = jsonValue_->ToString();
        ReportInspectorTreeValue(data);
    }
}

void UiSessionManager::ReportInspectorTreeValue(const std::string& data)
{
    for (auto pair : reportObjectMap_) {
        auto reportService = iface_cast<ReportService>(pair.second);
        if (reportService != nullptr) {
            size_t partSize = data.size() / ONCE_IPC_SEND_DATA_MAX_SIZE;
            for (size_t i = 0; i <= partSize; i++) {
                if (i != partSize) {
                    reportService->ReportInspectorTreeValue(
                        data.substr(i * ONCE_IPC_SEND_DATA_MAX_SIZE, ONCE_IPC_SEND_DATA_MAX_SIZE), i + 1, false);
                } else {
                    reportService->ReportInspectorTreeValue(data.substr(i * ONCE_IPC_SEND_DATA_MAX_SIZE), i + 1, true);
                }
            }
        } else {
            LOGW("report component event failed,process id:%{public}d", pair.first);
        }
    }
}

void UiSessionManager::NotifyAllWebPattern(bool isRegister)
{
    webFocusEventRegistered = isRegister;
    notifyWebFunction_(isRegister);
}

void UiSessionManager::SaveRegisterForWebFunction(NotifyAllWebFunction&& function)
{
    notifyWebFunction_ = std::move(function);
}

bool UiSessionManager::GetWebFocusRegistered()
{
    return webFocusEventRegistered;
}

void UiSessionManager::OnRouterChange(const std::string& path, const std::string& event)
{
    if (GetRouterChangeEventRegistered()) {
        auto value = InspectorJsonUtil::Create(true);
        value->Put("path", path.c_str());
        value->Put("event", event.c_str());
        ReportRouterChangeEvent(value->ToString());
    }
}

void UiSessionManager::SaveBaseInfo(const std::string& info)
{
    baseInfo_ = info;
}

void UiSessionManager::SendBaseInfo(int32_t processId)
{
    auto reportService = iface_cast<ReportService>(reportObjectMap_[processId]);
    if (reportService != nullptr) {
        reportService->SendBaseInfo(baseInfo_);
    }
}
} // namespace OHOS::Ace