/*
 * 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>
#include <QAbstractItemModel>
#include "bookmarkitem.h"
#include "bookmarkmodel.h"

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


void readChildElements(QXmlStreamReader &reader, BookmarkItem *parent)
{
    while(reader.readNextStartElement()) {
        if(reader.name() == "title") {
            parent->setData(BookmarkItem::Title, reader.readElementText());

        } else if(reader.name() == "folder") {
            auto *item = new BookmarkItem({}, BookmarkItem::Folder, parent);
            item->setExpanded(!(reader.attributes().value("folded") == QLatin1Literal("yes")));
            parent->appendChild(item);
            readChildElements(reader, item);

        } else if(reader.name() == "bookmark") {
            auto *item = new BookmarkItem({}, BookmarkItem::Bookmark, parent);
            item->setData(BookmarkItem::Href, reader.attributes().value("href").toString());
            parent->appendChild(item);
            readChildElements(reader, item);

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

void XbelReader::read(BookmarkItem *item)
{
    QXmlStreamReader qXmlStreamReader(m_file);

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

        readChildElements(qXmlStreamReader, item);
    }
}

XbelWriter::XbelWriter()
{
}

inline void writeChildElement(QXmlStreamWriter &writer, const BookmarkItem *item)
{
    switch (item->type()) {
        case BookmarkItem::Folder:
            writer.writeStartElement("folder");
            writer.writeAttribute("folded", !item->isExpanded() ? "yes" : "no");
            writer.writeTextElement("title", item->data(BookmarkItem::Title).toString());

            for(int i = 0; i < item->childCount(); ++i) {
                writeChildElement(writer, item->child(i));
            }

            writer.writeEndElement();
            break;

        case BookmarkItem::Bookmark:
            writer.writeStartElement("bookmark");
            writer.writeAttribute("href", item->data(BookmarkItem::Href).toString());
            writer.writeTextElement("title", item->data(BookmarkItem::Title).toString());
            writer.writeEndElement();
            break;
    }
}
void XbelWriter::write(QIODevice *device, const BookmarkItem *item)
{
    QXmlStreamWriter xmlWriter(device);
    xmlWriter.setAutoFormatting(true);

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

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

    writeChildElement(xmlWriter, item);

    xmlWriter.writeEndDocument();
}