3 files changed, 90 insertions, 0 deletions
diff --git a/Swiften/Disco/CapsInfoGenerator.cpp b/Swiften/Disco/CapsInfoGenerator.cpp
new file mode 100644
index 0000000..8d9e407
--- /dev/null
+++ b/Swiften/Disco/CapsInfoGenerator.cpp
@@ -0,0 +1,34 @@
+#include "Swiften/Disco/CapsInfoGenerator.h"
+
+#include <algorithm>
+
+#include "Swiften/Base/foreach.h"
+#include "Swiften/Elements/DiscoInfo.h"
+#include "Swiften/StringCodecs/SHA1.h"
+#include "Swiften/StringCodecs/Base64.h"
+
+namespace Swift {
+
+CapsInfoGenerator::CapsInfoGenerator(const String& node) : node_(node) {
+}
+
+CapsInfo CapsInfoGenerator::generateCapsInfo(const DiscoInfo& discoInfo) const {
+	String serializedCaps;
+
+	std::vector<DiscoInfo::Identity> identities(discoInfo.getIdentities());
+	std::sort(identities.begin(), identities.end());
+	foreach (const DiscoInfo::Identity& identity, identities) {
+		serializedCaps += identity.getCategory() + "/" + identity.getType() + "/" + identity.getLanguage() + "/" + identity.getName() + "<";
+	}
+
+	std::vector<String> features(discoInfo.getFeatures());
+	std::sort(features.begin(), features.end());
+	foreach (const String& feature, features) {
+		serializedCaps += feature + "<";
+	}
+
+	String version(Base64::encode(SHA1::getHash(serializedCaps)));
+	return CapsInfo(node_, version, "sha-1");
+}
+
+}
diff --git a/Swiften/Disco/CapsInfoGenerator.h b/Swiften/Disco/CapsInfoGenerator.h
new file mode 100644
index 0000000..66949ab
--- /dev/null
+++ b/Swiften/Disco/CapsInfoGenerator.h
@@ -0,0 +1,21 @@
+#ifndef SWIFTEN_CapsInfoGenerator_H
+#define SWIFTEN_CapsInfoGenerator_H
+
+#include "Swiften/Base/String.h"
+#include "Swiften/Elements/CapsInfo.h"
+
+namespace Swift {
+	class DiscoInfo;
+
+	class CapsInfoGenerator {
+		public:
+			CapsInfoGenerator(const String& node);
+
+			CapsInfo generateCapsInfo(const DiscoInfo& discoInfo) const;
+
+		private:
+			String node_;
+	};
+}
+
+#endif
diff --git a/Swiften/Disco/UnitTest/CapsInfoGeneratorTest.cpp b/Swiften/Disco/UnitTest/CapsInfoGeneratorTest.cpp
new file mode 100644
index 0000000..94b9913
--- /dev/null
+++ b/Swiften/Disco/UnitTest/CapsInfoGeneratorTest.cpp
@@ -0,0 +1,35 @@
+#include <cppunit/extensions/HelperMacros.h>
+#include <cppunit/extensions/TestFactoryRegistry.h>
+
+#include "Swiften/Elements/DiscoInfo.h"
+#include "Swiften/Disco/CapsInfoGenerator.h"
+
+using namespace Swift;
+
+class CapsInfoGeneratorTest : public CppUnit::TestFixture
+{
+		CPPUNIT_TEST_SUITE(CapsInfoGeneratorTest);
+		CPPUNIT_TEST(testGenerate_XEP0115SimpleExample);
+		CPPUNIT_TEST_SUITE_END();
+
+	public:
+		CapsInfoGeneratorTest() {}
+
+		void testGenerate_XEP0115SimpleExample() {
+			DiscoInfo discoInfo;
+			discoInfo.addIdentity(DiscoInfo::Identity("Exodus 0.9.1", "client", "pc"));
+			discoInfo.addFeature("http://jabber.org/protocol/disco#items");
+			discoInfo.addFeature("http://jabber.org/protocol/caps");
+			discoInfo.addFeature("http://jabber.org/protocol/disco#info");
+			discoInfo.addFeature("http://jabber.org/protocol/muc");
+
+			CapsInfoGenerator testling("http://code.google.com/p/exodus");
+			CapsInfo result = testling.generateCapsInfo(discoInfo);
+
+			CPPUNIT_ASSERT_EQUAL(String("http://code.google.com/p/exodus"), result.getNode());
+			CPPUNIT_ASSERT_EQUAL(String("sha-1"), result.getHash());
+			CPPUNIT_ASSERT_EQUAL(String("QgayPKawpkPSDYmwT/WM94uAlu0="), result.getVersion());
+		}
+};
+
+CPPUNIT_TEST_SUITE_REGISTRATION(CapsInfoGeneratorTest);