X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=c%2B%2B0x%2Fxml-template.cpp;h=5bbcb9b5dbeec7079a1b436118fc3f7a858eaaeb;hb=6d6f69fcdf45c3afc82212c5c3d8085c93ec4717;hp=fda97fc7ca97b83e28b8b4ab1f3d35558cd3011f;hpb=4e40bd0f58047ceec9e5b162b16f54226bd45977;p=xml-template

diff --git a/c++0x/xml-template.cpp b/c++0x/xml-template.cpp
index fda97fc..5bbcb9b 100644
--- a/c++0x/xml-template.cpp
+++ b/c++0x/xml-template.cpp
@@ -12,7 +12,8 @@ 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) {
+	if (node->ns != NULL &&
+	    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;
@@ -47,9 +48,49 @@ 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<Directive *> &subdirectives)
 	: subdirectives(subdirectives) {}
-	
+
+Clone::Clone(const vector<Substitute *> &subdirectives_subs)
+{
+	for (auto it : subdirectives_subs) {
+		subdirectives.push_back(static_cast<Directive *>(it));
+	}
+}
+
 Clone::Clone(initializer_list<Directive *> subdirectives)
 	: subdirectives(subdirectives) {}
 
@@ -67,6 +108,9 @@ void Clone::process(xmlNode *node, bool clean)
 	vector<xmlNode *> 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);
@@ -89,6 +133,23 @@ void Clone::process(xmlNode *node, bool clean)
 		clean_node(node);
 	}
 }
+	
+Alternate::Alternate(const string &attribute,
+	             const vector<Substitute *> &subdirectives_subs,
+	             const vector<string> &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<string, Directive*> &substitution_map)
 	: substitution_map(substitution_map) {}
@@ -169,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);
 }