cli: Check for root privileges when needed

[pakfire.git] / src / cli / lib / dist.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 "command.h"
#include "dist.h"
#include "pakfire.h"

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

static const char* args_doc = "dist MAKEFILES...";

static const char* doc = "Creates source packages from makefiles";

#define MAX_MAKEFILES 32

struct config {
	const char* resultdir;

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

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

	switch (key) {
		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;
}

int cli_dist(void* data, int argc, char* argv[]) {
	struct pakfire* pakfire = NULL;
	struct config config = {
		.resultdir = NULL,
	};
	char cwd[PATH_MAX];
	int r;

	struct cli_config* cli_config = data;

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

	// Set result directory to PWD
	if (!config.resultdir)
		config.resultdir = getcwd(cwd, sizeof(cwd));

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

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

ERROR:
	if (pakfire)
		pakfire_unref(pakfire);

	return r;
}