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

**

** 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 tools applications 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 <QtXml>

#include <QHash>

#include <QMap>

#include "atom.h"

#include "helpprojectwriter.h"

#include "htmlgenerator.h"

#include "config.h"

#include "node.h"

#include "tree.h"

QT_BEGIN_NAMESPACE

HelpProjectWriter::HelpProjectWriter(const Config &config, const QString &defaultFileName)

{

    // The output directory should already have been checked by the calling

    // generator.

    outputDir = config.getString(CONFIG_OUTPUTDIR);

    QStringList names = config.getStringList(CONFIG_QHP + Config::dot + "projects");

    foreach (const QString &projectName, names) {

        HelpProject project;

        project.name = projectName;

        QString prefix = CONFIG_QHP + Config::dot + projectName + Config::dot;

        project.helpNamespace = config.getString(prefix + "namespace");

        project.virtualFolder = config.getString(prefix + "virtualFolder");

        project.fileName = config.getString(prefix + "file");

        if (project.fileName.isEmpty())

            project.fileName = defaultFileName;

        project.extraFiles = config.getStringSet(prefix + "extraFiles");

        project.indexTitle = config.getString(prefix + "indexTitle");

        project.indexRoot = config.getString(prefix + "indexRoot");

        project.filterAttributes = config.getStringList(prefix + "filterAttributes").toSet();

        QSet<QString> customFilterNames = config.subVars(prefix + "customFilters");

        foreach (const QString &filterName, customFilterNames) {

            QString name = config.getString(prefix + "customFilters" + Config::dot + filterName + Config::dot + "name");

            QSet<QString> filters = config.getStringList(prefix + "customFilters" + Config::dot + filterName + Config::dot + "filterAttributes").toSet();

            project.customFilters[name] = filters;

        }

        //customFilters = config.defs.

        foreach (QString name, config.getStringSet(prefix + "excluded"))

            project.excluded.insert(name.replace("\\", "/"));

        foreach (const QString &name, config.getStringList(prefix + "subprojects")) {

            SubProject subproject;

            QString subprefix = prefix + "subprojects" + Config::dot + name + Config::dot;

            subproject.title = config.getString(subprefix + "title");

            subproject.indexTitle = config.getString(subprefix + "indexTitle");

            subproject.sortPages = config.getBool(subprefix + "sortPages");

            readSelectors(subproject, config.getStringList(subprefix + "selectors"));

            project.subprojects[name] = subproject;

        }

        if (project.subprojects.isEmpty()) {

            SubProject subproject;

            readSelectors(subproject, config.getStringList(prefix + "selectors"));

            project.subprojects[""] = subproject;

        }

        projects.append(project);

    }

}

void HelpProjectWriter::readSelectors(SubProject &subproject, const QStringList &selectors)

{

    QHash<QString, Node::Type> typeHash;

    typeHash["namespace"] = Node::Namespace;

    typeHash["class"] = Node::Class;

    typeHash["fake"] = Node::Fake;

    typeHash["enum"] = Node::Enum;

    typeHash["typedef"] = Node::Typedef;

    typeHash["function"] = Node::Function;

    typeHash["property"] = Node::Property;

    typeHash["variable"] = Node::Variable;

    typeHash["target"] = Node::Target;

    QHash<QString, Node::SubType> subTypeHash;

    subTypeHash["example"] = Node::Example;

    subTypeHash["headerfile"] = Node::HeaderFile;

    subTypeHash["file"] = Node::File;

    subTypeHash["group"] = Node::Group;

    subTypeHash["module"] = Node::Module;

    subTypeHash["page"] = Node::Page;

    subTypeHash["externalpage"] = Node::ExternalPage;

#ifdef QDOC_QML

    subTypeHash["qmlclass"] = Node::QmlClass;

#endif

    QSet<Node::SubType> allSubTypes = QSet<Node::SubType>::fromList(subTypeHash.values());

    foreach (const QString &selector, selectors) {

        QStringList pieces = selector.split(":");

        if (pieces.size() == 1) {

            QString lower = selector.toLower();

            if (typeHash.contains(lower))

                subproject.selectors[typeHash[lower]] = allSubTypes;

        } else if (pieces.size() >= 2) {

            QString lower = pieces[0].toLower();

            pieces = pieces[1].split(",");

            if (typeHash.contains(lower)) {

                QSet<Node::SubType> subTypes;

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

                    QString lower = pieces[i].toLower();

                    if (subTypeHash.contains(lower))

                        subTypes.insert(subTypeHash[lower]);

                }

                subproject.selectors[typeHash[lower]] = subTypes;

            }

        }

    }

}

void HelpProjectWriter::addExtraFile(const QString &file)

{

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

        projects[i].extraFiles.insert(file);

}

void HelpProjectWriter::addExtraFiles(const QSet<QString> &files)

{

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

        projects[i].extraFiles.unite(files);

}

/*

    Returns a list of strings describing the keyword details for a given node.

    The first string is the human-readable name to be shown in Assistant.

    The second string is a unique identifier.

    The third string is the location of the documentation for the keyword.

*/

QStringList HelpProjectWriter::keywordDetails(const Node *node) const

{

    QStringList details;

    if (node->parent() && !node->parent()->name().isEmpty()) {

        // "name"

        if (node->type() == Node::Enum || node->type() == Node::Typedef)

            details << node->parent()->name()+"::"+node->name();

        else

            details << node->name();

        // "id"

        details << node->parent()->name()+"::"+node->name();

    } else if (node->type() == Node::Fake) {

        const FakeNode *fake = static_cast<const FakeNode *>(node);

#ifdef QDOC_QML

        if (fake->subType() == Node::QmlClass) {

            details << (QmlClassNode::qmlOnly ? fake->name() : fake->fullTitle());

            details << "QML." + fake->name();

        } else

#endif

        {

            details << fake->fullTitle();

            details << fake->fullTitle();

        }

    } else {

        details << node->name();

        details << node->name();

    }

    details << tree->fullDocumentLocation(node);

    return details;

}

bool HelpProjectWriter::generateSection(HelpProject &project,

                        QXmlStreamWriter & /* writer */, const Node *node)

{

    if (!node->url().isEmpty())

        return false;

    if (node->access() == Node::Private || node->status() == Node::Internal)

        return false;

    if (node->name().isEmpty())

        return true;

    QString docPath = node->doc().location().filePath();

    if (!docPath.isEmpty() && project.excluded.contains(docPath))

        return false;

    QString objName;

    if (node->type() == Node::Fake) {

        const FakeNode *fake = static_cast<const FakeNode *>(node);

        objName = fake->fullTitle();

    } else

        objName = tree->fullDocumentName(node);

    // Only add nodes to the set for each subproject if they match a selector.

    // Those that match will be listed in the table of contents.

    foreach (const QString &name, project.subprojects.keys()) {

        SubProject subproject = project.subprojects[name];

        // No selectors: accept all nodes.

        if (subproject.selectors.isEmpty())

            project.subprojects[name].nodes[objName] = node;

        else if (subproject.selectors.contains(node->type())) {

            // Accept only the node types in the selectors hash.

            if (node->type() != Node::Fake)

                project.subprojects[name].nodes[objName] = node;

            else {

                // Accept only fake nodes with subtypes contained in the selector's

                // mask.

                const FakeNode *fakeNode = static_cast<const FakeNode *>(node);

                if (subproject.selectors[node->type()].contains(fakeNode->subType()) &&

                    fakeNode->subType() != Node::ExternalPage &&

                    !fakeNode->fullTitle().isEmpty())

                    project.subprojects[name].nodes[objName] = node;

            }

        }

    }

    switch (node->type()) {

        case Node::Class:

            project.keywords.append(keywordDetails(node));

            project.files.insert(tree->fullDocumentLocation(node));

            break;

        case Node::Namespace:

            project.keywords.append(keywordDetails(node));

            project.files.insert(tree->fullDocumentLocation(node));

            break;

        case Node::Enum:

            project.keywords.append(keywordDetails(node));

            {

                const EnumNode *enumNode = static_cast<const EnumNode*>(node);

                foreach (const EnumItem &item, enumNode->items()) {

                    QStringList details;

                    if (enumNode->itemAccess(item.name()) == Node::Private)

                        continue;

                    if (!node->parent()->name().isEmpty()) {

                        details << node->parent()->name()+"::"+item.name(); // "name"

                        details << node->parent()->name()+"::"+item.name(); // "id"

                    } else {

                        details << item.name(); // "name"

                        details << item.name(); // "id"

                    }

                    details << tree->fullDocumentLocation(node);

                    project.keywords.append(details);

                }

            }

            break;

        case Node::Property:

            project.keywords.append(keywordDetails(node));

            break;

        case Node::Function:

            {

                const FunctionNode *funcNode = static_cast<const FunctionNode *>(node);

                // Only insert keywords for non-constructors. Constructors are covered

                // by the classes themselves.

                if (funcNode->metaness() != FunctionNode::Ctor)

                    project.keywords.append(keywordDetails(node));

                // Insert member status flags into the entries for the parent

                // node of the function, or the node it is related to.

                // Since parent nodes should have already been inserted into

                // the set of files, we only need to ensure that related nodes

                // are inserted.

                if (node->relates()) {

                    project.memberStatus[node->relates()].insert(node->status());

                    project.files.insert(tree->fullDocumentLocation(node->relates()));

                } else if (node->parent())

                    project.memberStatus[node->parent()].insert(node->status());

            }

            break;

        case Node::Typedef:

            {

                const TypedefNode *typedefNode = static_cast<const TypedefNode *>(node);

                QStringList typedefDetails = keywordDetails(node);

                const EnumNode *enumNode = typedefNode->associatedEnum();

                // Use the location of any associated enum node in preference

                // to that of the typedef.

                if (enumNode)

                    typedefDetails[2] = tree->fullDocumentLocation(enumNode);

                project.keywords.append(typedefDetails);

            }

            break;

        // Fake nodes (such as manual pages) contain subtypes, titles and other

        // attributes.

        case Node::Fake: {

            const FakeNode *fakeNode = static_cast<const FakeNode*>(node);

            if (fakeNode->subType() != Node::ExternalPage &&

                !fakeNode->fullTitle().isEmpty()) {

                if (fakeNode->subType() != Node::File) {

                    if (fakeNode->doc().hasKeywords()) {

                        foreach (const Atom *keyword, fakeNode->doc().keywords()) {

                            if (!keyword->string().isEmpty()) {

                                QStringList details;

                                details << keyword->string()

                                        << keyword->string()

                                        << tree->fullDocumentLocation(node) + "#" + Doc::canonicalTitle(keyword->string());

                                project.keywords.append(details);

                            } else

                                fakeNode->doc().location().warning(

                                    tr("Bad keyword in %1").arg(tree->fullDocumentLocation(node))

                                    );

                        }

                    }

                    project.keywords.append(keywordDetails(node));

                }

/*

                if (fakeNode->doc().hasTableOfContents()) {

                    foreach (const Atom *item, fakeNode->doc().tableOfContents()) {

                        QString title = Text::sectionHeading(item).toString();

                        if (!title.isEmpty()) {

                            QStringList details;

                            details << title

                                    << title

                                    << tree->fullDocumentLocation(node) + "#" + Doc::canonicalTitle(title);

                            project.keywords.append(details);

                        } else

                            fakeNode->doc().location().warning(

                                tr("Bad contents item in %1").arg(tree->fullDocumentLocation(node))

                                );

                    }

                }

*/

                project.files.insert(tree->fullDocumentLocation(node));

            }

            break;

            }

        default:

            ;

    }

    // Add all images referenced in the page to the set of files to include.

    const Atom *atom = node->doc().body().firstAtom();

    while (atom) {

        if (atom->type() == Atom::Image || atom->type() == Atom::InlineImage) {

            // Images are all placed within a single directory regardless of

            // whether the source images are in a nested directory structure.

            QStringList pieces = atom->string().split("/");

            project.files.insert("images/" + pieces.last());

        }

        atom = atom->next();

    }

    return true;

}

void HelpProjectWriter::generateSections(HelpProject &project,

                        QXmlStreamWriter &writer, const Node *node)

{

    if (!generateSection(project, writer, node))

        return;

    if (node->isInnerNode()) {

        const InnerNode *inner = static_cast<const InnerNode *>(node);

        // Ensure that we don't visit nodes more than once.

        QMap<QString, const Node*> childMap;

        foreach (const Node *node, inner->childNodes()) {

            if (node->access() == Node::Private)

                continue;

            if (node->type() == Node::Fake)

                childMap[static_cast<const FakeNode *>(node)->fullTitle()] = node;

            else {

                if (node->type() == Node::Function) {

                    const FunctionNode *funcNode = static_cast<const FunctionNode *>(node);

                    if (funcNode->isOverload())

                        continue;

                }

                childMap[tree->fullDocumentName(node)] = node;

            }

        }

        foreach (const Node *child, childMap)

            generateSections(project, writer, child);

    }

}

void HelpProjectWriter::generate(const Tree *tre)

{

    this->tree = tre;

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

        generateProject(projects[i]);

}

void HelpProjectWriter::writeNode(HelpProject &project, QXmlStreamWriter &writer,

                                  const Node *node)

{

    QString href = tree->fullDocumentLocation(node);

    QString objName = node->name();

    switch (node->type()) {

        case Node::Class:

            writer.writeStartElement("section");

            writer.writeAttribute("ref", href);

            if (node->parent() && !node->parent()->name().isEmpty())

                writer.writeAttribute("title", tr("%1::%2 Class Reference").arg(node->parent()->name()).arg(objName));

            else

                writer.writeAttribute("title", tr("%1 Class Reference").arg(objName));

            // Write subsections for all members, obsolete members and Qt 3

            // members.

            if (!project.memberStatus[node].isEmpty()) {

                QString membersPath = href.left(href.size()-5) + "-members.html";

                writer.writeStartElement("section");

                writer.writeAttribute("ref", membersPath);

                writer.writeAttribute("title", tr("List of all members"));

                writer.writeEndElement(); // section

                project.files.insert(membersPath);

            }

            if (project.memberStatus[node].contains(Node::Compat)) {

                QString compatPath = href.left(href.size()-5) + "-qt3.html";

                writer.writeStartElement("section");

                writer.writeAttribute("ref", compatPath);

                writer.writeAttribute("title", tr("Qt 3 support members"));

                writer.writeEndElement(); // section

                project.files.insert(compatPath);

            }

            if (project.memberStatus[node].contains(Node::Obsolete)) {

                QString obsoletePath = href.left(href.size()-5) + "-obsolete.html";

                writer.writeStartElement("section");

                writer.writeAttribute("ref", obsoletePath);

                writer.writeAttribute("title", tr("Obsolete members"));

                writer.writeEndElement(); // section

                project.files.insert(obsoletePath);

            }

            writer.writeEndElement(); // section

            break;

        case Node::Namespace:

            writer.writeStartElement("section");

            writer.writeAttribute("ref", href);

            writer.writeAttribute("title", objName);

            writer.writeEndElement(); // section

            break;

        case Node::Fake: {

            // Fake nodes (such as manual pages) contain subtypes, titles and other

            // attributes.

            const FakeNode *fakeNode = static_cast<const FakeNode*>(node);

            writer.writeStartElement("section");

            writer.writeAttribute("ref", href);

            writer.writeAttribute("title", fakeNode->fullTitle());

            //            qDebug() << "Title:" << fakeNode->fullTitle();

            if (fakeNode->subType() == Node::HeaderFile) {

                // Write subsections for all members, obsolete members and Qt 3

                // members.

                if (!project.memberStatus[node].isEmpty()) {

                    QString membersPath = href.left(href.size()-5) + "-members.html";

                    writer.writeStartElement("section");

                    writer.writeAttribute("ref", membersPath);

                    writer.writeAttribute("title", tr("List of all members"));

                    writer.writeEndElement(); // section

                    project.files.insert(membersPath);

                }

                if (project.memberStatus[node].contains(Node::Compat)) {

                    QString compatPath = href.left(href.size()-5) + "-qt3.html";

                    writer.writeStartElement("section");

                    writer.writeAttribute("ref", compatPath);

                    writer.writeAttribute("title", tr("Qt 3 support members"));

                    writer.writeEndElement(); // section

                    project.files.insert(compatPath);

                }

                if (project.memberStatus[node].contains(Node::Obsolete)) {

                    QString obsoletePath = href.left(href.size()-5) + "-obsolete.html";

                    writer.writeStartElement("section");

                    writer.writeAttribute("ref", obsoletePath);

                    writer.writeAttribute("title", tr("Obsolete members"));

                    writer.writeEndElement(); // section

                    project.files.insert(obsoletePath);

                }

            }