DataOutput.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 <stdexcept>

#include "QTextStream"
#include "QFile"

#include "Confdump/TextFormatter.hpp"
#include "Confdump/XmlFormatter.hpp"
#include "Confdump/DataOutput.hpp"

Confdump::DataOutput::DataOutput()
{
}

Confdump::DataOutput::~DataOutput()
{
}

boost::shared_ptr<Confdump::Formatter> Confdump::DataOutput::createFormatter( Options::OutputFormat format ) const
{
    boost::shared_ptr<Formatter> fmt;

    switch ( format )
    {
    case Options::TextFormat:
        fmt.reset( new TextFormatter );
        break;

    case Options::XmlFormat:
        fmt.reset( new XmlFormatter );
        break;

    }

    if ( !fmt )
        throw std::invalid_argument( "unhandled output format" );

    fmt->setDevice( device_ );

    return fmt;
}

QTextStream &Confdump::DataOutput::stream() const
{
    return *stream_;
}

void Confdump::DataOutput::setSink( FILE *file )
{
    boost::shared_ptr<QFile> device( new QFile );
    if ( !device->open( file, QIODevice::WriteOnly | QIODevice::Text ) )
        throw std::runtime_error( "unable to open data output stream" );
    setSink( device );
}

void Confdump::DataOutput::setSink( QString file )
{
    boost::shared_ptr<QFile> device( new QFile( file ) );
    if ( !device->open( QIODevice::WriteOnly | QIODevice::Text | QIODevice::Truncate ) )
        throw std::runtime_error( "unable to open output file" );
    setSink( device );
}

void Confdump::DataOutput::setSink( boost::shared_ptr<QIODevice> device )
{
    device_ = device;
    stream_.reset( new QTextStream( device.get() ) );
}