cli: Check for root privileges when needed

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

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

#define MAX_PACKAGES 128

struct config {
	int flags;

	// Packages
	const char* packages[MAX_PACKAGES];
	unsigned int num_packages;
};

static const char* doc =
	"Creates a build environment and launches an interactive shell in it";

enum {
	OPT_DISABLE_SNAPSHOT = 1,
	OPT_INSTALL          = 2,
};

static struct argp_option options[] = {
	{ "disable-snapshot", OPT_DISABLE_SNAPSHOT,      NULL, 0, "Do not use the snapshot", 0 },
	{ "install",          OPT_INSTALL,          "PACKAGE", 0, "Install this package", 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_SNAPSHOT:
			config->flags |= PAKFIRE_BUILD_DISABLE_SNAPSHOT;
			break;

		case ARGP_KEY_ARG:
			if (config->num_packages >= MAX_PACKAGES)
				return -ENOBUFS;

			config->packages[config->num_packages++] = arg;
			break;

		default:
			return ARGP_ERR_UNKNOWN;
	}

	return 0;
}

int cli_shell(void* data, int argc, char* argv[]) {
	struct pakfire* pakfire = NULL;
	int r;

	struct cli_config* cli_config = data;

	struct config config = {
		.flags        = 0,
		.packages     = { NULL },
		.num_packages = 0,
	};

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

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

	// Run the shell
	r = pakfire_shell(pakfire, config.packages, config.flags);
	if (r)
		goto ERROR;

ERROR:
	if (pakfire)
		pakfire_unref(pakfire);

	return r;
}