1 /* This file is part of Metaproxy.
2 Copyright (C) Index Data
4 Metaproxy is free software; you can redistribute it and/or modify it under
5 the terms of the GNU General Public License as published by the Free
6 Software Foundation; either version 2, or (at your option) any later
9 Metaproxy is distributed in the hope that it will be useful, but WITHOUT ANY
10 WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software
16 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
20 #include <metaproxy/filter.hpp>
21 #include <metaproxy/package.hpp>
23 #include <metaproxy/util.hpp>
24 #include "filter_query_rewrite.hpp"
28 #include <yaz/xmlquery.h>
29 #include <yaz/diagbib1.h>
30 #include <yaz/query-charset.h>
31 #include <yaz/tpath.h>
33 #include <libxslt/xsltutils.h>
34 #include <libxslt/transform.h>
36 namespace mp = metaproxy_1;
37 namespace yf = mp::filter;
39 namespace metaproxy_1 {
41 class QueryRewrite::Rep {
45 void process(mp::Package &package) const;
46 void configure(const xmlNode * ptr, bool test_only,
49 xsltStylesheetPtr m_stylesheet;
50 std::string charset_from;
51 std::string charset_to;
56 yf::QueryRewrite::Rep::Rep() : m_stylesheet(0), charset_from("UTF-8")
60 yf::QueryRewrite::Rep::~Rep()
63 xsltFreeStylesheet(m_stylesheet);
66 yf::QueryRewrite::QueryRewrite() : m_p(new Rep)
70 yf::QueryRewrite::~QueryRewrite()
71 { // must have a destructor because of boost::scoped_ptr
74 void mp::filter::QueryRewrite::configure(const xmlNode *ptr, bool test_only,
77 m_p->configure(ptr, test_only, path);
80 void yf::QueryRewrite::process(mp::Package &package) const
82 m_p->process(package);
85 void yf::QueryRewrite::Rep::process(mp::Package &package) const
87 Z_GDU *gdu = package.request().get();
89 if (gdu && gdu->which == Z_GDU_Z3950)
91 Z_APDU *apdu_req = gdu->u.z3950;
92 if (apdu_req->which == Z_APDU_searchRequest)
95 const char *addinfo = 0;
97 Z_SearchRequest *req = apdu_req->u.searchRequest;
101 xmlDocPtr doc_input = 0;
102 yaz_query2xml(req->query, &doc_input);
106 xmlDocPtr doc_res = xsltApplyStylesheet(m_stylesheet,
110 error_code = YAZ_BIB1_MALFORMED_QUERY;
111 addinfo = "XSLT transform failed for query";
115 const xmlNode *root_element = xmlDocGetRootElement(doc_res);
116 yaz_xml2query(root_element, &req->query, odr,
117 &error_code, &addinfo);
120 xmlFreeDoc(doc_input);
123 if (!error_code && charset_to.length() && charset_from.length() &&
124 (req->query->which == Z_Query_type_1
125 || req->query->which == Z_Query_type_101))
127 yaz_iconv_t cd = yaz_iconv_open(charset_to.c_str(),
128 charset_from.c_str());
131 int r = yaz_query_charset_convert_rpnquery_check(
132 req->query->u.type_1, odr, cd);
135 { /* query could not be char converted */
136 error_code = YAZ_BIB1_MALFORMED_QUERY;
137 addinfo = "could not convert query to target charset";
144 odr.create_searchResponse(apdu_req, error_code, addinfo);
145 package.response() = f_apdu;
148 package.request() = gdu;
154 void mp::filter::QueryRewrite::Rep::configure(const xmlNode *ptr,
155 bool test_only, const char *path)
157 for (ptr = ptr->children; ptr; ptr = ptr->next)
159 if (ptr->type != XML_ELEMENT_NODE)
162 if (mp::xml::is_element_mp(ptr, "xslt"))
166 throw mp::filter::FilterException
167 ("Only one xslt element allowed in query_rewrite filter");
172 for (struct _xmlAttr *attr = ptr->properties;
173 attr; attr = attr->next)
175 mp::xml::check_attribute(attr, "", "stylesheet");
176 fname = mp::xml::get_text(attr);
179 if (0 == fname.size())
180 throw mp::filter::FilterException
181 ("Attribute <xslt stylesheet=\""
183 + "\"> needs XSLT stylesheet path content"
184 + " in query_rewrite filter");
187 char *cp = yaz_filepath_resolve(fname.c_str(), path, 0, fullpath);
190 throw mp::filter::FilterException("Cannot read XSLT " + fname);
193 m_stylesheet = xsltParseStylesheetFile(BAD_CAST cp);
196 throw mp::filter::FilterException
197 ("Failed to read XSLT stylesheet '"
199 + "' in query_rewrite filter");
202 else if (mp::xml::is_element_mp(ptr, "charset"))
204 for (struct _xmlAttr *attr = ptr->properties;
205 attr; attr = attr->next)
207 if (!strcmp((const char *) attr->name, "from"))
209 charset_from = mp::xml::get_text(attr);
211 else if (!strcmp((const char *) attr->name, "to"))
213 charset_to = mp::xml::get_text(attr);
216 throw mp::filter::FilterException
217 ("Invalid attribute inside charset inside "
218 "query_rewrite filter");
223 throw mp::filter::FilterException
225 + std::string((const char *) ptr->name)
226 + " in query_rewrite filter");
231 static mp::filter::Base* filter_creator()
233 return new mp::filter::QueryRewrite;
237 struct metaproxy_1_filter_struct metaproxy_1_filter_query_rewrite = {
247 * c-file-style: "Stroustrup"
248 * indent-tabs-mode: nil
250 * vim: shiftwidth=4 tabstop=8 expandtab