utils.cpp

#include "stdafx.h"

#include <iostream>
#include <iterator>
#include <rapidxml.hpp>
#include <rapidxml_print.hpp>
#include <sstream>
#include <vector>

#include "utils.h"

namespace Utils
{
    QString formatXML(const QString &text)
    {
        // try to format XML document
        int startPos = text.indexOf(QChar('<'));
        int endPos   = text.lastIndexOf(QChar('>'));
        if (startPos > 0 && endPos > startPos)
        {
            QString header = text.mid(0, startPos);
            QString xmlIn  = text.mid(startPos, endPos - startPos + 1);
#ifndef QT_NO_DEBUG
            qDebug() << "raw text:" << text;
            qDebug() << "xml in:" << xmlIn;
#endif

            using namespace rapidxml;
            QString xmlOut;
            try
            {
                xml_document<>     doc;
                std::istringstream in(xmlIn.toStdString());
                std::vector<char>  buffer((std::istreambuf_iterator<char>(in)), std::istreambuf_iterator<char>());
                buffer.push_back('\0');
                doc.parse<0>(&buffer[0]);

                print(std::back_inserter(xmlOut), doc, 0);
            }
            catch (...)
            {
                return text;
            }

            header.append("\n");
            header.append(xmlOut);
            if (header.length() > text.length())
                return header;
        }

        return text;
    }
} // namespace Utils