downloader: Add a progressbar to single downloads

[pakfire.git] / src / libpakfire / package.c

/*#############################################################################
#                                                                             #
# Pakfire - The IPFire package management system                              #
# Copyright (C) 2013 Pakfire development team                                 #
#                                                                             #
# This program is free software: you can redistribute it and/or modify        #
# it under the terms of the GNU General Public License as published by        #
# the Free Software Foundation, either version 3 of the License, or           #
# (at your option) any later version.                                         #
#                                                                             #
# This program is distributed in the hope that it will be useful,             #
# but WITHOUT ANY WARRANTY; without even the implied warranty of              #
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               #
# GNU General Public License for more details.                                #
#                                                                             #
# You should have received a copy of the GNU General Public License           #
# along with this program.  If not, see <http://www.gnu.org/licenses/>.       #
#                                                                             #
#############################################################################*/

#include <linux/limits.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>

#include <solv/evr.h>
#include <solv/pool.h>
#include <solv/pooltypes.h>
#include <solv/repo.h>
#include <solv/solvable.h>

#include <pakfire/archive.h>
#include <pakfire/constants.h>
#include <pakfire/file.h>
#include <pakfire/filelist.h>
#include <pakfire/i18n.h>
#include <pakfire/logging.h>
#include <pakfire/package.h>
#include <pakfire/pakfire.h>
#include <pakfire/private.h>
#include <pakfire/relation.h>
#include <pakfire/relationlist.h>
#include <pakfire/repo.h>
#include <pakfire/util.h>

struct _PakfirePackage {
	Pakfire pakfire;
	Id id;
	int nrefs;

	PakfireRepo repo;
};

static Pool* pakfire_package_get_solv_pool(PakfirePackage pkg) {
	return pakfire_get_solv_pool(pkg->pakfire);
}

static void pakfire_package_add_self_provides(Pakfire pakfire, PakfirePackage pkg, const char* name, const char* evr) {
	PakfireRelation relation = pakfire_relation_create(pakfire, name, PAKFIRE_EQ, evr);
	pakfire_package_add_provides(pkg, relation);
	pakfire_relation_unref(relation);
}

PAKFIRE_EXPORT PakfirePackage pakfire_package_create_from_solvable(Pakfire pakfire, Id id) {
	PakfirePackage pkg = calloc(1, sizeof(*pkg));
	if (!pkg)
		return NULL;

	DEBUG(pakfire, "Allocated Package at %p\n", pkg);

	pkg->pakfire = pakfire_ref(pakfire);
	pkg->id = id;

	// Initialize reference counter
	pkg->nrefs = 1;

	return pkg;
}

PAKFIRE_EXPORT PakfirePackage pakfire_package_create(Pakfire pakfire, PakfireRepo repo, const char* name, const char* evr, const char* arch) {
	Id id = pakfire_repo_add_solvable(repo);
	if (!id)
		return NULL;

	PakfirePackage pkg = pakfire_package_create_from_solvable(pakfire, id);
	if (!pkg)
		return NULL;

	pkg->repo = pakfire_repo_ref(repo);

	// Set the given attributes
	pakfire_package_set_name(pkg, name);
	pakfire_package_set_evr(pkg, evr);
	pakfire_package_set_arch(pkg, arch);

	// Add self-provides
	pakfire_package_add_self_provides(pakfire, pkg, name, evr);

	return pkg;
}

static void pakfire_package_free(PakfirePackage pkg) {
	DEBUG(pkg->pakfire, "Releasing Package at %p\n", pkg);

	if (pkg->repo)
		pakfire_repo_unref(pkg->repo);

	pakfire_unref(pkg->pakfire);
	free(pkg);
}

PAKFIRE_EXPORT PakfirePackage pakfire_package_ref(PakfirePackage pkg) {
	pkg->nrefs++;

	return pkg;
}

PAKFIRE_EXPORT PakfirePackage pakfire_package_unref(PakfirePackage pkg) {
	if (!pkg)
		return NULL;

	if (--pkg->nrefs > 0)
		return pkg;

	pakfire_package_free(pkg);
	return NULL;
}

PAKFIRE_EXPORT Pakfire pakfire_package_get_pakfire(PakfirePackage pkg) {
	return pakfire_ref(pkg->pakfire);
}

static Solvable* get_solvable(PakfirePackage pkg) {
	Pool* pool = pakfire_package_get_solv_pool(pkg);

	return pool_id2solvable(pool, pkg->id);
}

static Repo* pakfire_package_solv_repo(PakfirePackage pkg) {
	Solvable* s = get_solvable(pkg);

	return s->repo;
}

static Id pakfire_package_get_handle(PakfirePackage pkg) {
	Pool* pool = pakfire_package_get_solv_pool(pkg);
	Solvable* s = get_solvable(pkg);

	return s - pool->solvables;
}

PAKFIRE_EXPORT int pakfire_package_identical(PakfirePackage pkg1, PakfirePackage pkg2) {
	return pkg1->id == pkg2->id;
}

PAKFIRE_EXPORT int pakfire_package_cmp(PakfirePackage pkg1, PakfirePackage pkg2) {
	Pool* pool = pakfire_package_get_solv_pool(pkg1);

	Solvable* s1 = get_solvable(pkg1);
	Solvable* s2 = get_solvable(pkg2);

	// Check names
	const char* str1 = pool_id2str(pool, s1->name);
	const char* str2 = pool_id2str(pool, s2->name);

	int ret = strcmp(str1, str2);
	if (ret)
		return ret;

	// Check the version string
	ret = pakfire_package_evr_cmp(pkg1, pkg2);
	if (ret)
		return ret;

	// Check repositories
	PakfireRepo repo1 = pakfire_package_get_repo(pkg1);
	PakfireRepo repo2 = pakfire_package_get_repo(pkg2);

	if (repo1 && repo2) {
		ret = pakfire_repo_cmp(repo1, repo2);
	}

	pakfire_repo_unref(repo1);
	pakfire_repo_unref(repo2);

	if (ret)
		return ret;

	// Check package architectures
	str1 = pool_id2str(pool, s1->arch);
	str2 = pool_id2str(pool, s2->arch);

	return strcmp(str1, str2);
}

PAKFIRE_EXPORT int pakfire_package_evr_cmp(PakfirePackage pkg1, PakfirePackage pkg2) {
	Pool* pool = pakfire_package_get_solv_pool(pkg1);

	Solvable* s1 = get_solvable(pkg1);
	Solvable* s2 = get_solvable(pkg2);

	return pool_evrcmp(pool, s1->evr, s2->evr, EVRCMP_COMPARE);
}

PAKFIRE_EXPORT Id pakfire_package_id(PakfirePackage pkg) {
	return pkg->id;
}

PAKFIRE_EXPORT char* pakfire_package_get_nevra(PakfirePackage pkg) {
	Pool* pool = pakfire_package_get_solv_pool(pkg);
	Solvable* s = get_solvable(pkg);

	const char* nevra = pool_solvable2str(pool, s);

	return strdup(nevra);
}

PAKFIRE_EXPORT const char* pakfire_package_get_name(PakfirePackage pkg) {
	Pool* pool = pakfire_package_get_solv_pool(pkg);
	Solvable* s = get_solvable(pkg);

	return pool_id2str(pool, s->name);
}

PAKFIRE_EXPORT void pakfire_package_set_name(PakfirePackage pkg, const char* name) {
	Pool* pool = pakfire_package_get_solv_pool(pkg);
	Solvable* s = get_solvable(pkg);

	s->name = pool_str2id(pool, name, 1);
}

PAKFIRE_EXPORT const char* pakfire_package_get_evr(PakfirePackage pkg) {
	Pool* pool = pakfire_package_get_solv_pool(pkg);
	Solvable* s = get_solvable(pkg);

	return pool_id2str(pool, s->evr);
}

PAKFIRE_EXPORT void pakfire_package_set_evr(PakfirePackage pkg, const char* evr) {
	Pool* pool = pakfire_package_get_solv_pool(pkg);
	Solvable* s = get_solvable(pkg);

	s->evr = pool_str2id(pool, evr, 1);
}

PAKFIRE_EXPORT char* pakfire_package_join_evr(const char* e, const char* v, const char* r) {
	size_t l = strlen(v) + strlen(r) + 2;

	// Do not include epoch when it is zero
	if (strncmp(e, "0", strlen("0")) == 0) {
		e = NULL;
	} else {
		l += strlen(e);
	}

	char* buffer = malloc(l + 1);
	if (e)
		snprintf(buffer, l + 1, "%s:%s-%s", e, v, r);
	else
		snprintf(buffer, l + 1, "%s-%s", v, r);

	return buffer;
}

static void split_evr(Pool* pool, const char* evr_c, char** epoch, char** version, char** release) {
    char *e, *v, *r;

	char* evr = pool_alloctmpspace(pool, strlen(evr_c) + 1);
	strcpy(evr, evr_c);

    for (e = evr + 1; *e != ':' && *e != '-'; ++e)
    	;

    if (*e == '-') {
		*e = '\0';
		v = evr;
		r = e + 1;
		e = NULL;
	} else { /* *e == ':' */
		*e = '\0';
		v = e + 1;
		e = evr;
		for (r = v + 1; *r != '-'; ++r)
			;
		*r = '\0';
		r++;
	}

    *epoch = e;
    *version = v;
    *release = r;
}

PAKFIRE_EXPORT unsigned long pakfire_package_get_epoch(PakfirePackage pkg) {
	Pool* pool = pakfire_package_get_solv_pool(pkg);
	char *e, *v, *r, *endptr;

    unsigned long epoch = 0;

	split_evr(pool, pakfire_package_get_evr(pkg), &e, &v, &r);

	if (e) {
		long int converted = strtol(e, &endptr, 10);
		epoch = converted;
	}

	return epoch;
}

PAKFIRE_EXPORT const char* pakfire_package_get_version(PakfirePackage pkg) {
	Pool* pool = pakfire_package_get_solv_pool(pkg);
	char *e, *v, *r;

	split_evr(pool, pakfire_package_get_evr(pkg), &e, &v, &r);
	return strdup(v);
}

PAKFIRE_EXPORT const char* pakfire_package_get_release(PakfirePackage pkg) {
	Pool* pool = pakfire_package_get_solv_pool(pkg);
	char *e, *v, *r;

	split_evr(pool, pakfire_package_get_evr(pkg), &e, &v, &r);
	return strdup(r);
}

PAKFIRE_EXPORT const char* pakfire_package_get_arch(PakfirePackage pkg) {
	Pool* pool = pakfire_package_get_solv_pool(pkg);
	Solvable* s = get_solvable(pkg);

	return pool_id2str(pool, s->arch);
}

PAKFIRE_EXPORT void pakfire_package_set_arch(PakfirePackage pkg, const char* arch) {
	Pool* pool = pakfire_package_get_solv_pool(pkg);
	Solvable* s = get_solvable(pkg);

	s->arch = pool_str2id(pool, arch, 1);
}

static void pakfire_package_internalize_repo(PakfirePackage pkg) {
	PakfireRepo repo = pakfire_package_get_repo(pkg);
	if (repo) {
		pakfire_repo_internalize(repo);
		pakfire_repo_unref(repo);
	}
}

static const char* pakfire_package_get_string(PakfirePackage pkg, int key) {
	pakfire_package_internalize_repo(pkg);

	Solvable* s = get_solvable(pkg);
	const char* str = solvable_lookup_str(s, key);

	if (!str)
		return NULL;

	if (strlen(str) == 0)
		return NULL;

	return str;
}

static void pakfire_package_set_string(PakfirePackage pkg, int key, const char* value) {
	Solvable* s = get_solvable(pkg);

	if (!value)
		value = "";

	solvable_set_poolstr(s, key, value);
}

static void pakfire_package_add_string_array(PakfirePackage pkg, int key, const char* value) {
	Solvable* s = get_solvable(pkg);

	if (!value)
		return;

	solvable_add_poolstr_array(s, key, value);
}

PAKFIRE_EXPORT const char* pakfire_package_get_uuid(PakfirePackage pkg) {
	return pakfire_package_get_string(pkg, SOLVABLE_PKGID);
}

PAKFIRE_EXPORT void pakfire_package_set_uuid(PakfirePackage pkg, const char* uuid) {
	pakfire_package_set_string(pkg, SOLVABLE_PKGID, uuid);
}

PAKFIRE_EXPORT const char* pakfire_package_get_checksum(PakfirePackage pkg) {
	return pakfire_package_get_string(pkg, SOLVABLE_CHECKSUM);
}

PAKFIRE_EXPORT void pakfire_package_set_checksum(PakfirePackage pkg, const char* checksum) {
	pakfire_package_set_string(pkg, SOLVABLE_CHECKSUM, checksum);
}

PAKFIRE_EXPORT const char* pakfire_package_get_summary(PakfirePackage pkg) {
	return pakfire_package_get_string(pkg, SOLVABLE_SUMMARY);
}

PAKFIRE_EXPORT void pakfire_package_set_summary(PakfirePackage pkg, const char* summary) {
	pakfire_package_set_string(pkg, SOLVABLE_SUMMARY, summary);
}

PAKFIRE_EXPORT const char* pakfire_package_get_description(PakfirePackage pkg) {
	return pakfire_package_get_string(pkg, SOLVABLE_DESCRIPTION);
}

PAKFIRE_EXPORT void pakfire_package_set_description(PakfirePackage pkg, const char* description) {
	pakfire_package_set_string(pkg, SOLVABLE_DESCRIPTION, description);
}

PAKFIRE_EXPORT const char* pakfire_package_get_license(PakfirePackage pkg) {
	return pakfire_package_get_string(pkg, SOLVABLE_LICENSE);
}

PAKFIRE_EXPORT void pakfire_package_set_license(PakfirePackage pkg, const char* license) {
	pakfire_package_set_string(pkg, SOLVABLE_LICENSE, license);
}

PAKFIRE_EXPORT const char* pakfire_package_get_url(PakfirePackage pkg) {
	return pakfire_package_get_string(pkg, SOLVABLE_URL);
}

PAKFIRE_EXPORT void pakfire_package_set_url(PakfirePackage pkg, const char* url) {
	pakfire_package_set_string(pkg, SOLVABLE_URL, url);
}

PAKFIRE_EXPORT const char* pakfire_package_get_groups(PakfirePackage pkg) {
	return pakfire_package_get_string(pkg, SOLVABLE_GROUP);
}

PAKFIRE_EXPORT void pakfire_package_set_groups(PakfirePackage pkg, const char* groups) {
	char** list = pakfire_split_string(groups, ' ');

	while (list && *list) {
		pakfire_package_add_string_array(pkg, SOLVABLE_GROUP, *list);
		list++;
	}
}

PAKFIRE_EXPORT const char* pakfire_package_get_vendor(PakfirePackage pkg) {
	return pakfire_package_get_string(pkg, SOLVABLE_VENDOR);
}

PAKFIRE_EXPORT void pakfire_package_set_vendor(PakfirePackage pkg, const char* vendor) {
	pakfire_package_set_string(pkg, SOLVABLE_VENDOR, vendor);
}

PAKFIRE_EXPORT const char* pakfire_package_get_maintainer(PakfirePackage pkg) {
	return pakfire_package_get_string(pkg, SOLVABLE_PACKAGER);
}

PAKFIRE_EXPORT void pakfire_package_set_maintainer(PakfirePackage pkg, const char* maintainer) {
	pakfire_package_set_string(pkg, SOLVABLE_PACKAGER, maintainer);
}

PAKFIRE_EXPORT const char* pakfire_package_get_filename(PakfirePackage pkg) {
	return pakfire_package_get_string(pkg, SOLVABLE_MEDIAFILE);
}

PAKFIRE_EXPORT void pakfire_package_set_filename(PakfirePackage pkg, const char* filename) {
	pakfire_package_set_string(pkg, SOLVABLE_MEDIAFILE, filename);
}

PAKFIRE_EXPORT int pakfire_package_is_installed(PakfirePackage pkg) {
	Pool* pool = pakfire_package_get_solv_pool(pkg);
	Solvable* s = get_solvable(pkg);

	return pool->installed == s->repo;
}

static unsigned long long pakfire_package_get_num(PakfirePackage pkg, Id type) {
	pakfire_package_internalize_repo(pkg);

	Solvable* s = get_solvable(pkg);
	return solvable_lookup_num(s, type, 0);
}

static void pakfire_package_set_num(PakfirePackage pkg, Id type, unsigned long long value) {
	Solvable* s = get_solvable(pkg);

	solvable_set_num(s, type, value);
}

PAKFIRE_EXPORT unsigned long long pakfire_package_get_downloadsize(PakfirePackage pkg) {
	return pakfire_package_get_num(pkg, SOLVABLE_DOWNLOADSIZE);
}

PAKFIRE_EXPORT void pakfire_package_set_downloadsize(PakfirePackage pkg, unsigned long long downloadsize) {
	return pakfire_package_set_num(pkg, SOLVABLE_DOWNLOADSIZE, downloadsize);
}

PAKFIRE_EXPORT unsigned long long pakfire_package_get_installsize(PakfirePackage pkg) {
	return pakfire_package_get_num(pkg, SOLVABLE_INSTALLSIZE);
}

PAKFIRE_EXPORT void pakfire_package_set_installsize(PakfirePackage pkg, unsigned long long installsize) {
	return pakfire_package_set_num(pkg, SOLVABLE_INSTALLSIZE, installsize);
}

PAKFIRE_EXPORT unsigned long long pakfire_package_get_size(PakfirePackage pkg) {
	if (pakfire_package_is_installed(pkg))
		return pakfire_package_get_installsize(pkg);

	return pakfire_package_get_downloadsize(pkg);
}

PAKFIRE_EXPORT const char* pakfire_package_get_build_host(PakfirePackage pkg) {
	return pakfire_package_get_string(pkg, SOLVABLE_BUILDHOST);
}

PAKFIRE_EXPORT void pakfire_package_set_build_host(PakfirePackage pkg, const char* build_host) {
	pakfire_package_set_string(pkg, SOLVABLE_BUILDHOST, build_host);
}

PAKFIRE_EXPORT time_t pakfire_package_get_build_time(PakfirePackage pkg) {
	return pakfire_package_get_num(pkg, SOLVABLE_BUILDTIME);
}

PAKFIRE_EXPORT void pakfire_package_set_build_time(PakfirePackage pkg, time_t build_time) {
	pakfire_package_set_num(pkg, SOLVABLE_BUILDTIME, build_time);
}

PAKFIRE_EXPORT time_t pakfire_package_get_install_time(PakfirePackage pkg) {
	return pakfire_package_get_num(pkg, SOLVABLE_INSTALLTIME);
}

PAKFIRE_EXPORT void pakfire_package_set_install_time(PakfirePackage pkg, time_t install_time) {
	pakfire_package_set_num(pkg, SOLVABLE_INSTALLTIME, install_time);
}

static PakfireRelationList pakfire_package_get_relationlist(
		PakfirePackage pkg, Id type, Id marker) {
	Queue q;
	queue_init(&q);

	Solvable* s = get_solvable(pkg);
	solvable_lookup_deparray(s, type, &q, marker);

	PakfireRelationList list;
	int r = pakfire_relationlist_from_queue(&list, pkg->pakfire, q);

	queue_free(&q);

	if (r)
		return NULL;

	return list;
}

static void pakfire_package_set_relationlist(PakfirePackage pkg, Id type,
		PakfireRelationList relationlist, Id marker) {
	Solvable* s = get_solvable(pkg);
	Queue* q = pakfire_relationlist_get_queue(relationlist);

#if 0
	// This implemention should be the fastest, but unfortunately does not work.
	solvable_set_deparray(s, type, q, marker);
#else
	for (int i = 0; i < q->count; i++)
		solvable_add_deparray(s, type, q->elements[i], marker);
#endif
}

static void pakfire_package_add_relation(PakfirePackage pkg, Id type,
		PakfireRelation relation, Id marker) {
	Solvable* s = get_solvable(pkg);

	solvable_add_deparray(s, type, pakfire_relation_get_id(relation), marker);
}

PAKFIRE_EXPORT PakfireRelationList pakfire_package_get_provides(PakfirePackage pkg) {
	return pakfire_package_get_relationlist(pkg, SOLVABLE_PROVIDES, -SOLVABLE_FILEMARKER);
}

PAKFIRE_EXPORT void pakfire_package_set_provides(PakfirePackage pkg, PakfireRelationList relationlist) {
	pakfire_package_set_relationlist(pkg, SOLVABLE_PROVIDES, relationlist, -SOLVABLE_FILEMARKER);
}

PAKFIRE_EXPORT void pakfire_package_add_provides(PakfirePackage pkg, PakfireRelation relation) {
	pakfire_package_add_relation(pkg, SOLVABLE_PROVIDES, relation, -SOLVABLE_FILEMARKER);
}

PAKFIRE_EXPORT PakfireRelationList pakfire_package_get_prerequires(PakfirePackage pkg) {
	return pakfire_package_get_relationlist(pkg, SOLVABLE_REQUIRES, SOLVABLE_PREREQMARKER);
}

PAKFIRE_EXPORT void pakfire_package_set_prerequires(PakfirePackage pkg, PakfireRelationList relationlist) {
	pakfire_package_set_relationlist(pkg, SOLVABLE_REQUIRES, relationlist, SOLVABLE_PREREQMARKER);
}

PAKFIRE_EXPORT PakfireRelationList pakfire_package_get_requires(PakfirePackage pkg) {
	return pakfire_package_get_relationlist(pkg, SOLVABLE_REQUIRES, -SOLVABLE_PREREQMARKER);
}

PAKFIRE_EXPORT void pakfire_package_set_requires(PakfirePackage pkg, PakfireRelationList relationlist) {
	pakfire_package_set_relationlist(pkg, SOLVABLE_REQUIRES, relationlist, -SOLVABLE_PREREQMARKER);
}

PAKFIRE_EXPORT void pakfire_package_add_requires(PakfirePackage pkg, PakfireRelation relation) {
	pakfire_package_add_relation(pkg, SOLVABLE_REQUIRES, relation, -SOLVABLE_PREREQMARKER);
}

PAKFIRE_EXPORT PakfireRelationList pakfire_package_get_conflicts(PakfirePackage pkg) {
	return pakfire_package_get_relationlist(pkg, SOLVABLE_CONFLICTS, 0);
}

PAKFIRE_EXPORT void pakfire_package_set_conflicts(PakfirePackage pkg, PakfireRelationList relationlist) {
	pakfire_package_set_relationlist(pkg, SOLVABLE_CONFLICTS, relationlist, 0);
}

PAKFIRE_EXPORT void pakfire_package_add_conflicts(PakfirePackage pkg, PakfireRelation relation) {
	pakfire_package_add_relation(pkg, SOLVABLE_CONFLICTS, relation, 0);
}

PAKFIRE_EXPORT PakfireRelationList pakfire_package_get_obsoletes(PakfirePackage pkg) {
	return pakfire_package_get_relationlist(pkg, SOLVABLE_OBSOLETES, 0);
}

PAKFIRE_EXPORT void pakfire_package_set_obsoletes(PakfirePackage pkg, PakfireRelationList relationlist) {
	pakfire_package_set_relationlist(pkg, SOLVABLE_OBSOLETES, relationlist, 0);
}

PAKFIRE_EXPORT void pakfire_package_add_obsoletes(PakfirePackage pkg, PakfireRelation relation) {
	pakfire_package_add_relation(pkg, SOLVABLE_OBSOLETES, relation, 0);
}

PAKFIRE_EXPORT PakfireRelationList pakfire_package_get_recommends(PakfirePackage pkg) {
	return pakfire_package_get_relationlist(pkg, SOLVABLE_RECOMMENDS, 0);
}

PAKFIRE_EXPORT void pakfire_package_set_recommends(PakfirePackage pkg, PakfireRelationList relationlist) {
	pakfire_package_set_relationlist(pkg, SOLVABLE_RECOMMENDS, relationlist, 0);
}

PAKFIRE_EXPORT void pakfire_package_add_recommends(PakfirePackage pkg, PakfireRelation relation) {
	pakfire_package_add_relation(pkg, SOLVABLE_RECOMMENDS, relation, 0);
}

PAKFIRE_EXPORT PakfireRelationList pakfire_package_get_suggests(PakfirePackage pkg) {
	return pakfire_package_get_relationlist(pkg, SOLVABLE_SUGGESTS, 0);
}

PAKFIRE_EXPORT void pakfire_package_set_suggests(PakfirePackage pkg, PakfireRelationList relationlist) {
	pakfire_package_set_relationlist(pkg, SOLVABLE_SUGGESTS, relationlist, 0);
}

PAKFIRE_EXPORT void pakfire_package_add_suggests(PakfirePackage pkg, PakfireRelation relation) {
	pakfire_package_add_relation(pkg, SOLVABLE_SUGGESTS, relation, 0);
}

PAKFIRE_EXPORT PakfireRelationList pakfire_package_get_supplements(PakfirePackage pkg) {
	return pakfire_package_get_relationlist(pkg, SOLVABLE_SUPPLEMENTS, 0);
}

PAKFIRE_EXPORT void pakfire_package_set_supplements(PakfirePackage pkg, PakfireRelationList relationlist) {
	pakfire_package_set_relationlist(pkg, SOLVABLE_SUPPLEMENTS, relationlist, 0);
}

PAKFIRE_EXPORT PakfireRelationList pakfire_package_get_enhances(PakfirePackage pkg) {
	return pakfire_package_get_relationlist(pkg, SOLVABLE_ENHANCES, 0);
}

PAKFIRE_EXPORT void pakfire_package_set_enhances(PakfirePackage pkg, PakfireRelationList relationlist) {
	pakfire_package_set_relationlist(pkg, SOLVABLE_ENHANCES, relationlist, 0);
}

/*
	Fast-path function to check if a package is in a certain repository
*/
int pakfire_package_is_in_repo(PakfirePackage pkg, Repo* repo) {
	Solvable* s = get_solvable(pkg);

	return s->repo == repo;
}

PAKFIRE_EXPORT PakfireRepo pakfire_package_get_repo(PakfirePackage pkg) {
	if (!pkg->repo) {
		Solvable* s = get_solvable(pkg);

		pkg->repo = pakfire_repo_create_from_repo(pkg->pakfire, s->repo);
	}

	return pakfire_repo_ref(pkg->repo);
}

PAKFIRE_EXPORT void pakfire_package_set_repo(PakfirePackage pkg, PakfireRepo repo) {
	// Free old repository
	if (pkg->repo)
		pakfire_repo_unref(pkg->repo);

	Solvable* s = get_solvable(pkg);
	s->repo = pakfire_repo_get_repo(repo);

	pkg->repo = pakfire_repo_ref(repo);
}

PAKFIRE_EXPORT char* pakfire_package_get_location(PakfirePackage pkg) {
	pakfire_package_internalize_repo(pkg);

	Solvable* s = get_solvable(pkg);

	const char* location = solvable_get_location(s, NULL);
	return strdup(location);
}

static void pakfire_package_dump_add_line(char** str, const char* key, const char* val) {
	if (val)
		asprintf(str, "%s%-15s: %s\n", *str, key ? key : "", val);
}

static void pakfire_package_dump_add_lines(char** str, const char* key, const char* val) {
	char** lines = pakfire_split_string(val, '\n');
	if (!lines)
		return;

	while (*lines) {
		pakfire_package_dump_add_line(str, key, *lines);
		lines++;
	}
}

static void pakfire_package_dump_add_line_date(char** str, const char* key, unsigned long long date) {
	// Convert from integer to tm struct.
	struct tm* timer = gmtime((time_t *)&date);

	char val[128];
	strftime(val, sizeof(val) - 1, "%a, %d %b %Y %T %z", timer);

	pakfire_package_dump_add_line(str, key, val);
}

static void pakfire_package_dump_add_line_relations(char** str, const char* key, PakfireRelationList deps) {
	pakfire_relationlist_sort(deps);

	size_t size = pakfire_relationlist_size(deps);
	for (unsigned int i = 0; i < size; i++) {
		PakfireRelation relation = pakfire_relationlist_get(deps, i);

		if (relation) {
			char* dep = pakfire_relation_str(relation);
			pakfire_relation_unref(relation);

			if (dep) {
				pakfire_package_dump_add_line(str, (i == 0) ? key : "", dep);
				free(dep);
			}
		}
	}
}

static void pakfire_package_dump_add_line_size(char** str, const char* key, unsigned long long size) {
	char* val = pakfire_format_size(size);

	if (val) {
		pakfire_package_dump_add_line(str, key, val);
		free(val);
	}
}

PAKFIRE_EXPORT char* pakfire_package_dump(PakfirePackage pkg, int flags) {
	char* string = "";

	// Name
	const char* name = pakfire_package_get_name(pkg);
	pakfire_package_dump_add_line(&string, _("Name"), name);

	// Version
	const char* version = pakfire_package_get_version(pkg);
	pakfire_package_dump_add_line(&string, _("Version"), version);

	// Release
	const char* release = pakfire_package_get_release(pkg);
	pakfire_package_dump_add_line(&string, _("Release"), release);

	// Size
	unsigned long long size = pakfire_package_get_size(pkg);
	pakfire_package_dump_add_line_size(&string, _("Size"), size);

	// Installed size
	if (pakfire_package_is_installed(pkg)) {
		unsigned long long installsize = pakfire_package_get_installsize(pkg);
		pakfire_package_dump_add_line_size(&string, _("Installed size"), installsize);

	// Downloadsize
	} else {
		unsigned long long downloadsize = pakfire_package_get_downloadsize(pkg);
		pakfire_package_dump_add_line_size(&string, _("Download size"), downloadsize);
	}

	PakfireRepo repo = pakfire_package_get_repo(pkg);
	if (repo) {
		const char* repo_name = pakfire_repo_get_name(repo);
		pakfire_package_dump_add_line(&string, _("Repo"), repo_name);

		pakfire_repo_unref(repo);
	}

	// Summary
	const char* summary = pakfire_package_get_summary(pkg);
	pakfire_package_dump_add_line(&string, _("Summary"), summary);

	// Description
	const char* description = pakfire_package_get_description(pkg);
	if (description)
		pakfire_package_dump_add_lines(&string, _("Description"), description);

	// Groups
	const char* groups = pakfire_package_get_groups(pkg);
	if (groups) {
		pakfire_package_dump_add_lines(&string, _("Groups"), groups);
	}

	// URL
	const char* url = pakfire_package_get_url(pkg);
	pakfire_package_dump_add_line(&string, _("URL"), url);

	// License
	const char* license = pakfire_package_get_license(pkg);
	pakfire_package_dump_add_line(&string, _("License"), license);

	if (flags & PAKFIRE_PKG_DUMP_LONG) {
		// Install Time
		time_t install_time = pakfire_package_get_install_time(pkg);
		if (install_time) {
			pakfire_package_dump_add_line_date(&string, _("Install Time"), install_time);
		}

		// Maintainer
		const char* maintainer = pakfire_package_get_maintainer(pkg);
		pakfire_package_dump_add_line(&string, _("Maintainer"), maintainer);

		// Vendor
		const char* vendor = pakfire_package_get_vendor(pkg);
		pakfire_package_dump_add_line(&string, _("Vendor"), vendor);

		// UUID
		const char* uuid = pakfire_package_get_uuid(pkg);
		pakfire_package_dump_add_line(&string, _("UUID"), uuid);

		// Build time
		time_t build_time = pakfire_package_get_build_time(pkg);
		pakfire_package_dump_add_line_date(&string, _("Build Time"), build_time);

		// Build host
		const char* build_host = pakfire_package_get_build_host(pkg);
		pakfire_package_dump_add_line(&string, _("Build Host"), build_host);

		// Dependencies

		const struct relation {
			const char* name;
			PakfireRelationList (*get)(PakfirePackage pkg);
		} relations[] = {
			{ _("Provides"),     pakfire_package_get_provides, },
			{ _("Pre-Requires"), pakfire_package_get_prerequires, },
			{ _("Requires"),     pakfire_package_get_requires, },
			{ _("Conflicts"),    pakfire_package_get_conflicts, },
			{ _("Obsoletes"),    pakfire_package_get_obsoletes, },
			{ _("Recommends"),   pakfire_package_get_recommends, },
			{ _("Suggests"),     pakfire_package_get_suggests, },
			{ _("Supplements"),  pakfire_package_get_supplements, },
			{ _("Enhances"),     pakfire_package_get_enhances, },
			{ NULL, NULL},
		};

		for (const struct relation* relation = relations; relation->name; relation++) {
			PakfireRelationList list = relation->get(pkg);
			if (list) {
				pakfire_package_dump_add_line_relations(&string, relation->name, list);
				pakfire_relationlist_unref(list);
			}
		}
	}

	if (flags & PAKFIRE_PKG_DUMP_FILELIST) {
		PakfireFilelist filelist = pakfire_package_get_filelist(pkg);

		const char* prefix = _("Filelist");

		for (unsigned int i = 0; i < pakfire_filelist_size(filelist); i++) {
			PakfireFile file = pakfire_filelist_get(filelist, i);

			const char* path = pakfire_file_get_path(file);
			pakfire_package_dump_add_line(&string, prefix, path);

			pakfire_file_unref(file);

			// Only prefix the first line
			prefix = NULL;
		}

		pakfire_filelist_unref(filelist);
	}

	return string;
}

PAKFIRE_EXPORT int pakfire_package_is_cached(PakfirePackage pkg) {
	char* path = pakfire_package_get_cache_path(pkg);

	// Check if the file is readable
	int r = access(path, R_OK);
	free(path);

	return (r == 0);
}

PAKFIRE_EXPORT char* pakfire_package_get_cache_path(PakfirePackage pkg) {
	char path[PATH_MAX];

	const char* filename = pakfire_package_get_filename(pkg);
	const char* checksum = pakfire_package_get_checksum(pkg);

	if (strlen(checksum) < 3)
		return NULL;

	pakfire_make_cache_path(pkg->pakfire, path, sizeof(path) - 1,
		"%c%c/%s/%s", checksum[0], checksum[1], checksum + 2, filename);

	return strdup(path);
}

PAKFIRE_EXPORT PakfireArchive pakfire_package_get_archive(PakfirePackage pkg) {
	// Otherwise open the archive from the cache
	char* path = pakfire_package_get_cache_path(pkg);
	if (!path)
		return NULL;

	PakfireArchive archive = NULL;

	// Open archive
	int r = pakfire_archive_open(&archive, pkg->pakfire, path);
	free(path);

	if (r == 0)
		return archive;

	return NULL;
}

static int pakfire_package_fetch_legacy_filelist(PakfirePackage pkg, PakfireFilelist filelist) {
	pakfire_package_internalize_repo(pkg);

	PakfireRepo repo = pakfire_package_get_repo(pkg);
	Solvable* s = get_solvable(pkg);
	Pool* p = pakfire_package_get_solv_pool(pkg);
	Repo* r = pakfire_repo_get_repo(repo);
	pakfire_repo_unref(repo);

	int found_marker = 0;

	Id id, *ids;
	ids = r->idarraydata + s->provides;
	while((id = *ids++) != 0) {
		const char* path = pool_dep2str(p, id);

		if (found_marker) {
			PakfireFile file;

			int r = pakfire_file_create(&file, pkg->pakfire);
			if (r)
				return r;

			// Set path
			pakfire_file_set_path(file, path);

			r = pakfire_filelist_append(filelist, file);
			if (r)
				return r;

			pakfire_file_unref(file);
			continue;
		}

		if (strcmp(path, PAKFIRE_SOLVABLE_FILEMARKER) == 0)
			++found_marker;
	}

	return 0;
}

static int pakfire_package_fetch_filelist(PakfirePackage pkg, PakfireFilelist filelist) {
	int r;

	pakfire_package_internalize_repo(pkg);

	Pool* pool = pakfire_package_get_solv_pool(pkg);
	Repo* repo = pakfire_package_solv_repo(pkg);
	Id handle = pakfire_package_get_handle(pkg);

	Dataiterator di;
	dataiterator_init(&di, pool, repo, handle,
		SOLVABLE_FILELIST, NULL, SEARCH_FILES | SEARCH_COMPLETE_FILELIST);

	while (dataiterator_step(&di)) {
		PakfireFile file;

		r = pakfire_file_create(&file, pkg->pakfire);
		if (r)
			return r;

		pakfire_file_set_path(file, di.kv.str);

		// Append to filelist
		pakfire_filelist_append(filelist, file);
		pakfire_file_unref(file);
	}
	dataiterator_free(&di);

	// If the file list is empty, we fall back to read files
	// in the older format.
	if (pakfire_filelist_is_empty(filelist)) {
		r = pakfire_package_fetch_legacy_filelist(pkg, filelist);
		if (r)
			return r;
	}

	// Sort the list
	pakfire_filelist_sort(filelist);

	return 0;
}

PAKFIRE_EXPORT PakfireFilelist pakfire_package_get_filelist(PakfirePackage pkg) {
	PakfireFilelist filelist;

	int r = pakfire_filelist_create(&filelist, pkg->pakfire);
	if (r)
		return NULL;

	r = pakfire_package_fetch_filelist(pkg, filelist);
	if (r) {
		pakfire_filelist_unref(filelist);
		return NULL;
	}

	return filelist;
}

static int pakfire_package_append_file(PakfirePackage pkg, const char* path) {
	// Fetch repodata
	PakfireRepo repo = pakfire_package_get_repo(pkg);
	Repodata* repodata = pakfire_repo_get_repodata(repo);
	pakfire_repo_unref(repo);

	Id handle = pakfire_package_get_handle(pkg);

	char* basename = pakfire_basename(path);
	char* dirname  = pakfire_dirname(path);

	// Convert directory into ID
	Id did = repodata_str2dir(repodata, dirname, 1);
	if (!did)
		did = repodata_str2dir(repodata, "/", 1);

	// Add data to list
	repodata_add_dirstr(repodata, handle,
		SOLVABLE_FILELIST, did, basename);

	free(basename);
	free(dirname);

	return 0;
}

PAKFIRE_EXPORT int pakfire_package_set_filelist(PakfirePackage pkg, PakfireFilelist filelist) {
	int r;

	for (unsigned int i = 0; i < pakfire_filelist_size(filelist); i++) {
		PakfireFile file = pakfire_filelist_get(filelist, i);

		const char* path = pakfire_file_get_path(file);
		if (path) {
			r = pakfire_package_append_file(pkg, path);
			if (r) {
				pakfire_file_unref(file);
				return r;
			}
		}

		pakfire_file_unref(file);

	}

	return 0;
}

int pakfire_package_set_filelist_from_string(PakfirePackage pkg, const char* files) {
	// Total length of the input string
	size_t size = strlen(files);

	// Copy files to the stack to be able to modify it
	char* buffer = alloca(size + 1);

	// Copy the content
	memcpy(buffer, files, size);

	// Replace all line breaks with NUL
	for (unsigned int i = 0; i < size; i++) {
		if (buffer[i] == '\n')
			buffer[i] = '\0';
	}

	size_t p = 0;
	while (p < size) {
		const char* name = &buffer[p];

		int r = pakfire_package_append_file(pkg, name);
		if (r)
			return r;

		p += strlen(name) + 1;
	}

	return 0;
}