/****************************************************************************

**

** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the examples of the Qt Toolkit.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include <QtGui>

#include "xbelgenerator.h"

XbelGenerator::XbelGenerator(QTreeWidget *treeWidget)

    : treeWidget(treeWidget)

{

}

bool XbelGenerator::write(QIODevice *device)

{

    out.setDevice(device);

    out.setCodec("UTF-8");

    out << "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"

        << "<!DOCTYPE xbel>\n"

        << "<xbel version=\"1.0\">\n";

    for (int i = 0; i < treeWidget->topLevelItemCount(); ++i)

        generateItem(treeWidget->topLevelItem(i), 1);

    out << "</xbel>\n";

    return true;

}

QString XbelGenerator::indent(int depth)

{

    const int IndentSize = 4;

    return QString(IndentSize * depth, ' ');

}

QString XbelGenerator::escapedText(const QString &str)

{

    QString result = str;

    result.replace("&", "&");

    result.replace("<", "&lt;");

    result.replace(">", ">");

    return result;

}

QString XbelGenerator::escapedAttribute(const QString &str)

{

    QString result = escapedText(str);

    result.replace("\"", """);

    result.prepend("\"");

    result.append("\"");

    return result;

}

void XbelGenerator::generateItem(QTreeWidgetItem *item, int depth)

{

    QString tagName = item->data(0, Qt::UserRole).toString();

    if (tagName == "folder") {

        bool folded = !treeWidget->isItemExpanded(item);

        out << indent(depth) << "<folder folded=\"" << (folded ? "yes" : "no")

                             << "\">\n"

            << indent(depth + 1) << "<title>" << escapedText(item->text(0))

                                 << "</title>\n";

        for (int i = 0; i < item->childCount(); ++i)

            generateItem(item->child(i), depth + 1);

        out << indent(depth) << "</folder>\n";

    } else if (tagName == "bookmark") {

        out << indent(depth) << "<bookmark";

        if (!item->text(1).isEmpty())

            out << " href=" << escapedAttribute(item->text(1));

        out << ">\n"

            << indent(depth + 1) << "<title>" << escapedText(item->text(0))

                                 << "</title>\n"

            << indent(depth) << "</bookmark>\n";

    } else if (tagName == "separator") {

        out << indent(depth) << "<separator/>\n";

    }

}