xref: /foundation/filemanagement/app_file_service/frameworks/native/backup_ext/src/ext_backup_js.cpp

/*
 * Copyright (c) 2022-2025 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 "ext_backup_js.h"

#include <cstdio>
#include <memory>
#include <sstream>
#include <sys/stat.h>
#include <sys/types.h>

#include "bundle_mgr_client.h"
#include "ext_backup_context_js.h"
#include "js_extension_context.h"
#include "js_native_api.h"
#include "js_native_api_types.h"
#include "js_runtime.h"
#include "js_runtime_utils.h"
#include "napi/native_api.h"
#include "napi/native_node_api.h"
#include "napi_common_util.h"
#include "napi_common_want.h"
#include "napi_remote_object.h"
#include "unique_fd.h"

#include "b_anony/b_anony.h"
#include "b_error/b_error.h"
#include "b_error/b_excep_utils.h"
#include "b_json/b_json_cached_entity.h"
#include "b_json/b_json_entity_extension_config.h"
#include "b_radar/b_radar.h"
#include "b_resources/b_constants.h"
#include "directory_ex.h"
#include "ext_extension.h"
#include "filemgmt_libhilog.h"

namespace OHOS::FileManagement::Backup {
using namespace std;
constexpr size_t ARGC_ONE = 1;
static std::mutex g_extBackupValidLock;
static int32_t g_extBackupCount = 0;

static string GetSrcPath(const AppExecFwk::AbilityInfo &info)
{
    using AbilityRuntime::Extension;
    stringstream ss;

    // API9(stage model) 中通过 $(module)$(name)/$(srcEntrance/(.*$)/(.abc)) 获取自定义插件路径
    if (!info.srcEntrance.empty()) {
        ss << info.moduleName << '/' << string(info.srcEntrance, 0, info.srcEntrance.rfind(".")) << ".abc";
        return ss.str();
    }
    return "";
}

static napi_status DealNapiStrValue(napi_env env, const napi_value napi_StrValue, std::string &result)
{
    HILOGI("Start DealNapiStrValue");
    std::string buffer = "";
    size_t bufferSize = 0;
    napi_status status = napi_ok;
    status = napi_get_value_string_utf8(env, napi_StrValue, nullptr, -1, &bufferSize);
    if (status != napi_ok) {
        HILOGE("Can not get buffer size");
        return status;
    }
    buffer.reserve(bufferSize + 1);
    buffer.resize(bufferSize);
    if (bufferSize > 0) {
        status = napi_get_value_string_utf8(env, napi_StrValue, buffer.data(), bufferSize + 1, &bufferSize);
        if (status != napi_ok) {
            HILOGE("Can not get buffer value");
            return status;
        }
    }
    result = buffer;
    return status;
}

static napi_status DealNapiException(napi_env env, napi_value &exception, std::string &exceptionInfo)
{
    HILOGI("call DealNapiException start.");
    napi_status status = napi_get_and_clear_last_exception(env, &exception);
    if (status != napi_ok) {
        HILOGE("call napi_get_and_clear_last_exception failed.");
        return status;
    }
    status = DealNapiStrValue(env, exception, exceptionInfo);
    if (status != napi_ok) {
        HILOGE("call DealNapiStrValue failed.");
        return status;
    }
    HILOGI("call DealNapiException end, exception info = %{public}s.", exceptionInfo.c_str());
    return status;
}

static napi_value PromiseCallback(napi_env env, napi_callback_info info)
{
    std::lock_guard<std::mutex> lock(g_extBackupValidLock);
    if (g_extBackupCount <= 0) {
        HILOGE("ExtBackup is invalid, count=%{public}d", g_extBackupCount);
        return nullptr;
    }
    HILOGI("Promise callback.");
    void *data = nullptr;
    if (napi_get_cb_info(env, info, nullptr, 0, nullptr, &data) != napi_ok) {
        HILOGE("Failed to get callback info.");
        return nullptr;
    }
    auto *callbackInfo = static_cast<CallbackInfo *>(data);
    if (callbackInfo == nullptr) {
        HILOGE("CallbackInfo is nullptr");
        return nullptr;
    }
    string str;
    callbackInfo->callback(BError(BError::Codes::OK), str);
    data = nullptr;
    return nullptr;
}

static napi_value PromiseCatchCallback(napi_env env, napi_callback_info info)
{
    HILOGI("Promise catch callback begin.");
    size_t argc = 1;
    napi_value argv = {nullptr};
    void *data = nullptr;
    NAPI_CALL_NO_THROW(napi_get_cb_info(env, info, &argc, &argv, nullptr, &data), nullptr);
    string exceptionInfo;
    DealNapiStrValue(env, argv, exceptionInfo);
    HILOGI("Catch exception info is %{public}s.", exceptionInfo.c_str());
    auto *callbackInfo = static_cast<CallbackInfo *>(data);
    if (callbackInfo == nullptr) {
        HILOGE("CallbackInfo is nullptr");
        return nullptr;
    }
    napi_status throwStatus = napi_fatal_exception(env, argv);
    if (throwStatus != napi_ok) {
        HILOGE("Failed to throw an exception, %{public}d", throwStatus);
        return nullptr;
    }
    std::lock_guard<std::mutex> lock(g_extBackupValidLock);
    if (g_extBackupCount <= 0) {
        HILOGE("ExtBackup is invalid, count=%{public}d", g_extBackupCount);
        data = nullptr;
        return nullptr;
    }
    callbackInfo->callback(BError(BError::Codes::EXT_THROW_EXCEPTION), exceptionInfo);
    data = nullptr;
    HILOGI("Promise catch callback end.");
    return nullptr;
}

static napi_value PromiseCallbackEx(napi_env env, napi_callback_info info)
{
    std::lock_guard<std::mutex> lock(g_extBackupValidLock);
    if (g_extBackupCount <= 0) {
        HILOGE("ExtBackup is invalid, count=%{public}d", g_extBackupCount);
        return nullptr;
    }
    HILOGI("PromiseEx callback.");
    void *data = nullptr;
    std::string str;
    size_t argc = 1;
    napi_value argv = {nullptr};
    NAPI_CALL_NO_THROW(napi_get_cb_info(env, info, &argc, &argv, nullptr, &data), nullptr);
    auto *callbackInfoEx = static_cast<CallbackInfoEx *>(data);
    if (callbackInfoEx == nullptr) {
        HILOGE("CallbackInfo is nullPtr");
        return nullptr;
    }
    DealNapiStrValue(env, argv, str);
    callbackInfoEx->callbackParam(BError(BError::Codes::OK), str);
    data = nullptr;
    return nullptr;
}

static napi_value PromiseCatchCallbackEx(napi_env env, napi_callback_info info)
{
    HILOGI("PromiseEx catch callback begin.");
    void *data = nullptr;
    size_t argc = 1;
    napi_value argv = {nullptr};
    NAPI_CALL_NO_THROW(napi_get_cb_info(env, info, &argc, &argv, nullptr, &data), nullptr);
    string exceptionInfo;
    DealNapiStrValue(env, argv, exceptionInfo);
    HILOGI("Catch exception info is %{public}s.", exceptionInfo.c_str());
    auto *callbackInfoEx = static_cast<CallbackInfoEx *>(data);
    if (callbackInfoEx == nullptr) {
        HILOGE("CallbackInfo is nullPtr");
        return nullptr;
    }
    napi_status throwStatus = napi_fatal_exception(env, argv);
    if (throwStatus != napi_ok) {
        HILOGE("Failed to throw an exception, %{public}d", throwStatus);
        return nullptr;
    }
    std::lock_guard<std::mutex> lock(g_extBackupValidLock);
    if (g_extBackupCount <= 0) {
        HILOGE("ExtBackup is invalid, count=%{public}d", g_extBackupCount);
        data = nullptr;
        return nullptr;
    }
    callbackInfoEx->callbackParam(BError(BError::Codes::EXT_THROW_EXCEPTION), exceptionInfo);
    data = nullptr;
    HILOGI("PromiseEx catch callback end.");
    return nullptr;
}

static bool CheckPromise(napi_env env, napi_value value)
{
    if (value == nullptr) {
        HILOGE("CheckPromise, result is null, no need to call promise.");
        return false;
    }
    bool isPromise = false;
    if (napi_is_promise(env, value, &isPromise) != napi_ok) {
        HILOGE("CheckPromise, result is not promise, no need to call promise.");
        return false;
    }
    return isPromise;
}

static bool CallCatchPromise(AbilityRuntime::JsRuntime &jsRuntime, napi_value result, CallbackInfo *callbackInfo)
{
    HILOGI("CallCatchPromise Begin.");
    AbilityRuntime::HandleScope handleScope(jsRuntime);
    auto env = jsRuntime.GetNapiEnv();
    napi_value method = nullptr;
    if (napi_get_named_property(env, result, "catch", &method) != napi_ok) {
        HILOGE("CallCatchPromise, Failed to get method catch");
        return false;
    }
    bool isCallable = false;
    if (napi_is_callable(env, method, &isCallable) != napi_ok) {
        HILOGE("CallCatchPromise, Failed to check method then is callable");
        return false;
    }
    if (!isCallable) {
        HILOGE("CallCatchPromise, property then is not callable.");
        return false;
    }
    napi_value ret;
    if (napi_create_function(env, "promiseCatchCallback", strlen("promiseCatchCallback"), PromiseCatchCallback,
        callbackInfo, &ret) != napi_ok) {
        HILOGE("napi create function failed");
        return false;
    }
    napi_value argv[1] = {ret};
    if (napi_call_function(env, result, method, 1, argv, nullptr) != napi_ok) {
        HILOGE("napi call function failed");
        return false;
    }
    return true;
}

static bool CallPromise(AbilityRuntime::JsRuntime &jsRuntime, napi_value result, CallbackInfo *callbackInfo)
{
    AbilityRuntime::HandleScope handleScope(jsRuntime);
    auto env = jsRuntime.GetNapiEnv();
    napi_value method = nullptr;
    if (napi_get_named_property(env, result, "then", &method) != napi_ok) {
        HILOGE("CallPromise, Failed to get method then");
        return false;
    }
    bool isCallable = false;
    if (napi_is_callable(env, method, &isCallable) != napi_ok) {
        HILOGE("CallPromise, Failed to check method then is callable");
        return false;
    }
    if (!isCallable) {
        HILOGE("CallPromise, property then is not callable.");
        return false;
    }
    napi_value ret;
    if (napi_create_function(env, "promiseCallback", strlen("promiseCallback"), PromiseCallback, callbackInfo, &ret) !=
        napi_ok) {
        HILOGE("napi create function failed");
        return false;
    }
    napi_value argv[1] = {ret};
    if (napi_call_function(env, result, method, 1, argv, nullptr) != napi_ok) {
        HILOGE("napi call function failed");
        return false;
    }
    if (!CallCatchPromise(jsRuntime, result, callbackInfo)) {
        HILOGE("CallCatchPromise failed.");
        return false;
    }
    return true;
}

static bool CallCatchPromiseEx(AbilityRuntime::JsRuntime &jsRuntime, napi_value result, CallbackInfoEx *callbackInfoEx)
{
    HILOGI("CallCatchPromiseEx Begin.");
    AbilityRuntime::HandleScope handleScope(jsRuntime);
    auto env = jsRuntime.GetNapiEnv();
    napi_value method = nullptr;
    if (napi_get_named_property(env, result, "catch", &method) != napi_ok) {
        HILOGE("CallCatchPromiseEx, Failed to get method catch");
        return false;
    }
    bool isCallable = false;
    if (napi_is_callable(env, method, &isCallable) != napi_ok) {
        HILOGE("CallCatchPromiseEx, Failed to check method then is callable");
        return false;
    }
    if (!isCallable) {
        HILOGE("CallCatchPromiseEx, property then is not callable.");
        return false;
    }
    napi_value ret;
    if (napi_create_function(env, "promiseCatchCallbackEx", strlen("promiseCatchCallbackEx"), PromiseCatchCallbackEx,
        callbackInfoEx, &ret) != napi_ok) {
        HILOGE("napi create function failed");
        return false;
    }
    napi_value argv[1] = {ret};
    if (napi_call_function(env, result, method, 1, argv, nullptr) != napi_ok) {
        HILOGE("napi call function failed");
        return false;
    }
    return true;
}

static bool CallPromiseEx(AbilityRuntime::JsRuntime &jsRuntime, napi_value result, CallbackInfoEx *callbackInfoEx)
{
    AbilityRuntime::HandleScope handleScope(jsRuntime);
    auto env = jsRuntime.GetNapiEnv();
    napi_value method = nullptr;
    if (napi_get_named_property(env, result, "then", &method) != napi_ok) {
        HILOGE("CallPromise, Failed to get method then");
        return false;
    }
    bool isCallable = false;
    if (napi_is_callable(env, method, &isCallable) != napi_ok) {
        HILOGE("CallPromise, Failed to check method then is callable");
        return false;
    }
    if (!isCallable) {
        HILOGE("CallPromise, property then is not callable.");
        return false;
    }
    napi_value ret;
    if (napi_create_function(env, "promiseCallbackEx", strlen("promiseCallbackEx"), PromiseCallbackEx, callbackInfoEx,
        &ret) != napi_ok) {
        HILOGE("napi create function failed");
        return false;
    }
    napi_value argv[1] = {ret};
    if (napi_call_function(env, result, method, 1, argv, nullptr) != napi_ok) {
        HILOGE("napi call function failed");
        return false;
    }
    if (!CallCatchPromiseEx(jsRuntime, result, callbackInfoEx)) {
        HILOGE("CallCatchPromiseEx failed.");
        return false;
    }
    return true;
}

static bool CallPromiseEx(AbilityRuntime::JsRuntime &jsRuntime, napi_value result,
    CallbackInfoBackup *callbackInfoBackup)
{
    AbilityRuntime::HandleScope handleScope(jsRuntime);
    auto env = jsRuntime.GetNapiEnv();
    napi_value method = nullptr;
    if (napi_get_named_property(env, result, "then", &method) != napi_ok) {
        HILOGE("CallPromise, Failed to get method then");
        return false;
    }
    bool isCallable = false;
    if (napi_is_callable(env, method, &isCallable) != napi_ok) {
        HILOGE("CallPromise, Failed to check method then is callable");
        return false;
    }
    if (!isCallable) {
        HILOGE("CallPromise, property then is not callable.");
        return false;
    }
    napi_value ret;
    if (napi_create_function(env, "promiseCallbackEx", strlen("promiseCallbackEx"), PromiseCallbackEx,
        callbackInfoBackup, &ret) != napi_ok) {
        HILOGE("napi create function failed");
        return false;
    }
    napi_value argv[1] = {ret};
    if (napi_call_function(env, result, method, 1, argv, nullptr) != napi_ok) {
        HILOGE("napi call function failed");
        return false;
    }
    return true;
}

void ExtBackupJs::Init(const shared_ptr<AppExecFwk::AbilityLocalRecord> &record,
                       const shared_ptr<AppExecFwk::OHOSApplication> &application,
                       shared_ptr<AppExecFwk::AbilityHandler> &handler,
                       const sptr<IRemoteObject> &token)
{
    HILOGI("Init the BackupExtensionAbility(JS)");
    try {
        ExtBackup::Init(record, application, handler, token);
        BExcepUltils::BAssert(abilityInfo_, BError::Codes::EXT_BROKEN_FRAMEWORK, "Invalid abilityInfo_");
        // 获取应用扩展的 BackupExtensionAbility 的路径
        const AppExecFwk::AbilityInfo &info = *abilityInfo_;
        string bundleName = info.bundleName;
        InitTempPath(bundleName);
        string moduleName(info.moduleName + "::" + info.name);
        string modulePath = GetSrcPath(info);
        int moduleType = static_cast<int>(info.type);
        HILOGI("Try to load %{public}s's %{public}s(type %{public}d) from %{public}s", bundleName.c_str(),
               moduleName.c_str(), moduleType, modulePath.c_str());

        // 加载用户扩展 BackupExtensionAbility 到 JS 引擎，并将之暂存在 jsObj_ 中。注意，允许加载失败，往后执行默认逻辑
        AbilityRuntime::HandleScope handleScope(jsRuntime_);
        jsObj_ = jsRuntime_.LoadModule(moduleName, modulePath, info.hapPath,
                                       abilityInfo_->compileMode == AbilityRuntime::CompileMode::ES_MODULE);
        if (jsObj_ == nullptr) {
            HILOGW("Oops! There's no custom BackupExtensionAbility");
            return;
        }
        HILOGI("Wow! Here's a custsom BackupExtensionAbility");
        ExportJsContext();
    } catch (const BError &e) {
        HILOGE("%{public}s", e.what());
    } catch (const exception &e) {
        HILOGE("%{public}s", e.what());
    }
}

napi_value AttachBackupExtensionContext(napi_env env, void *value, void *)
{
    HILOGI("AttachBackupExtensionContext");
    if (value == nullptr || env == nullptr) {
        HILOG_WARN("invalid parameter.");
        return nullptr;
    }
    auto ptr = reinterpret_cast<std::weak_ptr<ExtBackupContext> *>(value)->lock();
    if (ptr == nullptr) {
        HILOGE("invalid context.");
        return nullptr;
    }
    auto object = CreateExtBackupJsContext(env, ptr);
    if (object == nullptr) {
        HILOGE("Failed to get js backup extension context");
        return nullptr;
    }
    auto contextRef =
        AbilityRuntime::JsRuntime::LoadSystemModuleByEngine(env, "application.BackupExtensionContext", &object, 1);
    if (contextRef == nullptr) {
        HILOGE("Failed to load BackupExtensionContext.");
        return nullptr;
    }
    napi_value contextObj = contextRef->GetNapiValue();
    napi_coerce_to_native_binding_object(env, contextObj, AbilityRuntime::DetachCallbackFunc,
                                         AttachBackupExtensionContext, value, nullptr);

    auto workContext = new (std::nothrow) std::weak_ptr<ExtBackupContext>(ptr);
    if (workContext == nullptr) {
        HILOGE("Failed to get backup extension context");
        return nullptr;
    }
    napi_status status = napi_wrap(
        env, contextObj, workContext,
        [](napi_env, void *data, void *) {
            HILOG_DEBUG("Finalizer for weak_ptr base context is called");
            delete static_cast<std::weak_ptr<ExtBackupContext> *>(data);
        },
        nullptr, nullptr);
    if (status != napi_ok) {
        HILOG_DEBUG("Failed to wrap js instance");
        delete workContext;
        workContext = nullptr;
    }
    return contextObj;
}

ExtBackupJs::ExtBackupJs(AbilityRuntime::JsRuntime &jsRuntime) : jsRuntime_(jsRuntime)
{
    std::lock_guard<std::mutex> lock(g_extBackupValidLock);
    g_extBackupCount += 1;
    HILOGI("ExtBackupJs::ExtBackupJs, count=%{public}d.", g_extBackupCount);
}

ExtBackupJs::~ExtBackupJs()
{
    jsRuntime_.FreeNativeReference(std::move(jsObj_));
    std::lock_guard<std::mutex> lock(g_extBackupValidLock);
    g_extBackupCount -= 1;
    HILOGI("ExtBackupJs::~ExtBackupJs, count=%{public}d.", g_extBackupCount);
}

void ExtBackupJs::ExportJsContext(void)
{
    auto env = jsRuntime_.GetNapiEnv();
    if (jsObj_ == nullptr) {
        HILOGE("Failed to get js object.");
        return;
    }
    napi_value obj = jsObj_->GetNapiValue();
    if (obj == nullptr) {
        HILOGE("Failed to get BackupExtAbility object");
        return;
    }

    auto context = GetContext();
    if (context == nullptr) {
        HILOGE("Failed to get context");
        return;
    }

    HILOGI("CreateBackupExtAbilityContext");
    napi_value contextObj = CreateExtBackupJsContext(env, context);
    auto contextRef = jsRuntime_.LoadSystemModule("application.BackupExtensionContext", &contextObj, ARGC_ONE);
    if (!contextRef) {
        HILOGE("context is nullptr");
        return;
    }
    contextObj = contextRef->GetNapiValue();
    HILOGI("Bind context");
    context->Bind(jsRuntime_, contextRef.release());
    napi_set_named_property(env, obj, "context", contextObj);

    auto workContext = new (std::nothrow) std::weak_ptr<ExtBackupContext>(context);
    if (workContext == nullptr) {
        HILOGE("Failed to create ExtBackupContext.");
        return;
    }
    napi_coerce_to_native_binding_object(env, contextObj, AbilityRuntime::DetachCallbackFunc,
                                         AttachBackupExtensionContext, workContext, nullptr);
    HILOGI("Set backup extension ability context pointer is nullptr: %{public}d", context.get() == nullptr);
    napi_status status = napi_wrap(
        env, contextObj, workContext,
        [](napi_env, void *data, void *) {
            HILOG_DEBUG("Finalizer for weak_ptr base context is called");
            delete static_cast<std::weak_ptr<ExtBackupContext> *>(data);
        },
        nullptr, nullptr);
    if (status != napi_ok) {
        HILOG_DEBUG("Failed to wrap js instance");
        delete workContext;
        workContext = nullptr;
    }
}

[[maybe_unused]] tuple<ErrCode, napi_value> ExtBackupJs::CallObjectMethod(string_view name,
                                                                          const vector<napi_value> &argv)
{
    HILOGI("Call %{public}s", name.data());
    return {BError(BError::Codes::OK).GetCode(), nullptr};
}

ExtBackupJs *ExtBackupJs::Create(const unique_ptr<AbilityRuntime::Runtime> &runtime)
{
    HILOGI("Create as an BackupExtensionAbility(JS)");
    return new ExtBackupJs(static_cast<AbilityRuntime::JsRuntime &>(*runtime));
}

ErrCode ExtBackupJs::OnBackup(function<void(ErrCode, std::string)> callback,
    std::function<void(ErrCode, const std::string)> callbackEx)
{
    HILOGI("BackupExtensionAbility(JS) OnBackup ex");
    BExcepUltils::BAssert(jsObj_, BError::Codes::EXT_BROKEN_FRAMEWORK,
        "The app does not provide the onBackup interface.");
    BExcepUltils::BAssert(callback, BError::Codes::EXT_BROKEN_FRAMEWORK, "OnBackup callback is nullptr.");
    BExcepUltils::BAssert(callbackEx, BError::Codes::EXT_BROKEN_FRAMEWORK, "OnBackup callbackEx is nullptr.");
    callExtDefaultFunc_.store(false);
    callJsExMethodDone_.store(false);
    callbackInfo_ = std::make_shared<CallbackInfo>(callback);
    callbackInfoEx_ = std::make_shared<CallbackInfoEx>(callbackEx);
    return CallJsOnBackupEx();
}

ErrCode ExtBackupJs::CallJsOnBackupEx()
{
    HILOGI("Start call app js method onBackupEx");
    auto retParser = [jsRuntime{ &jsRuntime_ }, callbackInfoEx { callbackInfoEx_ }](napi_env envir,
        napi_value result) -> bool {
        if (!CheckPromise(envir, result)) {
            string str;
            bool isExceptionPending;
            napi_is_exception_pending(envir, &isExceptionPending);
            HILOGI("napi exception pending = %{public}d.", isExceptionPending);
            if (!callbackInfoEx) {
                HILOGE("callbackInfoEx is nullptr");
                return false;
            }
            if (isExceptionPending) {
                napi_value exception;
                DealNapiException(envir, exception, str);
                napi_fatal_exception(envir, exception);
                callbackInfoEx->callbackParam(BError(BError::Codes::EXT_THROW_EXCEPTION), str);
            } else {
                DealNapiStrValue(envir, result, str);
                callbackInfoEx->callbackParam(BError(BError::Codes::OK), str);
            }
            return true;
        }
        HILOGI("CheckPromise onBackupEx ok");
        return CallPromiseEx(*jsRuntime, result, callbackInfoEx.get());
    };
    auto errCode = CallJsMethod("onBackupEx", jsRuntime_, jsObj_.get(), ParseBackupExInfo(), retParser);
    if (errCode != ERR_OK) {
        HILOGE("Call onBackupEx error");
        return errCode;
    }
    HILOGI("Check call onBackupEx load");
    std::unique_lock<std::mutex> lock(callJsMutex_);
    callJsCon_.wait(lock, [this] { return callJsExMethodDone_.load(); });

    if (!callExtDefaultFunc_.load()) {
        HILOGI("Call Js method onBackupEx done");
        return ERR_OK;
    }
    return CallJsOnBackup();
}

ErrCode ExtBackupJs::CallJsOnBackup()
{
    HILOGI("Start call app js method onBackup");
    auto retParser = [jsRuntime {&jsRuntime_}, callbackInfo {callbackInfo_}](napi_env env,
        napi_value result) -> bool {
        if (!CheckPromise(env, result)) {
            string str;
            bool isExceptionPending;
            napi_is_exception_pending(env, &isExceptionPending);
            HILOGI("napi exception pending = %{public}d.", isExceptionPending);
            if (!callbackInfo) {
                HILOGE("callbackInfo is nullptr");
                return false;
            }
            if (isExceptionPending) {
                napi_value exception;
                DealNapiException(env, exception, str);
                napi_fatal_exception(env, exception);
                callbackInfo->callback(BError(BError::Codes::EXT_THROW_EXCEPTION), str);
            } else {
                callbackInfo->callback(BError(BError::Codes::OK), str);
            }
            return true;
        }
        HILOGI("CheckPromise Js Method onBackup ok.");
        return CallPromise(*jsRuntime, result, callbackInfo.get());
    };
    auto errCode = CallJsMethod("onBackup", jsRuntime_, jsObj_.get(), {}, retParser);
    if (errCode != ERR_OK) {
        HILOGE("CallJsMethod error, code:%{public}d.", errCode);
    }
    return errCode;
}

ErrCode ExtBackupJs::OnRestore(function<void(ErrCode, std::string)> callback,
    std::function<void(ErrCode, const std::string)> callbackEx)
{
    HILOGI("BackupExtensionAbility(JS) OnRestore.");
    BExcepUltils::BAssert(jsObj_, BError::Codes::EXT_BROKEN_FRAMEWORK,
        "The app does not provide the onRestore interface.");
    BExcepUltils::BAssert(callback, BError::Codes::EXT_BROKEN_FRAMEWORK, "OnRestore callback is nullptr.");
    BExcepUltils::BAssert(callbackEx, BError::Codes::EXT_BROKEN_FRAMEWORK, "OnRestore callbackEx is nullptr.");
    callExtDefaultFunc_.store(false);
    callJsExMethodDone_.store(false);
    callbackInfo_ = std::make_shared<CallbackInfo>(callback);
    callbackInfoEx_ = std::make_shared<CallbackInfoEx>(callbackEx);
    return CallJSRestoreEx();
}

ErrCode ExtBackupJs::CallJSRestoreEx()
{
    HILOGI("Start call app js method onRestoreEx");
    auto retParser = [jsRuntime {&jsRuntime_}, callbackInfoEx {callbackInfoEx_}](napi_env envir, napi_value result) ->
        bool {
        if (!CheckPromise(envir, result)) {
            string str;
            bool isExceptionPending;
            napi_is_exception_pending(envir, &isExceptionPending);
            HILOGI("napi exception pending = %{public}d.", isExceptionPending);
            if (!callbackInfoEx) {
                HILOGE("callbackInfoEx is nullptr");
                return false;
            }
            if (isExceptionPending) {
                napi_value exception;
                DealNapiException(envir, exception, str);
                napi_fatal_exception(envir, exception);
                callbackInfoEx->callbackParam(BError(BError::Codes::EXT_THROW_EXCEPTION), str);
            } else {
                DealNapiStrValue(envir, result, str);
                callbackInfoEx->callbackParam(BError(BError::Codes::OK), str);
            }
            return true;
        }
        HILOGI("CheckPromise onRestoreEx ok");
        return CallPromiseEx(*jsRuntime, result, callbackInfoEx.get());
    };
    auto errCode = CallJsMethod("onRestoreEx", jsRuntime_, jsObj_.get(), ParseRestoreExInfo(), retParser);
    if (errCode != ERR_OK) {
        HILOGE("Call onRestoreEx error");
        return errCode;
    }
    HILOGI("Check callRestoreExDone load");
    std::unique_lock<std::mutex> lock(callJsMutex_);
    callJsCon_.wait(lock, [this] { return callJsExMethodDone_.load(); });
    HILOGI("Check needCallOnRestore load");
    if (!callExtDefaultFunc_.load()) {
        HILOGI("Call Js method onRestoreEx done");
        return ERR_OK;
    }
    return CallJSRestore();
}

ErrCode ExtBackupJs::CallJSRestore()
{
    HILOGI("Start call app js method onRestore");
    auto retParser = [jsRuntime {&jsRuntime_}, callbackInfo {callbackInfo_}](napi_env env, napi_value result) -> bool {
        if (!CheckPromise(env, result)) {
            string str;
            bool isExceptionPending;
            napi_is_exception_pending(env, &isExceptionPending);
            HILOGI("napi exception pending = %{public}d.", isExceptionPending);
            if (!callbackInfo) {
                HILOGE("callbackInfo is nullptr");
                return false;
            }
            if (isExceptionPending) {
                napi_value exception;
                DealNapiException(env, exception, str);
                napi_fatal_exception(env, exception);
                callbackInfo->callback(BError(BError::Codes::EXT_THROW_EXCEPTION), str);
            } else {
                callbackInfo->callback(BError(BError::Codes::OK), str);
            }
            return true;
        }
        HILOGI("CheckPromise Js Method onRestore ok.");
        return CallPromise(*jsRuntime, result, callbackInfo.get());
    };
    auto errCode = CallJsMethod("onRestore", jsRuntime_, jsObj_.get(), ParseRestoreInfo(), retParser);
    if (errCode != ERR_OK) {
        HILOGE("CallJsMethod error, code:%{public}d.", errCode);
        return errCode;
    }
    return ERR_OK;
}

ErrCode ExtBackupJs::GetBackupInfo(std::function<void(ErrCode, const std::string)> callback)
{
    HILOGI("BackupExtensionAbility(JS) GetBackupInfo begin.");
    BExcepUltils::BAssert(jsObj_, BError::Codes::EXT_BROKEN_FRAMEWORK,
                          "The app does not provide the GetBackupInfo interface.");
    callbackInfoBackup_ = std::make_shared<CallbackInfoBackup>(callback);
    auto retParser = [jsRuntime {&jsRuntime_}, callBackInfo {callbackInfoBackup_}](napi_env env,
        napi_value result) -> bool {
        if (!CheckPromise(env, result)) {
            bool isExceptionPending;
            napi_is_exception_pending(env, &isExceptionPending);
            HILOGI("napi exception pending = %{public}d.", isExceptionPending);
            if (!callBackInfo) {
                HILOGE("callBackInfo is nullptr");
                return false;
            }
            if (isExceptionPending) {
                string str;
                napi_value exception;
                DealNapiException(env, exception, str);
                callBackInfo->callbackParam(BError(BError::Codes::EXT_THROW_EXCEPTION), str);
                return false;
            }
            size_t strLen = 0;
            napi_status status = napi_get_value_string_utf8(env, result, nullptr, -1, &strLen);
            if (status != napi_ok) {
                return false;
            }
            size_t bufLen = strLen + 1;
            unique_ptr<char[]> str = make_unique<char[]>(bufLen);
            status = napi_get_value_string_utf8(env, result, str.get(), bufLen, &strLen);
            callBackInfo->callbackParam(BError(BError::Codes::OK), str.get());
            return true;
        }
        HILOGI("BackupExtensionAbulity(JS) GetBackupInfo ok.");
        return CallPromiseEx(*jsRuntime, result, callBackInfo.get());
    };

    auto errCode = CallJsMethod("getBackupInfo", jsRuntime_, jsObj_.get(), {}, retParser);
    if (errCode != ERR_OK) {
        HILOGE("CallJsMethod error, code:%{public}d.", errCode);
    }
    HILOGI("BackupExtensionAbulity(JS) GetBackupInfo end.");
    return errCode;
}

static int InvokeJsMethod(CallJsParam *param, AbilityRuntime::HandleEscape& handleEscape, napi_env env,
    napi_handle_scope& scope, vector<napi_value>& argv)
{
    if (param == nullptr || param->jsObj == nullptr) {
        HILOGE("param or jsObj is nullptr");
        return EINVAL;
    }
    napi_value value = param->jsObj->GetNapiValue();
    if (value == nullptr) {
        HILOGE("failed to get napi value object.");
        return EINVAL;
    }
    napi_status status;
    napi_value method;
    status = napi_get_named_property(env, value, param->funcName.c_str(), &method);
    if (status != napi_ok || param->retParser == nullptr) {
        HILOGE("ResultValueParser must not null.");
        return EINVAL;
    }
    napi_value result;
    HILOGI("Extension start do call current js method, methodName:%{public}s", param->funcName.c_str());
    if (napi_call_function(env, value, method, argv.size(), argv.data(), &result) != napi_ok) {
        HILOGE("napi call function failed");
    }
    if (!param->retParser(env, handleEscape.Escape(result))) {
        HILOGE("Parser js result fail.");
        return EINVAL;
    }
    return ERR_OK;
}

static int DoCallJsMethod(CallJsParam *param)
{
    if (param == nullptr) {
        HILOGE("param is nullptr");
        return EINVAL;
    }
    AbilityRuntime::JsRuntime *jsRuntime = param->jsRuntime;
    HILOGI("Start execute DoCallJsMethod");
    if (jsRuntime == nullptr) {
        HILOGE("failed to get jsRuntime");
        return EINVAL;
    }
    AbilityRuntime::HandleEscape handleEscape(*jsRuntime);
    auto env = jsRuntime->GetNapiEnv();
    napi_handle_scope scope = nullptr;
    napi_open_handle_scope(env, &scope);
    if (scope == nullptr) {
        HILOGE("scope is nullptr");
        return EINVAL;
    }
    vector<napi_value> argv = {};
    if (param->argParser != nullptr) {
        if (!param->argParser(env, argv)) {
            HILOGE("failed to get params.");
            napi_close_handle_scope(env, scope);
            return EINVAL;
        }
    }
    auto ret = InvokeJsMethod(param, handleEscape, env, scope, argv);
    napi_close_handle_scope(env, scope);
    HILOGI("End execute DoCallJsMethod");
    return ret;
}

int ExtBackupJs::CallJsMethod(const std::string &funcName,
                              AbilityRuntime::JsRuntime &jsRuntime,
                              NativeReference *jsObj,
                              InputArgsParser argParser,
                              ResultValueParser retParser)
{
    uv_loop_s *loop = nullptr;
    napi_status status = napi_get_uv_event_loop(jsRuntime.GetNapiEnv(), &loop);
    if (status != napi_ok) {
        HILOGE("failed to get uv event loop.");
        return EINVAL;
    }
    auto param = std::make_shared<CallJsParam>(funcName, &jsRuntime, jsObj, argParser, retParser);
    BExcepUltils::BAssert(param, BError::Codes::EXT_BROKEN_FRAMEWORK, "failed to new param.");

    auto work = std::make_shared<uv_work_t>();
    BExcepUltils::BAssert(work, BError::Codes::EXT_BROKEN_FRAMEWORK, "failed to new uv_work_t.");

    work->data = reinterpret_cast<void *>(param.get());
    HILOGI("Will execute current js method");
    auto task = [work {work.get()}]() {
        if (work == nullptr) {
            HILOGE("failed to get work.");
            return;
        }
        HILOGI("AsyncWork Enter, %{public}zu", (size_t)work);
        CallJsParam *param = reinterpret_cast<CallJsParam *>(work->data);
        do {
            if (param == nullptr) {
                HILOGE("failed to get CallJsParam.");
                break;
            }
            HILOGI("Start call current js method");
            if (DoCallJsMethod(param) != ERR_OK) {
                HILOGE("failed to call DoCallJsMethod.");
            }
        } while (false);
        HILOGI("will notify current thread info");
        std::unique_lock<std::mutex> lock(param->backupOperateMutex);
        param->isReady.store(true);
        param->backupOperateCondition.notify_all();
    };
    auto ret = napi_send_event(jsRuntime.GetNapiEnv(), task, napi_eprio_high);
    if (ret != napi_status::napi_ok) {
        HILOGE("failed to napi_send_event, ret:%{public}d.", ret);
        return EINVAL;
    }
    HILOGI("Wait execute current js method");
    std::unique_lock<std::mutex> lock(param->backupOperateMutex);
    param->backupOperateCondition.wait(lock, [param]() { return param->isReady.load(); });
    HILOGI("End do call current js method");
    return ERR_OK;
}

std::function<bool(napi_env env, std::vector<napi_value> &argv)> ExtBackupJs::ParseBackupExInfo()
{
    auto onBackupExFun = [backupExtInfo(backupExtInfo_)](napi_env env, vector<napi_value> &argv) -> bool {
        napi_value backupExtInfoVal = nullptr;
        napi_create_object(env, &backupExtInfoVal);
        HILOGI("backupExtInfo is:%{public}s", GetAnonyString(backupExtInfo).c_str());
        if (napi_create_string_utf8(env, backupExtInfo.c_str(), backupExtInfo.size(), &backupExtInfoVal) != napi_ok) {
            HILOGE("create napi string failed");
            return false;
        }
        argv.emplace_back(backupExtInfoVal);
        return true;
    };
    return onBackupExFun;
}

std::function<bool(napi_env env, std::vector<napi_value> &argv)> ExtBackupJs::ParseRestoreExInfo()
{
    auto onRestoreExFun = [appVersionCode(appVersionCode_), appVersionStr(appVersionStr_),
        restoreExtInfo(restoreExtInfo_)](napi_env env, vector<napi_value> &argv) -> bool {
        HILOGI("restoreExtInfo is:%{public}s", GetAnonyString(restoreExtInfo).c_str());
        napi_value objValue = nullptr;
        napi_value restoreRetValue = nullptr;
        napi_create_object(env, &objValue);
        napi_create_object(env, &restoreRetValue);
        napi_set_named_property(env, objValue, "code", AbilityRuntime::CreateJsValue(env, appVersionCode));
        napi_set_named_property(env, objValue, "name", AbilityRuntime::CreateJsValue(env, appVersionStr.c_str()));
        if (napi_create_string_utf8(env, restoreExtInfo.c_str(), restoreExtInfo.size(), &restoreRetValue) != napi_ok) {
            HILOGE("create napi string failed");
            return false;
        }
        argv.emplace_back(objValue);
        argv.emplace_back(restoreRetValue);
        return true;
    };
    return onRestoreExFun;
}

std::function<bool(napi_env env, std::vector<napi_value> &argv)> ExtBackupJs::ParseRestoreInfo()
{
    auto onRestoreFun = [appVersionCode(appVersionCode_), appVersionStr(appVersionStr_)](napi_env env,
        vector<napi_value> &argv) -> bool {
        std::string appVersionStrFlag = appVersionStr;
        int64_t appVersionCodeFlag = appVersionCode;
        napi_value objValue = nullptr;
        napi_create_object(env, &objValue);
        auto pos = appVersionStrFlag.find_first_of(BConstants::VERSION_NAME_SEPARATOR_CHAR);
        std::string appVersionFlag = "";
        if (pos != string::npos) {
            appVersionFlag = appVersionStrFlag.substr(0, pos);
            if (appVersionFlag == BConstants::DEFAULT_VERSION_NAME) {
                appVersionStrFlag = appVersionFlag;
                appVersionCodeFlag = 0;
            }
        }
        napi_set_named_property(env, objValue, "code", AbilityRuntime::CreateJsValue(env, appVersionCodeFlag));
        napi_set_named_property(env, objValue, "name", AbilityRuntime::CreateJsValue(env, appVersionStrFlag.c_str()));
        argv.emplace_back(objValue);
        return true;
    };
    return onRestoreFun;
}

ErrCode ExtBackupJs::InvokeAppExtMethod(ErrCode errCode, const std::string result)
{
    HILOGI("Start Get onBackupEx/onRestoreEx method result, errCode: %{public}d, result: %{public}s",
        errCode, result.c_str());
    if ((result.size() == 0) && (errCode == BError(BError::Codes::OK))) {
        callExtDefaultFunc_.store(true);
    } else {
        callExtDefaultFunc_.store(false);
    }
    callJsExMethodDone_.store(true);
    callJsCon_.notify_one();
    HILOGI("End Get App onBackupEx/onRestoreEx method result");
    return ERR_OK;
}

ErrCode ExtBackupJs::OnProcess(std::function<void(ErrCode, const std::string)> callback)
{
    HILOGI("BackupExtensionAbility(JS) OnProcess begin.");
    BExcepUltils::BAssert(jsObj_, BError::Codes::EXT_BROKEN_FRAMEWORK,
                          "The app does not provide the OnProcess interface.");
    onProcessCallback_ = std::make_shared<OnProcessCallBackInfo>(callback);
    auto retParser = [jsRuntime {&jsRuntime_}, callBackInfo {onProcessCallback_}](napi_env env,
        napi_value result) -> bool {
            string processStr;
            bool isExceptionPending;
            napi_is_exception_pending(env, &isExceptionPending);
            HILOGI("napi exception pending = %{public}d.", isExceptionPending);
            if (!callBackInfo) {
                HILOGE("callbackInfo is nullptr");
                return false;
            }
            if (isExceptionPending) {
                napi_value exception;
                napi_get_and_clear_last_exception(env, &exception);
                callBackInfo->onProcessCallback(BError(BError::Codes::EXT_THROW_EXCEPTION), processStr);
            } else {
                DealNapiStrValue(env, result, processStr);
                callBackInfo->onProcessCallback(BError(BError::Codes::OK), processStr);
            }
            return true;
    };
    auto errCode = CallJsMethod("onProcess", jsRuntime_, jsObj_.get(), {}, retParser);
    if (errCode != ERR_OK) {
        HILOGE("CallJsMethod error, code:%{public}d.", errCode);
    }
    HILOGI("BackupExtensionAbulity(JS) OnProcess end.");
    return errCode;
}

void ExtBackupJs::InitTempPath(const std::string &bundleName)
{
    std::string el2BackupDir(BConstants::PATH_BUNDLE_BACKUP_HOME);
    if (access(el2BackupDir.c_str(), F_OK) != 0) {
        HILOGW("backup home el2 dir not exits, try to create");
        if (!ForceCreateDirectory(el2BackupDir.data())) {
            HILOGE("Failed to create directory, err = %{public}d", errno);
            AppRadar::Info info(bundleName, "", "Create backup home el2 dir failed");
            AppRadar::GetInstance().RecordDefaultFuncRes(info, "ExtBackupJs::InitTempPath",
                AppRadar::GetInstance().GetUserId(), BizStageBackup::BIZ_STAGE_DEFAULT,
                static_cast<int32_t>(BError::Codes::EXT_CREATE_DIR_ERROR));
        }
    }
    std::string el1BackupDir(BConstants::PATH_BUNDLE_BACKUP_HOME_EL1);
    if (access(el1BackupDir.c_str(), F_OK) != 0) {
        HILOGW("backup home el1 dir not exits, try to create");
        if (!ForceCreateDirectory(el1BackupDir.data())) {
            HILOGE("Failed to create home el1 dir, err = %{public}d", errno);
            AppRadar::Info info(bundleName, "", "Create backup home el1 dir failed");
            AppRadar::GetInstance().RecordDefaultFuncRes(info, "ExtBackupJs::InitTempPath",
                AppRadar::GetInstance().GetUserId(), BizStageBackup::BIZ_STAGE_DEFAULT,
                static_cast<int32_t>(BError::Codes::EXT_CREATE_DIR_ERROR));
        }
    }
}
} // namespace OHOS::FileManagement::Backup