/*
 * This file is part of smolbote. It's copyrighted by the contributors recorded
 * in the version control history of the file, available from its original
 * location: https://neueland.iserlohn-fortress.net/smolbote.hg
 *
 * SPDX-License-Identifier: GPL-3.0
 */

#include "xbel.h"
#include <QFile>

XbelReader::XbelReader(QIODevice *file)
{
    Q_CHECK_PTR(file);
    m_file = file;
}

void readChildElements(QXmlStreamReader &reader, BookmarksView *widget, QTreeWidgetItem *parentItem = nullptr)
{
    while(reader.readNextStartElement()) {
        if(reader.name() == "title") {
            Q_CHECK_PTR(parentItem);
            parentItem->setText(0, reader.readElementText());

        } else if(reader.name() == "folder") {
            auto *item = widget->createFolder(parentItem);
            widget->setItemExpanded(item, !(reader.attributes().value("folded") == QLatin1String("yes")));
            readChildElements(reader, widget, item);

        } else if(reader.name() == "bookmark") {
            auto *item = widget->createBookmark(parentItem);
            item->setText(1, reader.attributes().value("href").toString());
            readChildElements(reader, widget, item);

        } else {
            reader.skipCurrentElement();
        }
    }
}

void XbelReader::read(BookmarksView *treeWidget)
{
    QXmlStreamReader qXmlStreamReader(m_file);

    if(qXmlStreamReader.readNextStartElement()) {
        if(!(qXmlStreamReader.name() == "xbel" && qXmlStreamReader.attributes().value("version") == "1.0")) {
            return;
        }

        readChildElements(qXmlStreamReader, treeWidget);
    }
}

XbelWriter::XbelWriter(QIODevice *file)
{
    Q_CHECK_PTR(file);
    m_file = file;
}

void writeChildElements(QXmlStreamWriter &writer, BookmarksView *treeWidget, QTreeWidgetItem *parentItem)
{
    switch(treeWidget->itemType(parentItem)) {
    case BookmarksView::Folder:
        writer.writeStartElement("folder");
        writer.writeAttribute("folded", !treeWidget->isItemExpanded(parentItem) ? "yes" : "no");
        writer.writeTextElement("title", parentItem->text(0));

        for(int i = 0; i < parentItem->childCount(); ++i) {
            writeChildElements(writer, treeWidget, parentItem->child(i));
        }

        writer.writeEndElement();
        break;

    case BookmarksView::Bookmark:
        writer.writeStartElement("bookmark");
        writer.writeAttribute("href", parentItem->text(1));
        writer.writeTextElement("title", parentItem->text(0));
        writer.writeEndElement();
        break;

    case BookmarksView::Invalid:
        break;
    }
}

void XbelWriter::write(BookmarksView *treeWidget)
{
    QXmlStreamWriter xmlWriter(m_file);
    xmlWriter.setAutoFormatting(true);

    xmlWriter.writeStartDocument();
    xmlWriter.writeDTD("<!DOCTYPE xbel>");

    xmlWriter.writeStartElement("xbel");
    xmlWriter.writeAttribute("version", "1.0");

    for(int i = 0; i < treeWidget->topLevelItemCount(); ++i) {
        writeChildElements(xmlWriter, treeWidget, treeWidget->topLevelItem(i));
    }

    xmlWriter.writeEndDocument();
}