/*
 * Copyright 2022-2025, Andrew Lindesay <apl@lindesay.co.nz>.
 * All rights reserved. Distributed under the terms of the MIT License.
 */

#include "PackageKitUtils.h"


using namespace BPackageKit;


/*static*/ status_t
PackageKitUtils::DeriveLocalFilePath(const PackageInfoRef package, BPath& path)
{
	if (!package.IsSet())
		return B_ERROR;

	PackageLocalInfoRef localInfo = package->LocalInfo();

	if (localInfo.IsSet() && localInfo->IsLocalFile()) {
		path.SetTo(localInfo->LocalFilePath());
		return B_OK;
	}

	path.Unset();
	BPackageInstallationLocation installationLocation = DeriveInstallLocation(package);
	directory_which which;
	status_t result = _DeriveDirectoryWhich(installationLocation, &which);

	if (result == B_OK)
		result = find_directory(which, &path);

	if (result == B_OK)
		path.Append(localInfo->FileName());

	return result;
}


/*static*/ status_t
PackageKitUtils::_DeriveDirectoryWhich(BPackageInstallationLocation location,
	directory_which* which)
{
	switch (location) {
		case B_PACKAGE_INSTALLATION_LOCATION_SYSTEM:
			*which = B_SYSTEM_PACKAGES_DIRECTORY;
			break;
		case B_PACKAGE_INSTALLATION_LOCATION_HOME:
			*which = B_USER_PACKAGES_DIRECTORY;
			break;
		default:
			debugger("illegal state: unsupported package installation location");
			return B_BAD_VALUE;
	}
	return B_OK;
}


/*static*/ BPackageInstallationLocation
PackageKitUtils::DeriveInstallLocation(const PackageInfoRef package)
{
	if (package.IsSet()) {

		PackageLocalInfoRef localInfo = package->LocalInfo();

		if (localInfo.IsSet()) {
			const PackageInstallationLocationSet& locations = localInfo->InstallationLocations();

			// If the package is already installed, return its first installed location
			if (locations.size() != 0)
				return static_cast<BPackageInstallationLocation>(*locations.begin());
		}
	}

	return B_PACKAGE_INSTALLATION_LOCATION_SYSTEM;
}


/*static*/ PackageInfoRef
PackageKitUtils::CreatePackageInfo(const BPackageInfo& info)
{
	PackageCoreInfoBuilder coreInfoBuilder = PackageCoreInfoBuilder()
												 .WithArchitecture(info.ArchitectureName())
												 .WithVersion(new PackageVersion(info.Version()))
												 .WithPublisher(CreatePublisherInfo(info));

	PackageLocalizedTextBuilder localizedTextBuilder = PackageLocalizedTextBuilder()
														   .WithTitle(info.Name())
														   .WithSummary(info.Summary())
														   .WithDescription(info.Description());

	// TODO: Retrieve local file size
	PackageLocalInfoBuilder localInfoBuilder
		= PackageLocalInfoBuilder().WithFlags(info.Flags()).WithFileName(info.FileName());

	return PackageInfoBuilder(info.Name())
		.WithCoreInfo(coreInfoBuilder.BuildRef())
		.WithLocalizedText(localizedTextBuilder.BuildRef())
		.WithLocalInfo(localInfoBuilder.BuildRef())
		.BuildRef();
}


/*static*/ PackagePublisherInfoRef
PackageKitUtils::CreatePublisherInfo(const BPackageInfo& info)
{
	BString publisherURL;
	if (info.URLList().CountStrings() > 0)
		publisherURL = info.URLList().StringAt(0);

	BString publisherName = info.Vendor();
	const BStringList& copyrightList = info.CopyrightList();
	if (!copyrightList.IsEmpty()) {
		publisherName = "";

		for (int32 i = 0; i < copyrightList.CountStrings(); i++) {
			if (!publisherName.IsEmpty())
				publisherName << ", ";
			publisherName << copyrightList.StringAt(i);
		}
	}
	if (!publisherName.IsEmpty())
		publisherName.Prepend("© ");

	PackagePublisherInfoRef result(new PackagePublisherInfo(publisherName, publisherURL), true);
	return result;
}