source: trunk/FACT++/src/ServiceList.h@ 10290

#ifndef FACT_ServiceList
#define FACT_ServiceList

#include <map>
#include <vector>
#include <string>
#include <iostream>

#include "dic.hxx"
#include "dis.hxx"

class ServiceList : public DimClient
{
public:
    typedef std::map<const std::string, DimInfo*>                 ServerMap;
    typedef std::map<const std::string, std::vector<std::string>> ServiceMap;
    typedef std::map<const std::string, std::string>              FormatMap;

private:
    std::ostream &wout;        /// stream for redirection of the output

    DimInfo fDimServers;       /// A DimInfo to retrieve the SERVER_LIST from teh DNS server

    const std::string fType;   /// A filter for the type of the services to be collected, e.g. CMD

    ServerMap  fServerList;   /// A map storing server names and a DimInfo for their SERVICE_LIST
    ServiceMap fServiceList;  /// A map storing server names and vector with all their available commands
    FormatMap  fFormatList;   /// A map storing all commands and their format strings

    DimInfoHandler *fHandler;  /// A callback to signal updates

    DimInfo *CreateDimInfo(const std::string &str) const;

    void ProcessServerList();
    void ProcessServiceList(DimInfo &info);

    void infoHandler();

public:
    ServiceList(const char *type, std::ostream &out=std::cout);
    ServiceList(std::ostream &out=std::cout);
    ~ServiceList();

    void SetHandler(DimInfoHandler *handler=0) { fHandler=handler; }

    std::vector<std::string> GetServerList() const;
    std::vector<std::string> GetServiceList(const std::string &server) const;

    bool HasServer(const std::string &name) const;
    bool HasService(const std::string &server, const std::string &service) const;

    ServiceMap::const_iterator begin() const { return fServiceList.begin(); }
    ServiceMap::const_iterator end() const  { return fServiceList.end(); }

    std::vector<std::string>::const_iterator begin(const std::string &server) const;
    std::vector<std::string>::const_iterator end(const std::string &server) const;

    std::string GetFormat(const std::string &server, const std::string &name) const;
    std::string GetFormat(const std::string &service) const;

    void PrintServerList(std::ostream &out) const;
    void PrintServiceList(std::ostream &out) const;
    void PrintServerList() const { PrintServerList(wout); }
    void PrintServiceList() const { PrintServiceList(wout); }

    static void DumpServiceList(std::ostream &out);
    void DumpServiceList() const { DumpServiceList(wout); }

    bool SendDimCommand(std::ostream &lout, const std::string &server, const std::string &str) const;
    bool SendDimCommand(const std::string &server, const std::string &str) const
    {
        return SendDimCommand(std::cout, server, str);
    }
};

#endif