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

#include "log/log.h"
#include "package/cert_verify.h"
#include "package/package.h"
#include "package/pkg_manager.h"
#include "pkg_verify.h"
#include "updater/updater.h"
#include "utils.h"

namespace OHOS {
namespace SysInstaller {
using namespace Hpackage;
using namespace Updater;

int32_t ABUpdate::StartABUpdate(const std::string &pkgPath)
{
    LOG(INFO) << "StartABUpdate start";
    CertVerify::GetInstance().RegisterCertHelper(std::make_unique<SingleCertHelper>());
    if (statusManager_ == nullptr) {
        LOG(ERROR) << "statusManager_ nullptr";
        return -1;
    }

    Hpackage::PkgManager::PkgManagerPtr pkgManager = Hpackage::PkgManager::GetPackageInstance();
    if (pkgManager == nullptr) {
        LOG(ERROR) << "pkgManager is nullptr";
        return UPDATE_ERROR;
    }

    STAGE(UPDATE_STAGE_BEGIN) << "StartABUpdate start";
    LOG(INFO) << "ABUpdate start, pkg updaterPath : " << pkgPath.c_str();
    statusManager_->UpdateCallback(UPDATE_STATE_ONGOING, 10); // 10 : start install

    UpdaterParams upParams {
        false, false, 0, 0, 0, 0, {pkgPath}
    };
    UpdaterStatus updateRet = DoInstallUpdaterPackage(pkgManager, upParams, HOTA_UPDATE);
    if (updateRet != UPDATE_SUCCESS) {
        LOG(INFO) << "Install package failed!";
        STAGE(UPDATE_STAGE_FAIL) << "Install package failed";
        statusManager_->UpdateCallback(UPDATE_STATE_FAILED, 100); // 100 : failed
        Hpackage::PkgManager::ReleasePackageInstance(pkgManager);
        return -1;
    }

    LOG(INFO) << "Install package successfully!";
    STAGE(UPDATE_STAGE_SUCCESS) << "Install package success";
    statusManager_->UpdateCallback(UPDATE_STATE_SUCCESSFUL, 100); // 100 : success
    Hpackage::PkgManager::ReleasePackageInstance(pkgManager);
    if (!DeleteUpdaterPath(GetWorkPath())) {
        LOG(WARNING) << "Delete Updater Path fail.";
    }
    return 0;
}
} // namespace SysInstaller
} // namespace OHOS