xapian-dump: Add values to the dump as well.

[notmuch] / xapian-dump.cc

diff --git a/xapian-dump.cc b/xapian-dump.cc
index 6d1fc1be6150863155b52b81123b1c6b905de5ed..79a018d6de967709fea6696555bbae3fdd4ecd85 100644 (file)
--- a/xapian-dump.cc
+++ b/xapian-dump.cc
@@ -1,4 +1,5 @@
-/*
+/* xapian-dump: Create a textual dump of a Xapian database.
+ *
  * Copyright © 2009 Carl Worth
  *
  * This program is free software: you can redistribute it and/or modify
@@ -17,18 +18,93 @@
  * Author: Carl Worth <cworth@cworth.org>
  */
 
+/* Currently the dumped data includes:
+ *
+ *     All document IDs
+ *
+ * And for each document ID:
+ *
+ *     All terms
+ *     All values
+ *
+ * Things not yet dumped include:
+ *
+ * Data associated with a document.
+ */
+
 #include <cstdlib>
+#include <iostream>
 
 #include <xapian.h>
 
+using namespace std;
+
+static void
+print_document_terms (Xapian::Document doc)
+{
+    Xapian::TermIterator i;
+
+    printf ("Terms:\n");
+
+    for (i = doc.termlist_begin (); i != doc.termlist_end (); i++)
+       cout << "\t" << *i << endl;
+}
+
+static void
+print_document_values (Xapian::Document doc)
+{
+    Xapian::ValueIterator i;
+
+    printf ("Values:\n");
+
+    for (i = doc.values_begin (); i != doc.values_end (); i++)
+       cout << "\t" << i.get_valueno() << ": " << *i << endl;
+}
+
+static void
+print_document (Xapian::Database db, Xapian::docid id)
+{
+    Xapian::Document doc;
+
+    printf ("Document %u:\n", id);
+
+    doc = db.get_document (id);
+
+    print_document_terms (doc);
+
+    print_document_values (doc);
+}
+
 int
 main (int argc, char *argv[])
 {
+    const char *database_path;
+
     if (argc < 2) {
        fprintf (stderr, "Usage: %s <path-to-xapian-database>\n",
                 argv[0]);
        exit (1);
     }
 
+    database_path = argv[1];
+
+    try {
+
+       Xapian::Database db;
+        Xapian::PostingIterator i;
+       Xapian::docid doc_id;
+
+       db = Xapian::Database (database_path);
+       for (i = db.postlist_begin (""); i != db.postlist_end (""); i++) {
+           doc_id = *i;
+
+           print_document (db, doc_id);
+       }
+
+    } catch (const Xapian::Error &error) {
+       cerr << "A Xapian exception occurred: " << error.get_msg () << endl;
+       exit (1);
+    }
+
     return 0;
 }