[people/stevee/pakfire.git] / src / libpakfire / build.c

/*#############################################################################
#                                                                             #
# Pakfire - The IPFire package management system                              #
# Copyright (C) 2021 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 <errno.h>
#include <stdlib.h>
#include <sys/mount.h>
#include <unistd.h>
#include <uuid/uuid.h>

#include <pakfire/build.h>
#include <pakfire/cgroup.h>
#include <pakfire/dist.h>
#include <pakfire/file.h>
#include <pakfire/jail.h>
#include <pakfire/logging.h>
#include <pakfire/mount.h>
#include <pakfire/package.h>
#include <pakfire/packager.h>
#include <pakfire/parser.h>
#include <pakfire/private.h>
#include <pakfire/repo.h>
#include <pakfire/request.h>
#include <pakfire/scriptlet.h>
#include <pakfire/util.h>

struct pakfire_build {
	struct pakfire* pakfire;
	int nrefs;

	// Flags
	int flags;

	// Build ID
	uuid_t id;
	char _id[UUID_STR_LEN];

	char target[PATH_MAX];

	// cgroup
	struct pakfire_cgroup* cgroup;

	// Jail
	struct pakfire_jail* jail;
};

static const char* stages[] = {
	"prepare",
	"build",
	"test",
	"install",
	NULL,
};

#define TEMPLATE \
	"#!/bin/bash --login\n" \
	"\n" \
	"set -e\n" \
	"set -x\n" \
	"\n" \
	"%%{_%s}\n" \
	"\n" \
	"exit 0\n"

/*
	This function creates a new jail which is pre-configured for a build job.

	TODO Add resource limits
*/
static struct pakfire_jail* pakfire_build_make_jail(struct pakfire* pakfire) {
	struct pakfire_jail* jail = NULL;
	int r;

	// Create a new jail
	r = pakfire_jail_create(&jail, pakfire, 0);
	if (r)
		goto ERROR;

	return jail;

ERROR:
	if (jail)
		pakfire_jail_unref(jail);

	return NULL;
}

static int pakfire_build_run_script(struct pakfire* pakfire, const char* filename,
		const char* args[], char*** output) {
	struct pakfire_jail* jail = NULL;
	char* script = NULL;
	size_t size = 0;
	char path[PATH_MAX];

	DEBUG(pakfire, "Running build script '%s'...\n", filename);

	// Make the source path
	pakfire_path_join(path, PAKFIRE_SCRIPTS_DIR, filename);

	// Open the source script
	FILE* f = fopen(path, "r");
	if (!f) {
		ERROR(pakfire, "Could not open %s: %m\n", path);
		return 1;
	}

	// Read the script into memory
	int r = pakfire_read_file_into_buffer(f, &script, &size);
	if (r) {
		ERROR(pakfire, "Could not read script into memory: %m\n");
		goto ERROR;
	}

	// Create a new jail
	jail = pakfire_build_make_jail(pakfire);
	if (!jail)
		goto ERROR;

	// Execute the script
	r = pakfire_jail_exec_script(jail, script, size, args, output);
	if (r) {
		ERROR(pakfire, "Script '%s' failed with status %d\n", filename, r);
	}

ERROR:
	if (jail)
		pakfire_jail_unref(jail);
	if (script)
		free(script);

	return r;
}

static int find_dependency(char** haystack, const char* needle) {
	if (!needle) {
		errno = EINVAL;
		return -1;
	}

	if (!haystack)
		return 0;

	for (char** element = haystack; *element; element++) {
		if (strcmp(needle, *element) == 0)
			return 1;
	}

	return 0;
}

static int pakfire_build_find_dependencies(struct pakfire* pakfire,
		struct pakfire_package* pkg, struct pakfire_filelist* filelist, const char* buildroot) {
	char** provides = NULL;
	char** requires = NULL;
	char path[PATH_MAX];

	// Allocate path to write the filelist to
	int r = pakfire_make_path(pakfire, path, "tmp/.pakfire-filelist.XXXXXX");
	if (r < 0)
		return 1;

	// Create a temporary file
	FILE* f = pakfire_mktemp(path);
	if (!f)
		goto ERROR;

	// Write filelist to the temporary file
	r = pakfire_filelist_export(filelist, f);
	fclose(f);
	if (r) {
		ERROR(pakfire, "Could not export filelist: %m\n");
		goto ERROR;
	}

	const char* root = pakfire_get_path(pakfire);

	// Pass buildroot and the filelist as arguments
	const char* args[] = {
		buildroot, pakfire_path_relpath(root, path), NULL
	};

	// Find all provides
	r = pakfire_build_run_script(pakfire, "find-provides", args, &provides);
	if (r) {
		ERROR(pakfire, "find-provides returned with error %d\n", r);
		goto ERROR;
	}

	// Find all requires
	r = pakfire_build_run_script(pakfire, "find-requires", args, &requires);
	if (r) {
		ERROR(pakfire, "find-requires returned with error %d\n", r);
		goto ERROR;
	}

	// Add all provides to the package
	if (provides) {
		for (char** element = provides; *element; element++) {
			DEBUG(pakfire, "Adding provides: %s\n", *element);
			pakfire_package_add_provides(pkg, *element);
		}
	}

	// Add all requires to the package
	if (requires) {
		for (char** element = requires; *element; element++) {
			// Skip adding this requirement if also provided by this package
			if (find_dependency(provides, *element))
				continue;

			DEBUG(pakfire, "Adding requires: %s\n", *element);
			pakfire_package_add_requires(pkg, *element);
		}
	}

	// Success
	r = 0;

ERROR:
	if (provides) {
		for (char** element = provides; *element; element++)
			free(*element);
		free(provides);
	}
	if (requires) {
		for (char** element = requires; *element; element++)
			free(*element);
		free(requires);
	}
	unlink(path);

	return r;
}

static int append_to_array(const char*** array, const char* s) {
	unsigned int length = 0;

	// Determine the length of the existing array
	if (*array) {
		for (const char** element = *array; *element; element++)
			length++;
	}

	// Allocate space
	*array = reallocarray(*array, length + 2, sizeof(**array));
	if (!*array)
		return 1;

	// Append s and terminate the array
	(*array)[length] = s;
	(*array)[length + 1] = NULL;

	return 0;
}

static int pakfire_build_package_add_files(struct pakfire* pakfire, struct pakfire_parser* makefile,
		const char* buildroot, const char* namespace, struct pakfire_package* pkg,
		struct pakfire_packager* packager) {
	struct pakfire_filelist* filelist = NULL;
	char path[PATH_MAX];
	int r = 1;

	char** files = NULL;
	const char** includes = NULL;
	const char** excludes = NULL;

	// Fetch filelist from makefile
	files = pakfire_parser_get_split(makefile, namespace, "files", '\n');

	// No files to package?
	if (!files)
		return 0;

	// Convert to absolute path
	pakfire_make_path(pakfire, path, buildroot);

	// Split into includes and excludes
	for (char** file = files; *file; file++) {
		if (**file == '!')
			r = append_to_array(&excludes, *file);
		else
			r = append_to_array(&includes, *file);
		if (r)
			goto ERROR;
	}

	// Allocate a new filelist
	r = pakfire_filelist_create(&filelist, pakfire);
	if (r)
		goto ERROR;

	// Scan for files
	r = pakfire_filelist_scan(filelist, path, includes, excludes);
	if (r)
		goto ERROR;

	const size_t length = pakfire_filelist_size(filelist);
	DEBUG(pakfire, "%zu file(s) found\n", length);

	// Nothing to do if the filelist is empty
	if (!length)
		goto ERROR;

	// Find dependencies
	r = pakfire_build_find_dependencies(pakfire, pkg, filelist, buildroot);
	if (r) {
		ERROR(pakfire, "Finding dependencies failed: %m\n");
		goto ERROR;
	}

	// Add all files to the package
	for (unsigned int i = 0; i < length; i++) {
		struct pakfire_file* file = pakfire_filelist_get(filelist, i);

		// Add the file to the package
		r = pakfire_packager_add(
			packager,
			pakfire_file_get_abspath(file),
			pakfire_file_get_path(file)
		);
		if (r) {
			pakfire_file_unref(file);
			goto ERROR;
		}

		// Remove the file after it was packaged
		r = pakfire_file_cleanup(file);
		if (r) {
			pakfire_file_unref(file);
			goto ERROR;
		}

		pakfire_file_unref(file);
	}

ERROR:
	if (filelist)
		pakfire_filelist_unref(filelist);
	if (files) {
		for (char** file = files; *file; file++)
			free(*file);
		free(files);
	}
	if (includes)
		free(includes);
	if (excludes)
		free(excludes);

	return r;
}

static int pakfire_build_add_scriptlet_requires(struct pakfire* pakfire, struct pakfire_package* pkg,
		struct pakfire_scriptlet* scriptlet) {
	char** prerequires = NULL;
	char path[PATH_MAX];
	size_t size;
	int r;

	const char* root = pakfire_get_path(pakfire);

	// Make filename
	r = pakfire_make_path(pakfire, path, "tmp/.pakfire-scriptlet.XXXXXX");
	if (r < 0)
		return r;

	// Fetch scriptlet payload
	const char* data = pakfire_scriptlet_get_data(scriptlet, &size);

	// Create a temporary file
	FILE* f = pakfire_mktemp(path);
	if (!f)
		return 1;

	// Write scriptlet
	ssize_t bytes_written = fwrite(data, 1, size, f);
	if (bytes_written < 0) {
		ERROR(pakfire, "Could not write to %s: %m\n", path);
		fclose(f);
		goto ERROR;
	}

	// Close file
	fclose(f);

	// Build commandline
	const char* args[] = {
		pakfire_path_relpath(root, path),
		NULL,
	};

	// Find all pre-requires
	r = pakfire_build_run_script(pakfire, "find-prerequires", args, &prerequires);
	if (r)
		goto ERROR;

	// Add all pre-requires to the package
	if (prerequires) {
		for (char** element = prerequires; *element; element++) {
			DEBUG(pakfire, "Adding pre-requires: %s\n", *element);
			pakfire_package_add_prerequires(pkg, *element);
		}
	}

ERROR:
	if (r && *path)
		unlink(path);
	if (prerequires) {
		for (char** element = prerequires; *element; element++)
			free(*element);
		free(prerequires);
	}
	return r;
}

static int pakfire_build_package_add_scriptlet(struct pakfire* pakfire, struct pakfire_package* pkg,
		struct pakfire_packager* packager, const char* type, const char* data) {
	struct pakfire_scriptlet* scriptlet = NULL;
	char* shell = NULL;
	int r;

	// Wrap scriptlet into a shell script
	r = asprintf(&shell, "#!/bin/sh\n\nset -e\n\n%s\n\nexit 0\n", data);
	if (r < 0)
		goto ERROR;

	// Create a scriptlet
	r = pakfire_scriptlet_create(&scriptlet, pakfire, type, shell, 0);
	if (r)
		goto ERROR;

	// Add it to the package
	r = pakfire_packager_add_scriptlet(packager, scriptlet);
	if (r) {
		ERROR(pakfire, "Could not add scriptlet %s\n", type);
		goto ERROR;
	}

	// Add scriptlet requirements
	r = pakfire_build_add_scriptlet_requires(pakfire, pkg, scriptlet);
	if (r) {
		ERROR(pakfire, "Could not add scriptlet requirements: %m\n");
		goto ERROR;
	}

	// Success
	r = 0;

ERROR:
	if (scriptlet)
		pakfire_scriptlet_unref(scriptlet);
	if (shell)
		free(shell);

	return r;
}

static int pakfire_build_package_add_scriptlets(struct pakfire* pakfire, struct pakfire_parser* makefile,
		const char* namespace, struct pakfire_package* pkg, struct pakfire_packager* packager) {
	char name[NAME_MAX];
	int r;

	for (const char** type = pakfire_scriptlet_types; *type; type++) {
		r = pakfire_string_format(name, "scriptlet:%s", *type);
		if (r < 0)
			return r;

		// Fetch the scriptlet
		char* data = pakfire_parser_get(makefile, namespace, name);
		if (!data)
			continue;

		// Add it to the package
		r = pakfire_build_package_add_scriptlet(pakfire, pkg, packager, *type, data);
		if (r) {
			free(data);
			return r;
		}

		free(data);
	}

	return 0;
}

static int pakfire_build_package(struct pakfire_build* build, struct pakfire_parser* makefile,
		const char* buildroot, const char* namespace) {
	struct pakfire_package* pkg = NULL;
	struct pakfire_packager* packager = NULL;

	int r = 1;

	// Expand the handle into the package name
	char* name = pakfire_parser_expand(makefile, "packages", namespace);
	if (!name) {
		ERROR(build->pakfire, "Could not get package name: %m\n");
		goto ERROR;
	}

	INFO(build->pakfire, "Building package '%s'...\n", name);

	// Fetch build architecture
	const char* arch = pakfire_get_arch(build->pakfire);
	if (!arch)
		goto ERROR;

	// Fetch package from makefile
	r = pakfire_parser_create_package(makefile, &pkg, NULL, namespace, arch);
	if (r) {
		ERROR(build->pakfire, "Could not create package from makefile: %m\n");
		goto ERROR;
	}

	// Set distribution
	const char* distribution = pakfire_parser_get(makefile, NULL, "DISTRO_NAME");
	if (distribution)
		pakfire_package_set_distribution(pkg, distribution);

	// Set build ID
	pakfire_package_set_build_id_from_uuid(pkg, &build->id);

	// Set source package
	const char* source_name = pakfire_parser_get(makefile, NULL, "name");
	if (source_name)
		pakfire_package_set_source_name(pkg, source_name);

	// Set source EVR
	const char* source_evr = pakfire_parser_get(makefile, NULL, "evr");
	if (source_evr)
		pakfire_package_set_source_evr(pkg, source_evr);

	// Set source arch
	pakfire_package_set_source_arch(pkg, "src");

	// Create a packager
	r = pakfire_packager_create(&packager, build->pakfire, pkg);
	if (r)
		goto ERROR;

	// Add files
	r = pakfire_build_package_add_files(build->pakfire, makefile, buildroot, namespace,
		pkg, packager);
	if (r)
		goto ERROR;

	// Add scriptlets
	r = pakfire_build_package_add_scriptlets(build->pakfire, makefile, namespace,
		pkg, packager);
	if (r)
		goto ERROR;

	// Write the finished package
	r = pakfire_packager_finish_to_directory(packager, build->target, NULL);
	if (r) {
		ERROR(build->pakfire, "pakfire_packager_finish() failed: %m\n");
		goto ERROR;
	}

#ifdef ENABLE_DEBUG
	char* dump = pakfire_package_dump(pkg, PAKFIRE_PKG_DUMP_LONG);
	if (dump) {
		DEBUG(build->pakfire, "%s\n", dump);
		free(dump);
	}
#endif

	// Success
	r = 0;

ERROR:
	if (packager)
		pakfire_packager_unref(packager);
	if (pkg)
		pakfire_package_unref(pkg);
	if (name)
		free(name);

	return r;
}

static int pakfire_build_packages(struct pakfire_build* build,
		struct pakfire_parser* makefile, const char* buildroot) {
	DEBUG(build->pakfire, "Creating packages...");
	int r = 1;

	// Fetch a list all all packages
	char** packages = pakfire_parser_list_namespaces(makefile, "packages.package:*");
	if (!packages) {
		ERROR(build->pakfire, "Could not find any packages: %m\n");
		goto ERROR;
	}

	unsigned int num_packages = 0;

	// Count how many packages we have
	for (char** package = packages; *package; package++)
		num_packages++;

	DEBUG(build->pakfire, "Found %d package(s)\n", num_packages);

	// Build packages in reverse order
	for (int i = num_packages - 1; i >= 0; i--) {
		r = pakfire_build_package(build, makefile, buildroot, packages[i]);
		if (r)
			goto ERROR;
	}

	// Success
	r = 0;

ERROR:
	if (packages)
		free(packages);

	return r;
}

static int pakfire_build_stage(struct pakfire_build* build,
		struct pakfire_parser* makefile, const char* stage) {
	struct pakfire_jail* jail = NULL;
	char template[1024];

	// Prepare template for this stage
	int r = pakfire_string_format(template, TEMPLATE, stage);
	if (r < 0)
		return r;

	// Fetch the environment
	char** envp = pakfire_parser_make_environ(makefile);

	// Create the build script
	char* script = pakfire_parser_expand(makefile, "build", template);
	if (!script) {
		ERROR(build->pakfire, "Could not generate the build script for stage '%s': %m\n",
			stage);
		goto ERROR;
	}

	INFO(build->pakfire, "Running build stage '%s'\n", stage);

	// Create a new jail
	jail = pakfire_build_make_jail(build->pakfire);
	if (!jail)
		goto ERROR;

	// Import environment
	r = pakfire_jail_import_env(jail, (const char**)envp);
	if (r) {
		ERROR(build->pakfire, "Could not import environment: %m\n");
		goto ERROR;
	}

	// Run the script
	r = pakfire_jail_exec_script(jail, script, strlen(script), NULL, NULL);
	if (r) {
		ERROR(build->pakfire, "Build stage '%s' failed with status %d\n", stage, r);
	}

ERROR:
	if (jail)
		pakfire_jail_unref(jail);
	if (envp) {
		for (char** e = envp; *e; e++)
			free(*e);
		free(envp);
	}
	if (script)
		free(script);

	return r;
}

static const char* post_build_scripts[] = {
	"remove-static-libs",
	"check-symlinks",
	"check-unsafe-files",
	"check-libraries",
	"check-rpaths",
	"check-buildroot",
	"check-include",
	"check-hardening",
	"check-interpreters",
	"check-fhs",
	"compress-man-pages",
	"strip",
	NULL,
};

static int pakfire_build_run_post_build_scripts(
		struct pakfire_build* build, const char* buildroot) {
	// Set default arguments for build scripts
	const char* args[] = {
		buildroot, NULL
	};

	// Run them one by one
	for (const char** script = post_build_scripts; *script; script++) {
		int r = pakfire_build_run_script(build->pakfire, *script, args, NULL);
		if (r)
			return r;
	}

	return 0;
}

static int pakfire_build_makefile(struct pakfire_build* build, const char* path) {
	struct pakfire_parser* makefile = NULL;
	char buildroot[PATH_MAX];
	struct pakfire_parser_error* error = NULL;
	int r = 1;

	const char* root = pakfire_get_path(build->pakfire);

	// Create BUILDROOT
	pakfire_make_path(build->pakfire, buildroot, "/tmp/.buildroot.XXXXXX");
	if (!pakfire_mkdtemp(buildroot))
		goto ERROR;

	// Make relative BUILDROOT path
	const char* buildroot_rel = pakfire_path_relpath(root, buildroot);
	if (!buildroot_rel)
		goto ERROR;

	// Read makefile
	r = pakfire_read_makefile(&makefile, build->pakfire, path, &error);
	if (r) {
		if (error) {
			ERROR(build->pakfire, "Could not parse makefile %s: %s\n", path,
				pakfire_parser_error_get_message(error));
			pakfire_parser_error_unref(error);
		} else {
			ERROR(build->pakfire, "Could not parse makefile %s: %m\n", path);
		}

		goto ERROR;
	}

	// Set BUILDROOT
	pakfire_parser_set(makefile, NULL, "BUILDROOT", buildroot_rel, 0);

	// Run through all build stages
	for (const char** stage = stages; *stage; stage++) {
		r = pakfire_build_stage(build, makefile, *stage);
		if (r) {
			// Drop to a shell for debugging
			if (pakfire_has_flag(build->pakfire, PAKFIRE_FLAGS_INTERACTIVE))
				pakfire_jail_shell(build->pakfire);

			goto ERROR;
		}
	}

	// Run post build scripts
	r = pakfire_build_run_post_build_scripts(build, buildroot_rel);
	if (r)
		goto ERROR;

	// Create the packages
	r = pakfire_build_packages(build, makefile, buildroot_rel);
	if (r) {
		ERROR(build->pakfire, "Could not create packages: %m\n");
		goto ERROR;
	}

ERROR:
	if (makefile)
		pakfire_parser_unref(makefile);

	// Remove buildroot
	pakfire_rmtree(buildroot, 0);

	return r;
}

static void pakfire_build_free(struct pakfire_build* build) {
	if (build->jail)
		pakfire_jail_unref(build->jail);

	if (build->cgroup) {
		// Destroy the cgroup
		pakfire_cgroup_destroy(build->cgroup);

		// Free it
		pakfire_cgroup_unref(build->cgroup);
	}

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

static int pakfire_build_parse_id(struct pakfire_build* build, const char* id) {
	int r;

	// Try parsing the Build ID
	if (id) {
		r = uuid_parse(id, build->id);
		if (r) {
			ERROR(build->pakfire, "Could not parse build ID '%s'\n", id);
			errno = EINVAL;
			return r;
		}

	// Otherwise initialize the Build ID with something random
	} else {
		uuid_generate_random(build->id);
	}

	// Store the ID as string, too
	uuid_unparse_lower(build->id, build->_id);

	return 0;
}

/*
	Sets the default target
*/
static int pakfire_build_set_default_target(struct pakfire_build* build) {
	// Look for the "local" repository
	struct pakfire_repo* repo = pakfire_get_repo(build->pakfire, PAKFIRE_REPO_LOCAL);
	if (repo) {
		const char* target = pakfire_repo_get_path(repo);
		if (target)
			pakfire_string_set(build->target, target);

		pakfire_repo_unref(repo);
	}

	// Default to the current working directory
	if (!*build->target) {
		char* cwd = getcwd(build->target, sizeof(build->target));
		if (!cwd) {
			ERROR(build->pakfire, "getcwd() failed: %m\n");
			return 1;
		}
	}

	return 0;
}

/*
	Sets up a new cgroup for this build
*/
static int pakfire_build_setup_cgroup(struct pakfire_build* build) {
	char path[PATH_MAX];
	int r;

	// Compose path
	r = pakfire_string_format(path, "pakfire/build-%s", build->_id);
	if (r < 0) {
		ERROR(build->pakfire, "Could not compose path for cgroup: %m\n");
		return 1;
	}

	// Create a new cgroup
	r = pakfire_cgroup_open(&build->cgroup, build->pakfire, path,
		PAKFIRE_CGROUP_ENABLE_ACCOUNTING);
	if (r) {
		ERROR(build->pakfire, "Could not create cgroup for build %s: %m\n", build->_id);
		return r;
	}

	// Done
	return 0;
}

/*
	Sets up a new jail for this build
*/
static int pakfire_build_setup_jail(struct pakfire_build* build) {
	int r;

	// Create a new jail
	r = pakfire_jail_create(&build->jail, build->pakfire, 0);
	if (r) {
		ERROR(build->pakfire, "Could not create jail for build %s: %m\n", build->_id);
		return r;
	}

	// Connect the jail to our cgroup
	r = pakfire_jail_set_cgroup(build->jail, build->cgroup);
	if (r) {
		ERROR(build->pakfire, "Could not set cgroup for jail: %m\n");
		return r;
	}

	// Done
	return 0;
}

PAKFIRE_EXPORT int pakfire_build_create(struct pakfire_build** build,
		struct pakfire* pakfire, const char* id, int flags) {
	int r;

	// Allocate build object
	struct pakfire_build* b = calloc(1, sizeof(*b));
	if (!b)
		return 1;

	// Reference pakfire
	b->pakfire = pakfire_ref(pakfire);

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

	// Copy flags
	b->flags = flags;

	// Parse ID
	r = pakfire_build_parse_id(b, id);
	if (r)
		goto ERROR;

	// Set target
	r = pakfire_build_set_default_target(b);
	if (r)
		goto ERROR;

	// Create cgroup
	r = pakfire_build_setup_cgroup(b);
	if (r)
		goto ERROR;

	// Create jail
	r = pakfire_build_setup_jail(b);
	if (r)
		goto ERROR;

	*build = b;
	return 0;

ERROR:
	pakfire_build_free(b);
	return r;
}

PAKFIRE_EXPORT struct pakfire_build* pakfire_build_ref(struct pakfire_build* build) {
	++build->nrefs;

	return build;
}

PAKFIRE_EXPORT struct pakfire_build* pakfire_build_unref(struct pakfire_build* build) {
	if (--build->nrefs > 0)
		return build;

	pakfire_build_free(build);
	return NULL;
}

PAKFIRE_EXPORT int pakfire_build_set_target(
		struct pakfire_build* build, const char* target) {
	pakfire_string_set(build->target, target);

	return 0;
}

PAKFIRE_EXPORT int pakfire_build_exec(struct pakfire_build* build, const char* path) {
	struct pakfire_archive* archive = NULL;
	struct pakfire_package* package = NULL;

	char makefile[PATH_MAX];
	int r;

	// Open source archive
	r = pakfire_archive_open(&archive, build->pakfire, path);
	if (r) {
		ERROR(build->pakfire, "Could not open source archive %s: %m\n", path);
		goto ERROR;
	}

	// Fetch package metadata
	r = pakfire_archive_make_package(archive, NULL, &package);
	if (r) {
		ERROR(build->pakfire, "Could not read package metadata: %m\n");
		goto ERROR;
	}

	// Fetch some information
	const char* nevra = pakfire_package_get_nevra(package);
	const char* name  = pakfire_package_get_name(package);

	INFO(build->pakfire, "Building %s...\n", nevra);

	// Setup build environment
	r = pakfire_build_setup(build->pakfire);
	if (r)
		goto ERROR;

	const char* packages[] = {
		path, NULL
	};

	// Install the package into the build environment
	r = pakfire_install(build->pakfire, 0, 0, packages, NULL, PAKFIRE_REQUEST_ESSENTIAL,
		NULL, NULL, NULL);
	if (r) {
		ERROR(build->pakfire, "Could not install %s\n", path);
		goto ERROR;
	}

	// Compose path to makefile
	r = pakfire_string_format(makefile, "/usr/src/packages/%s/%s.nm", name, name);
	if (r < 0) {
		ERROR(build->pakfire, "Could not compose makefile path: %m\n");
		goto ERROR;
	}

	// Run build
	r = pakfire_build_makefile(build, makefile);
	if (r) {
		ERROR(build->pakfire, "Could not build %s: %m\n", nevra);
		goto ERROR;
	}

ERROR:
	if (archive)
		pakfire_archive_unref(archive);
	if (package)
		pakfire_package_unref(package);

	return r;
}

/*
	Compatibility function to keep the legacy API.
*/
PAKFIRE_EXPORT int pakfire_build(struct pakfire* pakfire, const char* path,
		const char* target, const char* id, int flags) {
	struct pakfire_build* build = NULL;
	int r;

	// Check if path is set
	if (!path) {
		errno = EINVAL;
		return 1;
	}

	// Create a new build environment
	r = pakfire_build_create(&build, pakfire, id, flags);
	if (r)
		goto ERROR;

	// Set target
	if (target) {
		r = pakfire_build_set_target(build, target);
		if (r)
			goto ERROR;
	}

	// Run build
	r = pakfire_build_exec(build, path);

ERROR:
	if (build)
		pakfire_build_unref(build);

	return r;
}

/*
	This function enables the local repository so that it can be access by
	a pakfire instance running inside the chroot.
*/
static int pakfire_build_enable_repos(struct pakfire* pakfire) {
	char repopath[PATH_MAX];
	FILE* f = NULL;
	int r = 1;

	// Fetch the local repository
	struct pakfire_repo* repo = pakfire_get_repo(pakfire, PAKFIRE_REPO_LOCAL);
	if (!repo) {
		DEBUG(pakfire, "Could not find repository '%s': %m\n", PAKFIRE_REPO_LOCAL);
		return 0;
	}

	// Fetch repository configuration
	char* config = pakfire_repo_get_config(repo);
	if (!config) {
		ERROR(pakfire, "Could not generate repository configuration: %m\n");
		goto ERROR;
	}

	const char* path = pakfire_repo_get_path(repo);

	// Make sure the source directory exists
	r = pakfire_mkdir(path, 0700);
	if (r && errno != EEXIST) {
		ERROR(pakfire, "Could not create repository directory %s: %m\n", path);
		goto ERROR;
	}

	// Bind-mount the repository data read-only
	r = pakfire_bind(pakfire, path, NULL, MS_RDONLY);
	if (r) {
		ERROR(pakfire, "Could not bind-mount the repository at %s: %m\n", path);
		goto ERROR;
	}

	// Make path for configuration file
	r = pakfire_make_path(pakfire, repopath,
		PAKFIRE_CONFIG_DIR "/repos/" PAKFIRE_REPO_LOCAL ".repo");
	if (r < 0)
		goto ERROR;

	// Open configuration file for writing
	f = fopen(repopath, "w");
	if (!f) {
		ERROR(pakfire, "Could not open %s for writing: %m\n", path);
		r = 1;
		goto ERROR;
	}

	// Write configuration
	size_t bytes_written = fprintf(f, "%s", config);
	if (bytes_written < strlen(config)) {
		ERROR(pakfire, "Could not write repository configuration: %m\n");
		r = 1;
		goto ERROR;
	}

	// Success
	r = 0;

ERROR:
	if (f)
		fclose(f);
	if (config)
		free(config);
	pakfire_repo_unref(repo);

	return r;
}

PAKFIRE_EXPORT int pakfire_shell(struct pakfire* pakfire, const char** packages) {
	int r;

	// Setup build environment
	r = pakfire_build_setup(pakfire);
	if (r)
		return r;

	// Install any additional packages
	if (packages) {
		r = pakfire_install(pakfire, 0, 0, packages, NULL, 0, NULL, NULL, NULL);
		if (r)
			return r;
	}

	// Export local repository if running in interactive mode
	r = pakfire_build_enable_repos(pakfire);
	if (r)
		return r;

	return pakfire_jail_shell(pakfire);
}

int pakfire_build_clean(struct pakfire* pakfire, int flags) {
	struct pakfire_repo* local = NULL;
	int r = 0;

	// Fetch local repository
	local = pakfire_get_repo(pakfire, PAKFIRE_REPO_LOCAL);
	if (!local) {
		ERROR(pakfire, "Could not find repository %s: %m\n", PAKFIRE_REPO_LOCAL);
		goto ERROR;
	}

	// Destroy everything in it
	r = pakfire_repo_clean(local, PAKFIRE_REPO_CLEAN_FLAGS_DESTROY);
	if (r)
		goto ERROR;

ERROR:
	if (local)
		pakfire_repo_unref(local);

	return r;
}