Implement attribute handling in C++0x.

[xml-template] / c++0x / xml-template.cpp

diff --git a/c++0x/xml-template.cpp b/c++0x/xml-template.cpp
index c1d43edb421bada23f539026b60f60577d6f5124..febece21a0b70a2b2bcfd923b350bba2fbedc9e1 100644 (file)
--- a/c++0x/xml-template.cpp
+++ b/c++0x/xml-template.cpp
@@ -23,9 +23,17 @@ void clean_node(xmlNode *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) {}
@@ -37,6 +45,8 @@ void Replace::process(xmlNode *node, bool clean) {
        }
 }
 
+string Replace::get_contents() { return str; }
+
 Clone::Clone(const vector<Directive *> &subdirectives)
        : subdirectives(subdirectives) {}
        
@@ -119,11 +129,32 @@ void Substitute::process(xmlNode *node, bool clean)
 
                        // 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;
                                }
                        }
                }