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

#include <string>
#include <unistd.h>
#include "init_reboot.h"
#include "misc_info/misc_info.h"
#include "securec.h"
#include "utils.h"

using namespace Updater;

static bool WriteToMiscAndRebootToUpdater(const struct UpdateMessage &updateMsg)
{
    // Write package name to misc, then trigger reboot.
    const char *bootCmd = "boot_updater";
    int ret = strncpy_s(const_cast<char*>(updateMsg.command), sizeof(updateMsg.command), bootCmd,
        sizeof(updateMsg.command) - 1);
    if (ret != 0) {
        return false;
    }
#ifndef UPDATER_UT
    WriteUpdaterMiscMsg(updateMsg);
    DoReboot("updater");
    while (true) {
        pause();
    }
#else
    return true;
#endif
}

bool RebootAndInstallUpgradePackage(const std::string &miscFile, const std::vector<std::string> &packageName)
{
    if (packageName.size() == 0) {
        std::cout << "updaterkits: invalid argument. one of arugments is empty\n";
        return false;
    }

    struct UpdateMessage updateMsg {};
    size_t updateOffset = 0;
    for (auto path : packageName) {
        if (access(path.c_str(), R_OK) < 0) {
            std::cout << "updaterkits: " << path << " is not readable\n";
            return false;
        }
        if (updateOffset > sizeof(updateMsg.update)) {
            std::cout << "updaterkits: updateOffset > updateMsg.update, return false\n";
            return false;
        }
        int ret = snprintf_s(updateMsg.update + updateOffset, sizeof(updateMsg.update) - updateOffset,
            sizeof(updateMsg.update) - 1 - updateOffset, "--update_package=%s\n", path.c_str());
        if (ret < 0) {
            std::cout << "updaterkits: copy updater message failed\n";
            return false;
        }
        updateOffset += static_cast<size_t>(ret);
    }

    WriteToMiscAndRebootToUpdater(updateMsg);

    // Never get here.
    return true;
}

bool RebootAndCleanUserData(const std::string &miscFile, const std::string &cmd)
{
    if (miscFile.empty() || cmd.empty()) {
        std::cout << "updaterkits: invalid argument. one of arugments is empty\n";
        return false;
    }

    // Write package name to misc, then trigger reboot.
    struct UpdateMessage updateMsg {};
    if (strncpy_s(updateMsg.update, sizeof(updateMsg.update), cmd.c_str(), cmd.size()) != EOK) {
        std::cout << "updaterkits: copy updater message failed\n";
        return false;
    }

    WriteToMiscAndRebootToUpdater(updateMsg);

    // Never get here.
    return true;
}