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

**

** Copyright (C) 2009 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 "ui3reader.h"

#include "parser.h"

#include "domtool.h"

#include "ui4.h"

#include "widgetinfo.h"

#include "globaldefs.h"

#include "qt3to4.h"

#include "utils.h"

#include "option.h"

#include "cppextractimages.h"

#include <QtDebug>

#include <QFile>

#include <QHash>

#include <QPair>

#include <QStringList>

#include <QDateTime>

#include <QRegExp>

#include <QSizePolicy>

#include <stdio.h>

#include <stdlib.h>

QT_BEGIN_NAMESPACE

enum { warnHeaderGeneration = 0 };

#define CONVERT_PROPERTY(o, n) \

    do { \

        if (name == QLatin1String(o) \

                && !WidgetInfo::isValidProperty(className, (o)) \

                && WidgetInfo::isValidProperty(className, (n))) { \

            prop->setAttributeName((n)); \

        } \

    } while (0)

static QString classNameForObjectName(const QDomElement &widget, const QString &objectName)

{

    QList<QDomElement> widgetStack;

    widgetStack.append(widget);

    while (!widgetStack.isEmpty()) {

        QDomElement w = widgetStack.takeFirst();

        QDomElement child = w.firstChild().toElement();

        while (!child.isNull()) {

            if (child.tagName() == QLatin1String("property")

                && child.attribute(QLatin1String("name")) == QLatin1String("name")) {

                QDomElement name = child.firstChild().toElement();

                DomString str;

                str.read(name);

                if (str.text() == objectName)

                    return w.attribute(QLatin1String("class"));

            } else if (child.tagName() == QLatin1String("widget")

                || child.tagName() == QLatin1String("vbox")

                || child.tagName() == QLatin1String("hbox")

                || child.tagName() == QLatin1String("grid")) {

                widgetStack.prepend(child);

            }

            child = child.nextSibling().toElement();

        }

    }

    return QString();

}

// Check for potential KDE classes like

//  K3ListView or KLineEdit as precise as possible

static inline bool isKDEClass(const QString &className)

{

    if (className.indexOf(QLatin1Char(':')) != -1)

        return false;

    const int size = className.size();

    if (size < 3 || className.at(0) != QLatin1Char('K'))

        return false;

    // K3ListView

    if (className.at(1) == QLatin1Char('3')) {

        if (size < 4)

            return false;

        return className.at(2).isUpper() &&  className.at(3).isLower();

    }

    // KLineEdit

    return className.at(1) .isUpper() && className.at(2).isLower();

}

DomUI *Ui3Reader::generateUi4(const QDomElement &widget, bool implicitIncludes)

{

    QDomNodeList nl;

    candidateCustomWidgets.clear();

    QString objClass = getClassName(widget);

    if (objClass.isEmpty())

        return 0;

    QString objName = getObjectName(widget);

    DomUI *ui = new DomUI;

    ui->setAttributeVersion(QLatin1String("4.0"));

    QString pixmapFunction = QLatin1String("qPixmapFromMimeSource");

    QStringList ui_tabstops;

    QStringList ui_custom_slots;

    QList<DomInclude*> ui_includes;

    QList<DomWidget*> ui_toolbars;

    QList<DomWidget*> ui_menubars;

    QList<DomAction*> ui_action_list;

    QList<DomActionGroup*> ui_action_group_list;

    QList<DomCustomWidget*> ui_customwidget_list;

    QList<DomConnection*> ui_connection_list;

    QList<QPair<int, int> > ui_connection_lineinfo_list;

    QString author, comment, exportMacro;

    QString klass;

    for (QDomElement n = root.firstChild().toElement(); !n.isNull(); n = n.nextSibling().toElement()) {

        QString tagName = n.tagName().toLower();

        if (tagName == QLatin1String("tabstops")) {

            QDomElement n2 = n.firstChild().toElement();

            while (!n2.isNull()) {

                if (n2.tagName().toLower() == QLatin1String("tabstop")) {

                    QString name = n2.firstChild().toText().data();

                    ui_tabstops.append(name);

                }

                n2 = n2.nextSibling().toElement();

            }

        } else if (tagName == QLatin1String("pixmapfunction")) {

            pixmapFunction = n.firstChild().toText().data();

        } else if (tagName == QLatin1String("class")) {

            klass = n.firstChild().toText().data();

        } else if (tagName == QLatin1String("author")) {

            author = n.firstChild().toText().data();

        } else if (tagName == QLatin1String("comment")) {

            comment = n.firstChild().toText().data();

        } else if (tagName == QLatin1String("exportmacro")) {

            exportMacro = n.firstChild().toText().data();

        } else if ( n.tagName() == QLatin1String("includehints") ) {

            QDomElement n2 = n.firstChild().toElement();

            while ( !n2.isNull() ) {

                if ( n2.tagName() == QLatin1String("includehint") ) {

                    QString name = n2.firstChild().toText().data();

                    DomInclude *incl = new DomInclude();

                    incl->setText(fixHeaderName(name));

                    incl->setAttributeLocation(n.attribute(QLatin1String("location"), QLatin1String("local")));

                    ui_includes.append(incl);

                }

                n2 = n2.nextSibling().toElement();

            }

        } else if (tagName == QLatin1String("includes")) {

            QDomElement n2 = n.firstChild().toElement();

            while (!n2.isNull()) {

                if (n2.tagName().toLower() == QLatin1String("include")) {

                    QString name = n2.firstChild().toText().data();

                    if (n2.attribute(QLatin1String("impldecl"), QLatin1String("in implementation")) == QLatin1String("in declaration")) {

                        if (name.right(5) == QLatin1String(".ui.h"))

                            continue;

                        DomInclude *incl = new DomInclude();

                        incl->setText(fixHeaderName(name));

                        incl->setAttributeLocation(n2.attribute(QLatin1String("location"), QLatin1String("global")));

                        ui_includes.append(incl);

                    }

                }

                n2 = n2.nextSibling().toElement();

            }

        } else if (tagName == QLatin1String("include")) {

            QString name = n.firstChild().toText().data();

            if (n.attribute(QLatin1String("impldecl"), QLatin1String("in implementation")) == QLatin1String("in declaration")) {

                if (name.right(5) == QLatin1String(".ui.h"))

                    continue;

                DomInclude *incl = new DomInclude();

                incl->setText(fixHeaderName(name));

                incl->setAttributeLocation(n.attribute(QLatin1String("location"), QLatin1String("global")));

                ui_includes.append(incl);

            }

        } else if (tagName == QLatin1String("layoutdefaults")) {

            QString margin = n.attribute(QLatin1String("margin"));

            QString spacing = n.attribute(QLatin1String("spacing"));

            DomLayoutDefault *layoutDefault = new DomLayoutDefault();

            if (!margin.isEmpty())

                layoutDefault->setAttributeMargin(margin.toInt());

            if (!spacing.isEmpty())

                layoutDefault->setAttributeSpacing(spacing.toInt());

            ui->setElementLayoutDefault(layoutDefault);

        } else if (tagName == QLatin1String("layoutfunctions")) {

            QString margin = n.attribute(QLatin1String("margin"));

            QString spacing = n.attribute(QLatin1String("spacing"));

            DomLayoutFunction *layoutDefault = new DomLayoutFunction();

            if (!margin.isEmpty())

                layoutDefault->setAttributeMargin(margin);

            if (!spacing.isEmpty())

                layoutDefault->setAttributeSpacing(spacing);

            ui->setElementLayoutFunction(layoutDefault);

        } else if (tagName == QLatin1String("images")) {

            QDomNodeList nl = n.elementsByTagName(QLatin1String("image"));

            QList<DomImage*> ui_image_list;

            for (int i=0; i<(int)nl.length(); i++) {

                QDomElement e = nl.item(i).toElement();

                QDomElement tmp = e.firstChild().toElement();

                if (tmp.tagName().toLower() != QLatin1String("data"))

                    continue;

                // create the image

                DomImage *img = new DomImage();

                img->setAttributeName(e.attribute(QLatin1String("name")));

                // create the data

                DomImageData *data = new DomImageData();

                img->setElementData(data);

                if (tmp.hasAttribute(QLatin1String("format")))

                    data->setAttributeFormat(tmp.attribute(QLatin1String("format"), QLatin1String("PNG")));

                if (tmp.hasAttribute(QLatin1String("length")))

                    data->setAttributeLength(tmp.attribute(QLatin1String("length")).toInt());

                data->setText(tmp.firstChild().toText().data());

                ui_image_list.append(img);

                QString format = img->elementData()->attributeFormat();

                QString extension = format.left(format.indexOf('.')).toLower();

                m_imageMap[img->attributeName()] = img->attributeName() + QLatin1Char('.') + extension;

            }

            if (ui_image_list.size()) {

                DomImages *images = new DomImages();

                images->setElementImage(ui_image_list);

                ui->setElementImages(images);

            }

        } else if (tagName == QLatin1String("actions")) {

            QDomElement n2 = n.firstChild().toElement();

            while (!n2.isNull()) {

                QString tag = n2.tagName().toLower();

                if (tag == QLatin1String("action")) {

                    DomAction *action = new DomAction();

                    action->read(n2);

                    QList<DomProperty*> properties = action->elementProperty();

                    QString actionName = fixActionProperties(properties);

                    action->setAttributeName(actionName);

                    action->setElementProperty(properties);

                    if (actionName.isEmpty()) {

                        delete action;

                    } else

                        ui_action_list.append(action);

                } else if (tag == QLatin1String("actiongroup")) {

                    DomActionGroup *g= new DomActionGroup();

                    g->read(n2);

                    fixActionGroup(g);

                    ui_action_group_list.append(g);

                }

                n2 = n2.nextSibling().toElement();

            }

        } else if (tagName == QLatin1String("toolbars")) {

            QDomElement n2 = n.firstChild().toElement();

            while (!n2.isNull()) {

                if (n2.tagName().toLower() == QLatin1String("toolbar")) {

                    DomWidget *tb = createWidget(n2, QLatin1String("QToolBar"));

                    ui_toolbars.append(tb);

                }

                n2 = n2.nextSibling().toElement();

            }

        } else if (tagName == QLatin1String("menubar")) {

            DomWidget *tb = createWidget(n, QLatin1String("QMenuBar"));

            ui_menubars.append(tb);

        } else if (tagName == QLatin1String("customwidgets")) {

            QDomElement n2 = n.firstChild().toElement();

            while (!n2.isNull()) {

                if (n2.tagName().toLower() == QLatin1String("customwidget")) {

                    DomCustomWidget *customWidget = new DomCustomWidget;

                    customWidget->read(n2);

                    if (!customWidget->hasElementExtends())

                        customWidget->setElementExtends(QLatin1String("QWidget"));

                    QDomElement n3 = n2.firstChild().toElement();

                    QString cl;

                    QList<DomPropertyData*> ui_property_list;

                    while (!n3.isNull()) {

                        QString tagName = n3.tagName().toLower();

                        if (tagName == QLatin1String("property")) {

                            DomPropertyData *p = new DomPropertyData();

                            p->read(n3);

                            ui_property_list.append(p);

                        }

                        n3 = n3.nextSibling().toElement();

                    }

                    if (ui_property_list.size()) {

                        DomProperties *properties = new DomProperties();

                        properties->setElementProperty(ui_property_list);

                        customWidget->setElementProperties(properties);

                    }

                    ui_customwidget_list.append(customWidget);

                }

                n2 = n2.nextSibling().toElement();

            }

        } else if (tagName == QLatin1String("connections")) {

            QDomElement n2 = n.firstChild().toElement();

            while (!n2.isNull()) {

                if (n2.tagName().toLower() == QLatin1String("connection")) {

                    DomConnection *connection = new DomConnection;

                    connection->read(n2);

                    QString signal = fixMethod(connection->elementSignal());

                    QString slot = fixMethod(connection->elementSlot());

                    connection->setElementSignal(signal);

                    connection->setElementSlot(slot);

                    ui_connection_list.append(connection);

                    ui_connection_lineinfo_list.append(

                        QPair<int, int>(n2.lineNumber(), n2.columnNumber()));

                }

                n2 = n2.nextSibling().toElement();

            }

        } else if (tagName == QLatin1String("slots")) {

            QDomElement n2 = n.firstChild().toElement();

            while (!n2.isNull()) {

                if (n2.tagName().toLower() == QLatin1String("slot")) {

                    QString name = n2.firstChild().toText().data();

                    ui_custom_slots.append(fixMethod(Parser::cleanArgs(name)));

                }

                n2 = n2.nextSibling().toElement();

            }

        }

    }

    // validate the connections

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

        DomConnection *conn = ui_connection_list.at(i);

        QPair<int, int> lineinfo = ui_connection_lineinfo_list.at(i);

        QString sender = conn->elementSender();

        QString senderClass = fixClassName(classNameForObjectName(widget, sender));

        QString signal = conn->elementSignal();

        QString receiver = conn->elementReceiver();

        QString receiverClass = fixClassName(classNameForObjectName(widget, receiver));

        QString slot = conn->elementSlot();

        if (!WidgetInfo::isValidSignal(senderClass, signal)) {

            errorInvalidSignal(signal, sender, senderClass,

                               lineinfo.first, lineinfo.second);

        } else if (!WidgetInfo::isValidSlot(receiverClass, slot)) {

            bool resolved = false;

            if (objName == receiver) {

                // see if it's a custom slot

                foreach (QString cs, ui_custom_slots) {

                    if (cs == slot) {

                        resolved = true;

                        break;

                    }

                }

            }

            if (!resolved) {

                errorInvalidSlot(slot, receiver, receiverClass,

                                 lineinfo.first, lineinfo.second);

            }

        }

    }

    DomWidget *w = createWidget(widget);

    Q_ASSERT(w != 0);

    QList<DomWidget*> l = w->elementWidget();

    l += ui_toolbars;

    l += ui_menubars;

    w->setElementWidget(l);

    if (ui_action_group_list.size())

        w->setElementActionGroup(ui_action_group_list);

    if (ui_action_list.size())

        w->setElementAction(ui_action_list);

    ui->setElementWidget(w);

    if (klass.isEmpty())

        klass = w->attributeName();

    ui->setElementClass(klass);

    ui->setElementAuthor(author);

    ui->setElementComment(comment);

    ui->setElementExportMacro(exportMacro);

    if (!ui->elementImages())

        ui->setElementPixmapFunction(pixmapFunction);

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

        const QString name = ui_customwidget_list.at(i)->elementClass();

        if (candidateCustomWidgets.contains(name))

            candidateCustomWidgets.remove(name);

    }

    QMapIterator<QString, bool> it(candidateCustomWidgets);

    while (it.hasNext()) {

        it.next();

        const QString customClass = it.key();

        QString baseClass;

        if (customClass.endsWith(QLatin1String("ListView")))

            baseClass = QLatin1String("Q3ListView");

        else if (customClass.endsWith(QLatin1String("ListBox")))

            baseClass = QLatin1String("Q3ListBox");

        else if (customClass.endsWith(QLatin1String("IconView")))

            baseClass = QLatin1String("Q3IconView");

        else if (customClass.endsWith(QLatin1String("ComboBox")))

            baseClass = QLatin1String("QComboBox");

        if (baseClass.isEmpty())

            continue;

        DomCustomWidget *customWidget = new DomCustomWidget();

        customWidget->setElementClass(customClass);

        customWidget->setElementExtends(baseClass);

        // Magic header generation feature for legacy KDE forms

        // (for example, filesharing/advanced/kcm_sambaconf/share.ui)

        if (implicitIncludes && isKDEClass(customClass)) {

            QString header = customClass.toLower();

            header += QLatin1String(".h");

            DomHeader *domHeader = new DomHeader;

            domHeader->setText(header);

            domHeader->setAttributeLocation(QLatin1String("global"));

            customWidget->setElementHeader(domHeader);

            if (warnHeaderGeneration) {

                const QString msg = QString::fromUtf8("Warning: generated header '%1' for class '%2'.").arg(header).arg(customClass);

                qWarning("%s", qPrintable(msg));

            }

        }

        ui_customwidget_list.append(customWidget);

    }

    if (ui_customwidget_list.size()) {

        DomCustomWidgets *customWidgets = new DomCustomWidgets();

        customWidgets->setElementCustomWidget(ui_customwidget_list);

        ui->setElementCustomWidgets(customWidgets);

    }

    if (ui_tabstops.size()) {

        DomTabStops *tabStops = new DomTabStops();

        tabStops->setElementTabStop(ui_tabstops);

        ui->setElementTabStops(tabStops);

    }

    if (ui_includes.size()) {

        DomIncludes *includes = new DomIncludes();

        includes->setElementInclude(ui_includes);

        ui->setElementIncludes(includes);

    }

    if (ui_connection_list.size()) {

        DomConnections *connections = new DomConnections();

        connections->setElementConnection(ui_connection_list);

        ui->setElementConnections(connections);

    }

    ui->setAttributeStdSetDef(stdsetdef);

    if (m_extractImages) {

        Option opt;

        opt.extractImages = m_extractImages;

        opt.qrcOutputFile = m_qrcOutputFile;

        CPP::ExtractImages(opt).acceptUI(ui);

        ui->clearElementImages();

        DomResources *res = ui->elementResources();

        if (!res) {

            res = new DomResources();