/*
 * Copyright 2013, Stephan Aßmus <superstippi@gmx.de>.
 * Copyright 2013, Rene Gollent, <rene@gollent.com>
 * Copyright 2020-2025, Andrew Lindesay <apl@lindesay.co.nz>
 *
 * All rights reserved. Distributed under the terms of the MIT License.
 */


#include "AbstractPackageProcess.h"

#include <Autolock.h>

#include "Logger.h"
#include "Model.h"
#include "PackageKitUtils.h"
#include "PackageManager.h"
#include "PackageUtils.h"


using namespace BPackageKit;


AbstractPackageProcess::AbstractPackageProcess(const BString& packageName, Model* model)
        :
        fPackageName(packageName),
        fModel(model)
{
}


AbstractPackageProcess::~AbstractPackageProcess()
{
}


int32
AbstractPackageProcess::InstallLocation() const
{
        PackageInfoRef package = FindPackageByName(fPackageName);
        if (package.IsSet())
                return PackageKitUtils::DeriveInstallLocation(package.Get());
        return B_PACKAGE_INSTALLATION_LOCATION_SYSTEM;
}


PackageInfoRef
AbstractPackageProcess::FindPackageByName(const BString& packageName) const
{
        return fModel->PackageForName(packageName);
}


void
AbstractPackageProcess::SetPackageState(const BString& packageName, PackageState state)
{
        PackageInfoRef package = fModel->PackageForName(packageName);

        if (package.IsSet()) {
                PackageLocalInfoRef localInfo
                        = PackageLocalInfoBuilder(package->LocalInfo()).WithState(state).BuildRef();

                PackageInfoRef updatedPackage
                        = PackageInfoBuilder(package).WithLocalInfo(localInfo).BuildRef();

                fModel->AddPackage(updatedPackage);

        } else {
                HDERROR("setting state, but the package [%s] is not present", packageName.String());
        }
}


void
AbstractPackageProcess::SetPackageDownloadProgress(const BString& packageName, float value)
{
        PackageInfoRef package = fModel->PackageForName(packageName);

        if (package.IsSet()) {
                PackageLocalInfoRef localInfo
                        = PackageLocalInfoBuilder(package->LocalInfo()).WithDownloadProgress(value).BuildRef();

                PackageInfoRef updatedPackage
                        = PackageInfoBuilder(package).WithLocalInfo(localInfo).BuildRef();

                fModel->AddPackage(updatedPackage);

        } else {
                HDERROR("setting progress, but the package [%s] is not present", packageName.String());
        }
}


void
AbstractPackageProcess::ClearPackageInstallationLocations(const BString& packageName)
{
        PackageInfoRef package = fModel->PackageForName(packageName);

        if (package.IsSet()) {
                PackageLocalInfoRef localInfo
                        = PackageLocalInfoBuilder(package->LocalInfo()).ClearInstallationLocations().BuildRef();

                PackageInfoRef updatedPackage
                        = PackageInfoBuilder(package).WithLocalInfo(localInfo).BuildRef();

                fModel->AddPackage(updatedPackage);

        } else {
                HDERROR("clearing installation locations, but the package [%s] is not present",
                        packageName.String());
        }
}