util/yaz-xmlquery.c

   1 /*
   2  * Copyright (C) 1995-2005, Index Data ApS
   3  * See the file LICENSE for details.
   4  *
   5  * $Id: yaz-xmlquery.c,v 1.2 2006-03-06 08:11:41 adam Exp $
   6  */
   7
   8 #include <stdlib.h>
   9 #include <stdio.h>
  10
  11 #include <yaz/options.h>
  12 #include <yaz/querytowrbuf.h>
  13 #include <yaz/xmlquery.h>
  14 #include <yaz/pquery.h>
  15 #include <yaz/test.h>
  16
  17 #if HAVE_XML2
  18 #include <libxml/parser.h>
  19 #endif
  20
  21 static char *prog = "yaz-xmlquery";
  22
  23 #if HAVE_XML2
  24 void pqftoxmlquery(const char *pqf)
  25 {
  26     YAZ_PQF_Parser parser = yaz_pqf_create();
  27     ODR odr = odr_createmem(ODR_ENCODE);
  28     Z_RPNQuery *rpn;
  29
  30     if (!parser)
  31     {
  32         fprintf(stderr, "%s: cannot create parser\n", prog);
  33         exit(1);
  34     }
  35
  36     if (!odr)
  37     {
  38         fprintf(stderr, "%s: cannot create parser\n", prog);
  39         exit(1);
  40     }
  41
  42     rpn = yaz_pqf_parse(parser, odr, pqf);
  43
  44     yaz_pqf_destroy(parser);
  45
  46     if (!rpn)
  47     {
  48         fprintf(stderr, "%s: pqf parse error for query %s\n", prog, pqf);
  49         exit(2);
  50     }
  51     else
  52     {
  53         xmlDocPtr doc = 0;
  54
  55         yaz_rpnquery2xml(rpn, &doc);
  56
  57         if (!doc)
  58         {
  59             fprintf(stderr, "%s: yaz_rpnquery2xml failed for query %s\n",
  60                     prog, pqf);
  61             exit(3);
  62         }
  63         else
  64         {
  65             char *buf_out = 0;
  66             int len_out = 0;
  67
  68             xmlDocDumpMemory(doc, (xmlChar **) &buf_out, &len_out);
  69
  70             if (!len_out || !buf_out)
  71             {
  72                 fprintf(stderr, "%s: xmlDocDumpMemory failed for query %s\n",
  73                         prog, pqf);
  74                 exit(4);
  75             }
  76             else
  77                 fwrite(buf_out, len_out, 1, stdout);
  78             xmlFreeDoc(doc);
  79         }
  80     }
  81     odr_destroy(odr);
  82 }
  83
  84
  85 void xmlquerytopqf(const char *xmlstr)
  86 {
  87     xmlDocPtr doc;
  88
  89     doc = xmlParseMemory(xmlstr, strlen(xmlstr));
  90     if (!doc)
  91     {
  92         fprintf(stderr, "%s: xml parse error for XML:\n%s\n", prog, xmlstr);
  93         exit(1);
  94     }
  95     else
  96     {
  97         int error_code = 0;
  98         const char *addinfo = 0;
  99         Z_Query *query = 0;
 100         ODR odr = odr_createmem(ODR_ENCODE);
 101
 102         const xmlNode *root_element = xmlDocGetRootElement(doc);
 103         yaz_xml2query(root_element, &query, odr, &error_code, &addinfo);
 104         if (error_code)
 105         {
 106             fprintf(stderr, "%s: yaz_xml2query failed code=%d addinfo=%s\n",
 107                     prog, error_code, addinfo);
 108             exit(1);
 109         }
 110         else if (!query)
 111         {
 112             fprintf(stderr, "%s: yaz_xml2query no query result\n",
 113                     prog);
 114             exit(1);
 115         }
 116         else
 117         {
 118             WRBUF w = wrbuf_alloc();
 119             yaz_query_to_wrbuf(w, query);
 120             printf("%s\n", wrbuf_buf(w));
 121             wrbuf_free(w, 1);
 122         }
 123         odr_destroy(odr);
 124         xmlFreeDoc(doc);
 125     }
 126 }
 127
 128 void xmlfiletopqf(const char *xmlfile)
 129 {
 130     long sz;
 131     char *xmlstr;
 132     FILE *f = fopen(xmlfile, "rb");
 133     if (!f)
 134     {
 135         fprintf(stderr, "%s: cannot open %s\n", prog, xmlfile);
 136         exit(1);
 137     }
 138     fseek(f, 0, SEEK_END);
 139     sz = ftell(f);
 140     if (sz <= 0 || sz >= 1<<18)
 141     {
 142         fprintf(stderr, "%s: bad size for file %s\n", prog, xmlfile);
 143         exit(1);
 144     }
 145     rewind(f);
 146     xmlstr = xmalloc(sz+1);
 147     xmlstr[sz] = '\0';
 148     fread(xmlstr, sz, 1, f);
 149     fclose(f);
 150
 151     xmlquerytopqf(xmlstr);
 152     xfree(xmlstr);
 153 }
 154 #endif
 155
 156 void usage()
 157 {
 158     fprintf(stderr, "%s [-p pqf] [-x xmlfile]\n", prog);
 159     fprintf(stderr, " -p pqf      reads pqf. write xml to stdout\n");
 160     fprintf(stderr, " -x xmlfile  reads XML from file. write pqf to stdout\n");
 161     exit(1);
 162 }
 163
 164 int main (int argc, char **argv)
 165 {
 166 #if HAVE_XML2
 167     char *arg;
 168     int r;
 169     int active = 0;
 170
 171     while ((r = options("-p:x:", argv, argc, &arg)) != -2)
 172     {
 173         switch(r)
 174         {
 175         case 'p':
 176             pqftoxmlquery(arg);
 177             active = 1;
 178             break;
 179         case 'x':
 180             xmlfiletopqf(arg);
 181             active = 1;
 182             break;
 183         case 0:
 184             break;
 185         }
 186     }
 187     if (!active)
 188     {
 189         fprintf(stderr, "%s: nothing to do\n", prog);
 190         usage();
 191     }
 192 #else
 193     fprintf(stderr, "%s: XML support not enabled.\n", prog);
 194     exit(1);
 195 #endif
 196     return 0;
 197 }
 198