Introduce -D, --database

The request of FS#12950 is implemented.

On the backend side, I introduced a new function, alpm_db_set_pkgreason(),
to modify the install reason of a package in the local database. On the
front-end side, I introduced a new main operation, -D/--database, which has
two options, --asdeps and --asexplicit. I documented this in pacman manual.
I've created two pactests to test -D: database001.py and database002.py.

Signed-off-by: Nagy Gabor <ngaba@bibl.u-szeged.hu>
Signed-off-by: Dan McGee <dan@archlinux.org>

6 files changed, 113 insertions, 6 deletions

diff --git a/src/pacman/Makefile.am b/src/pacman/Makefile.am
index 220ee9cb..8c14562c 100644
--- a/src/pacman/Makefile.am
+++ b/src/pacman/Makefile.am
@@ -24,6 +24,7 @@ endif
 
 pacman_SOURCES = \
 	conf.h conf.c \
+	database.c \
 	deptest.c \
 	package.h package.c \
 	pacman.h pacman.c \
diff --git a/src/pacman/conf.h b/src/pacman/conf.h
index 1520c3b4..8b1bc2e1 100644
--- a/src/pacman/conf.h
+++ b/src/pacman/conf.h
@@ -86,7 +86,8 @@ enum {
 	PM_OP_UPGRADE,
 	PM_OP_QUERY,
 	PM_OP_SYNC,
-	PM_OP_DEPTEST
+	PM_OP_DEPTEST,
+	PM_OP_DATABASE
 };
 
 /* Long Operations */
diff --git a/src/pacman/database.c b/src/pacman/database.c
new file mode 100644
index 00000000..9e335fe1
--- /dev/null
+++ b/src/pacman/database.c
@@ -0,0 +1,90 @@
+/*
+ *  database.c
+ *
+ *  Copyright (c) 2006-2010 Pacman Development Team <pacman-dev@archlinux.org>
+ *  Copyright (c) 2002-2006 by Judd Vinet <jvinet@zeroflux.org>
+ *
+ *  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, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "config.h"
+
+#include <stdlib.h>
+#include <stdio.h>
+
+#include <alpm.h>
+#include <alpm_list.h>
+
+/* pacman */
+#include "pacman.h"
+#include "conf.h"
+#include "util.h"
+
+extern pmdb_t *db_local;
+
+/**
+ * @brief Modify the 'local' package database.
+ *
+ * @param targets a list of packages (as strings) to modify
+ *
+ * @return 0 on success, 1 on failure
+ */
+int pacman_database(alpm_list_t *targets)
+{
+	alpm_list_t *i;
+	int retval = 0;
+	pmpkgreason_t reason;
+
+	if(targets == NULL) {
+		pm_printf(PM_LOG_ERROR, _("no targets specified (use -h for help)\n"));
+		return(1);
+	}
+
+	if(config->flags & PM_TRANS_FLAG_ALLDEPS) { /* --asdeps */
+		reason = PM_PKG_REASON_DEPEND;
+	} else if(config->flags & PM_TRANS_FLAG_ALLEXPLICIT) { /* --asexplicit */
+		reason = PM_PKG_REASON_EXPLICIT;
+	} else {
+		pm_printf(PM_LOG_ERROR, _("no install reason specified (use -h for help)\n"));
+		return(1);
+	}
+
+	/* Lock database */
+	if(trans_init(0) == -1) {
+		return(1);
+	}
+
+	for(i = targets; i; i = alpm_list_next(i)) {
+		char *pkgname = i->data;
+		if(alpm_db_set_pkgreason(db_local, pkgname, reason) == -1) {
+			pm_printf(PM_LOG_ERROR, _("could not set install reason for package %s (%s)\n"),
+							pkgname, alpm_strerrorlast());
+			retval = 1;
+		} else {
+			if(reason == PM_PKG_REASON_DEPEND) {
+				printf(_("%s: install reason has been set to 'installed as dependency'\n"), pkgname);
+			} else {
+				printf(_("%s: install reason has been set to 'explicitly installed'\n"), pkgname);
+			}
+		}
+	}
+
+	/* Unlock database */
+	if(trans_release() == -1) {
+		return(1);
+	}
+	return(retval);
+}
+
+/* vim: set ts=2 sw=2 noet: */
diff --git a/src/pacman/pacman.c b/src/pacman/pacman.c
index e601a14d..19d27a94 100644
--- a/src/pacman/pacman.c
+++ b/src/pacman/pacman.c
@@ -77,10 +77,11 @@ static void usage(int op, const char * const myname)
 		printf(_("operations:\n"));
 		printf("    %s {-h --help}\n", myname);
 		printf("    %s {-V --version}\n", myname);
-		printf("    %s {-Q --query}   [%s] [%s]\n", myname, str_opt, str_pkg);
-		printf("    %s {-R --remove}  [%s] <%s>\n", myname, str_opt, str_pkg);
-		printf("    %s {-S --sync}    [%s] [%s]\n", myname, str_opt, str_pkg);
-		printf("    %s {-U --upgrade} [%s] <%s>\n", myname, str_opt, str_file);
+		printf("    %s {-D --database} <%s> <%s>\n", myname, str_opt, str_pkg);
+		printf("    %s {-Q --query}    [%s] [%s]\n", myname, str_opt, str_pkg);
+		printf("    %s {-R --remove}   [%s] <%s>\n", myname, str_opt, str_pkg);
+		printf("    %s {-S --sync}     [%s] [%s]\n", myname, str_opt, str_pkg);
+		printf("    %s {-U --upgrade}  [%s] <%s>\n", myname, str_opt, str_file);
 		printf(_("\nuse '%s {-h --help}' with an operation for available options\n"),
 				myname);
 	} else {
@@ -147,6 +148,11 @@ static void usage(int op, const char * const myname)
 			printf(_("      --print-format <string>\n"
 			         "                       specify how the targets should be printed\n"));
 			printf(_("  -q, --quiet          show less information for query and search\n"));
+		} else if (op == PM_OP_DATABASE) {
+			printf("%s:  %s {-D --database} <%s> <%s>\n", str_usg, myname, str_opt, str_pkg);
+			printf("%s:\n", str_opt);
+			printf(_("      --asdeps         mark packages as non-explicitly installed\n"));
+			printf(_("      --asexplicit     mark packages as explicitly installed\n"));
 		}
 		printf(_("      --config <path>  set an alternate configuration file\n"));
 		printf(_("      --logfile <path> set an alternate log file\n"));
@@ -355,6 +361,7 @@ static int parseargs(int argc, char *argv[])
 	int option_index = 0;
 	static struct option opts[] =
 	{
+		{"database",   no_argument,       0, 'D'},
 		{"query",      no_argument,       0, 'Q'},
 		{"remove",     no_argument,       0, 'R'},
 		{"sync",       no_argument,       0, 'S'},
@@ -409,7 +416,7 @@ static int parseargs(int argc, char *argv[])
 		{0, 0, 0, 0}
 	};
 
-	while((opt = getopt_long(argc, argv, "RUQSTr:b:vkhscVfmnoldepqituwygz", opts, &option_index))) {
+	while((opt = getopt_long(argc, argv, "RUDQSTr:b:vkhscVfmnoldepqituwygz", opts, &option_index))) {
 		alpm_list_t *list = NULL, *item = NULL; /* lists for splitting strings */
 
 		if(opt < 0) {
@@ -498,6 +505,7 @@ static int parseargs(int argc, char *argv[])
 				check_optarg();
 				config->print_format = strdup(optarg);
 				break;
+			case 'D': config->op = (config->op != PM_OP_MAIN ? 0 : PM_OP_DATABASE); break;
 			case 'Q': config->op = (config->op != PM_OP_MAIN ? 0 : PM_OP_QUERY); break;
 			case 'R': config->op = (config->op != PM_OP_MAIN ? 0 : PM_OP_REMOVE); break;
 			case 'S': config->op = (config->op != PM_OP_MAIN ? 0 : PM_OP_SYNC); break;
@@ -1184,6 +1192,9 @@ int main(int argc, char *argv[])
 
 	/* start the requested operation */
 	switch(config->op) {
+		case PM_OP_DATABASE:
+			ret = pacman_database(pm_targets);
+			break;
 		case PM_OP_REMOVE:
 			ret = pacman_remove(pm_targets);
 			break;
diff --git a/src/pacman/pacman.h b/src/pacman/pacman.h
index 964fe7d8..f8443cf6 100644
--- a/src/pacman/pacman.h
+++ b/src/pacman/pacman.h
@@ -22,6 +22,8 @@
 
 #include <alpm_list.h>
 
+/* database.c */
+int pacman_database(alpm_list_t *targets);
 /* query.c */
 int pacman_query(alpm_list_t *targets);
 /* remove.c */
diff --git a/src/pacman/util.c b/src/pacman/util.c
index 5099237e..0cae6d7c 100644
--- a/src/pacman/util.c
+++ b/src/pacman/util.c
@@ -80,6 +80,8 @@ int trans_release(void)
 int needs_root(void)
 {
 	switch(config->op) {
+		case PM_OP_DATABASE:
+			return(1);
 		case PM_OP_UPGRADE:
 		case PM_OP_REMOVE:
 			return(!config->print);