1 files changed, 61 insertions, 0 deletions
diff --git a/Swiften/Parser/StreamFeaturesParser.cpp b/Swiften/Parser/StreamFeaturesParser.cpp
new file mode 100644
index 0000000..5072e7c
--- /dev/null
+++ b/Swiften/Parser/StreamFeaturesParser.cpp
@@ -0,0 +1,61 @@
+#include "Swiften/Parser/StreamFeaturesParser.h"
+
+namespace Swift {
+
+StreamFeaturesParser::StreamFeaturesParser() : GenericElementParser<StreamFeatures>(), currentDepth_(0), inMechanisms_(false), inMechanism_(false), inCompression_(false), inCompressionMethod_(false) {
+}
+
+void StreamFeaturesParser::handleStartElement(const String& element, const String& ns, const AttributeMap&) {
+	if (currentDepth_ == 1) {
+		if (element == "starttls" && ns == "urn:ietf:params:xml:ns:xmpp-tls") {
+			getElementGeneric()->setHasStartTLS();
+		}
+		else if (element == "session" && ns == "urn:ietf:params:xml:ns:xmpp-session") {
+			getElementGeneric()->setHasSession();
+		}
+		else if (element == "bind" && ns == "urn:ietf:params:xml:ns:xmpp-bind") {
+			getElementGeneric()->setHasResourceBind();
+		}
+		else if (element == "mechanisms" && ns == "urn:ietf:params:xml:ns:xmpp-sasl") {
+			inMechanisms_ = true;
+		}
+		else if (element == "compression" && ns == "http://jabber.org/features/compress") {
+			inCompression_ = true;
+		}
+	}
+	else if (currentDepth_ == 2) {
+		if (inCompression_ && element == "method") {
+			inCompressionMethod_ = true;
+			currentText_ = "";
+		}
+		else if (inMechanisms_ && element == "mechanism") {
+			inMechanism_ = true;
+			currentText_ = "";
+		}
+	}
+	++currentDepth_;
+}
+
+void StreamFeaturesParser::handleEndElement(const String&, const String&) {
+	--currentDepth_;
+	if (currentDepth_ == 1) {
+		inCompression_ = false;
+		inMechanisms_ = false;
+	}
+	else if (currentDepth_ == 2) {
+		if (inCompressionMethod_) {
+			getElementGeneric()->addCompressionMethod(currentText_);
+			inCompressionMethod_ = false;
+		}
+		else if (inMechanism_) {
+			getElementGeneric()->addAuthenticationMechanism(currentText_);
+			inMechanism_ = false;
+		}
+	}
+}
+
+void StreamFeaturesParser::handleCharacterData(const String& data) {
+	currentText_ += data;
+}
+
+}