# HG changeset patch
# User Kristian H?gsberg <krh@redhat.com>
# Date 1190225343 14400
# Node ID c8958f67afd8f57673c46c5db118386d0fefe89f
# Parent  5dbd81809d264b96eb717bc7f5dfa534b19476d5
Add importer for system rpm database.

diff -r 5dbd81809d26 -r c8958f67afd8 Makefile
--- a/Makefile	Tue Sep 18 15:02:04 2007 -0400
+++ b/Makefile	Wed Sep 19 14:09:03 2007 -0400
@@ -1,5 +1,5 @@
 CFLAGS = -Wall -g -O2
-LDLIBS = -lexpat -g
+LDLIBS = -lexpat -g -lrpm
 
 razor : razor.o import.o sha1.o
 
diff -r 5dbd81809d26 -r c8958f67afd8 import.c
--- a/import.c	Tue Sep 18 15:02:04 2007 -0400
+++ b/import.c	Wed Sep 19 14:09:03 2007 -0400
@@ -9,6 +9,8 @@
 #include <errno.h>
 
 #include <expat.h>
+#include <rpm/rpmlib.h>
+#include <rpm/rpmdb.h>
 #include "sha1.h"
 #include "razor.h"
 
@@ -257,3 +259,62 @@
 
 	return razor_finish_import(&ctx.ctx);
 }
+
+struct razor_set *
+razor_set_create_from_rpmdb(void)
+{
+	struct import_context ctx;
+	rpmdbMatchIterator iter;
+	Header h;
+	int_32 type, count, i;
+	char *name, *version, *release;
+	char **properties, **property_versions;
+	rpmdb db;
+
+	rpmReadConfigFiles(NULL, NULL);
+
+	if (rpmdbOpen("", &db, O_RDONLY, 0644) != 0) {
+		fprintf(stderr, "cannot open rpm database\n");
+		exit(1);
+	}
+
+	razor_prepare_import(&ctx);
+
+	iter = rpmdbInitIterator(db, 0, NULL, 0);
+	while (h = rpmdbNextIterator(iter), h != NULL) {
+		headerGetEntry(h, RPMTAG_NAME, &type,
+			       (void **) &name, &count);
+		headerGetEntry(h, RPMTAG_VERSION, &type,
+			       (void **) &version, &count);
+		headerGetEntry(h, RPMTAG_RELEASE, &type,
+			       (void **) &release, &count);
+		import_context_add_package(&ctx, name, version);
+
+
+		headerGetEntry(h, RPMTAG_REQUIRES, &type,
+			       (void **) &properties, &count);
+		headerGetEntry(h, RPMTAG_REQUIREVERSION, &type,
+			       (void **) &property_versions, &count);
+		for (i = 0; i < count; i++)
+			import_context_add_property(&ctx,
+						    &ctx.requires,
+						    properties[i],
+						    property_versions[i]);
+
+		headerGetEntry(h, RPMTAG_PROVIDES, &type,
+			       (void **) &properties, &count);
+		headerGetEntry(h, RPMTAG_PROVIDEVERSION, &type,
+			       (void **) &property_versions, &count);
+		for (i = 0; i < count; i++)
+			import_context_add_property(&ctx,
+						    &ctx.provides,
+						    properties[i],
+						    property_versions[i]);
+
+		import_context_finish_package(&ctx);
+	}
+
+	rpmdbClose(db);
+
+	return razor_finish_import(&ctx);
+}
diff -r 5dbd81809d26 -r c8958f67afd8 razor.c
--- a/razor.c	Tue Sep 18 15:02:04 2007 -0400
+++ b/razor.c	Wed Sep 19 14:09:03 2007 -0400
@@ -928,6 +928,13 @@
 		razor_set_write(set, rawhide_repo_filename);
 		razor_set_destroy(set);
 		printf("wrote %s\n", rawhide_repo_filename);
+	} else if (strcmp(argv[1], "import-rpmdb") == 0) {
+		set = razor_set_create_from_rpmdb();
+		if (set == NULL)
+			return 1;
+		razor_set_write(set, repo_filename);
+		razor_set_destroy(set);
+		printf("wrote %s\n", repo_filename);
 	} else if (strcmp(argv[1], "validate") == 0) {
 		set = razor_set_open(repo_filename);
 		if (set == NULL)
diff -r 5dbd81809d26 -r c8958f67afd8 razor.h
--- a/razor.h	Tue Sep 18 15:02:04 2007 -0400
+++ b/razor.h	Wed Sep 19 14:09:03 2007 -0400
@@ -79,5 +79,6 @@
 
 struct razor_set *razor_import_rzr_files(int count, const char **files);
 struct razor_set *razor_set_create_from_yum_filelist(int fd);
+struct razor_set *razor_set_create_from_rpmdb(void);
 
 #endif /* _RAZOR_H_ */