X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=c%2B%2B0x%2Fxml-template.cpp;h=5bbcb9b5dbeec7079a1b436118fc3f7a858eaaeb;hb=6d6f69fcdf45c3afc82212c5c3d8085c93ec4717;hp=c2b62cd77ecbb6da8441e938809bb4042120c453;hpb=83a41af3311ec494205205ec093ffff0aad56fd0;p=xml-template diff --git a/c++0x/xml-template.cpp b/c++0x/xml-template.cpp index c2b62cd..5bbcb9b 100644 --- a/c++0x/xml-template.cpp +++ b/c++0x/xml-template.cpp @@ -12,17 +12,29 @@ void clean_node(xmlNode *node) if (node->type != XML_ELEMENT_NODE) { return; } - if (strcmp(reinterpret_cast(node->ns->href), "http://template.sesse.net/") == 0) { + if (node->ns != NULL && + strcmp(reinterpret_cast(node->ns->href), "http://template.sesse.net/") == 0) { xmlNode *frag = xmlNewDocFragment(node->doc); xmlReplaceNode(node, frag); frag->children = node->children; frag->last = node->last; + + node->children = node->last = NULL; + xmlFreeNode(node); } } +// Does A begin with B? +bool begins_with(const string &a, const string &b) +{ + return (a.compare(0, b.size(), b) == 0); +} + } // namespace Directive::~Directive() {} + +string Directive::get_contents() { return ""; } Replace::Replace(const string &str) : str(str) {} @@ -34,9 +46,51 @@ void Replace::process(xmlNode *node, bool clean) { } } +string Replace::get_contents() { return str; } + +ReplaceInclude::ReplaceInclude(xmlNodePtr included_node) + : included_node(included_node), + included_doc(NULL) {} + +ReplaceInclude::ReplaceInclude(xmlDocPtr included_doc) + : included_node(xmlDocGetRootElement(included_doc)), + included_doc(included_doc) {} + +ReplaceInclude::~ReplaceInclude() +{ + if (included_doc != NULL) { + xmlFreeDoc(included_doc); + } else { + xmlFreeNode(included_node); + } +} + +void ReplaceInclude::process(xmlNode *node, bool clean) { + xmlFreeNodeList(node->children); + node->content = NULL; + node->children = node->last = NULL; + + xmlNode *new_node; + xmlDOMWrapCloneNode(NULL, NULL, included_node, &new_node, node->doc, node, 1, 0); + xmlAddChild(node, new_node); + + if (clean) { + Substitute dummy_subs {}; + dummy_subs.process(new_node, clean); + clean_node(node); + } +} + Clone::Clone(const vector &subdirectives) : subdirectives(subdirectives) {} - + +Clone::Clone(const vector &subdirectives_subs) +{ + for (auto it : subdirectives_subs) { + subdirectives.push_back(static_cast(it)); + } +} + Clone::Clone(initializer_list subdirectives) : subdirectives(subdirectives) {} @@ -54,12 +108,18 @@ void Clone::process(xmlNode *node, bool clean) vector new_nodes; for (auto it : subdirectives) { + if (it == NULL) { + continue; + } xmlNode *new_node; xmlDOMWrapCloneNode(NULL, node->doc, node, &new_node, node->doc, NULL, 1, 0); it->process(new_node, clean); - for (xmlNode *child = new_node->children; child != NULL; child = child->next) { + while (new_node->children != NULL) { + xmlNode *child = new_node->children; + xmlUnlinkNode(child); new_nodes.push_back(child); } + xmlFreeNode(new_node); } xmlFreeNodeList(node->children); @@ -73,6 +133,23 @@ void Clone::process(xmlNode *node, bool clean) clean_node(node); } } + +Alternate::Alternate(const string &attribute, + const vector &subdirectives_subs, + const vector &alternatives) + : Clone(subdirectives_subs) +{ + unsigned jx = 0; + for (unsigned ix = 0; ix < subdirectives_subs.size(); ++ix) { + if (subdirectives_subs[ix] == NULL) { + continue; + } + string value = alternatives[jx++ % alternatives.size()]; + subdirectives_subs[ix]->substitution_map.insert(make_pair( + attribute, + new Replace { value })); + } +} Substitute::Substitute(const unordered_map &substitution_map) : substitution_map(substitution_map) {} @@ -89,38 +166,61 @@ Substitute::~Substitute() void Substitute::process(xmlNode *node, bool clean) { - for (xmlNode *child = node->children; child != NULL; child = child->next) { + xmlNode *next_child; + for (xmlNode *child = node->children; child != NULL; child = next_child) { + next_child = child->next; bool processed = false; if (child->type == XML_ELEMENT_NODE) { // Find the ID, if any. string id; + xmlAttr *id_attr = NULL; for (xmlAttr *attr = child->properties; attr != NULL; attr = attr->next) { - if (strcmp(reinterpret_cast(attr->ns->href), "http://template.sesse.net/") == 0 && + if (attr->ns != NULL && + strcmp(reinterpret_cast(attr->ns->href), "http://template.sesse.net/") == 0 && strcmp(reinterpret_cast(attr->name), "id") == 0) { - id = reinterpret_cast(xmlNodeGetContent(attr->children)); - - if (clean) { - if (attr->prev == NULL) { - child->properties = attr->next; - } else { - attr->prev->next = attr->next; - } - } + xmlChar *id_buf = xmlNodeGetContent(attr->children); + id = reinterpret_cast(id_buf); + xmlFree(id_buf); + id_attr = attr; } } + if (clean && id_attr != NULL) { + xmlRemoveProp(id_attr); + } // Check all substitutions to see if we found anything appropriate. for (auto it : substitution_map) { - if (it.first == reinterpret_cast(child->name) || + string tag = reinterpret_cast(child->name); + + // Attribute substitution. + if (begins_with(it.first, tag + "/")) { + const xmlChar *attr_key = reinterpret_cast( + it.first.substr(tag.size() + 1).c_str()); + const xmlChar *attr_value = reinterpret_cast( + it.second->get_contents().c_str()); + xmlSetProp(child, attr_key, attr_value); + } else if ((!id.empty() && begins_with(it.first, "#" + id + "/"))) { + const xmlChar *attr_key = reinterpret_cast( + it.first.substr(tag.size() + 2).c_str()); + const xmlChar *attr_value = reinterpret_cast( + it.second->get_contents().c_str()); + xmlSetProp(child, attr_key, attr_value); + } + + if (processed) { + continue; + } + + // Regular substitution. + if (it.first == tag || (!id.empty() && it.first == ("#" + id))) { it.second->process(child, clean); processed = true; - break; } } } - + if (!processed) { process(child, clean); } @@ -130,17 +230,22 @@ void Substitute::process(xmlNode *node, bool clean) } } -void process_file(const string &input_filename, - const string &output_filename, - Directive *root_directive) +xmlDocPtr process_file(const string &input_filename, + Directive *root_directive, + bool clean) { LIBXML_TEST_VERSION xmlDocPtr doc = xmlParseFile(input_filename.c_str()); - root_directive->process(xmlDocGetRootElement(doc), true); - xmlSaveFile(output_filename.c_str(), doc); - xmlFreeDoc(doc); - + root_directive->process(xmlDocGetRootElement(doc), clean); xmlCleanupParser(); xmlMemoryDump(); + return doc; +} + +void output_to_fd_and_free(xmlDocPtr doc, int fd) +{ + xmlOutputBufferPtr buf = xmlOutputBufferCreateFd(fd, NULL); + xmlSaveFileTo(buf, doc, NULL); + xmlFreeDoc(doc); }