cli: Check for root privileges when needed

[pakfire.git] / src / cli / lib / build.c

/*#############################################################################
#                                                                             #
# Pakfire - The IPFire package management system                              #
# Copyright (C) 2023 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 <argp.h>
#include <errno.h>
#include <stdlib.h>
#include <sys/syslog.h>

#include "build.h"
#include "color.h"
#include "command.h"
#include "pakfire.h"

#include <pakfire/build.h>
#include <pakfire/pakfire.h>

#define MAX_MAKEFILES 32

struct config {
	const char* id;
	const char* target;
	int flags;

	// Makefiles
	char* makefiles[MAX_MAKEFILES];
	unsigned int num_makefiles;
};

enum {
	OPT_DISABLE_CCACHE   = 1,
	OPT_DISABLE_SNAPSHOT = 2,
	OPT_DISABLE_TESTS    = 3,
	OPT_ID               = 4,
	OPT_NON_INTERACTIVE  = 5,
	OPT_TARGET           = 6,
};

static struct argp_option options[] = {
	{ "disable-ccache",   OPT_DISABLE_CCACHE,      NULL, 0, "Disable the ccache",     0 },
	{ "disable-snapshot", OPT_DISABLE_SNAPSHOT,    NULL, 0, "Do not use the snapshot", 0 },
	{ "disable-tests",    OPT_DISABLE_TESTS,       NULL, 0, "Do not run tests",        0 },
	{ "id",               OPT_ID,                  "ID", 0, "Use this build ID",       1 },
	{ "non-interactive",  OPT_NON_INTERACTIVE,     NULL, 0, "Run the build non-interactively", 0 },
	{ "target",           OPT_TARGET,          "TARGET", 0, "Output all packages into this directory", 0 },
	{ NULL },
};

static error_t parse(int key, char* arg, struct argp_state* state, void* data) {
	struct config* config = data;

	switch (key) {
		case OPT_DISABLE_CCACHE:
			config->flags |= PAKFIRE_BUILD_DISABLE_CCACHE;
			break;

		case OPT_DISABLE_SNAPSHOT:
			config->flags |= PAKFIRE_BUILD_DISABLE_SNAPSHOT;
			break;

		case OPT_DISABLE_TESTS:
			config->flags |= PAKFIRE_BUILD_DISABLE_TESTS;
			break;

		case OPT_ID:
			config->id = arg;
			break;

		case OPT_NON_INTERACTIVE:
			config->flags &= ~PAKFIRE_BUILD_INTERACTIVE;
			break;

		case OPT_TARGET:
			config->target = arg;
			break;

		case ARGP_KEY_ARG:
			if (config->num_makefiles >= MAX_MAKEFILES)
				return -ENOBUFS;

			config->makefiles[config->num_makefiles++] = arg;
			break;

		default:
			return ARGP_ERR_UNKNOWN;
	}

	return 0;
}

static int log_callback(struct pakfire_build* build, void* data,
		int priority, int error, const char* file, int line, const char* function, const char* format, ...) {
	char* buffer = NULL;
	va_list args;
	int r;

	// Format the message
	va_start(args, format);
	r = vasprintf(&buffer, format, args);
	va_end(args);

	// Fail if we could not format the message
	if (r < 0)
		return r;

	switch (priority) {
		// Highlight any error messages
		case LOG_ERR:
			fprintf(stderr, "%s%s%s", color_highlight(), buffer, color_reset());
			break;

		// Print the rest to stdout
		default:
			fprintf(stdout, "%s", buffer);
			break;
	}

	// Cleanup
	if (buffer)
		free(buffer);

	return 0;
}

int cli_build(void* data, int argc, char* argv[]) {
	struct pakfire* pakfire = NULL;
	struct pakfire_build* build = NULL;
	struct config config = {
		.id     = NULL,
		.target = NULL,
		.flags  = PAKFIRE_BUILD_INTERACTIVE,
	};
	int r;

	struct cli_config* cli_config = data;

	// Parse the command line
	r = cli_parse(options, NULL, NULL, NULL, parse, 0, argc, argv, &config);
	if (r)
		goto ERROR;

	// Setup pakfire
	r = cli_setup_pakfire(&pakfire, cli_config);
	if (r)
		goto ERROR;

	// Setup the build environment
	r = pakfire_build_create(&build, pakfire, config.id, config.flags);
	if (r) {
		fprintf(stderr, "Could not setup the build environment: %m\n");
		goto ERROR;
	}

	// Set log callback
	pakfire_build_set_log_callback(build, log_callback, NULL);

	// Set target
	if (config.target) {
		r = pakfire_build_set_target(build, config.target);
		if (r) {
			fprintf(stderr, "Could not set target directory %s: %m\n", config.target);
			goto ERROR;
		}
	}

	// Process all packages
	for (unsigned int i = 0; i < config.num_makefiles; i++) {
		// Run the build
		r = pakfire_build_exec(build, config.makefiles[i]);
		if (r) {
			fprintf(stderr, "Could not build %s\n", config.makefiles[i]);
			goto ERROR;
		}
	}

ERROR:
	if (build)
		pakfire_build_unref(build);
	if (pakfire)
		pakfire_unref(pakfire);

	return r;
}