1 files changed, 64 insertions, 0 deletions
diff --git a/Swiften/LinkLocal/DNSSD/Avahi/AvahiBrowseQuery.h b/Swiften/LinkLocal/DNSSD/Avahi/AvahiBrowseQuery.h
new file mode 100644
index 0000000..229f97e
--- /dev/null
+++ b/Swiften/LinkLocal/DNSSD/Avahi/AvahiBrowseQuery.h
@@ -0,0 +1,64 @@
+#pragma once
+
+#include <boost/bind.hpp>
+
+#include "Swiften/LinkLocal/DNSSD/Avahi/AvahiQuery.h"
+#include "Swiften/LinkLocal/DNSSD/DNSSDBrowseQuery.h"
+#include "Swiften/EventLoop/MainEventLoop.h"
+
+namespace Swift {
+	class AvahiQuerier;
+
+	class AvahiBrowseQuery : public DNSSDBrowseQuery, public AvahiQuery {
+		public:	
+			AvahiBrowseQuery(boost::shared_ptr<AvahiQuerier> q) : AvahiQuery(q) {
+			}
+
+			void startBrowsing() {
+				std::cout << "Start browsing" << std::endl;
+				avahi_threaded_poll_lock(querier->getThreadedPoll());
+				std::cout << "Creating browser" << std::endl;
+				AvahiServiceBrowser* browser = avahi_service_browser_new(querier->getClient(), AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, "_presence._tcp", NULL, (AvahiLookupFlags) 0, &handleServiceDiscoveredStatic, this);
+				if (!browser) {
+					std::cout << "Error" << std::endl;
+					MainEventLoop::postEvent(boost::bind(boost::ref(onError)), shared_from_this());
+				}
+				std::cout << "Unlocking" << std::endl;
+				avahi_threaded_poll_unlock(querier->getThreadedPoll());
+				std::cout << "Browse started" << std::endl;
+			}
+
+			void stopBrowsing() {
+				// TODO
+			}
+
+		private:
+			static void handleServiceDiscoveredStatic(AvahiServiceBrowser *b, AvahiIfIndex interfaceIndex, AvahiProtocol protocol, AvahiBrowserEvent event, const char *name, const char *type, const char *domain, AvahiLookupResultFlags flags, void* context) {
+				static_cast<AvahiBrowseQuery*>(context)->handleServiceDiscovered(b, interfaceIndex, protocol, event, name, type, domain, flags);
+			}
+
+			void handleServiceDiscovered(AvahiServiceBrowser *, AvahiIfIndex interfaceIndex, AvahiProtocol, AvahiBrowserEvent event, const char *name, const char *type, const char *domain, AvahiLookupResultFlags) {
+				switch (event) {
+					case AVAHI_BROWSER_FAILURE:	
+						std::cout << "Service browse error" << std::endl;
+						MainEventLoop::postEvent(boost::bind(boost::ref(onError)), shared_from_this());
+						break;
+					case AVAHI_BROWSER_NEW: {
+						DNSSDServiceID service(name, domain, type, interfaceIndex);
+						std::cout << "Service discovered " << name << " " << type << " " << domain << std::endl;
+						MainEventLoop::postEvent(boost::bind(boost::ref(onServiceAdded), service), shared_from_this());
+						break;
+					}
+					case AVAHI_BROWSER_REMOVE: {
+						std::cout << "Service went away " << name << " " << type << " " << domain << std::endl;
+						DNSSDServiceID service(name, domain, type, interfaceIndex);
+						MainEventLoop::postEvent(boost::bind(boost::ref(onServiceRemoved), service), shared_from_this());
+						break;
+					}
+					case AVAHI_BROWSER_ALL_FOR_NOW:
+					case AVAHI_BROWSER_CACHE_EXHAUSTED:
+						break;
+				}
+			}
+	};
+}