OpenHarmony-v4.1-Release/s

/*
 * Copyright (c) 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 <mutex>
#include "napi_base_context.h"
#include "napi_print_utils.h"
#include "print_async_call.h"
#include "print_log.h"
#include "print_manager_client.h"
#include "print_task.h"
#include "napi_print_task.h"
#include "print_attributes_helper.h"
#include "print_callback.h"
#include "iprint_callback.h"

static constexpr const char *FUNCTION_ON = "on";
static constexpr const char *FUNCTION_OFF = "off";

namespace OHOS::Print {
__thread napi_ref NapiPrintTask::globalCtor = nullptr;
std::mutex g_printTaskMutex;

napi_value NapiPrintTask::Print(napi_env env, napi_callback_info info)
{
    napi_value argv[NapiPrintUtils::MAX_ARGC] = { nullptr };
    size_t paramCount = NapiPrintUtils::GetJsVal(env, info, argv, NapiPrintUtils::MAX_ARGC);
    napi_valuetype type;
    PRINT_CALL(env, napi_typeof(env, argv[0], &type));
    if ((paramCount > NapiPrintUtils::ARGC_THREE) && type == napi_string) {
        return PrintByAdapter(env, info);
    }

    auto context = std::make_shared<PrintTaskContext>();
    auto input = [context](napi_env env, size_t argc, napi_value *argv, napi_value self) -> napi_status {
        PRINT_ASSERT_BASE(env, argc == NapiPrintUtils::ARGC_ONE || argc == NapiPrintUtils::ARGC_TWO,
            "need 1 or 2 parameter!", napi_invalid_arg);
        napi_status checkStatus = VerifyParameters(env, argc, argv, context);
        if (checkStatus != napi_ok) {
            return checkStatus;
        }

        napi_value proxy = nullptr;
        napi_status status = napi_new_instance(env, GetCtor(env), argc, argv, &proxy);
        if ((proxy == nullptr) || (status != napi_ok)) {
            PRINT_HILOGE("Failed to create print task");
            context->SetErrorIndex(E_PRINT_GENERIC_FAILURE);
            return napi_generic_failure;
        }

        PrintTask *task;
        PRINT_CALL_BASE(env, napi_unwrap(env, proxy, reinterpret_cast<void **>(&task)), napi_invalid_arg);
        uint32_t ret = E_PRINT_GENERIC_FAILURE;
        if (task != nullptr) {
            ret = task->Start();
        }
        if (ret != E_PRINT_NONE) {
            PRINT_HILOGE("Failed to start print task");
            context->SetErrorIndex(ret);
            return napi_generic_failure;
        }
        napi_create_reference(env, proxy, 1, &(context->ref));
        return napi_ok;
    };
    auto output = [context](napi_env env, napi_value *result) -> napi_status {
        if (context->ref == nullptr) {
            *result = NapiPrintUtils::GetUndefined(env);
            return napi_generic_failure;
        }
        napi_status status = napi_get_reference_value(env, context->ref, result);
        napi_delete_reference(env, context->ref);
        return status;
    };
    context->SetAction(std::move(input), std::move(output));
    PrintAsyncCall asyncCall(env, info, std::dynamic_pointer_cast<PrintAsyncCall::Context>(context));
    return asyncCall.Call(env);
}

napi_value NapiPrintTask::PrintByAdapter(napi_env env, napi_callback_info info)
{
    PRINT_HILOGI("PrintByAdapter start ---->");
    auto context = std::make_shared<PrintTaskContext>();
    auto input = [context](napi_env env, size_t argc, napi_value *argv, napi_value self) -> napi_status {
        PRINT_ASSERT_BASE(env, argc == NapiPrintUtils::ARGC_FOUR, "need 4 parameter!", napi_invalid_arg);
        napi_status checkStatus = VerifyParameters(env, argc, argv, context);
        if (checkStatus != napi_ok) {
            return checkStatus;
        }

        napi_value proxy = nullptr;
        napi_status status = napi_new_instance(env, GetCtor(env), argc, argv, &proxy);
        if ((proxy == nullptr) || (status != napi_ok)) {
            PRINT_HILOGE("Failed to create print task");
            context->SetErrorIndex(E_PRINT_GENERIC_FAILURE);
            return napi_generic_failure;
        }

        PrintTask *task;
        PRINT_CALL_BASE(env, napi_unwrap(env, proxy, reinterpret_cast<void **>(&task)), napi_invalid_arg);
        uint32_t ret = E_PRINT_GENERIC_FAILURE;
        if (task != nullptr) {
            ret = task->StartPrintAdapter();
        }
        if (ret != E_PRINT_NONE) {
            PRINT_HILOGE("Failed to start print task");
            context->SetErrorIndex(ret);
            return napi_generic_failure;
        }
        napi_create_reference(env, proxy, 1, &(context->ref));
        return napi_ok;
    };
    auto output = [context](napi_env env, napi_value *result) -> napi_status {
        if (context->ref == nullptr) {
            *result = NapiPrintUtils::GetUndefined(env);
            return napi_generic_failure;
        }
        napi_status status = napi_get_reference_value(env, context->ref, result);
        napi_delete_reference(env, context->ref);
        return status;
    };
    context->SetAction(std::move(input), std::move(output));
    PrintAsyncCall asyncCall(env, info, std::dynamic_pointer_cast<PrintAsyncCall::Context>(context));
    return asyncCall.Call(env);
}

napi_value NapiPrintTask::ParsePrintAdapterParameter(napi_env env, size_t argc, napi_value *argv, napi_value self)
{
    if (argc > NapiPrintUtils::ARGC_THREE && argc > NapiPrintUtils::ARGC_TWO) {
        std::string printJobName = NapiPrintUtils::GetStringFromValueUtf8(env, argv[0]);

        std::shared_ptr<PrintAttributes> printAttributes =
            PrintAttributesHelper::BuildFromJs(env, argv[NapiPrintUtils::ARGC_TWO]);
        if (printAttributes == nullptr) {
            PRINT_HILOGE("printAdapter parameter error");
            return nullptr;
        }

        napi_ref adapterRef = NapiPrintUtils::CreateReference(env, argv[1]);
        sptr<IPrintCallback> callback = new (std::nothrow) PrintCallback(env, adapterRef);
        if (callback == nullptr) {
            PRINT_HILOGE("callback parameter error");
            return nullptr;
        }

        std::shared_ptr<OHOS::AbilityRuntime::AbilityContext> abilityContext;
        sptr<IRemoteObject> callerToken = nullptr;
        if (GetAbilityContext(env, argv[NapiPrintUtils::ARGC_THREE], abilityContext) != nullptr) {
            callerToken = abilityContext->GetToken();
        }
        auto task = new (std::nothrow) PrintTask(printJobName, callback, printAttributes, callerToken);

        if (task == nullptr) {
            PRINT_HILOGE("print task fail");
            return nullptr;
        }
        auto finalize = [](napi_env env, void *data, void *hint) {
            PRINT_HILOGD("destructed print task");
            PrintTask *task = reinterpret_cast<PrintTask *>(data);
            delete task;
        };
        if (napi_wrap(env, self, task, finalize, nullptr, nullptr) != napi_ok) {
            finalize(env, task, nullptr);
            return nullptr;
        }
        PRINT_HILOGD("Succeed to allocate print task");
        return self;
    }
    return nullptr;
}

napi_value NapiPrintTask::GetAbilityContext(
    napi_env env, napi_value value, std::shared_ptr<OHOS::AbilityRuntime::AbilityContext> &abilityContext)
{
    bool stageMode = false;
    napi_status status = OHOS::AbilityRuntime::IsStageContext(env, value, stageMode);
    if (status != napi_ok || !stageMode) {
        PRINT_HILOGE("GetAbilityContext it is not a stage mode");
        return nullptr;
    } else {
        auto context = OHOS::AbilityRuntime::GetStageModeContext(env, value);
        if (context == nullptr) {
            PRINT_HILOGE("GetAbilityContext get context failed");
            return nullptr;
        }
        abilityContext = OHOS::AbilityRuntime::Context::ConvertTo<OHOS::AbilityRuntime::AbilityContext>(context);
        if (abilityContext == nullptr) {
            PRINT_HILOGE("GetAbilityContext get Stage model ability context failed.");
            return nullptr;
        }
        return WrapVoidToJS(env);
    }
}

napi_value NapiPrintTask::WrapVoidToJS(napi_env env)
{
    napi_value result = nullptr;
    NAPI_CALL(env, napi_get_null(env, &result));
    return result;
}

napi_value NapiPrintTask::GetCtor(napi_env env)
{
    std::lock_guard<std::mutex> lock(g_printTaskMutex);
    napi_value cons;
    if (globalCtor != nullptr) {
        PRINT_CALL(env, napi_get_reference_value(env, globalCtor, &cons));
        return cons;
    }

    napi_property_descriptor clzDes[] = {
        { FUNCTION_ON, 0, PrintTask::On, 0, 0, 0, napi_default, 0 },
        { FUNCTION_OFF, 0, PrintTask::Off, 0, 0, 0, napi_default, 0 },
    };
    PRINT_CALL(env, napi_define_class(env, "NapiPrintTask", NAPI_AUTO_LENGTH, Initialize, nullptr,
                       sizeof(clzDes) / sizeof(napi_property_descriptor), clzDes, &cons));
    PRINT_CALL(env, napi_create_reference(env, cons, 1, &globalCtor));
    return cons;
}

napi_value NapiPrintTask::Initialize(napi_env env, napi_callback_info info)
{
    PRINT_HILOGD("constructor print task!");
    napi_value self = nullptr;
    size_t argc = NapiPrintUtils::MAX_ARGC;
    napi_value argv[NapiPrintUtils::MAX_ARGC] = { nullptr };
    PRINT_CALL(env, napi_get_cb_info(env, info, &argc, argv, &self, nullptr));

    if (argc > NapiPrintUtils::ARGC_THREE) {
        return ParsePrintAdapterParameter(env, argc, argv, self);
    } else {
        std::vector<std::string> printfiles;
        uint32_t arrayReLength = 0;
        PRINT_CALL(env, napi_get_array_length(env, argv[0], &arrayReLength));
        for (uint32_t index = 0; index < arrayReLength; index++) {
            napi_value filesValue;
            napi_get_element(env, argv[0], index, &filesValue);
            std::string files = NapiPrintUtils::GetStringFromValueUtf8(env, filesValue);
            PRINT_HILOGD("file[%{public}d] %{private}s.", index, files.c_str());
            if (IsValidFile(files)) {
                printfiles.emplace_back(files);
            }
        }

        std::shared_ptr<OHOS::AbilityRuntime::AbilityContext> abilityContext;
        sptr<IRemoteObject> callerToken;
        if (argc == NapiPrintUtils::ARGC_TWO && GetAbilityContext(env, argv[1], abilityContext) != nullptr) {
            callerToken = abilityContext->GetToken();
            PRINT_HILOGI("get callerToken:%{public}s", callerToken != nullptr ? "success" : "failed");
        }
        auto task = new (std::nothrow) PrintTask(printfiles, callerToken);
        if (task == nullptr) {
            PRINT_HILOGE("print task fail");
            return nullptr;
        }
        auto finalize = [](napi_env env, void *data, void *hint) {
            PRINT_HILOGD("destructed print task");
            PrintTask *task = reinterpret_cast<PrintTask *>(data);
            delete task;
        };
        if (napi_wrap(env, self, task, finalize, nullptr, nullptr) != napi_ok) {
            finalize(env, task, nullptr);
            return nullptr;
        }
        PRINT_HILOGD("Succeed to allocate print task");
        return self;
    }
}

bool NapiPrintTask::IsValidFile(const std::string &fileName)
{
    if (fileName == "") {
        PRINT_HILOGE("invalid file name");
        return false;
    }
    auto file = fopen(fileName.c_str(), "rb");
    if (file != nullptr) {
        fclose(file);
        return true;
    }
    if (fileName.find("file://") == 0 || fileName.find("fd://") == 0) {
        return true;
    }
    PRINT_HILOGE("invalid file name");
    return false;
}

napi_status NapiPrintTask::VerifyParameters(napi_env env, size_t argc, napi_value *argv,
    const std::shared_ptr<PrintTaskContext> context)
{
    if (argc > NapiPrintUtils::ARGC_THREE) {
        std::shared_ptr<OHOS::AbilityRuntime::AbilityContext> abilityContext;
        if (GetAbilityContext(env, argv[NapiPrintUtils::ARGC_THREE], abilityContext) == nullptr) {
            PRINT_HILOGE("Print adapter Ability Context is null.");
            context->SetErrorIndex(E_PRINT_INVALID_PARAMETER);
        }
        return napi_ok;
    }

    bool isFileArray = false;

    napi_is_array(env, argv[0], &isFileArray);
    PRINT_ASSERT_BASE(env, isFileArray == true, "parameter type isn't list", napi_invalid_arg);

    uint32_t len = 0;
    PRINT_CALL_BASE(env, napi_get_array_length(env, argv[0], &len), napi_invalid_arg);

    if (!isFileArray || len == 0) {
        context->SetErrorIndex(E_PRINT_INVALID_PARAMETER);
        return napi_invalid_arg;
    }

    std::shared_ptr<OHOS::AbilityRuntime::AbilityContext> abilityContext;
    if (argc == NapiPrintUtils::ARGC_TWO) {
        if (GetAbilityContext(env, argv[1], abilityContext) == nullptr) {
            PRINT_HILOGE("Print, Ability Context is null.");
        }
    }
    return napi_ok;
}
} // namespace OHOS::Print