X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=recctrl%2Fxslt.c;h=6544b37baeab3e015c1239715e41fe72efeff3ad;hb=1722bbb1db5ba2b1cf096388cf3206863ae0a203;hp=421a307330480c94ebce38543b1c52ccd08d8e52;hpb=fd5356062a1e7ed5d750b8ffc97d0052c1c6af85;p=idzebra-moved-to-github.git diff --git a/recctrl/xslt.c b/recctrl/xslt.c index 421a307..6544b37 100644 --- a/recctrl/xslt.c +++ b/recctrl/xslt.c @@ -1,4 +1,4 @@ -/* $Id: xslt.c,v 1.2 2005-04-28 12:34:21 adam Exp $ +/* $Id: xslt.c,v 1.24 2006-05-24 12:56:56 marc Exp $ Copyright (C) 1995-2005 Index Data ApS @@ -25,23 +25,57 @@ Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA #include #include +#include + +#include +#include +#include +#include #include #include #include #include -struct filter_info { +struct filter_xslt_schema { + const char *name; + const char *identifier; + const char *stylesheet; + struct filter_xslt_schema *next; + const char *default_schema; + const char *include_snippet; xsltStylesheetPtr stylesheet_xsp; - xmlTextReaderPtr reader; +}; + +struct filter_xslt_info { + xmlDocPtr doc; char *fname; - int split_depth; + char *full_name; + const char *profile_path; + const char *split_level; + const char *split_path; ODR odr; + struct filter_xslt_schema *schemas; + xmlTextReaderPtr reader; }; -#define ZEBRA_INDEX_NS "http://indexdata.dk/zebra/indexing/1" -#define ZEBRA_SCHEMA_IDENTITY_NS "http://indexdata.dk/zebra/identity/1" -static const char *zebra_index_ns = ZEBRA_INDEX_NS; + +#define ZEBRA_SCHEMA_XSLT_NS "http://indexdata.dk/zebra/xslt/1" + +#define XML_STRCMP(a,b) strcmp((char*)a, b) +#define XML_STRLEN(a) strlen((char*)a) + +static const char *zebra_xslt_ns = ZEBRA_SCHEMA_XSLT_NS; + +static void set_param_xml(const char **params, const char *name, + const char *value, ODR odr) +{ + while (*params) + params++; + params[0] = name; + params[1] = value; + params[2] = 0; +} static void set_param_str(const char **params, const char *name, const char *value, ODR odr) @@ -59,47 +93,234 @@ static void set_param_int(const char **params, const char *name, zint value, ODR odr) { char *quoted = odr_malloc(odr, 30); /* 25 digits enough for 2^64 */ + while (*params) + params++; sprintf(quoted, "'" ZINT_FORMAT "'", value); params[0] = name; params[1] = quoted; params[2] = 0; } +#define ENABLE_INPUT_CALLBACK 0 + +#if ENABLE_INPUT_CALLBACK +static int zebra_xmlInputMatchCallback (char const *filename) +{ + yaz_log(YLOG_LOG, "match %s", filename); + return 0; +} -static void *filter_init (Res res, RecType recType) +static void * zebra_xmlInputOpenCallback (char const *filename) { - struct filter_info *tinfo = (struct filter_info *) xmalloc(sizeof(*tinfo)); - tinfo->stylesheet_xsp = 0; + return 0; +} + +static int zebra_xmlInputReadCallback (void * context, char * buffer, int len) +{ + return 0; +} + +static int zebra_xmlInputCloseCallback (void * context) +{ + return 0; +} +#endif + +static void *filter_init(Res res, RecType recType) +{ + struct filter_xslt_info *tinfo + = (struct filter_xslt_info *) xmalloc(sizeof(*tinfo)); tinfo->reader = 0; tinfo->fname = 0; - tinfo->split_depth = 1; + tinfo->full_name = 0; + tinfo->profile_path = 0; + tinfo->split_level = 0; + tinfo->split_path = 0; tinfo->odr = odr_createmem(ODR_ENCODE); + tinfo->doc = 0; + tinfo->schemas = 0; + +#if ENABLE_INPUT_CALLBACK + xmlRegisterDefaultInputCallbacks(); + xmlRegisterInputCallbacks(zebra_xmlInputMatchCallback, + zebra_xmlInputOpenCallback, + zebra_xmlInputReadCallback, + zebra_xmlInputCloseCallback); +#endif return tinfo; } -static void filter_config(void *clientData, Res res, const char *args) +static int attr_content(struct _xmlAttr *attr, const char *name, + const char **dst_content) { - struct filter_info *tinfo = clientData; - if (!args || !*args) - args = "default.xsl"; - if (!tinfo->fname || strcmp(args, tinfo->fname)) + if (!XML_STRCMP(attr->name, name) && attr->children && + attr->children->type == XML_TEXT_NODE) { - /* different filename so must reread stylesheet */ - xfree(tinfo->fname); - tinfo->fname = xstrdup(args); - if (tinfo->stylesheet_xsp) - xsltFreeStylesheet(tinfo->stylesheet_xsp); - tinfo->stylesheet_xsp = - xsltParseStylesheetFile((const xmlChar*) tinfo->fname); + *dst_content = (const char *)(attr->children->content); + return 1; } + return 0; } -static void filter_destroy(void *clientData) +static void destroy_schemas(struct filter_xslt_info *tinfo) { - struct filter_info *tinfo = clientData; - if (tinfo->stylesheet_xsp) - xsltFreeStylesheet(tinfo->stylesheet_xsp); + struct filter_xslt_schema *schema = tinfo->schemas; + while (schema) + { + struct filter_xslt_schema *schema_next = schema->next; + if (schema->stylesheet_xsp) + xsltFreeStylesheet(schema->stylesheet_xsp); + xfree(schema); + schema = schema_next; + } + tinfo->schemas = 0; xfree(tinfo->fname); + if (tinfo->doc) + xmlFreeDoc(tinfo->doc); + tinfo->doc = 0; +} + +static ZEBRA_RES create_schemas(struct filter_xslt_info *tinfo, + const char *fname) +{ + char tmp_full_name[1024]; + xmlNodePtr ptr; + tinfo->fname = xstrdup(fname); + + if (yaz_filepath_resolve(tinfo->fname, tinfo->profile_path, + NULL, tmp_full_name)) + tinfo->full_name = xstrdup(tmp_full_name); + else + tinfo->full_name = xstrdup(tinfo->fname); + + yaz_log(YLOG_LOG, "xslt filter: loading config file %s", tinfo->full_name); + + tinfo->doc = xmlParseFile(tinfo->full_name); + if (!tinfo->doc) { + yaz_log(YLOG_WARN, "xslt filter: could not parse config file %s", + tinfo->full_name); + return ZEBRA_FAIL; + } + + ptr = xmlDocGetRootElement(tinfo->doc); + if (!ptr || ptr->type != XML_ELEMENT_NODE || + XML_STRCMP(ptr->name, "schemaInfo")){ + yaz_log(YLOG_WARN, + "xslt filter: config file %s :" + " expected root element ", + tinfo->full_name); + return ZEBRA_FAIL; + } + + for (ptr = ptr->children; ptr; ptr = ptr->next) + { + if (ptr->type != XML_ELEMENT_NODE) + continue; + if (!XML_STRCMP(ptr->name, "schema")) + { + char tmp_xslt_full_name[1024]; + struct _xmlAttr *attr; + struct filter_xslt_schema *schema = xmalloc(sizeof(*schema)); + schema->name = 0; + schema->identifier = 0; + schema->stylesheet = 0; + schema->default_schema = 0; + schema->next = tinfo->schemas; + schema->stylesheet_xsp = 0; + schema->include_snippet = 0; + tinfo->schemas = schema; + for (attr = ptr->properties; attr; attr = attr->next) + { + attr_content(attr, "identifier", &schema->identifier); + attr_content(attr, "name", &schema->name); + attr_content(attr, "stylesheet", &schema->stylesheet); + attr_content(attr, "default", &schema->default_schema); + attr_content(attr, "snippet", &schema->include_snippet); + } + if (schema->stylesheet){ + yaz_filepath_resolve(schema->stylesheet, tinfo->profile_path, + NULL, tmp_xslt_full_name); + schema->stylesheet_xsp + = xsltParseStylesheetFile((const xmlChar*) tmp_xslt_full_name); + if (!schema->stylesheet_xsp) + yaz_log(YLOG_WARN, + "xslt filter: could not parse xslt stylesheet %s", + tmp_xslt_full_name); + } + + } + else if (!XML_STRCMP(ptr->name, "split")) + { + struct _xmlAttr *attr; + for (attr = ptr->properties; attr; attr = attr->next) + { + attr_content(attr, "level", &tinfo->split_level); + attr_content(attr, "path", &tinfo->split_path); + } + } + else + { + yaz_log(YLOG_WARN, "Bad element %s in %s", ptr->name, fname); + return ZEBRA_FAIL; + } + } + return ZEBRA_OK; +} + +static struct filter_xslt_schema *lookup_schema(struct filter_xslt_info *tinfo, + const char *est) +{ + struct filter_xslt_schema *schema; + for (schema = tinfo->schemas; schema; schema = schema->next) + { + /* find requested schema */ + if (est) + { + if (schema->identifier && !strcmp(schema->identifier, est)) + return schema; + + if (schema->name && !strcmp(schema->name, est)) + return schema; + } + /* or return default schema if defined */ + else if (schema->default_schema) + return schema; + } + + /* return first schema if no default schema defined */ + if (tinfo->schemas) + return tinfo->schemas; + + return 0; +} + +static ZEBRA_RES filter_config(void *clientData, Res res, const char *args) +{ + struct filter_xslt_info *tinfo = clientData; + if (!args || !*args){ + yaz_log(YLOG_WARN, "xslt filter: need config file"); + return ZEBRA_FAIL; + } + + if (tinfo->fname && !strcmp(args, tinfo->fname)) + return ZEBRA_OK; + + tinfo->profile_path + /* = res_get_def(res, "profilePath", DEFAULT_PROFILE_PATH); */ + = res_get(res, "profilePath"); + yaz_log(YLOG_LOG, "xslt filter: profilePath %s", tinfo->profile_path); + + destroy_schemas(tinfo); + create_schemas(tinfo, args); + return ZEBRA_OK; +} + +static void filter_destroy(void *clientData) +{ + struct filter_xslt_info *tinfo = clientData; + destroy_schemas(tinfo); + if (tinfo->reader) + xmlFreeTextReader(tinfo->reader); odr_destroy(tinfo->odr); xfree(tinfo); } @@ -115,64 +336,140 @@ static int ioclose_ex(void *context) return 0; } -static void index_field(struct filter_info *tinfo, struct recExtractCtrl *ctrl, +static void index_cdata(struct filter_xslt_info *tinfo, struct recExtractCtrl *ctrl, xmlNodePtr ptr, RecWord *recWord) { for(; ptr; ptr = ptr->next) { - index_field(tinfo, ctrl, ptr->children, recWord); + index_cdata(tinfo, ctrl, ptr->children, recWord); if (ptr->type != XML_TEXT_NODE) continue; - recWord->term_buf = ptr->content; - recWord->term_len = strlen(ptr->content); + recWord->term_buf = (const char *)ptr->content; + recWord->term_len = XML_STRLEN(ptr->content); (*ctrl->tokenAdd)(recWord); } } -static void index_node(struct filter_info *tinfo, struct recExtractCtrl *ctrl, +static void index_node(struct filter_xslt_info *tinfo, struct recExtractCtrl *ctrl, xmlNodePtr ptr, RecWord *recWord) { for(; ptr; ptr = ptr->next) { index_node(tinfo, ctrl, ptr->children, recWord); if (ptr->type != XML_ELEMENT_NODE || !ptr->ns || - strcmp(ptr->ns->href, zebra_index_ns)) + XML_STRCMP(ptr->ns->href, zebra_xslt_ns)) continue; - if (!strcmp(ptr->name, "index")) + if (!XML_STRCMP(ptr->name, "index")) { - char *field_str = 0; + const char *name_str = 0; + const char *type_str = 0; const char *xpath_str = 0; struct _xmlAttr *attr; for (attr = ptr->properties; attr; attr = attr->next) { - if (!strcmp(attr->name, "field") - && attr->children && attr->children->type == XML_TEXT_NODE) - field_str = attr->children->content; - if (!strcmp(attr->name, "xpath") - && attr->children && attr->children->type == XML_TEXT_NODE) - xpath_str = attr->children->content; + attr_content(attr, "name", &name_str); + attr_content(attr, "xpath", &xpath_str); + attr_content(attr, "type", &type_str); } - if (field_str) + if (name_str) { - recWord->attrStr = field_str; - index_field(tinfo, ctrl, ptr->children, recWord); + int prev_type = recWord->index_type; /* save default type */ + + if (type_str && *type_str) + recWord->index_type = *type_str; /* type was given */ + recWord->index_name = name_str; + index_cdata(tinfo, ctrl, ptr->children, recWord); + + recWord->index_type = prev_type; /* restore it again */ } } } } -static int filter_extract(void *clientData, struct recExtractCtrl *p) +static void index_record(struct filter_xslt_info *tinfo,struct recExtractCtrl *ctrl, + xmlNodePtr ptr, RecWord *recWord) +{ + if (ptr && ptr->type == XML_ELEMENT_NODE && ptr->ns && + !XML_STRCMP(ptr->ns->href, zebra_xslt_ns) + && !XML_STRCMP(ptr->name, "record")) + { + const char *type_str = "update"; + const char *id_str = 0; + const char *rank_str = 0; + struct _xmlAttr *attr; + for (attr = ptr->properties; attr; attr = attr->next) + { + attr_content(attr, "type", &type_str); + attr_content(attr, "id", &id_str); + attr_content(attr, "rank", &rank_str); + } + if (id_str) + sscanf(id_str, "%255s", ctrl->match_criteria); + if (rank_str) + { + ctrl->staticrank = atoi(rank_str); + yaz_log(YLOG_LOG, "rank=%d",ctrl->staticrank); + } + else + yaz_log(YLOG_LOG, "no rank"); + + ptr = ptr->children; + } + index_node(tinfo, ctrl, ptr, recWord); +} + +static int extract_doc(struct filter_xslt_info *tinfo, struct recExtractCtrl *p, + xmlDocPtr doc) { - const char *params[10]; - struct filter_info *tinfo = clientData; RecWord recWord; - int ret; + const char *params[10]; + xmlChar *buf_out; + int len_out; + + struct filter_xslt_schema *schema = lookup_schema(tinfo, zebra_xslt_ns); params[0] = 0; + set_param_str(params, "schema", zebra_xslt_ns, tinfo->odr); - odr_reset(tinfo->odr); - set_param_str(params, "schema", ZEBRA_INDEX_NS, tinfo->odr); + (*p->init)(p, &recWord); + if (schema && schema->stylesheet_xsp) + { + xmlNodePtr root_ptr; + xmlDocPtr resDoc = + xsltApplyStylesheet(schema->stylesheet_xsp, + doc, params); + if (p->flagShowRecords) + { + xmlDocDumpMemory(resDoc, &buf_out, &len_out); + fwrite(buf_out, len_out, 1, stdout); + xmlFree(buf_out); + } + root_ptr = xmlDocGetRootElement(resDoc); + if (root_ptr) + index_record(tinfo, p, root_ptr, &recWord); + else + { + yaz_log(YLOG_WARN, "No root for index XML record." + " split_level=%s stylesheet=%s", + tinfo->split_level, schema->stylesheet); + } + xmlFreeDoc(resDoc); + } + xmlDocDumpMemory(doc, &buf_out, &len_out); + if (p->flagShowRecords) + fwrite(buf_out, len_out, 1, stdout); + (*p->setStoreData)(p, buf_out, len_out); + xmlFree(buf_out); + + xmlFreeDoc(doc); + return RECCTRL_EXTRACT_OK; +} + +static int extract_split(struct filter_xslt_info *tinfo, struct recExtractCtrl *p) +{ + int ret; + int split_depth = 0; if (p->first_record) { if (tinfo->reader) @@ -186,51 +483,24 @@ static int filter_extract(void *clientData, struct recExtractCtrl *p) if (!tinfo->reader) return RECCTRL_EXTRACT_ERROR_GENERIC; - if (!tinfo->stylesheet_xsp) - return RECCTRL_EXTRACT_ERROR_GENERIC; - - (*p->init)(p, &recWord); - recWord.reg_type = 'w'; - + if (tinfo->split_level) + split_depth = atoi(tinfo->split_level); ret = xmlTextReaderRead(tinfo->reader); while (ret == 1) { int type = xmlTextReaderNodeType(tinfo->reader); int depth = xmlTextReaderDepth(tinfo->reader); - if (tinfo->split_depth == 0 || - (type == XML_READER_TYPE_ELEMENT && tinfo->split_depth == depth)) + if (split_depth == 0 || + (split_depth > 0 && + type == XML_READER_TYPE_ELEMENT && split_depth == depth)) { - xmlChar *buf_out; - int len_out; - xmlNodePtr ptr = xmlTextReaderExpand(tinfo->reader); xmlNodePtr ptr2 = xmlCopyNode(ptr, 1); - xmlDocPtr doc = xmlNewDoc("1.0"); + xmlDocPtr doc = xmlNewDoc((const xmlChar*) "1.0"); xmlDocSetRootElement(doc, ptr2); - - if (tinfo->stylesheet_xsp) - { - xmlDocPtr resDoc = - xsltApplyStylesheet(tinfo->stylesheet_xsp, - doc, params); - if (p->flagShowRecords) - { - xmlDocDumpMemory(resDoc, &buf_out, &len_out); - fwrite(buf_out, len_out, 1, stdout); - xmlFree(buf_out); - } - index_node(tinfo, p, xmlDocGetRootElement(resDoc), &recWord); - xmlFreeDoc(resDoc); - } - xmlDocDumpMemory(doc, &buf_out, &len_out); - if (p->flagShowRecords) - fwrite(buf_out, len_out, 1, stdout); - (*p->setStoreData)(p, buf_out, len_out); - xmlFree(buf_out); - xmlFreeDoc(doc); - return RECCTRL_EXTRACT_OK; - } + return extract_doc(tinfo, p, doc); + } ret = xmlTextReaderRead(tinfo->reader); } xmlFreeTextReader(tinfo->reader); @@ -238,6 +508,38 @@ static int filter_extract(void *clientData, struct recExtractCtrl *p) return RECCTRL_EXTRACT_EOF; } +static int extract_full(struct filter_xslt_info *tinfo, struct recExtractCtrl *p) +{ + if (p->first_record) /* only one record per stream */ + { + xmlDocPtr doc = xmlReadIO(ioread_ex, ioclose_ex, p /* I/O handler */, + 0 /* URL */, + 0 /* encoding */, + XML_PARSE_XINCLUDE); + if (!doc) + { + return RECCTRL_EXTRACT_ERROR_GENERIC; + } + return extract_doc(tinfo, p, doc); + } + else + return RECCTRL_EXTRACT_EOF; +} + +static int filter_extract(void *clientData, struct recExtractCtrl *p) +{ + struct filter_xslt_info *tinfo = clientData; + + odr_reset(tinfo->odr); + + if (tinfo->split_level == 0 && tinfo->split_path == 0) + return extract_full(tinfo, p); + else + { + return extract_split(tinfo, p); + } +} + static int ioread_ret(void *context, char *buffer, int len) { struct recRetrieveCtrl *p = context; @@ -249,38 +551,114 @@ static int ioclose_ret(void *context) return 0; } + +static const char *snippet_doc(struct recRetrieveCtrl *p, int text_mode, + int window_size) +{ + const char *xml_doc_str; + int ord = 0; + WRBUF wrbuf = wrbuf_alloc(); + zebra_snippets *res = + zebra_snippets_window(p->doc_snippet, p->hit_snippet, window_size); + zebra_snippet_word *w = zebra_snippets_list(res); + + if (text_mode) + wrbuf_printf(wrbuf, "\'"); + else + wrbuf_printf(wrbuf, "\n", zebra_xslt_ns); + for (; w; w = w->next) + { + if (ord == 0) + ord = w->ord; + else if (ord != w->ord) + + break; + if (text_mode) + wrbuf_printf(wrbuf, "%s%s%s ", + w->match ? "*" : "", + w->term, + w->match ? "*" : ""); + else + { + wrbuf_printf(wrbuf, " ", + w->ord, w->seqno, + (w->match ? "match='1'" : "")); + wrbuf_xmlputs(wrbuf, w->term); + wrbuf_printf(wrbuf, "\n"); + } + } + if (text_mode) + wrbuf_printf(wrbuf, "\'"); + else + wrbuf_printf(wrbuf, "\n"); + + xml_doc_str = odr_strdup(p->odr, wrbuf_buf(wrbuf)); + + zebra_snippets_destroy(res); + wrbuf_free(wrbuf, 1); + return xml_doc_str; +} + static int filter_retrieve (void *clientData, struct recRetrieveCtrl *p) { - const char *esn = ZEBRA_SCHEMA_IDENTITY_NS; - const char *params[10]; - struct filter_info *tinfo = clientData; + const char *esn = 0; + const char *params[32]; + struct filter_xslt_info *tinfo = clientData; xmlDocPtr resDoc; xmlDocPtr doc; + struct filter_xslt_schema *schema; + int window_size = -1; if (p->comp) { - if (p->comp->which != Z_RecordComp_simple - || p->comp->u.simple->which != Z_ElementSetNames_generic) + if (p->comp->which == Z_RecordComp_simple + && p->comp->u.simple->which == Z_ElementSetNames_generic) + { + esn = p->comp->u.simple->u.generic; + } + else if (p->comp->which == Z_RecordComp_complex + && p->comp->u.complex->generic->elementSpec + && p->comp->u.complex->generic->elementSpec->which == + Z_ElementSpec_elementSetName) { - p->diagnostic = YAZ_BIB1_PRESENT_COMP_SPEC_PARAMETER_UNSUPP; - return 0; + esn = p->comp->u.complex->generic->elementSpec->u.elementSetName; } - esn = p->comp->u.simple->u.generic; } - + schema = lookup_schema(tinfo, esn); + if (!schema) + { + p->diagnostic = + YAZ_BIB1_SPECIFIED_ELEMENT_SET_NAME_NOT_VALID_FOR_SPECIFIED_; + return 0; + } + + if (schema->include_snippet) + window_size = atoi(schema->include_snippet); + params[0] = 0; - set_param_str(params, "schema", esn, p->odr); + set_param_int(params, "id", p->localno, p->odr); if (p->fname) set_param_str(params, "filename", p->fname, p->odr); + if (p->staticrank >= 0) + set_param_int(params, "rank", p->staticrank, p->odr); + + if (esn) + set_param_str(params, "schema", esn, p->odr); + else + if (schema->name) + set_param_str(params, "schema", schema->name, p->odr); + else if (schema->identifier) + set_param_str(params, "schema", schema->identifier, p->odr); + else + set_param_str(params, "schema", "", p->odr); + if (p->score >= 0) set_param_int(params, "score", p->score, p->odr); set_param_int(params, "size", p->recordSize, p->odr); - - if (!tinfo->stylesheet_xsp) - { - p->diagnostic = YAZ_BIB1_SYSTEM_ERROR_IN_PRESENTING_RECORDS; - return 0; - } + + if (window_size >= 0) + set_param_xml(params, "snippet", snippet_doc(p, 1, window_size), + p->odr); doc = xmlReadIO(ioread_ret, ioclose_ret, p /* I/O handler */, 0 /* URL */, 0 /* encoding */, @@ -291,11 +669,18 @@ static int filter_retrieve (void *clientData, struct recRetrieveCtrl *p) return 0; } - if (!strcmp(esn, ZEBRA_SCHEMA_IDENTITY_NS)) + if (window_size >= 0) + { + xmlNodePtr node = xmlDocGetRootElement(doc); + const char *snippet_str = snippet_doc(p, 0, window_size); + xmlDocPtr snippet_doc = xmlParseMemory(snippet_str, strlen(snippet_str)); + xmlAddChild(node, xmlDocGetRootElement(snippet_doc)); + } + if (!schema->stylesheet_xsp) resDoc = doc; else { - resDoc = xsltApplyStylesheet(tinfo->stylesheet_xsp, + resDoc = xsltApplyStylesheet(schema->stylesheet_xsp, doc, params); xmlFreeDoc(doc); } @@ -358,3 +743,11 @@ idzebra_filter &filter_type, 0, }; +/* + * Local variables: + * c-basic-offset: 4 + * indent-tabs-mode: nil + * End: + * vim: shiftwidth=4 tabstop=8 expandtab + */ +