#pragma once

#include <map>
#include <boost/enable_shared_from_this.hpp>
#include <avahi-client/client.h>
#include <avahi-client/lookup.h>
#include <avahi-common/thread-watch.h>
#include <avahi-common/watch.h>
#include <avahi-common/malloc.h>
#include <avahi-common/error.h>


#include "Swiften/EventLoop/EventOwner.h"
#include "Swiften/LinkLocal/DNSSDService.h"

namespace Swift {
	class AvahiDNSSDService : public DNSSDService, public EventOwner, public boost::enable_shared_from_this<AvahiDNSSDService> {
		public:
			AvahiDNSSDService();
			~AvahiDNSSDService();

			virtual void start();
			virtual void stop();

			virtual void registerService(const String& name, int port, const LinkLocalServiceInfo&);
			virtual void updateService(const LinkLocalServiceInfo&);
			virtual void unregisterService();

			virtual void startResolvingService(const DNSSDServiceID&);
			virtual void stopResolvingService(const DNSSDServiceID&);

			virtual void resolveHostname(const String& hostname, int interfaceIndex = 0);
		
		private:
			static void handleServiceDiscoveredGlobal(AvahiServiceBrowser *b, AvahiIfIndex networkInterface, AvahiProtocol protocol, AvahiBrowserEvent event, const char *name, const char *type, const char *domain, AvahiLookupResultFlags flags, void* userdata) {
				static_cast<AvahiDNSSDService*>(userdata)->handleServiceDiscovered(b, networkInterface, protocol, event, name, type, domain, flags);
			}
			void handleServiceDiscovered(AvahiServiceBrowser *b, AvahiIfIndex interface, AvahiProtocol protocol, AvahiBrowserEvent event, const char *name, const char *type, const char *domain, AVAHI_GCC_UNUSED AvahiLookupResultFlags flags);

			static void handleServiceResolvedGlobal(AvahiServiceResolver *r, AvahiIfIndex interfaceIndex, AvahiProtocol protocol, AvahiResolverEvent event, const char *name, const char *type, const char *domain, const char *hostname, const AvahiAddress *address, uint16_t port, AvahiStringList *txt, AvahiLookupResultFlags flags, void* userdata) {
				static_cast<AvahiDNSSDService*>(userdata)->handleServiceResolved(r, interfaceIndex, protocol, event, name, type, domain, hostname, address, port, txt, flags);
			}
			void handleServiceResolved(AvahiServiceResolver *r, AvahiIfIndex interface, AvahiProtocol protocol, AvahiResolverEvent event, const char *name, const char *type, const char *domain, const char *hostname, const AvahiAddress *address, uint16_t port, AvahiStringList *txt, AvahiLookupResultFlags flags);


		private:
			AvahiClient* client;
			AvahiThreadedPoll* threadedPoll;
			AvahiServiceBrowser* serviceBrowser;
			typedef std::map<DNSSDServiceID, AvahiServiceResolver*> ServiceResolverMap;
			ServiceResolverMap serviceResolvers;
			typedef std::map<String, HostAddress> HostnameAddressMap;
			HostnameAddressMap hostnameAddresses;
	};
}