20 #include <metaproxy/filter.hpp>
21 #include <metaproxy/package.hpp>
23 #include <metaproxy/util.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>
37 namespace yf = mp::filter;
45 void process(mp::Package &package)
const;
46 void configure(
const xmlNode * ptr,
bool test_only,
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()
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);
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");
233 return new mp::filter::QueryRewrite;
static mp::filter::Base * filter_creator()
struct metaproxy_1_filter_struct metaproxy_1_filter_query_rewrite