TokenMapLoader.cpp

Go to the documentation of this file.

#include "robot_instance/TokenMapLoader.h"

using namespace std;
using namespace log4cpp;

void TokenMapLoader::loadElements(RobotInstancePtr instance, TokenMapPtr tokenMap)
{
    FileMap::const_iterator itr;

    tokenMap->clear();

    if (not instance->masterTokenFile.empty())
    {
        Private::loadElementsFromFile(instance->configurationSafetyBasePath + RobotInstance::TOKEN_PATH + instance->masterTokenFile, "", tokenMap);
    }

    for (itr = instance->nodeTokenMap.begin(); itr != instance->nodeTokenMap.end(); ++itr)
    {
        //cout << "Key: " << (*itr).first << " Value: " << instance->configurationSafetyBasePath + RobotInstance::TOKEN_PATH << (*itr).second << endl;
        if (not (*itr).second.empty())
        {
            Private::loadElementsFromFile(instance->configurationSafetyBasePath + RobotInstance::TOKEN_PATH + (*itr).second, (*itr).first, tokenMap);
        }
    }
}

void TokenMapLoader::Private::loadElementsFromFile(const string& filename, const string& namePrefix, TokenMapPtr tokenMap)
{
    TiXmlDocument file(filename.c_str());
    bool loadOkay = file.LoadFile();

    if (!loadOkay)
    {
        stringstream err;
        err << "Failed to load file [" << filename << "]";
        NasaCommonLogging::Logger::log("gov.nasa.robonet.TokenMapLoader", Priority::FATAL, err.str());
        throw runtime_error(err.str());
    }

    TiXmlHandle doc(&file);
    NasaCommonLogging::Logger::log("gov.nasa.robonet.TokenMapLoader", Priority::INFO, "TokenFile [" + filename + "] successfully loaded.");

    string elementString;

    // Check for TokenData
    if (!(doc.FirstChildElement("TokenData").ToElement()))
    {
        stringstream err;
        err << "The file " << filename << " has no element named [TokenData]";
        NasaCommonLogging::Logger::log("gov.nasa.robonet.TokenMapLoader", Priority::ERROR, err.str());
        throw runtime_error(err.str());
    }

    // Check for Channels
    TiXmlHandle tokensElement(doc.FirstChildElement("TokenData").FirstChildElement("Tokens"));

    if (tokensElement.ToElement())
    {
        TiXmlHandle currentTokenElement = 0;

        // Iterate over channels
        currentTokenElement = tokensElement.FirstChild("Token");

        while (currentTokenElement.ToElement())
        {
            elementString.clear();
            elementString << *(currentTokenElement.ToElement());
            elementString.append("\n");

            std::pair<TokenMap::iterator, bool> ret = tokenMap->insert(TokenFactory::fromXml(elementString, namePrefix));

            if (not ret.second)
            {
                NasaCommonLogging::Logger::log("gov.nasa.robonet.TokenMapLoader", Priority::NOTICE, "Duplicate token added: " + ret.first->first);
            }

            currentTokenElement = currentTokenElement.ToElement()->NextSiblingElement("Token");
        }
    }
    else
    {
        stringstream err;
        err << "The file " << filename << " has no element named [Tokens]";
        NasaCommonLogging::Logger::log("gov.nasa.robonet.TokenMapLoader", Priority::ERROR, err.str());
        throw runtime_error(err.str());
    }
}