Client.hpp

Go to the documentation of this file.

#ifndef _SERVICE_DISCOVERY_AVAHICLIENT_H_
#define _SERVICE_DISCOVERY_AVAHICLIENT_H_

#define DEFAULT_THREAD_POLL

#ifdef DEFAULT_THREAD_POLL
#define DEFAULT_POLL ThreadPoll
#else
#define DEFAULT_POLL SimplePoll
#endif

#include <avahi-client/client.h>
#include <sigc++/sigc++.h>

#include <string>
#include <iostream>
#include <cassert>
#include <vector>

#include <avahi-common/error.h>
#include <avahi-common/thread-watch.h>
#include <base-logging/Singleton.hpp>

namespace servicediscovery {
namespace avahi {

class ClientObserver
{
public:
    // Client events
    enum EventType { UNKNOWN, DISCONNECTED, RECOVERED };

    // Event structure
    struct Event 
    {
        EventType type;
    };

    virtual ~ClientObserver() {}

    virtual void update(const Event& event) = 0;

    virtual void stop() = 0;
};

class Client : public base::Singleton<Client> 
{
        friend class base::Singleton<Client>;
        friend class UniqueClientLock;


protected:
        Client();

private:
        // Internal modes in order to handle reconnection to the avahi-daemon
        enum Mode { INIT, NORMAL, RECOVERY };

        static AvahiClient* msAvahiClient;

        static AvahiThreadedPoll* msPoll; 

        std::vector<ClientObserver*> msObservers;

        Mode mMode;
        
        static void stateUpdateCallback(AvahiClient* client, AvahiClientState state, void* userdata);

        void start(bool runningDaemonRequired);

        void cleanup();

public:
        virtual ~Client();

        static AvahiClient* getAvahiClient();

        static AvahiClientState getAvahiClientState();

        void addObserver(ClientObserver* clientObserver);

        void removeObserver(ClientObserver* clientObserver);

protected:
        static void lock();

        static void unlock();
};

class UniqueClientLock
{
    public: 
        UniqueClientLock();

        ~UniqueClientLock();
};

} // end namespace avahi
} // end namespace servicediscovery
#endif /* _SERVICE_DISCOVERY_AVAHICLIENT_H_ */