xref: /foundation/arkui/ace_engine/frameworks/bridge/declarative_frontend/engine/jsi/modules/jsi_router_module.cpp

/*
 * 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 "frameworks/bridge/declarative_frontend/engine/jsi/modules/jsi_router_module.h"

#include "base/json/json_util.h"
#include "base/log/log.h"
#include "frameworks/bridge/common/utils/engine_helper.h"

namespace OHOS::Ace::Framework {

std::string DeclarativeParseRouteUrl(const shared_ptr<JsRuntime>& runtime, const shared_ptr<JsValue>& arg,
    const std::string& key)
{
    std::string argStr = arg->GetJsonString(runtime);
    if (argStr.empty()) {
        return argStr;
    }

    std::string pageRoute;
    std::unique_ptr<JsonValue> argsPtr = JsonUtil::ParseJsonString(argStr);
    if (argsPtr != nullptr && argsPtr->GetValue(key) != nullptr && argsPtr->GetValue(key)->IsString()) {
        pageRoute = argsPtr->GetValue(key)->GetString();
    }

    return pageRoute;
}

std::string DeclarativeParseRouteParams(const shared_ptr<JsRuntime>& runtime, const shared_ptr<JsValue>& arg,
    const std::string& key)
{
    std::string argStr = arg->GetJsonString(runtime);
    std::unique_ptr<JsonValue> argsPtr = JsonUtil::ParseJsonString(argStr);
    std::string params;
    if (argsPtr != nullptr && argsPtr->Contains(key) && argsPtr->GetValue(key)->IsObject()) {
        params = argsPtr->GetValue(key)->ToString();
    }
    return params;
}

shared_ptr<JsValue> PagePush(const shared_ptr<JsRuntime>& runtime, const shared_ptr<JsValue>& thisObj,
    const std::vector<shared_ptr<JsValue>>& argv, int32_t argc)
{
    if (argc != 1) {
        return runtime->NewNull();
    }

    std::string uri = DeclarativeParseRouteUrl(runtime, argv[0], ROUTE_KEY_URI);
    std::string params = DeclarativeParseRouteParams(runtime, argv[0], ROUTE_KEY_PARAMS);
    auto delegate = EngineHelper::GetCurrentDelegate();
    if (delegate == nullptr) {
        return runtime->NewNull();
    }

    delegate->Push(uri, params);
    return runtime->NewNull();
}

shared_ptr<JsValue> PageReplace(const shared_ptr<JsRuntime>& runtime, const shared_ptr<JsValue>& thisObj,
    const std::vector<shared_ptr<JsValue>>& argv, int32_t argc)
{
    if (argc != 1) {
        return runtime->NewNull();
    }

    std::string uri = DeclarativeParseRouteUrl(runtime, argv[0], ROUTE_KEY_URI);
    std::string params = DeclarativeParseRouteParams(runtime, argv[0], ROUTE_KEY_PARAMS);
    auto delegate = EngineHelper::GetCurrentDelegate();
    if (delegate == nullptr) {
        return runtime->NewNull();
    }

    delegate->Replace(uri, params);
    return runtime->NewNull();
}

shared_ptr<JsValue> PageBack(const shared_ptr<JsRuntime>& runtime, const shared_ptr<JsValue>& thisObj,
    const std::vector<shared_ptr<JsValue>>& argv, int32_t argc)
{
    if (argc != 1 && argc != 0) {
        return runtime->NewNull();
    }

    std::string uri;
    std::string params;
    if (argc == 1) {
        uri = DeclarativeParseRouteUrl(runtime, argv[0], ROUTE_KEY_URI);
        params = DeclarativeParseRouteParams(runtime, argv[0], ROUTE_KEY_PARAMS);
    }
    auto delegate = EngineHelper::GetCurrentDelegate();
    if (delegate == nullptr) {
        return runtime->NewNull();
    }

    delegate->Back(uri, params);
    return runtime->NewNull();
}

shared_ptr<JsValue> PageClear(const shared_ptr<JsRuntime>& runtime, const shared_ptr<JsValue>& thisObj,
    const std::vector<shared_ptr<JsValue>>& argv, int32_t argc)
{
    auto delegate = EngineHelper::GetCurrentDelegate();
    if (delegate == nullptr) {
        return runtime->NewNull();
    }

    delegate->Clear();
    return runtime->NewNull();
}

shared_ptr<JsValue> PageGetLength(const shared_ptr<JsRuntime>& runtime, const shared_ptr<JsValue>& thisObj,
    const std::vector<shared_ptr<JsValue>>& argv, int32_t argc)
{
    auto delegate = EngineHelper::GetCurrentDelegate();
    if (delegate == nullptr) {
        return runtime->NewNull();
    }

    int32_t routeLength = delegate->GetStackSize();
    return runtime->NewString(std::to_string(routeLength));
}

shared_ptr<JsValue> PageGetState(const shared_ptr<JsRuntime>& runtime, const shared_ptr<JsValue>& thisObj,
    const std::vector<shared_ptr<JsValue>>& argv, int32_t argc)
{
    auto delegate = EngineHelper::GetCurrentDelegate();
    if (delegate == nullptr) {
        return runtime->NewNull();
    }

    int32_t routeIndex = 0;
    std::string routeName;
    std::string routePath;
    delegate->GetState(routeIndex, routeName, routePath);
    shared_ptr<JsValue> jsState = runtime->NewObject();
    jsState->SetProperty(runtime, "index", runtime->NewNumber(routeIndex));
    jsState->SetProperty(runtime, "name", runtime->NewString(routeName));
    jsState->SetProperty(runtime, "path", runtime->NewString(routePath));
    return jsState;
}

shared_ptr<JsValue> PageGetParams(const shared_ptr<JsRuntime>& runtime, const shared_ptr<JsValue>& thisObj,
    const std::vector<shared_ptr<JsValue>>& argv, int32_t argc)
{
    auto delegate = EngineHelper::GetCurrentDelegate();
    if (delegate == nullptr) {
        return runtime->NewNull();
    }

    std::string paramsStr = delegate->GetParams();
    if (paramsStr.empty()) {
        return runtime->NewNull();
    }
    return runtime->ParseJson(paramsStr);
}

void InitRouterModule(const shared_ptr<JsRuntime>& runtime, shared_ptr<JsValue>& moduleObj)
{
    moduleObj->SetProperty(runtime, ROUTE_PAGE_PUSH, runtime->NewFunction(PagePush));
    moduleObj->SetProperty(runtime, ROUTE_PAGE_REPLACE, runtime->NewFunction(PageReplace));
    moduleObj->SetProperty(runtime, ROUTE_PAGE_BACK, runtime->NewFunction(PageBack));
    moduleObj->SetProperty(runtime, ROUTE_PAGE_CLEAR, runtime->NewFunction(PageClear));
    moduleObj->SetProperty(runtime, ROUTE_PAGE_GET_LENGTH, runtime->NewFunction(PageGetLength));
    moduleObj->SetProperty(runtime, ROUTE_PAGE_GET_STATE, runtime->NewFunction(PageGetState));
    moduleObj->SetProperty(runtime, ROUTE_PAGE_GET_PARAMS, runtime->NewFunction(PageGetParams));

    shared_ptr<JsValue> global = runtime->GetGlobal();
    shared_ptr<JsValue> requireNapiFunc = global->GetProperty(runtime, "requireNapi");
    if (!requireNapiFunc || !requireNapiFunc->IsFunction(runtime)) {
        return;
    }
    std::vector<shared_ptr<JsValue>> argv = { runtime->NewString("router") };
    shared_ptr<JsValue> napiObj = requireNapiFunc->Call(runtime, global, argv, argv.size());
    moduleObj->SetProperty(runtime,
        ROUTE_ENABLE_ALERT_BEFORE_BACK_PAGE, napiObj->GetProperty(runtime, ROUTE_ENABLE_ALERT_BEFORE_BACK_PAGE));
    moduleObj->SetProperty(runtime,
        ROUTE_DISABLE_ALERT_BEFORE_BACK_PAGE, napiObj->GetProperty(runtime, ROUTE_DISABLE_ALERT_BEFORE_BACK_PAGE));
}

} // namespace OHOS::Ace::Framework