1 files changed, 5 insertions, 2 deletions
diff --git a/Swiften/Disco/CapsInfoGenerator.h b/Swiften/Disco/CapsInfoGenerator.h
index d1b2663..17a01dd 100644
--- a/Swiften/Disco/CapsInfoGenerator.h
+++ b/Swiften/Disco/CapsInfoGenerator.h
@@ -8,12 +8,14 @@
 
 #include <string>
+#include <Swiften/Base/API.h>
 #include <Swiften/Elements/CapsInfo.h>
 
 namespace Swift {
 	class DiscoInfo;
+	class CryptoProvider;
 
-	class CapsInfoGenerator {
+	class SWIFTEN_API CapsInfoGenerator {
 		public:
-			CapsInfoGenerator(const std::string& node);
+			CapsInfoGenerator(const std::string& node, CryptoProvider* crypto);
 
 			CapsInfo generateCapsInfo(const DiscoInfo& discoInfo) const;
@@ -21,4 +23,5 @@ namespace Swift {
 		private:
 			std::string node_;
+			CryptoProvider* crypto_;
 	};
 }