DumperFactory.cpp

Go to the documentation of this file.

/* 
 * Confdump-Agent - Dump static and runtime system configuration
 * Copyright (C) 2009-2012  Straton IT, SAS
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 3 as
 * published by the Free Software Foundation.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "boost/foreach.hpp"

#include "Confdump/Dumper.hpp"
#include "Confdump/DumperSpecification.hpp"
#include "Confdump/DumperFactory.hpp"
#include "Confdump/RuntimeError.hpp"

#if CD_UNIX
#include "Confdump/System/LinuxDumper.hpp"
#elif CD_WIN32
#include "Confdump/System/WmiDumper.hpp"
#include "Confdump/System/Win32Dumper.hpp"
#endif

using boost::shared_ptr;

namespace Confdump
{

DumperFactory::DumperFactory()
{
    specifications_["main"] = mainDumperSpecification();
}

DumperFactory::~DumperFactory()
{
}

DumperSpecification DumperFactory::mainDumperSpecification()
{
    DumperSpecification spec;
    #if CD_UNIX
    spec.name = "linux";
    #elif CD_WIN32
    spec.name = "wmi";
    #endif
    return spec;
}

shared_ptr<Dumper> DumperFactory::createDumper( DumperSpecification spec )
{
    QString name = spec.name;

    #if CD_UNIX
    if ( name == "linux" )
        return shared_ptr<Dumper>( new System::LinuxDumper );
    #elif CD_WIN32
    if ( name == "wmi" )
    {
        QString ns = spec.parameters["namespace"];
        if ( !ns.isEmpty() )
            return shared_ptr<Dumper>( new System::WmiDumper( ns ) );
        else
            return shared_ptr<Dumper>( new System::WmiDumper );
    }
    else if ( name == "win32" )
    {
                return shared_ptr<Dumper>( new System::Win32Dumper );
        }
    #endif

    throw RuntimeError( "No such Dumper " + spec.name );
}

void DumperFactory::registerSpecifications( const SpecificationsMap &specs )
{
    typedef std::map<QString, DumperSpecification> collection;
    BOOST_FOREACH( collection::value_type pair, specs )
    {
        if ( pair.first == "main" )
            throw RuntimeError( "The 'main' dumper alias may not be redefined" );
        specifications_[pair.first] = pair.second;
    }
}

shared_ptr<Dumper> DumperFactory::dumper( QString alias )
{
    if ( cache_.find( alias ) != cache_.end() )
    {
        return cache_[alias];
    }
    else
    {
        if ( specifications_.find( alias ) == specifications_.end() )
            throw RuntimeError( "No such Dumper alias " + alias );
        cache_[alias] = createDumper( specifications_[alias] );
        return cache_[alias];
    }
}

QStringList DumperFactory::aliasList() const
{
    QStringList list;

    BOOST_FOREACH( SpecificationsMap::value_type spec, specifications_ )
        list << spec.first;

    return list;
}

}