/*

 * This file is part of the API Extractor project.

 *

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

 *

 * Contact: PySide team <contact@pyside.org>

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * version 2 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, write to the Free Software

 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA

 * 02110-1301 USA

 *

 */

#include "asttoxml.h"

#include "parser/control.h"

#include "parser/parser.h"

#include "parser/binder.h"

#include <QtCore/QXmlStreamWriter>

#include <QtCore/QTextStream>

#include <QtCore/QTextCodec>

#include <QtCore/QFile>

void astToXML(QString name)

{

    QFile file(name);

    if (!file.open(QFile::ReadOnly))

        return;

    QTextStream stream(&file);

    stream.setCodec(QTextCodec::codecForName("UTF-8"));

    QByteArray contents = stream.readAll().toUtf8();

    file.close();

    Control control;

    Parser p(&control);

    pool __pool;

    TranslationUnitAST *ast = p.parse(contents, contents.size(), &__pool);

    CodeModel model;

    Binder binder(&model, p.location());

    FileModelItem dom = binder.run(ast);

    QFile outputFile;

    if (!outputFile.open(stdout, QIODevice::WriteOnly))

        return;

    QXmlStreamWriter s(&outputFile);

    s.setAutoFormatting(true);

    s.writeStartElement("code");

    QHash<QString, NamespaceModelItem> namespaceMap = dom->namespaceMap();

    foreach (NamespaceModelItem item, namespaceMap.values())

        writeOutNamespace(s, item);

    QHash<QString, ClassModelItem> typeMap = dom->classMap();

    foreach (ClassModelItem item, typeMap.values())

        writeOutClass(s, item);

    s.writeEndElement();

}

void writeOutNamespace(QXmlStreamWriter &s, NamespaceModelItem &item)

{

    s.writeStartElement("namespace");

    s.writeAttribute("name", item->name());

    QHash<QString, NamespaceModelItem> namespaceMap = item->namespaceMap();

    foreach (NamespaceModelItem item, namespaceMap.values())

        writeOutNamespace(s, item);

    QHash<QString, ClassModelItem> typeMap = item->classMap();

    foreach (ClassModelItem item, typeMap.values())

        writeOutClass(s, item);

    QHash<QString, EnumModelItem> enumMap = item->enumMap();

    foreach (EnumModelItem item, enumMap.values())

        writeOutEnum(s, item);

    s.writeEndElement();

}

void writeOutEnum(QXmlStreamWriter &s, EnumModelItem &item)

{

    QString qualifiedName = item->qualifiedName().join("::");

    s.writeStartElement("enum");

    s.writeAttribute("name", qualifiedName);

    EnumeratorList enumList = item->enumerators();

    for (int i = 0; i < enumList.size() ; i++) {

        s.writeStartElement("enumerator");

        if (!enumList[i]->value().isEmpty())

            s.writeAttribute("value", enumList[i]->value());

        s.writeCharacters(enumList[i]->name());

        s.writeEndElement();

    }

    s.writeEndElement();

}

void writeOutFunction(QXmlStreamWriter &s, FunctionModelItem &item)

{

    QString qualifiedName = item->qualifiedName().join("::");

    s.writeStartElement("function");

    s.writeAttribute("name", qualifiedName);

    ArgumentList arguments = item->arguments();

    for (int i = 0; i < arguments.size() ; i++) {

        s.writeStartElement("argument");

        s.writeAttribute("type",  arguments[i]->type().qualifiedName().join("::"));

        s.writeEndElement();

    }

    s.writeEndElement();

}

void writeOutClass(QXmlStreamWriter &s, ClassModelItem &item)

{

    QString qualifiedName = item->qualifiedName().join("::");

    s.writeStartElement("class");

    s.writeAttribute("name", qualifiedName);

    QHash<QString, EnumModelItem> enumMap = item->enumMap();

    foreach (EnumModelItem item, enumMap.values())

        writeOutEnum(s, item);

    QHash<QString, FunctionModelItem> functionMap = item->functionMap();

    foreach (FunctionModelItem item, functionMap.values())

        writeOutFunction(s, item);

    QHash<QString, ClassModelItem> typeMap = item->classMap();

    foreach (ClassModelItem item, typeMap.values())

        writeOutClass(s, item);

    s.writeEndElement();

}