X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=c%2B%2B0x%2Fxml-template.cpp;h=c2b62cd77ecbb6da8441e938809bb4042120c453;hb=83a41af3311ec494205205ec093ffff0aad56fd0;hp=1e727613160acf49fb1ec860b88179b0fd7aecb5;hpb=e772f92e8c2124293715855f5f05ecba7d407ec2;p=xml-template

diff --git a/c++0x/xml-template.cpp b/c++0x/xml-template.cpp
index 1e72761..c2b62cd 100644
--- a/c++0x/xml-template.cpp
+++ b/c++0x/xml-template.cpp
@@ -16,25 +16,39 @@ void clean_node(xmlNode *node)
 		xmlNode *frag = xmlNewDocFragment(node->doc);
 		xmlReplaceNode(node, frag);
 		frag->children = node->children;
+		frag->last = node->last;
 	}
 }
 
 }  // namespace
 
+Directive::~Directive() {}
+
 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)
+Clone::Clone(const vector<Directive *> &subdirectives)
+	: subdirectives(subdirectives) {}
+	
+Clone::Clone(initializer_list<Directive *> subdirectives)
 	: subdirectives(subdirectives) {}
 
-void Clone::process(xmlNode *node, bool clean) {
+Clone::~Clone()
+{
+	for (auto it : subdirectives) {
+		delete it;
+	}
+}
+
+void Clone::process(xmlNode *node, bool clean)
+{
 	// We can't use xmlNewDocFragment, since xmlDOMWrapCloneNode only knows
 	// how to clone elements.
 	vector<xmlNode *> new_nodes;
@@ -48,7 +62,10 @@ void Clone::process(xmlNode *node, bool clean) {
 		}
 	}
 
-	node->children = NULL;
+	xmlFreeNodeList(node->children);
+	node->content = NULL;
+	node->children = node->last = NULL;
+
 	for (auto child : new_nodes) {
 		xmlAddChild(node, child);
 	}	
@@ -59,8 +76,19 @@ void Clone::process(xmlNode *node, bool clean) {
 
 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) {
+void Substitute::process(xmlNode *node, bool clean)
+{
 	for (xmlNode *child = node->children; child != NULL; child = child->next) {
 		bool processed = false;
 
@@ -111,6 +139,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();