1 files changed, 144 insertions, 0 deletions
diff --git a/SwifTools/URIHandler/XMPPURI.cpp b/SwifTools/URIHandler/XMPPURI.cpp
new file mode 100644
index 0000000..d9fde07
--- /dev/null
+++ b/SwifTools/URIHandler/XMPPURI.cpp
@@ -0,0 +1,144 @@
+/*
+ * Copyright (c) 2011 Remko Tronçon
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#include <SwifTools/URIHandler/XMPPURI.h>
+
+#include <boost/algorithm/string/predicate.hpp>
+#include <boost/algorithm/string/find_format.hpp>
+#include <boost/algorithm/string/formatter.hpp>
+#include <boost/algorithm/string/find_iterator.hpp>
+#include <boost/algorithm/string/split.hpp>
+#include <boost/algorithm/string/classification.hpp>
+#include <sstream>
+#include <stdexcept>
+#include <vector>
+
+using namespace Swift;
+
+namespace {
+
+	struct PercentEncodedCharacterFinder {
+		template<typename Iterator>
+		boost::iterator_range<Iterator> operator()(Iterator begin, Iterator end) {
+			boost::iterator_range<Iterator> r = boost::first_finder("%")(begin, end);
+			if (r.end() == end) {
+				return r;
+			}
+			else {
+				if (r.end() + 1 == end || r.end() + 2 == end) {
+					throw std::runtime_error("Incomplete escape character");
+				}
+				else {
+					r.advance_end(2);
+					return r;
+				}
+			}
+		}
+	};
+
+	struct PercentUnencodeFormatter {
+		template<typename FindResult>
+		std::string operator()(const FindResult& match) const {
+			std::stringstream s;
+			s << std::hex << std::string(match.begin() + 1, match.end());
+			unsigned int value;
+			s >> value;
+			if (s.fail() || s.bad()) {
+				throw std::runtime_error("Invalid escape character");
+			}
+			return std::string(reinterpret_cast<const char*>(&value), 1);
+		}
+	};
+	
+
+	std::string unescape(const std::string& s) {
+		try {
+			return boost::find_format_all_copy(s, PercentEncodedCharacterFinder(), PercentUnencodeFormatter());
+		}
+		catch (const std::exception&) {
+			return "";
+		}
+	}
+}
+
+XMPPURI::XMPPURI() {
+}
+
+XMPPURI XMPPURI::fromString(const std::string& s) {
+	XMPPURI result;
+	if (boost::starts_with(s, "xmpp:")) {
+		std::string uri = s.substr(5, s.npos);
+		bool parsePath = true;
+		bool parseQuery = true;
+		bool parseFragment = true;
+
+		// Parse authority
+		if (boost::starts_with(uri, "//")) {
+			size_t i = uri.find_first_of("/#?", 2);
+			result.setAuthority(JID(unescape(uri.substr(2, i - 2))));
+			if (i == uri.npos) {
+				uri = "";
+				parsePath = parseQuery = parseFragment = false;
+			}
+			else {
+				if (uri[i] == '?') {
+					parsePath = false;
+				}
+				else if (uri[i] == '#') {
+					parseQuery = parsePath = false;
+				}
+				uri = uri.substr(i + 1, uri.npos);
+			}
+		}
+
+		// Parse path
+		if (parsePath) {
+			size_t i = uri.find_first_of("#?");
+			result.setPath(JID(unescape(uri.substr(0, i))));
+			if (i == uri.npos) {
+				uri = "";
+				parseQuery = parseFragment = false;
+			}
+			else {
+				if (uri[i] == '#') {
+					parseQuery = false;
+				}
+				uri = uri.substr(i + 1, uri.npos);
+			}
+		}
+
+		// Parse query
+		if (parseQuery) {
+			size_t end = uri.find_first_of("#");
+			std::string query = uri.substr(0, end);
+			bool haveType = false;
+			typedef boost::split_iterator<std::string::iterator> split_iterator;
+	    for (split_iterator it = boost::make_split_iterator(query, boost::first_finder(";")); it != split_iterator(); ++it) {
+	    	if (haveType) {
+	    		std::vector<std::string> keyValue;
+	    		boost::split(keyValue, *it, boost::is_any_of("="));
+	    		if (keyValue.size() == 1) {
+	    			result.addQueryParameter(unescape(keyValue[0]), "");
+	    		}
+	    		else if (keyValue.size() >= 2) {
+	    			result.addQueryParameter(unescape(keyValue[0]), unescape(keyValue[1]));
+	    		}
+	    	}
+	    	else {
+	    		result.setQueryType(unescape(boost::copy_range<std::string>(*it)));
+	    		haveType = true;
+	    	}
+	    }
+	    uri = (end == uri.npos ? "" : uri.substr(end + 1, uri.npos));
+		}
+
+		// Parse fragment
+		if (parseFragment) {
+			result.setFragment(unescape(uri));
+		}
+	}
+	return result;
+}