/*

 * Copyright (C) 2008  Kristian Høgsberg <krh@redhat.com>

 * Copyright (C) 2008  Red Hat, Inc

 *

 * 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 2 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, write to the Free Software Foundation, Inc.,

 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

 */

#include <stdlib.h>

#include <string.h>

#include <stdio.h>

#include <dirent.h>

#include "razor.h"

enum option_type {

	OPTION_LAST,

	OPTION_GROUP,

	OPTION_BOOL,

	OPTION_STRING

};

struct option {

	enum option_type type;

	const char *name;

	char short_name;

	const char *arg_name;

	const char *description;

	void *data;

};

/* A note about all these options: rpm allows options to mean

 * different things depending on what other options are present on the

 * command line.  For example, if -q or --query is present, -i no

 * longer means install, but info.  The way we handle this is by

 * setting all the options that may match (ie if -i is given we set

 * install and info), and then look at the relevent one depending on

 * what else in on the command line. */

static int option_all, option_whatrequires, option_whatprovides;

static int option_package;

static const struct option query_options[] = {

	{ OPTION_BOOL, "configfiles", 'c', NULL, "list all configuration files", NULL },

	{ OPTION_BOOL, "docfiles", 'd', NULL, "list all documentation files", NULL },

	{ OPTION_BOOL, "dump", 0, NULL, "dump basic file information", NULL },

	{ OPTION_BOOL, "list", 0, NULL, "list files in package", NULL },

	{ OPTION_STRING, "queryformat", 0, "QUERYFORMAT", "use the following query format", NULL },

	{ OPTION_BOOL, "state", 's', NULL, "display the states of the listed files", NULL },

	{ OPTION_BOOL, "all", 'a', NULL, "query/verify all packages", &option_all },

	{ OPTION_BOOL, "file", 'f', NULL, "query/verify package(s) owning file", NULL },

	{ OPTION_BOOL, "group", 'g', NULL, "query/verify package(s) in group", NULL },

	{ OPTION_BOOL, "package", 'p', NULL, "query/verify a package file", &option_package },

	{ OPTION_BOOL, "ftswalk", 'W', NULL, "query/verify package(s) from TOP file tree walk", NULL },

	{ OPTION_BOOL, "pkgid", 0, NULL, "query/verify package(s) with package identifier", NULL },

	{ OPTION_BOOL, "hdrid", 0, NULL, "query/verify package(s) with header identifier", NULL },

	{ OPTION_BOOL, "fileid", 0, NULL, "query/verify package(s) with file identifier", NULL },

	{ OPTION_BOOL, "specfile", 0, NULL, "query a spec file", NULL, },

	{ OPTION_BOOL, "triggeredby", 0, NULL, "query the package(s) triggered by the package", NULL },

	{ OPTION_BOOL, "whatrequires", 0, NULL, "query/verify the package(s) which require a dependency", &option_whatrequires },

	{ OPTION_BOOL, "whatprovides", 0, NULL, "query/verify the package(s) which provide a dependency", &option_whatprovides },

	{ OPTION_BOOL, "nomanifest", 0, NULL, "do not process non-package files as manifests", NULL },

	{ }

};

static int option_nodeps;

static const struct option verify_options[] = {

	{ OPTION_BOOL, "nomd5", 0, NULL, "don't verify MD5 digest of files", NULL },

	{ OPTION_BOOL, "nofiles", 0, NULL, "don't verify files in package", NULL },

	{ OPTION_BOOL, "nodeps", 0, NULL, "don't verify package dependencies", &option_nodeps },

	{ OPTION_BOOL, "noscript", 0, NULL, "don't execute verify script(s)", NULL, },

	{ OPTION_BOOL, "all", 'a', NULL, "query/verify all packages", &option_all },

	{ OPTION_BOOL, "file", 'f', NULL, "query/verify package(s) owning file", NULL },

	{ OPTION_BOOL, "group", 'g', NULL, "query/verify package(s) in group", NULL },

	{ OPTION_BOOL, "package", 'p', NULL, "query/verify a package file", &option_package },

	{ OPTION_BOOL, "ftswalk", 'W', NULL, "query/verify package(s) from TOP file tree walk", NULL },

	{ OPTION_BOOL, "pkgid", 0, NULL, "query/verify package(s) with package identifier", NULL },

	{ OPTION_BOOL, "hdrid", 0, NULL, "query/verify package(s) with header identifier", NULL },

	{ OPTION_BOOL, "fileid", 0, NULL, "query/verify package(s) with file identifier", NULL },

	{ OPTION_BOOL, "specfile", 0, NULL, "query a spec file", NULL },

	{ OPTION_BOOL, "triggeredby", 0, NULL, "query the package(s) triggered by the package", NULL },

	{ OPTION_BOOL, "whatrequires", 0, NULL, "query/verify the package(s) which require a dependency", &option_whatrequires },

	{ OPTION_BOOL, "whatprovides", 0, NULL, "query/verify the package(s) which provide a dependency", &option_whatprovides },

	{ OPTION_BOOL, "nomanifest", 0, NULL, "do not process non-package files as manifests", NULL },

	{ }

};

static const struct option ftw_options[] = {

	{ OPTION_BOOL, "comfollow", 0, NULL, "FTS_COMFOLLOW: follow command line symlinks", NULL },

	{ OPTION_BOOL, "logical", 0, NULL, "FTS_LOGICAL: logical walk", NULL },

	{ OPTION_BOOL, "nochdir", 0, NULL, "FTS_NOCHDIR: don't change directories", NULL },

	{ OPTION_BOOL, "nostat", 0, NULL, "FTS_NOSTAT: don't get stat info", NULL },

	{ OPTION_BOOL, "physical", 0, NULL, "FTS_PHYSICAL: physical walk", NULL },

	{ OPTION_BOOL, "seedot", 0, NULL, "FTS_SEEDOT: return dot and dot-dot", NULL },

	{ OPTION_BOOL, "xdev", 0, NULL, "FTS_XDEV: don't cross devices", NULL },

	{ OPTION_BOOL, "whiteout", 0, NULL, "FTS_WHITEOUT: return whiteout information", NULL },

	{ }

};

static const struct option signature_options[] = {

	{ OPTION_BOOL, "addsign", 0, NULL, "sign package(s) (identical to --resign)", NULL, },

	{ OPTION_BOOL, "checksig", 'K', NULL, "verify package signature(s)", NULL, },

	{ OPTION_BOOL, "delsign", 0, NULL, "delete package signatures", NULL, },

	{ OPTION_BOOL, "import", 0, NULL, "import an armored public key", NULL, },

	{ OPTION_BOOL, "resign", 0, NULL, "sign package(s) (identical to --addsign)", NULL, },

	{ OPTION_BOOL, "nodigest", 0, NULL, "don't verify package digest(s)", NULL, },

	{ OPTION_BOOL, "nosignature", 0, NULL, "don't verify package signature(s)", NULL },

	{ }

};

static int option_initdb;

static const struct option database_options[] = {

	{ OPTION_BOOL, "initdb", 0, NULL, "initialize database", &option_initdb },

	{ OPTION_BOOL, "rebuilddb", 0, NULL, "rebuild database inverted lists from installed package headers", NULL },

	{ }

};

static int option_erase, option_install, option_upgrade, option_justdb;

static int option_test;

static const struct option install_options[] = {

	{ OPTION_BOOL, "aid", 0, NULL, "add suggested packages to transaction", NULL, },

	{ OPTION_BOOL, "allfiles", 0, NULL, "install all files, even configurations which might otherwise be skipped", NULL, },

	{ OPTION_BOOL, "allmatches", 0, NULL, "remove all packages which match <package> (normally an error is generated if <package> specified multiple packages)", NULL, },

	{ OPTION_BOOL, "badreloc", 0, NULL, "relocate files in non-relocatable package", NULL },

	{ OPTION_BOOL, "erase", 'e', "<package>", "erase (uninstall) package", &option_erase },

	{ OPTION_BOOL, "excludedocs", 0, NULL, "do not install documentation", NULL, },

	{ OPTION_BOOL, "excludepath", 0, "<path>", "skip files with leading component <path> ", NULL, },

	{ OPTION_BOOL, "fileconflicts", 0, NULL, "detect file conflicts between packages", NULL, },

	{ OPTION_BOOL, "force", 0, NULL, "short hand for --replacepkgs --replacefiles", NULL },

	{ OPTION_BOOL, "freshen", 'F', "<packagefile>+", "upgrade package(s) if already installed", NULL },

	{ OPTION_BOOL, "hash", 'h', NULL, "print hash marks as package installs (good with -v)", NULL },

	{ OPTION_BOOL, "ignorearch", 0, NULL, "don't verify package architecture", NULL, },

	{ OPTION_BOOL, "ignoreos", 0, NULL, "don't verify package operating system", NULL, },

	{ OPTION_BOOL, "ignoresize", 0, NULL, "don't check disk space before installing", NULL },

	{ OPTION_BOOL, "install", 'i', NULL, "install package(s)", &option_install },

	{ OPTION_BOOL, "justdb", 0, NULL, "update the database, but do not modify the filesystem", &option_justdb, },

	{ OPTION_BOOL, "nodeps", 0, NULL, "do not verify package dependencies", &option_nodeps, },

	{ OPTION_BOOL, "nomd5", 0, NULL, "don't verify MD5 digest of files", NULL, },

	{ OPTION_BOOL, "nocontexts", 0, NULL, "don't install file security contexts", NULL, },

	{ OPTION_BOOL, "noorder", 0, NULL, "do not reorder package installation to satisfy dependencies", NULL, },

	{ OPTION_BOOL, "nosuggest", 0, NULL, "do not suggest missing dependency resolution(s)", NULL, },

	{ OPTION_BOOL, "noscripts", 0, NULL, "do not execute package scriptlet(s)", NULL, },

	{ OPTION_BOOL, "notriggers", 0, NULL, "do not execute any scriptlet(s) triggered by this package", NULL, },

	{ OPTION_BOOL, "oldpackage", 0, NULL, "upgrade to an old version of the package (--force on upgrades does this automatically)", NULL },

	{ OPTION_BOOL, "percent", 0, NULL, "print percentages as package installs", NULL, },

	{ OPTION_STRING, "prefix", 0, "<dir>", "relocate the package to <dir>, if relocatable", NULL, },

	{ OPTION_STRING, "relocate", 0, "<old>=<new>", "relocate files from path <old> to <new>", NULL, },

	{ OPTION_BOOL, "repackage", 0, NULL, "save erased package files by repackaging", NULL, },

	{ OPTION_BOOL, "replacefiles", 0, NULL, "ignore file conflicts between packages", NULL, },

	{ OPTION_BOOL, "replacepkgs", 0, NULL, "reinstall if the package is already present", NULL, },

	{ OPTION_BOOL, "test", 0, NULL, "don't install, but tell if it would work or not", &option_test },

	{ OPTION_BOOL, "upgrade", 'U', "<packagefile>+", "upgrade package(s)", &option_upgrade },

	{ }

};

static int option_version;

static const char *option_root = "install";

static const struct option common_options[] = {

	{ OPTION_STRING, "define", 'D', "MACRO EXPR", "define MACRO with value EXPR", NULL, },

	{ OPTION_STRING, "eval", 'E', "EXPR", "print macro expansion of EXPR", NULL },

	{ OPTION_STRING, "macros", 0, "<FILE:...>", "read <FILE:...> instead of default file(s)", NULL },

	{ OPTION_BOOL, "nodigest", 0, NULL, "don't verify package digest(s)", NULL, },

	{ OPTION_BOOL, "nosignature", 0, NULL, "don't verify package signature(s)", NULL, },

	{ OPTION_STRING, "rcfile", 0, "<FILE:...>", "read <FILE:...> instead of default file(s)", NULL },

	{ OPTION_STRING, "root", 'r', "ROOT", "use ROOT as top level directory (default: \"/\")", &option_root },

	{ OPTION_BOOL, "querytags", 0, NULL, "display known query tags", NULL, },

	{ OPTION_BOOL, "showrc", 0, NULL, "display final rpmrc and macro configuration", NULL, },

	{ OPTION_BOOL, "quiet", 0, NULL, "provide less detailed output", NULL },

	{ OPTION_BOOL, "verbose", 'v', NULL, "provide more detailed output", NULL },

	{ OPTION_BOOL, "version", 0, NULL, "print the version of rpm being used", &option_version },

	{ }

};

static int option_conflicts, option_obsoletes, option_requires;

static int option_provides, option_info, option_changelog;

static const struct option alias_options[] = {

	{ OPTION_BOOL, "scripts", 0, NULL, "list install/erase scriptlets from package(s)", NULL, },

	{ OPTION_BOOL, "setperms", 0, NULL, "set permissions of files in a package", NULL, },

	{ OPTION_BOOL, "setugids", 0, NULL, "set user/group ownership of files in a package", NULL, },

	{ OPTION_BOOL, "conflicts", 0, NULL, "list capabilities this package conflicts with", &option_conflicts, },

	{ OPTION_BOOL, "obsoletes", 0, NULL, "list other packages removed by installing this package", &option_obsoletes, },

	{ OPTION_BOOL, "provides", 0, NULL, "list capabilities that this package provides", &option_provides, },

	{ OPTION_BOOL, "requires", 0, NULL, "list capabilities required by package(s)", &option_requires, },

	{ OPTION_BOOL, "info", 'i', NULL, "list descriptive information from package(s)", &option_info, },

	{ OPTION_BOOL, "changelog", 0, NULL, "list change logs for this package", &option_changelog, },

	{ OPTION_BOOL, "xml", 0, NULL, "list metadata in xml", NULL, },

	{ OPTION_BOOL, "triggers", 0, NULL, "list trigger scriptlets from package(s)", NULL, },

	{ OPTION_BOOL, "last", 0, NULL, "list package(s) by install time, most recent first", NULL, },

	{ OPTION_BOOL, "dupes", 0, NULL, "list duplicated packages", NULL, },

	{ OPTION_BOOL, "filesbypkg", 0, NULL, "list all files from each package", NULL, },

	{ OPTION_BOOL, "fileclass", 0, NULL, "list file names with classes", NULL, },

	{ OPTION_BOOL, "filecolor", 0, NULL, "list file names with colors", NULL, },

	{ OPTION_BOOL, "filecontext", 0, NULL, "list file names with security context from header", NULL, },

	{ OPTION_BOOL, "fscontext", 0, NULL, "list file names with security context from file system", NULL, },

	{ OPTION_BOOL, "recontext", 0, NULL, "list file names with security context from policy RE", NULL, },

	{ OPTION_BOOL, "fileprovide", 0, NULL, "list file names with provides", NULL, },

	{ OPTION_BOOL, "filerequire", 0, NULL, "list file names with requires", NULL, },

	{ OPTION_BOOL, "redhatprovides", 0, NULL, "find package name that contains a provided capability (needs rpmdb-redhat package installed)", NULL, },

	{ OPTION_BOOL, "redhatrequires", 0, NULL, "find package name that contains a required capability (needs rpmdb-redhat package installed)", NULL, },

	{ OPTION_STRING, "buildpolicy", 0, "<policy>", "set buildroot <policy> (e.g. compress man pages)", NULL, },

	{ OPTION_BOOL, "with", 0, "<option>", "enable configure <option> for build", NULL, },

	{ OPTION_BOOL, "without", 0, "<option>", "disable configure <option> for build", NULL },

	{ }

};

static int option_help, option_usage;

static const struct option help_options[] = {

	{ OPTION_BOOL, "help", '?', NULL, "Show this help message", &option_help },

	{ OPTION_BOOL, "usage", 0, NULL, "Display brief usage message", &option_usage},

	{ }

};

static int option_query, option_verify;

static const struct option rpm_options[] = {

	{ OPTION_BOOL, "query", 'q', NULL, "Query rpm database", &option_query },

	{ OPTION_BOOL, "verify", 'V', NULL, "Verify rpm database", &option_verify },

	{ OPTION_GROUP, NULL, 0, NULL, "Query options (with -q or --query):", &query_options },

	{ OPTION_GROUP, NULL, 0, NULL, "Verify options (with -V or --verify):", &verify_options },

	{ OPTION_GROUP, NULL, 0, NULL, "File tree walk options (with --ftswalk):", &ftw_options },

	{ OPTION_GROUP, NULL, 0, NULL, "Signature options:", &signature_options },

	{ OPTION_GROUP, NULL, 0, NULL, "Database options:", &database_options },

	{ OPTION_GROUP, NULL, 0, NULL, "Install/Upgrade/Erase options:", &install_options },

	{ OPTION_GROUP, NULL, 0, NULL, "Common options for all rpm modes and executables:", &common_options },

	{ OPTION_GROUP, NULL, 0, NULL, "Options implemented via popt alias/exec:", &alias_options },

	{ OPTION_GROUP, NULL, 0, NULL, "Help options", &help_options },

	{ }

};

static const char system_repo_filename[] = "system.repo";

static const char *repo_filename = system_repo_filename;

static void

command_initdb(int argc, const char *argv[])

{

	razor_root_create(option_root);

}

static struct razor_property *

add_property_packages(struct razor_set *set, 

		      struct razor_package_query *query,

		      const char *ref_name,

		      const char *ref_version,

		      enum razor_property_type ref_type)

{

	struct razor_property *property;

	struct razor_property_iterator *pi;

	struct razor_package_iterator *pkgi;

	const char *name, *version;

	enum razor_property_type type;

	enum razor_version_relation relation;

	pi = razor_property_iterator_create(set, NULL);

	while (razor_property_iterator_next(pi, &property, &name,

					    &relation, &version, &type)) {

		if (strcmp(ref_name, name) != 0)

			continue;

		if (ref_version && relation == RAZOR_VERSION_EQUAL &&

		    strcmp(ref_version, version) != 0)

			continue;

		if (ref_type != type)

			continue;

		pkgi = razor_package_iterator_create_for_property(set,

								  property);

		razor_package_query_add_iterator(query, pkgi);

		razor_package_iterator_destroy(pkgi);

	}

	razor_property_iterator_destroy(pi);

	return property;

}

static int

strcmpp(const void *p1, const void *p2)

{

	return strcmp(*(char * const *) p1, *(char * const *) p2);

}

static void

add_command_line_packages(struct razor_set *set,

			  struct razor_package_query *query,

			  int argc, const char **argv)

{

	struct razor_package *package;

	struct razor_package_iterator *pi;

	const char *name, *version, *arch;

	int i, cmp, errors;

	qsort(argv, argc, sizeof(*argv), strcmpp);

	i = 0;

	errors = 0;

	pi = razor_package_iterator_create(set);

	while (razor_package_iterator_next(pi, &package,

					   &name, &version, &arch)) {

		while (cmp = strcmp(argv[i], name), cmp < 0 && i < argc) {

			fprintf(stderr, "error: package %s is not installed\n",

				argv[i]);

			errors++;

			i++;

		}

		if (cmp == 0) {

			razor_package_query_add_package(query, package);

			i++;

		}

	}

	razor_package_iterator_destroy(pi);

	if (errors)

		exit(1);

}

static struct razor_package_iterator *

get_query_packages(struct razor_set *set, int argc, const char *argv[])

{

	struct razor_package_query *query;

	struct razor_package_iterator *pi;

	int i;

	if (option_all + option_whatprovides + option_whatrequires > 1) {

		printf("only one type of query/verify "

		       "may be performed at a time\n");

		exit(1);

	}

	query = razor_package_query_create(set);

	if (option_all) {

		pi = razor_package_iterator_create(set);

		razor_package_query_add_iterator(query, pi);

		razor_package_iterator_destroy(pi);

	} else if (option_whatrequires) {

		for (i = 0; i < argc; i++)

			add_property_packages(set, query,

					      argv[i], NULL,

					      RAZOR_PROPERTY_REQUIRES);

	} else if (option_whatprovides) {

		for (i = 0; i < argc; i++)

			add_property_packages(set, query,

					      argv[i], NULL,

					      RAZOR_PROPERTY_PROVIDES);

	} else if (argc > 0) {

		add_command_line_packages(set, query, argc, argv);

	} else {

		printf("no arguments given for query/verify\n");

		exit(1);

	}

	return razor_package_query_finish(query);

}

static const char *relation_string[] = { "<", "<=", "=", ">=", ">" };

static void

print_package_properties(struct razor_set *set,

			 struct razor_package *package,

			 enum razor_property_type ref_type)

{

	struct razor_property *property;

	struct razor_property_iterator *pi;

	const char *name, *version;

	enum razor_property_type type;

	enum razor_version_relation relation;

	pi = razor_property_iterator_create(set, package);

	while (razor_property_iterator_next(pi, &property,

					    &name, &relation, &version,

					    &type)) {

		if (type != ref_type)

			continue;

		if (version[0] == '\0')

			printf("%s\n", name);

		else

			printf("%s %s %s\n", name,

			       relation_string[relation], version);

	}

	razor_property_iterator_destroy(pi);

}

static void

print_package_info(struct razor_set *set, struct razor_package *package)

{

	printf("FIXME: Package info not tracked.\n");

}

static void

print_package_changelog(struct razor_set *set, struct razor_package *package)

{

	printf("FIXME: Package changelog not tracked.\n");

}

static struct razor_set *

create_set_from_command_line(int argc, const char *argv[])

{

	struct razor_importer *importer;

	struct razor_rpm *rpm;

	int i;

	importer = razor_importer_new();

	for (i = 0; i < argc; i++) {

		rpm = razor_rpm_open(argv[i]);

		if (rpm == NULL)

			continue;

		if (razor_importer_add_rpm(importer, rpm))

			printf("couldn't import %s\n", argv[i]);

		razor_rpm_close(rpm);

	}

	return razor_importer_finish(importer);

}

static void

command_query(int argc, const char *argv[])

{

	struct razor_set *set;

	struct razor_package_iterator *pi;

	struct razor_package *package;

	const char *name, *version, *arch;

	if (option_package) {

		set = create_set_from_command_line(argc, argv);

		argc = 0;

		option_all = 1;

	} else {

		set = razor_root_open_read_only(option_root);

	}

	pi = get_query_packages(set, argc, argv);

	while (razor_package_iterator_next(pi, &package,

					   &name, &version, &arch)) {

		if (option_conflicts)

			print_package_properties(set, package,

						 RAZOR_PROPERTY_CONFLICTS);

		if (option_obsoletes)

			print_package_properties(set, package,

						 RAZOR_PROPERTY_OBSOLETES);

		if (option_requires)

			print_package_properties(set, package,

						 RAZOR_PROPERTY_REQUIRES);

		if (option_provides)

			print_package_properties(set, package,

						 RAZOR_PROPERTY_PROVIDES);

		if (option_info)

			print_package_info(set, package);

		if (option_changelog)

			print_package_changelog(set, package);

		if (option_conflicts + option_obsoletes +

		    option_requires + option_provides +

		    option_info + option_changelog == 0)

			printf("%s-%s.%s\n", name, version, arch);

	}

	razor_package_iterator_destroy(pi);

	razor_set_destroy(set);

	return;

}

static void

command_verify(int argc, const char *argv[])

{

	struct razor_set *set;

	struct razor_package_iterator *pi;

	struct razor_package *package;

	const char *name, *version, *arch;

	if (option_package) {

		set = create_set_from_command_line(argc, argv);

		argc = 0;

		option_all = 1;

	} else {

		set = razor_root_open_read_only(option_root);

	}

	pi = get_query_packages(set, argc, argv);

	while (razor_package_iterator_next(pi, &package,

					   &name, &version, &arch)) {

		printf("verify %s-%s.%s - not implemented\n",

		       name, version, arch);

	}

	razor_package_iterator_destroy(pi);

}

static void

remove_package(const char *name,

	       const char *old_version, const char *new_version,

	       const char *arch, void *data)

{

	if (old_version)

		printf("remove %s-%s.%s\n", name, old_version, arch);

}

static void

command_erase(int argc, const char *argv[])

{

	struct razor_set *set, *upstream, *next;

	struct razor_transaction *trans;

	struct razor_package_query *query;

	struct razor_package_iterator *pi;

	struct razor_package *package;

	const char *name, *version, *arch;

	if (argc == 0) {

		printf("no packages given for erase\n");

		exit(1);

	}

	set = razor_set_open(repo_filename);

	upstream = razor_set_create();

	trans = razor_transaction_create(set, upstream);

	query = razor_package_query_create(set);

	add_command_line_packages(set, query, argc, argv);

	pi = razor_package_query_finish(query);

	while (razor_package_iterator_next(pi, &package,

					   &name, &version, &arch))

		razor_transaction_remove_package(trans, package);

	razor_package_iterator_destroy(pi);

	if (!option_nodeps && razor_transaction_describe(trans) > 0) {

		printf("unsatisfied dependencies.\n");

		exit(1);

	}

	if (option_test)

		exit(0);

	next = razor_transaction_finish(trans);

	if (!option_justdb)

		razor_set_diff(set, next, remove_package, NULL);

	razor_set_destroy(set);

	razor_set_destroy(upstream);

	razor_set_destroy(next);

}

static void

install_package(const char *name,

		const char *old_version, const char *new_version,

		const char *arch, void *data)

{

	if (new_version)

		printf("install %s-%s.%s\n", name, new_version, arch);

}

static void

command_install(int argc, const char *argv[])

{

	struct razor_set *set, *upstream, *next;

	struct razor_transaction *trans;

	struct razor_package_iterator *pi;

	struct razor_package *package;

	const char *name, *version, *arch;

	if (argc == 0) {

		printf("no packages given for install\n");

		exit(1);

	}

	set = razor_set_open(repo_filename);

	upstream = create_set_from_command_line(argc, argv);

	trans = razor_transaction_create(set, upstream);

	pi = razor_package_iterator_create(upstream);

	while (razor_package_iterator_next(pi, &package,

					   &name, &version, &arch))

		razor_transaction_install_package(trans, package);

	razor_package_iterator_destroy(pi);

	if (!option_nodeps && razor_transaction_describe(trans) > 0) {

		printf("unsatisfied dependencies.\n");

		exit(1);

	}

	if (option_test)

		exit(0);

	next = razor_transaction_finish(trans);

	if (!option_justdb)

		razor_set_diff(set, next, install_package, NULL);

	razor_set_destroy(set);

	razor_set_destroy(upstream);

	razor_set_destroy(next);

}

static void

update_package(const char *name,

	       const char *old_version, const char *new_version,

	       const char *arch, void *data)

{

	if (old_version)

		printf("remove %s-%s.%s\n", name, old_version, arch);

	if (new_version)

		printf("install %s-%s.%s\n", name, new_version, arch);

}

static void

command_update(int argc, const char *argv[])

{

	struct razor_set *set, *upstream, *next;

	struct razor_transaction *trans;

	struct razor_package_iterator *pi;

	struct razor_package *package;

	const char *name, *version, *arch;

	if (argc == 0) {

		printf("no packages given for update\n");

		exit(1);

	}

	set = razor_set_open(repo_filename);

	upstream = create_set_from_command_line(argc, argv);

	trans = razor_transaction_create(set, upstream);

	pi = razor_package_iterator_create(upstream);

	while (razor_package_iterator_next(pi, &package,

					   &name, &version, &arch))

		razor_transaction_update_package(trans, package);

	razor_package_iterator_destroy(pi);

	if (!option_nodeps && razor_transaction_describe(trans) > 0) {

		printf("unsatisfied dependencies.\n");

		exit(1);

	}

	if (option_test)

		exit(0);

	next = razor_transaction_finish(trans);

	if (!option_justdb)

		razor_set_diff(set, next, update_package, NULL);

	razor_set_destroy(set);

	razor_set_destroy(upstream);

	razor_set_destroy(next);

}

static int

for_each_option(const struct option *options,

		const char *name, char short_name,

		void (*fn)(const struct option *o,

			   const char *name, char short_name,

			   void *data), void *data)

{

	int i, count = 0;

	for (i = 0; options[i].type != OPTION_LAST; i++) {

		switch (options[i].type) {

		case OPTION_GROUP:

			count += for_each_option(options[i].data,

						 name, short_name, fn, data);

			break;

		case OPTION_BOOL:

		case OPTION_STRING:

			if (name && strcmp(options[i].name, name) == 0) {

				fn(&options[i], name, 0, data);

				count++;

				break;

			}

			if (short_name &&

			    short_name == options[i].short_name) {

				fn(&options[i], NULL, short_name, data);

				count++;

				break;

			}

			break;

		case OPTION_LAST:

			break;

		}

	}

	return count;

}

static void

handle_option(const struct option *o,

	      const char *name, char short_name, void *data)

{

	if (o->data == NULL) {

		if (name)

			printf("option --%s not supported\n", name);

		else

			printf("option -%c not supported\n", short_name);

		return;

	}

	switch (o->type) {

	case OPTION_BOOL:

		*(int *) o->data = 1;

		break;

	case OPTION_STRING:

		*(const char **) o->data = name + strlen(o->name) + 1;

		break;

	case OPTION_LAST:

	case OPTION_GROUP:

		/* Shouldn't happen. */

		break;

	}

}

static int

parse_options(const struct option *options, int argc, const char **argv)

{

	int i, j, k;

	for (i = 1, j = 0; i < argc; i++) {

		if (argv[i][0] != '-') {

			argv[j++] = argv[i];

			continue;

		}

		if (argv[i][1] == '-') {

			if (for_each_option(options, &argv[i][2], 0,

					    handle_option, NULL) == 0) {

				printf("unknown option: %s\n", argv[i]);

				exit(1);

			}

			continue;

		}

		for (k = 1; argv[i][k]; k++) {

			if (for_each_option(options, NULL, argv[i][k],

					    handle_option, NULL) == 0) {

				printf("unknown option: -%c\n", argv[i][k]);

				exit(1);

			}

		}

	}

	return j;

}

static void

print_options_help(const struct option *options)

{

	int i;

	for (i = 0; options[i].type != OPTION_LAST; i++) {

		switch (options[i].type) {

		case OPTION_GROUP:

			printf("%s\n", options[i].description);

			print_options_help(options[i].data);

			printf("\n");

			break;

		case OPTION_BOOL:

		case OPTION_STRING:

			printf("  ");

			if (options[i].short_name)

				printf("-%c", options[i].short_name);

			if (options[i].short_name && options[i].name)

				printf(", ");

			if (options[i].name)

				printf("--%s", options[i].name);

			if (options[i].arg_name)

				printf("=%s", options[i].arg_name);

			if (options[i].description)

				printf("\t\t%s", options[i].description);

			printf("\n");

			break;

		case OPTION_LAST:

			break;

		}

	}

}

static void

print_options_usage(const struct option *options)

{

	int i;

	for (i = 0; options[i].type != OPTION_LAST; i++) {

		switch (options[i].type) {

		case OPTION_GROUP:

			print_options_usage(options[i].data);

			break;

		case OPTION_BOOL:

			printf("[");

			if (options[i].short_name)

				printf("-%c", options[i].short_name);

			if (options[i].short_name && options[i].name)

				printf("|");

			if (options[i].name)

				printf("--%s", options[i].name);

			printf("] ");

			break;

		case OPTION_STRING:

			printf("[");

			if (options[i].short_name)

				printf("-%c", options[i].short_name);

			if (options[i].short_name && options[i].name)

				printf("|");

			if (options[i].name)

				printf("--%s", options[i].name);

			if (options[i].arg_name)

				printf("=%s", options[i].arg_name);

			printf("] ");

			break;

			break;

		case OPTION_LAST:

			break;

		}

	}

}

int

main(int argc, const char *argv[])

{

	argc = parse_options(rpm_options, argc, argv);

	if (option_version) {

		printf("razor rpm version hoopla.\n");

		exit(0);

	}

	if (option_help) {

		printf("Usage: rpm [OPTION...]\n");

		print_options_help(rpm_options);

		exit(0);

	}

	if (option_usage) {

		printf("Usage: rpm [OPTION...]\n");

		print_options_usage(rpm_options);

		printf("\n");

		exit(0);

	}

	if (option_initdb) {

		command_initdb(argc, argv);

	} else if (option_verify) {

		command_verify(argc, argv);

	} else if (option_query) {

		command_query(argc, argv);

	} else if (option_install) {

		command_install(argc, argv);

	} else if (option_upgrade) {

		command_update(argc, argv);

	} else if (option_erase) {

		command_erase(argc, argv);

	} else {

		print_options_usage(rpm_options);

		printf("\n");

		exit(0);

	}

	return 0;

}