X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=c%2B%2B0x%2Fxml-template.cpp;h=febece21a0b70a2b2bcfd923b350bba2fbedc9e1;hb=29f0ad69120092c93021e37f764777d5ccd84d3e;hp=7278236d911d6b50d4a1236a97fd6c8e5df8df7e;hpb=d8b19f8693d9521d2ccb6ef3c036d2cacc83a9ab;p=xml-template

diff --git a/c++0x/xml-template.cpp b/c++0x/xml-template.cpp
index 7278236..febece2 100644
--- a/c++0x/xml-template.cpp
+++ b/c++0x/xml-template.cpp
@@ -5,77 +5,167 @@
 
 using namespace std;
 
+namespace {
+
+void clean_node(xmlNode *node)
+{
+	if (node->type != XML_ELEMENT_NODE) {
+		return;
+	}
+	if (strcmp(reinterpret_cast<const char *>(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) {}
 
 void Replace::process(xmlNode *node, bool clean) {
-	node->children = xmlNewTextLen(reinterpret_cast<const xmlChar *>(str.data()), str.size());
+	xmlNodeSetContentLen(node, reinterpret_cast<const xmlChar *>(str.data()), str.size());
+	if (clean) {
+		clean_node(node);
+	}
 }
 
-Clone::Clone(const std::vector<Directive *> &subdirectives)
+string Replace::get_contents() { return str; }
+
+Clone::Clone(const vector<Directive *> &subdirectives)
 	: subdirectives(subdirectives) {}
+	
+Clone::Clone(initializer_list<Directive *> subdirectives)
+	: subdirectives(subdirectives) {}
+
+Clone::~Clone()
+{
+	for (auto it : subdirectives) {
+		delete it;
+	}
+}
 
-void Clone::process(xmlNode *node, bool clean) {
+void Clone::process(xmlNode *node, bool clean)
+{
 	// We can't use xmlNewDocFragment, since xmlDOMWrapCloneNode only knows
 	// how to clone elements.
 	vector<xmlNode *> new_nodes;
 
 	for (auto it : subdirectives) {
-		xmlDOMWrapCtxt ctx;
 		xmlNode *new_node;
-		int ret = xmlDOMWrapCloneNode(NULL, node->doc, node, &new_node, node->doc, NULL, 1, 0);
+		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);
 	}
 
-	node->children = NULL;
+	xmlFreeNodeList(node->children);
+	node->content = NULL;
+	node->children = node->last = NULL;
+
 	for (auto child : new_nodes) {
 		xmlAddChild(node, child);
 	}	
+	if (clean) {
+		clean_node(node);
+	}
 }
 
 Substitute::Substitute(const unordered_map<string, Directive*> &substitution_map)
 	: substitution_map(substitution_map) {}
+	
+Substitute::Substitute(initializer_list<pair<const string, Directive*>> substitution_map)
+	: substitution_map(substitution_map) {}
+
+Substitute::~Substitute()
+{
+	for (auto it : substitution_map) {
+		delete it.second;
+	}
+}
 
-void Substitute::process(xmlNode *node, bool clean) {
-	for (xmlNode *child = node->children; child != NULL; child = child->next) {
+void Substitute::process(xmlNode *node, bool clean)
+{
+	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<const char *>(attr->ns->href), "http://template.sesse.net/") == 0 &&
 				    strcmp(reinterpret_cast<const char *>(attr->name), "id") == 0) {
-					id = reinterpret_cast<const char *>(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<const char *>(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<const char *>(child->name) ||
+				string tag = reinterpret_cast<const char *>(child->name);
+
+				// Attribute substitution.
+				if (begins_with(it.first, tag + "/")) {
+					const xmlChar *attr_key = reinterpret_cast<const xmlChar *>(
+						it.first.substr(tag.size() + 1).c_str());
+					const xmlChar *attr_value = reinterpret_cast<const xmlChar *>(
+						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<const xmlChar *>(
+						it.first.substr(tag.size() + 2).c_str());
+					const xmlChar *attr_value = reinterpret_cast<const xmlChar *>(
+						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);
 		}
 	}
+	if (clean) {
+		clean_node(node);
+	}
 }
 	
 void process_file(const string &input_filename,
@@ -87,6 +177,7 @@ void process_file(const string &input_filename,
 	xmlDocPtr doc = xmlParseFile(input_filename.c_str());
 	root_directive->process(xmlDocGetRootElement(doc), true);
 	xmlSaveFile(output_filename.c_str(), doc);
+	xmlFreeDoc(doc);
 
 	xmlCleanupParser();
 	xmlMemoryDump();