XmlFormatter.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 "QXmlStreamWriter"
#include "QFile"
#include "QList"
#include "QIODevice"
#include "QDateTime"
#include "QVariant"
#include "QByteArray"
#include "QString"

#include "Confdump/Result.hpp"
#include "Confdump/XmlFormatter.hpp"

namespace
{

void writeCharacters( boost::shared_ptr<QXmlStreamWriter> writer, QVariant value )
{
    if ( value.type() == QVariant::ByteArray )
        writer->writeCharacters( value.toByteArray().toHex() );
        else if ( value.type() == QVariant::Invalid )
        writer->writeComment( "undefined" );
    else if ( value.canConvert( QVariant::String ) )
        writer->writeCharacters( value.toString() );
    else
        writer->writeComment( QString( "data of unhandled " ) + value.typeName() + " type" );
}

} // ns 

namespace Confdump
{

XmlFormatter::XmlFormatter()
{
}

XmlFormatter::~XmlFormatter()
{
}

void XmlFormatter::doPrintHeader( QString macAddress )
{
    writer_.reset( new QXmlStreamWriter( device().get() ) );
    writer_->setAutoFormatting( true );
    writer_->writeStartDocument();
    writer_->writeStartElement( "collect" );

    QDateTime now = QDateTime::currentDateTime();
    writer_->writeAttribute( "date", now.toUTC().toString( "ddMMyyyyhhmmss" ) );
    writer_->writeAttribute( "timestamp", now.toUTC().toString( Qt::ISODate ) + "Z" );
    writer_->writeAttribute( "uuid", macAddress );
}

void XmlFormatter::doPrintTable( QString title, const Result &result )
{
    writer_->writeStartElement( "query" );
    writer_->writeAttribute( "class", title );

    BOOST_FOREACH( Result::Row row, result )
    {
        QStringList names( result.propertyNames() );
        QList<QVariant> values( result.propertyValues( row ) );

        writer_->writeStartElement( "item" );
        if( values.size() == names.size() )
        {
            for ( int i = 0; i < names.size(); ++i )
            {
                writer_->writeStartElement( "attr" );
                writer_->writeAttribute( "name", names[i] );
                writer_->writeAttribute( "operator", "=" );

                if ( values[i].type() == QVariant::List )
                {
                    BOOST_FOREACH( QVariant element, values[i].toList() )
                    {
                                                writer_->writeStartElement( "element" );
                        writeCharacters( writer_, element );
                                                writer_->writeEndElement();
                                        }
                }
                else
                {
                    writeCharacters( writer_, values[i] );
                }

                writer_->writeEndElement();
            }
        }
        writer_->writeEndElement();
    }

    writer_->writeEndElement();
}

void XmlFormatter::doPrintFooter()
{
    writer_->writeEndElement();
}

}