1 files changed, 68 insertions, 0 deletions
diff --git a/Swiften/LinkLocal/LinkLocalServiceBrowser.h b/Swiften/LinkLocal/LinkLocalServiceBrowser.h
new file mode 100644
index 0000000..66973d5
--- /dev/null
+++ b/Swiften/LinkLocal/LinkLocalServiceBrowser.h
@@ -0,0 +1,68 @@
+#pragma once
+
+#include <boost/signal.hpp>
+#include <boost/shared_ptr.hpp>
+#include <boost/optional.hpp>
+#include <map>
+#include <vector>
+
+#include "Swiften/Base/String.h"
+#include "Swiften/LinkLocal/DNSSD/DNSSDQuerier.h"
+#include "Swiften/LinkLocal/DNSSD/DNSSDResolveServiceQuery.h"
+#include "Swiften/LinkLocal/DNSSD/DNSSDRegisterQuery.h"
+#include "Swiften/LinkLocal/DNSSD/DNSSDServiceID.h"
+#include "Swiften/LinkLocal/LinkLocalService.h"
+#include "Swiften/LinkLocal/LinkLocalServiceInfo.h"
+
+namespace Swift {
+	class LinkLocalServiceBrowser {
+		public:
+			LinkLocalServiceBrowser(boost::shared_ptr<DNSSDQuerier> querier);
+			~LinkLocalServiceBrowser();
+
+			void start();
+			void stop();
+			bool isRunning() const;
+			bool hasError() const;
+
+			void registerService(
+					const String& name, 
+					int port, 
+					const LinkLocalServiceInfo& info = LinkLocalServiceInfo());
+			void updateService(
+					const LinkLocalServiceInfo& info = LinkLocalServiceInfo());
+			void unregisterService();
+			bool isRegistered() const;
+
+			std::vector<LinkLocalService> getServices() const;
+
+			// FIXME: Ugly that we need this
+			boost::shared_ptr<DNSSDQuerier> getQuerier() const {
+				return querier;
+			}
+
+			boost::signal<void (const LinkLocalService&)> onServiceAdded;
+			boost::signal<void (const LinkLocalService&)> onServiceChanged;
+			boost::signal<void (const LinkLocalService&)> onServiceRemoved;
+			boost::signal<void (const DNSSDServiceID&)> onServiceRegistered;
+			boost::signal<void (bool)> onStopped;
+
+		private:
+			void handleServiceAdded(const DNSSDServiceID&);
+			void handleServiceRemoved(const DNSSDServiceID&);
+			void handleServiceResolved(const DNSSDServiceID& service, const boost::optional<DNSSDResolveServiceQuery::Result>& result);
+			void handleRegisterFinished(const boost::optional<DNSSDServiceID>&);
+			void handleBrowseError();
+
+		private:
+			boost::shared_ptr<DNSSDQuerier> querier;
+			boost::optional<DNSSDServiceID> selfService;
+			boost::shared_ptr<DNSSDBrowseQuery> browseQuery;
+			boost::shared_ptr<DNSSDRegisterQuery> registerQuery;
+			typedef std::map<DNSSDServiceID, boost::shared_ptr<DNSSDResolveServiceQuery> > ResolveQueryMap;
+			ResolveQueryMap resolveQueries;
+			typedef std::map<DNSSDServiceID, DNSSDResolveServiceQuery::Result> ServiceMap;
+			ServiceMap services;
+			bool haveError;
+	};
+}