ServiceList.cc

Go to the documentation of this file.

// **************************************************************************
// **************************************************************************
#include "ServiceList.h"

#include <sstream>

#include "WindowLog.h"
#include "Converter.h"

#include "tools.h"
#include "Time.h"

using namespace std;

// --------------------------------------------------------------------------
//
//
ServiceList::ServiceList(const char *type, ostream &out) :
    wout(out), fDimServers("DIS_DNS/SERVER_LIST", const_cast<char*>(""), this),
    fType(type), fHandler(0)
{
}

// --------------------------------------------------------------------------
//
//
ServiceList::ServiceList(ostream &out) :
    wout(out), fDimServers("DIS_DNS/SERVER_LIST", const_cast<char*>(""), this),
    fType(""), fHandler(0)
{
}

// --------------------------------------------------------------------------
//
//
ServiceList::~ServiceList()
{
    for (ServerMap::iterator i=fServerList.begin(); i!=fServerList.end(); i++)
    {
            delete i->second.first;
            delete i->second.second;
    }
}

// --------------------------------------------------------------------------
//
//
void ServiceList::infoHandler()
{
    if (getInfo()==&fDimServers)
        ProcessServerList();
    else
        ProcessServiceList(*getInfo());

    if (fHandler)
    {
        fHandler->itsService = 0;
        fHandler->infoHandler();
    }
}

// --------------------------------------------------------------------------
//
DimInfo *ServiceList::CreateDimInfo(const string &str, const string &svc) const
{
    return new DimInfo((str+'/'+svc).c_str(),
                       const_cast<char*>(""),
                       const_cast<ServiceList*>(this));
}

// --------------------------------------------------------------------------
//
void ServiceList::ProcessServerList()
{
    // Get the received string from the handler
    const string str = fDimServers.getString();

    // Check if it starts with + or -
    if (str[0]!='-' && str[0]!='+')
    {
        // If it doesn't start with + or - remove all existing servers
        // we have received a full server list
        for (ServerMap::iterator i=fServerList.begin(); i!=fServerList.end(); i++)
        {
            delete i->second.first;
            delete i->second.second;

            ServiceMap::iterator x = fServiceList.find(i->first);
            fServiceList.erase(x);
            //wout << "Delete: " << i->first << endl;
        }

        fServerList.clear();
    }

    // Create a stringstream to tokenize the received string
    stringstream stream(str);

    // Loop over the seperating tokens
    string buffer;
    while (getline(stream, buffer, '|'))
    {
        // The first part before the first @ is the server name
        const string server = buffer.substr(0, buffer.find_first_of('@'));
        if (server.empty())
            continue;

        // If it starts with a - we have to remove an entry
        if (server[0]=='-')
        {
            const string trunc = server.substr(1);

            // Check if this server is not found in the list.
            // This should never happen if Dim works reliable
            const ServerMap::iterator v = fServerList.find(trunc);
            if (v==fServerList.end())
            {
                wout << kRed << "Server '" << trunc << "' not in list as it ought to be." << endl;
                continue;
            }

            // Remove the server from the server list
            delete v->second.first;
            delete v->second.second;
            fServerList.erase(v);

            // Remove the server from the command list
            ServiceMap::iterator w = fServiceList.find(trunc);
            fServiceList.erase(w);

            wout << " -> " << Time().GetAsStr() << " - " << trunc << "/SERVICE_LIST: Disconnected." << endl;
            //wout << "Remove: " << server << endl;
            continue;
        }

        // If it starts with a + we have to add an entry
        if (server[0]=='+')
        {
            const string trunc = server.substr(1);

            // Check if this server is already in the list.
            // This should never happen if Dim works reliable
            const ServerMap::iterator v = fServerList.find(trunc);
            if (v!=fServerList.end())
            {
                wout << kRed << "Server '" << trunc << "' in list not as it ought to be." << endl;
                continue;
            }

            // Add the new server to the server list
            fServerList[trunc] = make_pair(CreateSL(trunc), CreateFMT(trunc));

            wout << " -> " << Time().GetAsStr() << " - " << trunc << "/SERVICE_LIST: Connected." << endl;
            //wout << "Add   : " << server << endl;
            continue;
        }

        // In any other case we just add the entry to the list
        fServerList[server] = make_pair(CreateSL(server), CreateFMT(server));
        //wout << "Add  0: " << server << endl;
    }
}

// --------------------------------------------------------------------------
//
void ServiceList::ProcessServiceList(DimInfo &info)
{
    const string str = info.getString();
    if (str.empty())
        return;

    // Get the name of the service
    string buffer = info.getName();

    if (buffer.find("SERVICE_DESC")!=buffer.length()-12)
    {
        // Get the server name from the service name
        const string server = buffer.substr(0, buffer.find_first_of('/'));

        // Initialize the entry with an empty list
        if (str[0]!='+')
            fServiceList[server] = vector<string>();

        // For easy and fast access get the corresponding reference
        vector<string> &list = fServiceList[server];

        // Tokenize the stream into lines
        stringstream stream(str);
        while (getline(stream, buffer, '\n'))
        {
            if (buffer.empty())
                continue;

            // Get the type and compare it with fType
            const string type = buffer.substr(buffer.find_last_of('|')+1);
            if (type!=fType)
                continue;

            // Get format, name and command name
            const string fmt  = buffer.substr(buffer.find_first_of('|')+1, buffer.find_last_of('|')-buffer.find_first_of('|')-1);
            const string name = buffer.substr(buffer.find_first_of('/')+1, buffer.find_first_of('|')-buffer.find_first_of('/')-1);
            const string cmd  = buffer.substr(0, buffer.find_first_of('|'));

            // Add name the the list
            list.push_back(name);

            // Add format to the list
            fFormatList[cmd] = fmt;
        }
    }
    else
    {
        fDescriptionMap.clear();

        stringstream stream(str);
        while (getline(stream, buffer, '\n'))
        {
            if (buffer.empty())
                continue;

            const vector<Description> v = Description::SplitDescription(buffer);

            const string svc = v[0].name;

            fDescriptionMap[svc]  = v[0].comment;
            fDescriptionList[svc] = vector<Description>(v.begin()+1, v.end());
        }
    }
}

// --------------------------------------------------------------------------
//
//
vector<string> ServiceList::GetServerList() const
{
    vector<string> v;
    for (ServerMap::const_iterator i=fServerList.begin(); i!=fServerList.end(); i++)
        v.push_back(i->first);

    return v;
}

vector<string> ServiceList::GetServiceList(const std::string &server) const
{
    const ServiceMap::const_iterator m = fServiceList.find(server);
    return m==end() ? vector<string>() : m->second;
}

vector<string> ServiceList::GetServiceList() const
{
    vector<string> vec;
    for (ServerMap::const_iterator i=fServerList.begin(); i!=fServerList.end(); i++)
    {
        const string server = i->first;

        const vector<string> v = GetServiceList(server);

        for (vector<string>::const_iterator s=v.begin(); s<v.end(); s++)
            vec.push_back(server+"/"+*s);
    }

    return vec;
}

// --------------------------------------------------------------------------
//
//
string ServiceList::GetFormat(const string &service) const
{
    const StringMap::const_iterator i = fFormatList.find(service);
    return i==fFormatList.end() ? "" : i->second;
}

// --------------------------------------------------------------------------
//
//
string ServiceList::GetFormat(const string &server, const string &name) const
{
    return GetFormat(server+'/'+name);
}

// --------------------------------------------------------------------------
//
//
vector<Description> ServiceList::GetDescriptions(const string &service) const
{
    const DescriptionMap::const_iterator i = fDescriptionList.find(service);
    return i==fDescriptionList.end() ? vector<Description>() : i->second;
}

// --------------------------------------------------------------------------
//
//
vector<Description> ServiceList::GetDescriptions(const string &server, const string &name) const
{
    return GetDescriptions(server+'/'+name);
}

// --------------------------------------------------------------------------
//
//
string ServiceList::GetComment(const string &service) const
{
    const StringMap::const_iterator i = fDescriptionMap.find(service);
    return i==fDescriptionMap.end() ? "" : i->second;
}

// --------------------------------------------------------------------------
//
//
string ServiceList::GetComment(const string &server, const string &name) const
{
    return GetComment(server+"/"+name);
}

// --------------------------------------------------------------------------
//
//
bool ServiceList::HasServer(const string &name) const
{
    return fServiceList.find(name)!=end();
}

// --------------------------------------------------------------------------
//
//
bool ServiceList::HasService(const string &server, const string &service) const
{
    ServiceMap::const_iterator v = fServiceList.find(server);
    if (v==end())
        return false;

    const vector<string> &w = v->second;
    return find(w.begin(), w.end(), service)!=w.end();
}

bool ServiceList::HasService(const string &svc) const
{
    const size_t p = svc.find_first_of('/');
    if (p==string::npos)
        return false;

    return HasService(svc.substr(0, p), svc.substr(p+1));
}

// --------------------------------------------------------------------------
//
//
vector<string>::const_iterator ServiceList::begin(const string &server) const
{
    ServiceMap::const_iterator i = fServiceList.find(server);
    if (i==end())
        return vector<string>().end();

    return i->second.begin();
}

// --------------------------------------------------------------------------
//
//
vector<string>::const_iterator ServiceList::end(const string &server) const
{
    ServiceMap::const_iterator i = fServiceList.find(server);
    if (i==end())
        return vector<string>().end();

    return i->second.end();
}


// --------------------------------------------------------------------------
//
void ServiceList::PrintServerList(ostream &out) const
{
    out << endl << kBold << "ServerList:" << endl;

    for (ServerMap::const_iterator i=fServerList.begin(); i!=fServerList.end(); i++)
    {
        const string &server = i->first;
        DimInfo *ptr1 = i->second.first;
        DimInfo *ptr2 = i->second.second;

        out << " " << server << " " << ptr1->getName() << "|" << ptr2->getName() << endl;
    }
    out << endl;
}

// --------------------------------------------------------------------------
//
void ServiceList::PrintServiceList(ostream &out) const
{
    out << endl << kBold << "ServiceList:" << endl;

    for (ServiceMap::const_iterator i=fServiceList.begin(); i!=fServiceList.end(); i++)
    {
        const string &server = i->first;
        const vector<string> &lst = i->second;

        out << " " << server << endl;

        for (vector<string>::const_iterator j=lst.begin(); j!=lst.end(); j++)
            out << "  " << *j << " [" << GetFormat(server, *j) << "]" << endl;
    }
    out << endl;
}

// --------------------------------------------------------------------------
//
//
int ServiceList::PrintDescription(std::ostream &out, const string &serv, const string &service) const
{
    int rc = 0;
    for (ServiceMap::const_iterator i=fServiceList.begin(); i!=fServiceList.end(); i++)
    {
        const string &server = i->first;

        if (!serv.empty() && server!=serv)
            continue;

        out << kRed << "----- " << server << " -----" << endl;

        const vector<string> &lst = i->second;
        for (vector<string>::const_iterator s=lst.begin(); s!=lst.end(); s++)
        {
            if (!service.empty() && *s!=service)
                continue;

            rc++;

            out << " " << *s;

            const string fmt = GetFormat(server, *s);
            if (!fmt.empty())
                out << '[' << fmt << ']';

            const string svc = server + '/' + *s;

            const DescriptionMap::const_iterator v = fDescriptionList.find(svc);
            if (v==fDescriptionList.end())
            {
                out << endl;
                continue;
            }

            for (vector<Description>::const_iterator j=v->second.begin();
                 j!=v->second.end(); j++)
                out << " <" << j->name << ">";
            out << endl;

            const StringMap::const_iterator d = fDescriptionMap.find(svc);
            if (d!=fDescriptionMap.end() && !d->second.empty())
                out << "    " << d->second << endl;

            for (vector<Description>::const_iterator j=v->second.begin();
                 j!=v->second.end(); j++)
            {
                out << "    " << kGreen << j->name;
                if (!j->comment.empty())
                    out << kReset << ": " << kBlue << j->comment;
                if (!j->unit.empty())
                    out << kYellow << " [" << j->unit << "]";
                out << endl;
            }
        }
        out << endl;
    }

    return rc;
}

// --------------------------------------------------------------------------
//
void ServiceList::DumpServiceList(ostream &out)
{
    DimCurrentInfo info1("DIS_DNS/SERVER_LIST", const_cast<char*>(""));

    stringstream stream(info1.getString());

    string buffer;
    while (getline(stream, buffer, '|'))
    {
        const string server = buffer.substr(0, buffer.find_first_of('@'));
        if (server.empty())
            continue;

        out << kBold << " " << server << endl;

        DimCurrentInfo info2((server+"/SERVICE_LIST").c_str(), const_cast<char*>(""));

        string buffer2;

        stringstream stream2(info2.getString());
        while (getline(stream2, buffer2, '\n'))
        {
            if (buffer2.empty())
                continue;

            out << "  " << buffer2 << endl;
        }
    }
}

// --------------------------------------------------------------------------
//
bool ServiceList::SendDimCommand(ostream &lout, const string &server, const string &str) const
{
    // Find the delimiter between the command name and the data
    size_t p0 = str.find_first_of(' ');
    if (p0==string::npos)
        p0 = str.length();

    // Get just the command name separated from the data
    const string name = str.substr(0, p0);

    // Compile the command which will be sent to the state-machine
    const string cmd = server + '/' + name;

    if (!HasService(server, name))
    {
        lout << kRed << "Unkown command '" << cmd << "'" << endl;
        return false;
    }

    // Get the format of the event data
    const string fmt = GetFormat(cmd);

    // Convert the user entered data according to the format string
    // into a data block which will be attached to the event
    const Converter conv(lout, fmt, false);
    if (!conv)
    {
        lout << kRed << "Couldn't properly parse the format... ignored." << endl;
        return false;
    }

    try
    {
        lout << kBlue << cmd;
        const vector<char> v = conv.GetVector(str.substr(p0));
        lout << endl;

        const int rc = DimClient::sendCommand(cmd.c_str(), (void*)v.data(), v.size());
        if (rc)
            lout << kGreen << "Command " << cmd << " emitted successfully to DimClient." << endl;
        else
            lout << kRed << "ERROR - Sending command " << cmd << " failed." << endl;
    }
    catch (const std::runtime_error &e)
    {
        lout << endl << kRed << e.what() << endl;
        return false;
    }

    return true;
}