util/ConfigManager.cpp

Go to the documentation of this file.

#include "ConfigManager.h"
#include "Error.h"
#include <fstream>
#include <iostream>
#include "utils.h"

ConfigManager::ConfigManager(std::string filename) :
    m_filename(filename)
{
}

ConfigSection &ConfigManager::getSection(std::string name)
{
    SectionMap::iterator it = m_sections.find(name);
    if (it == m_sections.end()) {
        std::string errorMsg = "getSection failed: Config section '";
        errorMsg += name;
        errorMsg += "' does not exists!";
        throw Error(errorMsg.c_str());
    }
    return it->second;
}

ConfigSection &ConfigManager::getSectionSilent(std::string name)
{
    SectionMap::iterator it = m_sections.find(name);
    if (it == m_sections.end()) {
        addEmptySection(name);
        return getSection(name);
    }
    return it->second;
}

void ConfigManager::setSection(std::string name, ConfigSection &section)
{
    SectionMap::iterator it = m_sections.find(name);
    if (it != m_sections.end()) {
        it->second = section;
    } else {
        m_sections.insert(std::pair<std::string, ConfigSection>(name, section));
    }
}

void ConfigManager::addEmptySection(std::string name)
{
    SectionMap::iterator it = m_sections.find(name);
    if (it != m_sections.end()) {
        std::string errorMsg = "addEmptySection failed: Config section '";
        errorMsg += name;
        errorMsg += "' already exists!";
        throw Error(errorMsg.c_str());
    } else {
        ConfigSection dummy;
        m_sections.insert(std::pair<std::string, ConfigSection>(name, dummy));
    }
}

void ConfigManager::save()
{
    std::ofstream outFile(m_filename.c_str());
    SectionMap::iterator it;
    for (it = m_sections.begin(); it != m_sections.end(); ++it) {
        outFile << "[" << it->first << "]" << std::endl;
        it->second.save(outFile);
        outFile << "[/" << it->first << "]" << std::endl << std::endl;
    }
}

void ConfigManager::load()
{ 
    std::ifstream inFile(m_filename.c_str());
    if (!inFile.good()) {
        std::string msg("ConfigManager::load(): cannot open file ");
        msg += m_filename;
        throw Error(msg.c_str());
    }
    std::string sectionString, sectionName, endSection, line;
    bool inSection = false;
    while (std::getline(inFile, line)) {
        if (!inSection && (line.find('[') != std::string::npos) && 
                (line.find(']') != std::string::npos)) {
            // begin section and extract section name
            inSection = true;
            sectionName = line;
            sectionName = sectionName.substr(1, sectionName.find(']')-1);
            endSection = "[/" + sectionName + "]";
        } else if (inSection) {
            if (line.find(endSection) != std::string::npos) {
                // section ended - add to db
                ConfigSection tmpSection;
                tmpSection.load(sectionString);
                setSection(sectionName, tmpSection);
                inSection = false;
            } else {
                // read information until section ends
                sectionString += line + "\n";
            }
        }
    }
    if (inSection) {
        std::string errMsg("Config file ");
        errMsg += m_filename;
        errMsg += " is malformated:\nSection ";
        errMsg += sectionName + " is not closed!";
        throw Error(errMsg.c_str());
    }
}

---