X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=src%2Fsolr.c;h=de46b5fbfb91858d2e6eb54766c1de55d3999f68;hb=03dccce903176e61f889b1558834aa5a09f9f8c0;hp=28b8f11b552e33bb3afb204c1f5a692069a77daa;hpb=7d2cc7589c630c539f114b24eaff53a98b696005;p=yaz-moved-to-github.git

diff --git a/src/solr.c b/src/solr.c
index 28b8f11..de46b5f 100644
--- a/src/solr.c
+++ b/src/solr.c
@@ -1,5 +1,5 @@
 /* This file is part of the YAZ toolkit.
- * Copyright (C) 1995-2011 Index Data
+ * Copyright (C) 1995-2012 Index Data
  * See the file LICENSE for details.
  */
 /**
@@ -27,6 +27,15 @@
 #include <libxml/parser.h>
 #include <libxml/tree.h>
 
+static void extract_text_node(xmlNodePtr node, WRBUF wrbuf) {
+    xmlNodePtr child;
+    for (child = node->children; child ; child = child->next)
+    {
+        if (child->type == XML_TEXT_NODE)
+            wrbuf_puts(wrbuf, (const char *) child->content);
+    }
+}
+
 static int match_xml_node_attribute(
     xmlNodePtr ptr,
     const char *node_name, const char *attribute_name, const char *value)
@@ -194,6 +203,64 @@ static int yaz_solr_decode_facet_counts(ODR o, xmlNodePtr root,
     return 0;
 }
 
+static void yaz_solr_decode_suggestion_values(xmlNodePtr listPptr, WRBUF wrbuf)
+{
+    xmlNodePtr node;
+    for (node = listPptr; node; node= node->next) {
+        if (!strcmp((char*) node->name, "lst")) {
+            xmlNodePtr child;
+            for (child = node->children; child; child= child->next) {
+                if (match_xml_node_attribute(child, "str", "name", "word")) {
+                    wrbuf_puts(wrbuf, "<suggestion>");
+                    extract_text_node(child, wrbuf);
+                    wrbuf_puts(wrbuf, "</suggestion>\n");
+                }
+            }
+        }
+    }
+}
+
+static void yaz_solr_decode_suggestion_lst(xmlNodePtr lstPtr, WRBUF wrbuf)
+{
+    xmlNodePtr node;
+    for (node = lstPtr; node; node= node->next) {
+        if (match_xml_node_attribute(node, "arr", "name", "suggestion")) {
+            yaz_solr_decode_suggestion_values(node->children, wrbuf);
+        }
+    }
+}
+
+static void yaz_solr_decode_misspelled(xmlNodePtr lstPtr, WRBUF wrbuf)
+{
+    xmlNodePtr node;
+    for (node = lstPtr; node; node= node->next)
+    {
+        if (!strcmp((const char*) node->name, "lst")) {
+            const char *misspelled = yaz_element_attribute_value_get(node, "lst", "name");
+            if (misspelled) {
+                wrbuf_printf(wrbuf, "<misspelled term=\"%s\">\n", misspelled);
+                yaz_solr_decode_suggestion_lst(node->children, wrbuf);
+                wrbuf_puts(wrbuf, "</misspelled>\n");
+            }
+        }
+    }
+}
+
+static int yaz_solr_decode_spellcheck(ODR o, xmlNodePtr spellcheckPtr, Z_SRW_searchRetrieveResponse *sr)
+{
+    xmlNodePtr ptr;
+    WRBUF wrbuf = wrbuf_alloc();
+    wrbuf_puts(wrbuf, "");
+    for (ptr = spellcheckPtr->children; ptr; ptr = ptr->next)
+    {
+        if (match_xml_node_attribute(ptr, "lst", "name", "suggestions"))
+        {
+            yaz_solr_decode_misspelled(ptr->children, wrbuf);
+        }
+    }
+    sr->suggestions = odr_strdup(o, wrbuf_cstr(wrbuf));
+    return 0;
+}
 #endif
 
 int yaz_solr_decode_response(ODR o, Z_HTTP_Response *hres, Z_SRW_PDU **pdup)
@@ -235,9 +302,12 @@ int yaz_solr_decode_response(ODR o, Z_HTTP_Response *hres, Z_SRW_PDU **pdup)
                 /* TODO The check on hits is a work-around to avoid garbled facets on zero results from the SOLR server.
                  * The work-around works because the results is before the facets in the xml. */
                 if (rc_result == 0 &&  *sr->numberOfRecords > 0 &&
-                    match_xml_node_attribute(ptr, "lst", "name",
-                                             "facet_counts"))
+                    match_xml_node_attribute(ptr, "lst", "name", "facet_counts"))
                     rc_facets =  yaz_solr_decode_facet_counts(o, ptr, sr);
+                if (rc_result == 0 &&  *sr->numberOfRecords == 0 &&
+                    match_xml_node_attribute(ptr, "lst", "name", "spellcheck"))
+                    rc_facets =  yaz_solr_decode_spellcheck(o, ptr, sr);
+
             }
             ret = rc_result + rc_facets;
         }