cli: Check for root privileges when needed

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

#include "command.h"
#include "pakfire.h"
#include "transaction.h"
#include "update.h"

static const char* args_doc = "update [OPTIONS...] PACKAGES...";

static const char* doc = "Update packages";

#define MAX_EXCLUDES 128
#define MAX_PACKAGES 128

struct config {
	int transaction_flags;

	// Excludes
	char* excludes[MAX_EXCLUDES];
	unsigned int num_excludes;

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

enum {
	OPT_ALLOW_DOWNGRADE = 1,
	OPT_ALLOW_UNINSTALL = 2,
};

static struct argp_option options[] = {
	{ "allow-downgrade", OPT_ALLOW_DOWNGRADE, NULL, 0, "Allow downgrading packages", 0 },
	{ "allow-uninstall", OPT_ALLOW_UNINSTALL, NULL, 0, "Allow uninstalling packages", 0 },
	{ "exclude",         'x',            "PACKAGE", 0, "Exclude a package from being updated", 0 },
	{ NULL },
};

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

	switch (key) {
		case OPT_ALLOW_DOWNGRADE:
			config->transaction_flags |= PAKFIRE_TRANSACTION_ALLOW_DOWNGRADE;
			break;

		case OPT_ALLOW_UNINSTALL:
			config->transaction_flags |= PAKFIRE_TRANSACTION_ALLOW_UNINSTALL;
			break;

		case 'x':
			if (config->num_excludes >= MAX_EXCLUDES)
				return -ENOBUFS;

			config->excludes[config->num_excludes++] = arg;
			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;
}

static int __cli_update(struct pakfire_transaction* transaction, int argc, char* argv[], void* data) {
	struct config* config = (struct config*)data;
	int r;

	// Exclude any packages
	for (unsigned int i = 0; i < config->num_excludes; i++) {
		r = pakfire_transaction_request(transaction, PAKFIRE_JOB_LOCK, config->excludes[i], 0);
		if (r)
			return r;
	}

	// Did the user pass any packages?
	if (argc) {
		// Add the remaining command line options as packages
		for (unsigned int i = 0; i < config->num_packages; i++) {
			r = pakfire_transaction_request(transaction,
				PAKFIRE_JOB_UPDATE, config->packages[i], 0);
			if (r) {
				fprintf(stderr, "Could not find '%s': %m\n", argv[i]);
				return r;
			}
		}

	// Otherwise update everything
	} else {
		r = pakfire_transaction_request(transaction, PAKFIRE_JOB_UPDATE_ALL, NULL, 0);
		if (r)
			return r;
	}

	return 0;
}

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

	struct cli_config* cli_config = data;

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

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

	r = cli_transaction(pakfire, argc, argv, config.transaction_flags, __cli_update, &config);

ERROR:
	if (pakfire)
		pakfire_unref(pakfire);

	return r;
}