/*
 * Copyright (c) 2021 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 "adapter/ohos/entrance/pa_container.h"

#include "adapter/ohos/entrance/ace_application_info.h"
#include "adapter/ohos/entrance/file_asset_provider_impl.h"
#include "adapter/ohos/entrance/hap_asset_provider_impl.h"
#include "adapter/ohos/entrance/pa_engine/engine/common/js_backend_engine_loader.h"
#include "adapter/ohos/entrance/pa_engine/pa_backend.h"
#include "base/log/ace_trace.h"
#include "base/log/event_report.h"
#include "base/log/log.h"
#include "base/subwindow/subwindow_manager.h"
#include "base/utils/system_properties.h"
#include "base/utils/utils.h"
#include "core/common/ace_engine.h"
#include "core/common/asset_manager_impl.h"
#include "core/common/platform_window.h"
#include "core/common/text_field_manager.h"
#include "core/common/window.h"

namespace OHOS::Ace::Platform {
namespace {

constexpr char ARK_PA_ENGINE_SHARED_LIB[] = "libace_engine_pa_ark.z.so";

const char* GetPaEngineSharedLibrary()
{
    return ARK_PA_ENGINE_SHARED_LIB;
}

} // namespace

PaContainer::PaContainer(int32_t instanceId, void* paAbility, const PaContainerOptions& options,
    std::unique_ptr<PlatformEventCallback> callback)
    : instanceId_(instanceId), paAbility_(paAbility)
{
    ACE_DCHECK(callback);
    type_ = options.type;
    hapPath_ = options.hapPath;
    workerPath_ =  options.workerPath;
    InitializeBackend(options.language);
    platformEventCallback_ = std::move(callback);
}

void PaContainer::InitializeBackend(SrcLanguage language)
{
    // create backend
    backend_ = Backend::Create();
    auto paBackend = AceType::DynamicCast<PaBackend>(backend_);
    CHECK_NULL_VOID(paBackend);

    // set JS engine, init in JS thread
    auto& loader = JsBackendEngineLoader::Get(GetPaEngineSharedLibrary());
    auto jsEngine = loader.CreateJsBackendEngine(instanceId_);
    jsEngine->AddExtraNativeObject("ability", paAbility_);
    jsEngine->SetNeedDebugBreakPoint(AceApplicationInfo::GetInstance().IsNeedDebugBreakPoint());
    jsEngine->SetDebugVersion(AceApplicationInfo::GetInstance().IsDebugVersion());
    jsEngine->SetHapPath(hapPath_);
    jsEngine->SetWorkerPath(workerPath_);
    paBackend->SetJsEngine(jsEngine);

    ACE_DCHECK(backend_);
    backend_->Initialize(type_, language);
}

RefPtr<PaContainer> PaContainer::GetContainer(int32_t instanceId)
{
    auto container = AceEngine::Get().GetContainer(instanceId);
    return AceType::DynamicCast<PaContainer>(container);
}

void PaContainer::CreateContainer(int32_t instanceId, void* paAbility, const PaContainerOptions& options,
    std::unique_ptr<PlatformEventCallback> callback)
{
    auto aceContainer = AceType::MakeRefPtr<PaContainer>(instanceId, paAbility, options, std::move(callback));
    AceEngine::Get().AddContainer(instanceId, aceContainer);

    auto back = aceContainer->GetBackend();
    CHECK_NULL_VOID(back);
    back->UpdateState(Backend::State::ON_CREATE);
}

bool PaContainer::RunPa(int32_t instanceId, const std::string& content, const OHOS::AAFwk::Want& want)
{
    LOGI("PA: PaContainer RunPa start");
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, false);
    ContainerScope scope(instanceId);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    CHECK_NULL_RETURN(aceContainer, false);
    auto paBackend = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(paBackend, false);
    paBackend->RunPa(content, want);
    return true;
}

bool PaContainer::OnCreate(int32_t instanceId, const OHOS::AAFwk::Want& want)
{
    LOGI("PA: PaContainer OnCreate start");
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, false);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    CHECK_NULL_RETURN(aceContainer, false);
    auto paBackend = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(paBackend, false);
    paBackend->OnCreate(want);
    return true;
}

bool PaContainer::OnDelete(int32_t instanceId, int64_t formId)
{
    LOGI("PA: PaContainer OnDelete start");
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, false);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    CHECK_NULL_RETURN(aceContainer, false);
    auto paBackend = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(paBackend, false);
    paBackend->OnDelete(formId);
    return true;
}

bool PaContainer::OnTriggerEvent(int32_t instanceId, int64_t formId, const std::string& message)
{
    LOGI("PA: PaContainer OnTriggerEvent start");
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, false);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    CHECK_NULL_RETURN(aceContainer, false);
    auto paBackend = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(paBackend, false);
    paBackend->OnTriggerEvent(formId, message);
    return true;
}

int32_t PaContainer::OnAcquireFormState(int32_t instanceId, const OHOS::AAFwk::Want& want)
{
    LOGI("PA: PaContainer OnAcquireFormState start");
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, -1);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    CHECK_NULL_RETURN(aceContainer, -1);
    auto paBackend = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(paBackend, -1);
    int32_t formState = paBackend->OnAcquireFormState(want);
    return formState;
}

bool PaContainer::OnUpdate(int32_t instanceId, int64_t formId)
{
    LOGI("PA: PaContainer OnUpdate start");
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, false);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    CHECK_NULL_RETURN(aceContainer, false);
    auto paBackend = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(paBackend, false);
    paBackend->OnUpdate(formId);
    return true;
}

bool PaContainer::OnCastTemptoNormal(int32_t instanceId, int64_t formId)
{
    LOGI("PA: PaContainer OnCastTemptoNormal start");
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, false);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    CHECK_NULL_RETURN(aceContainer, false);
    auto paBackend = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(paBackend, false);
    paBackend->OnCastTemptoNormal(formId);
    return true;
}

bool PaContainer::OnVisibilityChanged(int32_t instanceId, const std::map<int64_t, int32_t>& formEventsMap)
{
    LOGI("PA: PaContainer OnVisibilityChanged start");
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, false);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    CHECK_NULL_RETURN(aceContainer, false);
    auto paBackend = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(paBackend, false);
    paBackend->OnVisibilityChanged(formEventsMap);
    return true;
}

AppExecFwk::FormProviderData PaContainer::GetFormData(int32_t instanceId)
{
    LOGI("PA: PaContainer GetFormData start");
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, AppExecFwk::FormProviderData());
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    CHECK_NULL_RETURN(aceContainer, AppExecFwk::FormProviderData());
    auto paBackend = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(paBackend, AppExecFwk::FormProviderData());
    return paBackend->GetFormData();
}

void PaContainer::Destroy()
{
    RefPtr<Backend> backend;
    backend_.Swap(backend);
    CHECK_NULL_VOID(backend);
    backend->UpdateState(Backend::State::ON_DESTROY);
}

void PaContainer::DestroyContainer(int32_t instanceId)
{
    LOGI("DestroyContainer with id %{private}d", instanceId);
    SubwindowManager::GetInstance()->CloseDialog(instanceId);
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_VOID(container);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    CHECK_NULL_VOID(aceContainer);
    aceContainer->Destroy();
}

void PaContainer::AddAssetPath(int32_t instanceId, const std::string& packagePath, const std::string& hapPath,
    const std::vector<std::string>& paths)
{
    auto container = AceType::DynamicCast<PaContainer>(AceEngine::Get().GetContainer(instanceId));
    CHECK_NULL_VOID(container);
    RefPtr<AssetManagerImpl> assetManagerImpl;
    if (container->assetManager_) {
        assetManagerImpl = AceType::DynamicCast<AssetManagerImpl>(container->assetManager_);
    } else {
        assetManagerImpl = Referenced::MakeRefPtr<AssetManagerImpl>();
        container->assetManager_ = assetManagerImpl;
        AceType::DynamicCast<PaBackend>(container->GetBackend())->SetAssetManager(assetManagerImpl);
    }
    CHECK_NULL_VOID(assetManagerImpl);
    if (!hapPath.empty()) {
        auto assetProvider = AceType::MakeRefPtr<HapAssetProviderImpl>();
        if (assetProvider->Initialize(hapPath, paths)) {
            LOGI("Push AssetProvider to queue.");
            assetManagerImpl->PushBack(std::move(assetProvider));
        }
    }
    if (!packagePath.empty()) {
        auto assetProvider = AceType::MakeRefPtr<FileAssetProviderImpl>();
        if (assetProvider->Initialize(packagePath, paths)) {
            LOGI("Push AssetProvider to queue.");
            assetManagerImpl->PushBack(std::move(assetProvider));
        }
    }
}

void PaContainer::AddLibPath(int32_t instanceId, const std::vector<std::string>& libPath)
{
    auto container = AceType::DynamicCast<PaContainer>(AceEngine::Get().GetContainer(instanceId));
    CHECK_NULL_VOID(container);
    RefPtr<AssetManager> assetManagerImpl;
    if (container->assetManager_) {
        assetManagerImpl = AceType::DynamicCast<AssetManagerImpl>(container->assetManager_);
    } else {
        assetManagerImpl = Referenced::MakeRefPtr<AssetManagerImpl>();
        container->assetManager_ = assetManagerImpl;
        AceType::DynamicCast<PaBackend>(container->GetBackend())->SetAssetManager(assetManagerImpl);
    }
    CHECK_NULL_VOID(assetManagerImpl);
    assetManagerImpl->SetLibPath("default", libPath);
}

int32_t PaContainer::Insert(int32_t instanceId, const Uri& uri, const OHOS::NativeRdb::ValuesBucket& value)
{
    LOGI("Insert with id %{public}d", instanceId);
    int32_t ret = 0;
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, ret);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    auto back = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(back, ret);
    ret = back->Insert(uri, value);
    return ret;
}

std::shared_ptr<OHOS::NativeRdb::AbsSharedResultSet> PaContainer::Query(int32_t instanceId, const Uri& uri,
    const std::vector<std::string>& columns, const OHOS::NativeRdb::DataAbilityPredicates& predicates)
{
    LOGI("Query with id %{public}d", instanceId);
    std::shared_ptr<OHOS::NativeRdb::AbsSharedResultSet> ret;
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, ret);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    auto back = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(back, ret);
    ret = back->Query(uri, columns, predicates);
    return ret;
}

int32_t PaContainer::Update(int32_t instanceId, const Uri& uri, const OHOS::NativeRdb::ValuesBucket& value,
    const OHOS::NativeRdb::DataAbilityPredicates& predicates)
{
    LOGI("Update with id %{public}d", instanceId);
    int32_t ret = 0;
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, ret);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    auto back = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(back, ret);
    ret = back->Update(uri, value, predicates);
    return ret;
}

int32_t PaContainer::Delete(
    int32_t instanceId, const Uri& uri, const OHOS::NativeRdb::DataAbilityPredicates& predicates)
{
    LOGI("Delete with id %{public}d", instanceId);
    int32_t ret = 0;
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, ret);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    auto back = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(back, ret);
    ret = back->Delete(uri, predicates);
    return ret;
}

int32_t PaContainer::BatchInsert(
    int32_t instanceId, const Uri& uri, const std::vector<OHOS::NativeRdb::ValuesBucket>& values)
{
    LOGI("BatchInsert with id %{public}d", instanceId);
    int32_t ret = 0;
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, ret);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    auto back = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(back, ret);
    ret = back->BatchInsert(uri, values);
    return ret;
}

std::string PaContainer::GetType(int32_t instanceId, const Uri& uri)
{
    LOGI("GetType with id %{public}d", instanceId);
    std::string ret;
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, ret);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    auto back = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(back, ret);
    ret = back->GetType(uri);
    return ret;
}

std::vector<std::string> PaContainer::GetFileTypes(
    int32_t instanceId, const Uri& uri, const std::string& mimeTypeFilter)
{
    LOGI("GetFileTypes with id %{public}d", instanceId);
    std::vector<std::string> ret;
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, ret);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    auto back = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(back, ret);
    ret = back->GetFileTypes(uri, mimeTypeFilter);
    return ret;
}

int32_t PaContainer::OpenFile(int32_t instanceId, const Uri& uri, const std::string& mode)
{
    LOGI("OpenFile with id %{public}d", instanceId);
    int32_t ret = 0;
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, ret);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    auto back = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(back, ret);
    ret = back->OpenFile(uri, mode);
    return ret;
}

int32_t PaContainer::OpenRawFile(int32_t instanceId, const Uri& uri, const std::string& mode)
{
    LOGI("OpenRawFile with id %{public}d", instanceId);
    int32_t ret = 0;
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, ret);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    auto back = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(back, ret);
    ret = back->OpenRawFile(uri, mode);
    return ret;
}

Uri PaContainer::NormalizeUri(int32_t instanceId, const Uri& uri)
{
    LOGI("NormalizeUri with id %{public}d", instanceId);
    Uri ret("");
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, ret);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    auto back = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(back, ret);
    ret = back->NormalizeUri(uri);
    return ret;
}

Uri PaContainer::DenormalizeUri(int32_t instanceId, const Uri& uri)
{
    LOGI("DenormalizeUri with id %{public}d", instanceId);
    Uri ret("");
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, ret);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    auto back = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(back, ret);
    ret = back->DenormalizeUri(uri);
    return ret;
}

sptr<IRemoteObject> PaContainer::OnConnect(int32_t instanceId, const OHOS::AAFwk::Want& want)
{
    LOGI("OnConnect with id %{private}d", instanceId);
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, nullptr);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    auto back = aceContainer->GetBackend();
    CHECK_NULL_RETURN(back, nullptr);
    auto paBackend = AceType::DynamicCast<PaBackend>(back);
    CHECK_NULL_RETURN(paBackend, nullptr);
    return paBackend->OnConnect(want);
    return nullptr;
}

void PaContainer::OnDisConnect(int32_t instanceId, const OHOS::AAFwk::Want& want)
{
    LOGI("OnDisConnect with id %{private}d", instanceId);
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_VOID(container);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    auto back = aceContainer->GetBackend();
    CHECK_NULL_VOID(back);
    auto paBackend = AceType::DynamicCast<PaBackend>(back);
    CHECK_NULL_VOID(paBackend);
    paBackend->OnDisConnect(want);
}

void PaContainer::OnCommand(const OHOS::AAFwk::Want &want, int startId, int32_t instanceId)
{
    LOGI("OnCommand with id %{private}d", instanceId);
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_VOID(container);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    auto back = aceContainer->GetBackend();
    CHECK_NULL_VOID(back);
    auto paBackend = AceType::DynamicCast<PaBackend>(back);
    CHECK_NULL_VOID(paBackend);
    paBackend->OnCommand(want, startId);
}

bool PaContainer::OnShare(int32_t instanceId, int64_t formId, OHOS::AAFwk::WantParams &wantParams)
{
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, false);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    CHECK_NULL_RETURN(aceContainer, false);
    auto paBackend = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(paBackend, false);
    return paBackend->OnShare(formId, wantParams);
}

std::shared_ptr<AppExecFwk::PacMap> PaContainer::Call(int32_t instanceId, const Uri& uri,
    const std::string& method, const std::string& arg, const AppExecFwk::PacMap& pacMap)
{
    std::shared_ptr<AppExecFwk::PacMap> ret = nullptr;
    auto container = AceEngine::Get().GetContainer(instanceId);
    CHECK_NULL_RETURN(container, ret);
    auto aceContainer = AceType::DynamicCast<PaContainer>(container);
    auto back = AceType::DynamicCast<PaBackend>(aceContainer->GetBackend());
    CHECK_NULL_RETURN(back, ret);
    ret = back->Call(uri, method, arg, pacMap);
    return ret;
}
} // namespace OHOS::Ace::Platform