1 files changed, 53 insertions, 0 deletions
diff --git a/Swiften/Parser/GenericPayloadTreeParser.h b/Swiften/Parser/GenericPayloadTreeParser.h
new file mode 100644
index 0000000..df6d022
--- /dev/null
+++ b/Swiften/Parser/GenericPayloadTreeParser.h
@@ -0,0 +1,53 @@
+/*
+ * Copyright (c) 2011 Kevin Smith
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#pragma once
+
+#include <deque>
+
+#include <boost/shared_ptr.hpp>
+#include <boost/smart_ptr/make_shared.hpp>
+
+#include <Swiften/Parser/GenericPayloadParser.h>
+#include <Swiften/Parser/Tree/ParserElement.h>
+
+namespace Swift {
+	/**
+	 * Generic parser offering something a bit like a DOM to work with.
+	 */
+	template<typename PAYLOAD_TYPE>
+	class GenericPayloadTreeParser : public GenericPayloadParser<PAYLOAD_TYPE> {
+		public:
+			virtual void handleStartElement(const std::string& element, const std::string& xmlns, const AttributeMap& attributes) {
+				if (!root_) {
+					root_ = boost::make_shared<ParserElement>(element, xmlns, attributes);
+					elementStack_.push_back(root_);
+				} 
+				else {
+					ParserElement::ref current = *elementStack_.rbegin();
+					elementStack_.push_back(current->addChild(element, xmlns, attributes));
+				}
+			}
+
+			virtual void handleEndElement(const std::string& /*element*/, const std::string&) {
+				elementStack_.pop_back();
+				if (elementStack_.empty()) {
+					handleTree(root_);
+				}
+			}
+
+			virtual void handleCharacterData(const std::string& data) {
+				ParserElement::ref current = *elementStack_.rbegin();
+				current->appendCharacterData(data);
+			}
+
+			virtual void handleTree(ParserElement::ref root) = 0;
+
+		private:
+			std::deque<ParserElement::ref> elementStack_;
+			ParserElement::ref root_;
+	};
+}