/*
 * 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;
}