1 files changed, 116 insertions, 0 deletions
diff --git a/Swiften/JID/JID.cpp b/Swiften/JID/JID.cpp
new file mode 100644
index 0000000..dcd6dd1
--- /dev/null
+++ b/Swiften/JID/JID.cpp
@@ -0,0 +1,116 @@
+#include <stringprep.h>
+#include <vector>
+#include <iostream>
+
+#include "Swiften/JID/JID.h"
+
+namespace Swift {
+
+
+class StringPrepper {
+	private:
+		static const int MAX_STRINGPREP_SIZE = 1024;
+
+	public:
+		static String getNamePrepped(const String& name) {
+			return getStringPrepped(name, stringprep_nameprep);
+		}
+
+		static String getNodePrepped(const String& node) {
+			return getStringPrepped(node, stringprep_xmpp_nodeprep);
+		}
+
+		static String getResourcePrepped(const String& resource) {
+			return getStringPrepped(resource, stringprep_xmpp_resourceprep);
+		}
+
+		static String getStringPrepped(const String& s, const Stringprep_profile profile[]) {
+			std::vector<char> input(s.getUTF8String().begin(), s.getUTF8String().end());
+			input.resize(MAX_STRINGPREP_SIZE);
+			if (stringprep(&input[0], MAX_STRINGPREP_SIZE, static_cast<Stringprep_profile_flags>(0), profile) == 0) {
+				return String(&input[0]);
+			}
+			else {
+				return "";
+			}
+		}
+};
+
+
+JID::JID(const char* jid) {
+	initializeFromString(String(jid));
+}
+
+JID::JID(const String& jid) {
+	initializeFromString(jid);
+}
+
+JID::JID(const String& node, const String& domain) : hasResource_(false) {
+	nameprepAndSetComponents(node, domain, "");
+}
+
+JID::JID(const String& node, const String& domain, const String& resource) : hasResource_(true) {
+	nameprepAndSetComponents(node, domain, resource);
+}
+
+void JID::initializeFromString(const String& jid) {
+	if (jid.beginsWith('@')) {
+		return;
+	}
+
+	String bare, resource;
+	size_t slashIndex = jid.find('/');
+	if (slashIndex != jid.npos()) {
+		hasResource_ = true;
+		bare = jid.getSubstring(0, slashIndex);
+		resource = jid.getSubstring(slashIndex + 1, jid.npos());
+	}
+	else {
+		hasResource_ = false;
+		bare = jid;
+	}
+	std::pair<String,String> nodeAndDomain = bare.getSplittedAtFirst('@');
+	if (nodeAndDomain.second.isEmpty()) {
+		nameprepAndSetComponents("", nodeAndDomain.first, resource);
+	}
+	else {
+		nameprepAndSetComponents(nodeAndDomain.first, nodeAndDomain.second, resource);
+	}
+}
+
+
+void JID::nameprepAndSetComponents(const String& node, const String& domain, const String& resource) {
+	node_ = StringPrepper::getNamePrepped(node);
+	domain_ = StringPrepper::getNodePrepped(domain);
+	resource_ = StringPrepper::getResourcePrepped(resource);
+}
+
+String JID::toString() const {
+	String string;
+	if (!node_.isEmpty()) {
+		string += node_ + "@";
+	}
+	string += domain_;
+	if (!isBare()) {
+		string += "/" + resource_;
+	}
+	return string;
+}
+
+int JID::compare(const Swift::JID& o, CompareType compareType) const {
+	if (node_ < o.node_) { return -1; }
+	if (node_ > o.node_) { return 1; }
+	if (domain_ < o.domain_) { return -1; }
+	if (domain_ > o.domain_) { return 1; }
+	if (compareType == WithResource) {
+		if (hasResource_ != o.hasResource_) {
+			return hasResource_ ? 1 : -1;
+		}
+		if (resource_ < o.resource_) { return -1; }
+		if (resource_ > o.resource_) { return 1; }
+	}
+	return 0;
+}
+
+} // namespace Swift
+