main.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/config.hpp"
#include "boost/program_options.hpp"
#include "boost/shared_ptr.hpp"
#include "boost/foreach.hpp"

#include "QTextStream"

#include "Confdump/Options.hpp"
#include "Confdump/Result.hpp"
#include "Confdump/NoSuchTable.hpp"
#include "Confdump/DataOutput.hpp"
#include "Confdump/Formatter.hpp"
#include "Confdump/DumperFactory.hpp"
#include "Confdump/Dumper.hpp"

using namespace std;
using namespace Confdump;
using boost::shared_ptr;
namespace po = boost::program_options;

int main( int argc, char* argv[] )
{
    QTextStream error( stderr ); // bypass cerr, buggy on RHEL4
    try {
        DumperFactory factory;
        Options options( argc, argv, cout );
        factory.registerSpecifications( options.dumperSpecifications() );
        DataOutput output;
        if ( !options.outputFile().isEmpty() )
            output.setSink( options.outputFile() );
        else
            output.setSink( stdout );

        switch ( options.operatingMode() )
        {
        case Options::HelpTablesMode:
            BOOST_FOREACH( QString alias, factory.aliasList() )
            {
                shared_ptr<Dumper> dumper = factory.dumper( alias );
                BOOST_FOREACH( QString table, dumper->listTables() )
                    output.stream() << alias << "." << table << "\n";
            }
            break;

        case Options::NormalMode:
            if ( options.tables().empty() )
            {
                error << "No tables specified, try --help for usage information." << endl;
                #if CD_WIN32
                error << "\nPress a key to continue..." << endl;
                getchar();
                #endif
            }
            else
            {
                shared_ptr<Dumper> driver = factory.dumper( "main" );
                shared_ptr<Formatter> fmt = output.createFormatter( options.outputFormat() );

                QString identifier;
                try {
                    identifier = driver->computerIdentifier();
                }
                catch ( std::exception )
                {
                    // Let the identifier default to the empty string if it
                    // can't be queried. Until we can log, at least.
                }
                fmt->printHeader( identifier );

                BOOST_FOREACH( Options::TableList::value_type table, options.tables() )
                {
                    try {
                        shared_ptr<Dumper> driver = factory.dumper( table.first );
                        fmt->printTable( table.second, driver->queryTable( table.second ) );
                    }
                    catch ( NoSuchThing &e )
                    {
                        cerr << e.what() << endl;
                    }
                }

                #if CD_UNIX
                fmt->printTable( "CD_ConfigurationFile", driver->queryTable( "CD_ConfigurationFile" ) );
                #endif

                fmt->printFooter();
            }

            break;

        case Options::HelpMode:
            break;
        }

        return 0;
    }
    catch ( exception& e )
    {
        error << "error: " << e.what() << endl;
        return 1;
    }
    catch (...)
    {
        error << "Exception of unknown type!" << endl;
        return 1;
    }
}

namespace Confdump
{
}