- add query mode

diff --git a/tools/repomdxml2solv.c b/tools/repomdxml2solv.c
index a4bc097..2dcdb7e 100644 (file)
--- a/tools/repomdxml2solv.c
+++ b/tools/repomdxml2solv.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2007, Novell Inc.
+ * Copyright (c) 2007-2009, Novell Inc.
  *
  * This program is licensed under the BSD license, read LICENSE.BSD
  * for further information
@@ -11,6 +11,7 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <unistd.h>
 
 #include "pool.h"
 #include "repo.h"
@@ -18,43 +19,103 @@
 #include "common_write.h"
 
 static void
-usage(const char *err)
+usage(int status)
 {
-  if (err)
-    fprintf(stderr, "\n** Error:\n  %s\n", err);
   fprintf(stderr, "\nUsage:\n"
-          "repomdxml2solv [-a][-h][-k][-n <attrname>]\n"
+          "repomdxml2solv [-q query]\n"
          "  reads a 'repomd.xml' file from <stdin> and writes a .solv file to <stdout>\n"
+         "  -q : query a repomd data entry\n"
          "  -h : print help & exit\n"
         );
-   exit(0);
+   exit(status);
+}
+
+static void
+doquery(Pool *pool, Repo *repo, const char *query)
+{
+  Id id, type = 0;
+  char qbuf[256];
+  const char *qp;
+  Dataiterator di;
+
+  qp = strchr(query, ':');
+  if (qp)
+    {
+      type = strn2id(pool, query, qp - query, 0);
+      if (!type)
+       exit(0);
+      qp++;
+    }
+  else
+    qp = query;
+  snprintf(qbuf, sizeof(qbuf), "repository:repomd:%s", qp);
+  id = str2id(pool, qbuf, 0);
+  if (!id)
+    exit(0);
+  dataiterator_init(&di, pool, repo, SOLVID_META, id, 0, 0);
+  dataiterator_prepend_keyname(&di, REPOSITORY_REPOMD);
+  while (dataiterator_step(&di))
+    {
+      if (type)
+       {
+         dataiterator_setpos_parent(&di);
+         if (pool_lookup_id(pool, SOLVID_POS, REPOSITORY_REPOMD_TYPE) != type)
+           continue;
+       }
+      switch (di.key->type)
+       {
+       case REPOKEY_TYPE_ID:
+       case REPOKEY_TYPE_CONSTANTID:
+         printf("%s\n", id2str(pool, di.kv.id));
+         break;
+       case REPOKEY_TYPE_STR:
+         printf("%s\n", di.kv.str);
+         break;
+       case REPOKEY_TYPE_NUM:
+         printf("%d\n", di.kv.num);
+         break;
+       case REPOKEY_TYPE_SHA1:
+         printf("sha1:%s\n", repodata_chk2str(di.data, di.key->type, (unsigned char *)di.kv.str));
+         break;
+       case REPOKEY_TYPE_SHA256:
+         printf("sha256:%s\n", repodata_chk2str(di.data, di.key->type, (unsigned char *)di.kv.str));
+         break;
+       default:
+         break;
+       }
+    }
+  dataiterator_free(&di);
 }
 
 int
 main(int argc, char **argv)
 {
-  int flags = 0;
+  int c, flags = 0;
+  const char *query = 0;
   
   Pool *pool = pool_create();
   Repo *repo = repo_create(pool, "<stdin>");
 
-  argv++;
-  argc--;
-  while (argc--)
+  while ((c = getopt (argc, argv, "hq:")) >= 0)
     {
-      const char *s = argv[0];
-      if (*s++ == '-')
-        while (*s)
-          switch (*s++)
-           {
-             case 'h': usage(NULL); break;
-             default : break;
-           }
-      argv++;
+      switch(c)
+        {
+        case 'h':
+          usage(0);
+          break;
+        case 'q':
+         query = optarg;
+          break;
+       default:
+          usage(1);
+          break;
+        }
     }
-
   repo_add_repomdxml(repo, stdin, flags);
-  tool_write(repo, 0, 0);
+  if (query)
+    doquery(pool, repo, query);
+  else
+    tool_write(repo, 0, 0);
   pool_free(pool);
   exit(0);
 }