cli: Check for root privileges when needed

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

#include "command.h"
#include "image_create.h"
#include "pakfire.h"

struct config {
	const char* type;
	const char* path;
};

static const char* args_doc = "image create TYPE PATH";

static const char* doc = "Creates images";

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->type)
				config->type = arg;

			else if (!config->path)
				config->path = arg;

			break;

		default:
			return ARGP_ERR_UNKNOWN;
	}

	return 0;
}

int cli_image_create(void* data, int argc, char* argv[]) {
	struct pakfire* pakfire = NULL;
	struct pakfire_build* build = NULL;
	FILE* f = NULL;
	int r;

	struct cli_config* cli_config = data;

	struct config config = {
		.type = NULL,
		.path = NULL,
	};

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

	f = fopen(config.path, "w");
	if (!f) {
		fprintf(stderr, "Could not open %s: %m\n", config.path);
		r = -errno;
		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, NULL, 0);
	if (r)
		goto ERROR;

	// Create the image
	r = pakfire_build_mkimage(build, config.type, f);
	if (r)
		goto ERROR;

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

	return r;
}