/*
 * 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 "module_update_proxy.h"

#include "log/log.h"
#include "module_ipc_helper.h"
#include "string_ex.h"
#include "sys_installer_sa_ipc_interface_code.h"

namespace OHOS {
namespace SysInstaller {
using namespace Updater;

int32_t ModuleUpdateProxy::InstallModulePackage(const std::string &pkgPath)
{
    LOG(INFO) << "InstallModulePackage " << pkgPath;
    auto remote = Remote();
    if (remote == nullptr) {
        LOG(ERROR) << "Can not get remote";
        return ERR_FLATTEN_OBJECT;
    }

    MessageParcel data;
    if (!data.WriteInterfaceToken(GetDescriptor())) {
        LOG(ERROR) << "WriteInterfaceToken error";
        return ERR_FLATTEN_OBJECT;
    }
    data.WriteString16(Str8ToStr16(pkgPath));

    MessageParcel reply;
    MessageOption option;
    int32_t ret = remote->SendRequest(
        static_cast<uint32_t>(ModuleUpdateInterfaceCode::INSTALL_MODULE_PACKAGE), data, reply, option);
    if (ret != ERR_OK) {
        LOG(ERROR) << "SendRequest error";
        return ERR_FLATTEN_OBJECT;
    }

    return reply.ReadInt32();
}

int32_t ModuleUpdateProxy::UninstallModulePackage(const std::string &hmpName)
{
    LOG(INFO) << "UninstallModulePackage " << hmpName;
    auto remote = Remote();
    if (remote == nullptr) {
        LOG(ERROR) << "Can not get remote";
        return ERR_FLATTEN_OBJECT;
    }

    MessageParcel data;
    if (!data.WriteInterfaceToken(GetDescriptor())) {
        LOG(ERROR) << "WriteInterfaceToken error";
        return ERR_FLATTEN_OBJECT;
    }
    data.WriteString16(Str8ToStr16(hmpName));

    MessageParcel reply;
    MessageOption option;
    int32_t ret = remote->SendRequest(
        static_cast<uint32_t>(ModuleUpdateInterfaceCode::UNINSTALL_MODULE_PACKAGE), data, reply, option);
    if (ret != ERR_OK) {
        LOG(ERROR) << "SendRequest error";
        return ERR_FLATTEN_OBJECT;
    }

    return reply.ReadInt32();
}

int32_t ModuleUpdateProxy::GetModulePackageInfo(const std::string &hmpName,
    std::list<ModulePackageInfo> &modulePackageInfos)
{
    LOG(INFO) << "GetModulePackageInfo " << hmpName;
    auto remote = Remote();
    if (remote == nullptr) {
        LOG(ERROR) << "Can not get remote";
        return ERR_FLATTEN_OBJECT;
    }

    MessageParcel data;
    if (!data.WriteInterfaceToken(GetDescriptor())) {
        LOG(ERROR) << "WriteInterfaceToken error";
        return ERR_FLATTEN_OBJECT;
    }
    data.WriteString16(Str8ToStr16(hmpName));

    MessageParcel reply;
    MessageOption option;
    int32_t ret = remote->SendRequest(
        static_cast<uint32_t>(ModuleUpdateInterfaceCode::GET_MODULE_PACKAGE_INFO), data, reply, option);
    if (ret != ERR_OK) {
        LOG(ERROR) << "SendRequest error";
        return ERR_FLATTEN_OBJECT;
    }

    ModuleIpcHelper::ReadModulePackageInfos(reply, modulePackageInfos);
    return reply.ReadInt32();
}

int32_t ModuleUpdateProxy::ExitModuleUpdate()
{
    LOG(INFO) << "ExitModuleUpdate";
    auto remote = Remote();
    if (remote == nullptr) {
        LOG(ERROR) << "Can not get remote";
        return ERR_FLATTEN_OBJECT;
    }

    MessageParcel data;
    if (!data.WriteInterfaceToken(GetDescriptor())) {
        LOG(ERROR) << "WriteInterfaceToken error";
        return ERR_FLATTEN_OBJECT;
    }

    MessageParcel reply;
    MessageOption option;
    int32_t ret = remote->SendRequest(
        static_cast<uint32_t>(ModuleUpdateInterfaceCode::EXIT_MODULE_UPDATE), data, reply, option);
    if (ret != ERR_OK) {
        LOG(ERROR) << "SendRequest error";
        return ERR_FLATTEN_OBJECT;
    }

    return reply.ReadInt32();
}

std::vector<HmpVersionInfo> ModuleUpdateProxy::GetHmpVersionInfo()
{
    LOG(INFO) << "GetHmpVersionInfo";
    std::vector<HmpVersionInfo> versionInfo {};
    auto remote = Remote();
    if (remote == nullptr) {
        LOG(ERROR) << "Can not get remote";
        return versionInfo;
    }

    MessageParcel data;
    if (!data.WriteInterfaceToken(GetDescriptor())) {
        LOG(ERROR) << "WriteInterfaceToken error";
        return versionInfo;
    }

    MessageParcel reply;
    MessageOption option;
    int32_t ret = remote->SendRequest(
        static_cast<uint32_t>(ModuleUpdateInterfaceCode::GET_HMP_VERSION_INFO), data, reply, option);
    if (ret != ERR_OK) {
        LOG(ERROR) << "SendRequest error";
        return versionInfo;
    }

    int32_t count = reply.ReadInt32();
    for (int32_t i = 0; i < count; i++) {
        std::unique_ptr<HmpVersionInfo> infoPtr(reply.ReadParcelable<HmpVersionInfo>());
        if (infoPtr != nullptr) {
            versionInfo.push_back(*infoPtr);
        }
    }
    return versionInfo;
}

int32_t ModuleUpdateProxy::StartUpdateHmpPackage(const std::string &path,
    const sptr<ISysInstallerCallback> &updateCallback)
{
    LOG(INFO) << "StartUpdateHmpPackage";
    if (updateCallback == nullptr) {
        LOG(ERROR) <<  "Invalid param";
        return ERR_INVALID_VALUE;
    }
    auto remote = Remote();
    if (remote == nullptr) {
        LOG(ERROR) << "Can not get remote";
        return ERR_FLATTEN_OBJECT;
    }

    MessageParcel data;
    if (!data.WriteInterfaceToken(GetDescriptor())) {
        LOG(ERROR) << "WriteInterfaceToken error";
        return ERR_FLATTEN_OBJECT;
    }

    if (!data.WriteRemoteObject(updateCallback->AsObject())) {
        LOG(ERROR) << "WriteRemoteObject fail";
        return ERR_FLATTEN_OBJECT;
    }
    data.WriteString16(Str8ToStr16(path));

    MessageParcel reply;
    MessageOption option;
    int32_t ret = remote->SendRequest(
        static_cast<uint32_t>(ModuleUpdateInterfaceCode::START_UPDATE_HMP_PACKAGE), data, reply, option);
    if (ret != ERR_OK) {
        LOG(ERROR) << "SendRequest error";
        return ERR_FLATTEN_OBJECT;
    }

    return reply.ReadInt32();
}

std::vector<HmpUpdateInfo> ModuleUpdateProxy::GetHmpUpdateResult()
{
    LOG(INFO) << "GetHmpUpdateResult";
    std::vector<HmpUpdateInfo> updateInfo {};
    auto remote = Remote();
    if (remote == nullptr) {
        LOG(ERROR) << "Can not get remote";
        return updateInfo;
    }

    MessageParcel data;
    if (!data.WriteInterfaceToken(GetDescriptor())) {
        LOG(ERROR) << "WriteInterfaceToken error";
        return updateInfo;
    }

    MessageParcel reply;
    MessageOption option;
    int32_t ret = remote->SendRequest(
        static_cast<uint32_t>(ModuleUpdateInterfaceCode::GET_HMP_UPDATE_RESULT), data, reply, option);
    if (ret != ERR_OK) {
        LOG(ERROR) << "SendRequest error";
        return updateInfo;
    }

    int32_t count = reply.ReadInt32();
    for (int32_t i = 0; i < count; i++) {
        std::unique_ptr<HmpUpdateInfo> infoPtr(reply.ReadParcelable<HmpUpdateInfo>());
        if (infoPtr != nullptr) {
            updateInfo.push_back(*infoPtr);
        }
    }
    return updateInfo;
}
} // namespace SysInstaller
} // namespace OHOS