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

#include "fms_log_wrapper.h"
#include "form_bms_helper.h"
#include "form_data_mgr.h"
#include "form_data_proxy_record.h"
#include "form_util.h"
#include "form_mgr_errors.h"

namespace OHOS {
namespace AppExecFwk {
FormDataProxyMgr::FormDataProxyMgr()
{}

FormDataProxyMgr::~FormDataProxyMgr()
{}

ErrCode FormDataProxyMgr::SubscribeFormData(int64_t formId, const std::vector<FormDataProxy> &formDataProxies,
    const AAFwk::Want &want)
{
    HILOG_DEBUG("subscribe form data. formId:%{public}s, proxy data size:%{public}zu",
        std::to_string(formId).c_str(), formDataProxies.size());
    if (formDataProxies.empty()) {
        HILOG_DEBUG("empty formDataProxies");
        return ERR_OK;
    }

    FormRecord formRecord;
    if (!FormDataMgr::GetInstance().GetFormRecord(formId, formRecord)) {
        HILOG_ERROR("not exist such form:%{public}" PRId64 ".", formId);
        return ERR_APPEXECFWK_FORM_NOT_EXIST_ID;
    }

    if (!formRecord.isDataProxy) {
        HILOG_DEBUG("not data proxy form, formId:%{public}" PRId64 ".", formId);
        return ERR_OK;
    }

    {
        std::lock_guard<std::mutex> lock(formDataProxyRecordMutex_);
        auto search = formDataProxyRecordMap_.find(formId);
        if (search != formDataProxyRecordMap_.end()) {
            if (search->second != nullptr) {
                HILOG_DEBUG("the form has already subscribed, formId:%{public}s", std::to_string(formId).c_str());
                search->second->UnsubscribeFormData();
            }
        }
    }

    ApplicationInfo appInfo;
    if (FormBmsHelper::GetInstance().GetApplicationInfo(formRecord.bundleName, FormUtil::GetCurrentAccountId(),
        appInfo) != ERR_OK) {
        HILOG_ERROR("get app info failed");
        return ERR_APPEXECFWK_FORM_COMMON_CODE;
    }
    std::shared_ptr<FormDataProxyRecord> formDataProxyRecord = std::make_shared<FormDataProxyRecord>(formId,
        formRecord.bundleName, formRecord.uiSyntax, appInfo.accessTokenId, formRecord.uid);
    formDataProxyRecord->SetWant(want);
    auto ret = formDataProxyRecord->SubscribeFormData(formDataProxies);
    if (ret != ERR_OK) {
        HILOG_ERROR("SubscribeFormData failed");
        return ret;
    }
    std::lock_guard<std::mutex> lock(formDataProxyRecordMutex_);
    formDataProxyRecordMap_[formId] = formDataProxyRecord;
    return ERR_OK;
}

ErrCode FormDataProxyMgr::UnsubscribeFormData(int64_t formId)
{
    HILOG_DEBUG("unsubscribe form data. formId:%{public}s", std::to_string(formId).c_str());
    std::lock_guard<std::mutex> lock(formDataProxyRecordMutex_);
    auto search = formDataProxyRecordMap_.find(formId);
    if (search != formDataProxyRecordMap_.end()) {
        if (search->second != nullptr) {
            search->second->UnsubscribeFormData();
        }
        formDataProxyRecordMap_.erase(formId);
    }

    return ERR_OK;
}

void FormDataProxyMgr::UpdateSubscribeFormData(int64_t formId, const std::vector<FormDataProxy> &formDataProxies)
{
    HILOG_DEBUG("update subscribe form data. formId:%{public}s", std::to_string(formId).c_str());
    std::lock_guard<std::mutex> lock(formDataProxyRecordMutex_);
    auto search = formDataProxyRecordMap_.find(formId);
    if (search != formDataProxyRecordMap_.end()) {
        if (search->second != nullptr) {
            search->second->UpdateSubscribeFormData(formDataProxies);
        }
    }
}

void FormDataProxyMgr::ProduceFormDataProxies(int64_t formId, const std::vector<FormDataProxy> &formDataProxies)
{
    std::lock_guard<std::mutex> lock(formDataProxiesMutex_);
    formDataProxiesMap_[formId] = formDataProxies;
}

bool FormDataProxyMgr::ConsumeFormDataProxies(int64_t formId, std::vector<FormDataProxy> &formDataProxies)
{
    std::lock_guard<std::mutex> lock(formDataProxiesMutex_);
    auto search = formDataProxiesMap_.find(formId);
    if (search == formDataProxiesMap_.end()) {
        HILOG_DEBUG("no form data proxies, formId:%{public}s", std::to_string(formId).c_str());
        return false;
    }
    formDataProxies = search->second;
    formDataProxiesMap_.erase(search);
    return true;
}

void FormDataProxyMgr::EnableSubscribeFormData(const std::vector<int64_t> &formIds)
{
    std::lock_guard<std::mutex> lock(formDataProxyRecordMutex_);
    for (const auto &formId : formIds) {
        auto search = formDataProxyRecordMap_.find(formId);
        if (search != formDataProxyRecordMap_.end()) {
            if (search->second != nullptr) {
                search->second->EnableSubscribeFormData();
            }
        }
    }
}

void FormDataProxyMgr::DisableSubscribeFormData(const std::vector<int64_t> &formIds)
{
    std::lock_guard<std::mutex> lock(formDataProxyRecordMutex_);
    for (const auto &formId : formIds) {
        auto search = formDataProxyRecordMap_.find(formId);
        if (search != formDataProxyRecordMap_.end()) {
            if (search->second != nullptr) {
                search->second->DisableSubscribeFormData();
                HILOG_INFO("formId:%{public}s", std::to_string(formId).c_str());
            }
        }
    }
}

void FormDataProxyMgr::RetryFailureSubscribes()
{
    std::lock_guard<std::mutex> lock(formDataProxyRecordMutex_);
    for (auto& record : formDataProxyRecordMap_) {
        record.second->RetryFailureSubscribes();
    }
}

void FormDataProxyMgr::GetFormSubscribeInfo(
    const int64_t formId, std::vector<std::string> &subscribedKeys, int32_t &count)
{
    auto search = formDataProxyRecordMap_.find(formId);
    if (search != formDataProxyRecordMap_.end()) {
        if (search->second != nullptr) {
            search->second->GetFormSubscribeInfo(subscribedKeys, count);
        }
    }
}
} // namespace AppExecFwk
} // namespace OHOS