/*

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

    Copyright (C) 2009 Girish Ramakrishnan <girish@forwardbias.in>

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

    modify it under the terms of the GNU Library General Public

    License as published by the Free Software Foundation; either

    version 2 of the License, or (at your option) any later version.

    This library 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

    Library General Public License for more details.

    You should have received a copy of the GNU Library General Public License

    along with this library; see the file COPYING.LIB.  If not, write to

    the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,

    Boston, MA 02110-1301, USA.

*/

#include "config.h"

#include "qgraphicswebview.h"

#include "qwebframe.h"

#include "qwebpage.h"

#include "qwebpage_p.h"

#include "QWebPageClient.h"

#include <QtGui/QGraphicsScene>

#include <QtGui/QGraphicsView>

#include <QtGui/qapplication.h>

#include <QtGui/qgraphicssceneevent.h>

#include <QtGui/qstyleoption.h>

#if defined(Q_WS_X11)

#include <QX11Info>

#endif

class QGraphicsWebViewPrivate : public QWebPageClient {

public:

    QGraphicsWebViewPrivate(QGraphicsWebView* parent)

        : q(parent)

        , page(0)

    {}

    virtual ~QGraphicsWebViewPrivate();

    virtual void scroll(int dx, int dy, const QRect&);

    virtual void update(const QRect& dirtyRect);

    virtual void setInputMethodEnabled(bool enable);

    virtual bool inputMethodEnabled() const;

#if QT_VERSION >= 0x040600

    virtual void setInputMethodHint(Qt::InputMethodHint hint, bool enable);

#endif

#ifndef QT_NO_CURSOR

    virtual QCursor cursor() const;

    virtual void updateCursor(const QCursor& cursor);

#endif

    virtual QPalette palette() const;

    virtual int screenNumber() const;

    virtual QWidget* ownerWidget() const;

    virtual QObject* pluginParent() const;

    void _q_doLoadFinished(bool success);

    QGraphicsWebView* q;

    QWebPage* page;

};

QGraphicsWebViewPrivate::~QGraphicsWebViewPrivate()

{

}

void QGraphicsWebViewPrivate::_q_doLoadFinished(bool success)

{

    // If the page had no title, still make sure it gets the signal

    if (q->title().isEmpty())

        emit q->urlChanged(q->url());

    emit q->loadFinished(success);

}

void QGraphicsWebViewPrivate::scroll(int dx, int dy, const QRect& rectToScroll)

{

    q->scroll(qreal(dx), qreal(dy), QRectF(rectToScroll));

}

void QGraphicsWebViewPrivate::update(const QRect & dirtyRect)

{

    q->update(QRectF(dirtyRect));

}

void QGraphicsWebViewPrivate::setInputMethodEnabled(bool enable)

{

#if QT_VERSION >= QT_VERSION_CHECK(4, 6, 0)

    q->setFlag(QGraphicsItem::ItemAcceptsInputMethod, enable);

#endif

}

bool QGraphicsWebViewPrivate::inputMethodEnabled() const

{

#if QT_VERSION >= QT_VERSION_CHECK(4, 6, 0)

    return q->flags() & QGraphicsItem::ItemAcceptsInputMethod;

#else

    return false;

#endif

}

#if QT_VERSION >= 0x040600

void QGraphicsWebViewPrivate::setInputMethodHint(Qt::InputMethodHint hint, bool enable)

{

    if (enable)

        q->setInputMethodHints(q->inputMethodHints() | hint);

    else

        q->setInputMethodHints(q->inputMethodHints() & ~hint);

}

#endif

#ifndef QT_NO_CURSOR

QCursor QGraphicsWebViewPrivate::cursor() const

{

    return q->cursor();

}

void QGraphicsWebViewPrivate::updateCursor(const QCursor& cursor)

{

    q->setCursor(cursor);

}

#endif

QPalette QGraphicsWebViewPrivate::palette() const

{

    return q->palette();

}

int QGraphicsWebViewPrivate::screenNumber() const

{

#if defined(Q_WS_X11)

    const QList<QGraphicsView*> views = q->scene()->views();

    if (!views.isEmpty())

        return views.at(0)->x11Info().screen();

#endif

    return 0;

}

QWidget* QGraphicsWebViewPrivate::ownerWidget() const

{

    const QList<QGraphicsView*> views = q->scene()->views();

    return views.value(0);

}

QObject* QGraphicsWebViewPrivate::pluginParent() const

{

    return q;

}

/*!

    \class QGraphicsWebView

    \brief The QGraphicsWebView class allows Web content to be added to a GraphicsView.

    \since 4.6

    An instance of this class renders Web content from a URL or supplied as data, using

    features of the QtWebKit module.

    If the width and height of the item are not set, they will default to 800 and 600,

    respectively. If the Web page contents is larger than that, scrollbars will be shown

    if not disabled explicitly.

    \section1 Browser Features

    Many of the functions, signals and properties provided by QWebView are also available

    for this item, making it simple to adapt existing code to use QGraphicsWebView instead

    of QWebView.

    The item uses a QWebPage object to perform the rendering of Web content, and this can

    be obtained with the page() function, enabling the document itself to be accessed and

    modified.

    As with QWebView, the item records the browsing history using a QWebHistory object,

    accessible using the history() function. The QWebSettings object that defines the

    configuration of the browser can be obtained with the settings() function, enabling

    features like plugin support to be customized for each item.

    \sa QWebView, QGraphicsTextItem

*/

/*!

    \fn void QGraphicsWebView::titleChanged(const QString &title)

    This signal is emitted whenever the \a title of the main frame changes.

    \sa title()

*/

/*!

    \fn void QGraphicsWebView::urlChanged(const QUrl &url)

    This signal is emitted when the \a url of the view changes.

    \sa url(), load()

*/

/*!

    \fn void QGraphicsWebView::iconChanged()

    This signal is emitted whenever the icon of the page is loaded or changes.

    In order for icons to be loaded, you will need to set an icon database path

    using QWebSettings::setIconDatabasePath().

    \sa icon(), QWebSettings::setIconDatabasePath()

*/

/*!

    \fn void QGraphicsWebView::loadStarted()

    This signal is emitted when a new load of the page is started.

    \sa loadProgress(), loadFinished()

*/

/*!

    \fn void QGraphicsWebView::loadFinished(bool ok)

    This signal is emitted when a load of the page is finished.

    \a ok will indicate whether the load was successful or any error occurred.

    \sa loadStarted()

*/

/*!

    Constructs an empty QGraphicsWebView with parent \a parent.

    \sa load()

*/

QGraphicsWebView::QGraphicsWebView(QGraphicsItem* parent)

    : QGraphicsWidget(parent)

    , d(new QGraphicsWebViewPrivate(this))

{

#if QT_VERSION >= 0x040600

    setFlag(QGraphicsItem::ItemUsesExtendedStyleOption, true);

#endif

    setAcceptHoverEvents(true);

    setFocusPolicy(Qt::StrongFocus);

}

/*!

    Destroys the item.

*/

QGraphicsWebView::~QGraphicsWebView()

{

    if (d->page) {

#if QT_VERSION >= 0x040600

        d->page->d->view.clear();

#else

        d->page->d->view = 0;

#endif

        d->page->d->client = 0; // unset the page client

    }

    if (d->page && d->page->parent() == this)

        delete d->page;

    delete d;

}

/*!

    Returns a pointer to the underlying web page.

    \sa setPage()

*/

QWebPage* QGraphicsWebView::page() const

{

    if (!d->page) {

        QGraphicsWebView* that = const_cast<QGraphicsWebView*>(this);

        QWebPage* page = new QWebPage(that);

        // Default to not having a background, in the case

        // the page doesn't provide one.

        QPalette palette = QApplication::palette();

        palette.setBrush(QPalette::Base, QColor::fromRgbF(0, 0, 0, 0));

        page->setPalette(palette);

        that->setPage(page);

    }

    return d->page;

}

/*! \reimp

*/

void QGraphicsWebView::paint(QPainter* painter, const QStyleOptionGraphicsItem* option, QWidget*)

{

    page()->mainFrame()->render(painter, option->exposedRect.toRect());

}

/*! \reimp

*/

bool QGraphicsWebView::sceneEvent(QEvent* event)

{

    // Re-implemented in order to allows fixing event-related bugs in patch releases.

    return QGraphicsWidget::sceneEvent(event);

}

/*! \reimp

*/

QVariant QGraphicsWebView::itemChange(GraphicsItemChange change, const QVariant& value)

{

    switch (change) {

    // Differently from QWebView, it is interesting to QGraphicsWebView to handle

    // post mouse cursor change notifications. Reason: 'ItemCursorChange' is sent

    // as the first action in QGraphicsItem::setCursor implementation, and at that

    // item widget's cursor has not been effectively changed yet.

    // After cursor is properly set (at 'ItemCursorHasChanged' emission time), we

    // fire 'CursorChange'.

    case ItemCursorChange:

        return value;

    case ItemCursorHasChanged:

        QEvent event(QEvent::CursorChange);

        QApplication::sendEvent(this, &event);

        return value;

    }

    return QGraphicsWidget::itemChange(change, value);

}

/*! \reimp

*/

QSizeF QGraphicsWebView::sizeHint(Qt::SizeHint which, const QSizeF& constraint) const

{

    if (which == Qt::PreferredSize)

        return QSizeF(800, 600); // ###

    return QGraphicsWidget::sizeHint(which, constraint);

}

/*! \reimp

*/

QVariant QGraphicsWebView::inputMethodQuery(Qt::InputMethodQuery query) const

{

    if (d->page)

        return d->page->inputMethodQuery(query);

    return QVariant();

}

/*! \reimp

*/

bool QGraphicsWebView::event(QEvent* event)

{

    // Re-implemented in order to allows fixing event-related bugs in patch releases.

    if (d->page) {

#ifndef QT_NO_CONTEXTMENU

        if (event->type() == QEvent::GraphicsSceneContextMenu) {

            if (!isEnabled())

                return false;

            QGraphicsSceneContextMenuEvent* ev = static_cast<QGraphicsSceneContextMenuEvent*>(event);

            QContextMenuEvent fakeEvent(QContextMenuEvent::Reason(ev->reason()), ev->pos().toPoint());

            if (d->page->swallowContextMenuEvent(&fakeEvent)) {

                event->accept();

                return true;

            }

            d->page->updatePositionDependentActions(fakeEvent.pos());

        } else

#endif // QT_NO_CONTEXTMENU

        {

#ifndef QT_NO_CURSOR

#if QT_VERSION >= 0x040400

            if (event->type() == QEvent::CursorChange) {

                // An unsetCursor will set the cursor to Qt::ArrowCursor.

                // Thus this cursor change might be a QWidget::unsetCursor()

                // If this is not the case and it came from WebCore, the

                // QWebPageClient already has set its cursor internally

                // to Qt::ArrowCursor, so updating the cursor is always

                // right, as it falls back to the last cursor set by

                // WebCore.

                // FIXME: Add a QEvent::CursorUnset or similar to Qt.

                if (cursor().shape() == Qt::ArrowCursor)

                    d->resetCursor();

            }

#endif

#endif

        }

    }

    return QGraphicsWidget::event(event);

}

/*!

    Makes \a page the new web page of the web graphicsitem.

    The parent QObject of the provided page remains the owner

    of the object. If the current document is a child of the web

    view, it will be deleted.

    \sa page()

*/

void QGraphicsWebView::setPage(QWebPage* page)

{

    if (d->page == page)

        return;

    if (d->page) {

        d->page->d->client = 0; // unset the page client

        if (d->page->parent() == this)

            delete d->page;

        else

            d->page->disconnect(this);

    }

    d->page = page;

    if (!d->page)

        return;

    d->page->d->client = d; // set the page client

    QSize size = geometry().size().toSize();

    page->setViewportSize(size);

    QWebFrame* mainFrame = d->page->mainFrame();

    connect(mainFrame, SIGNAL(titleChanged(QString)),

            this, SIGNAL(titleChanged(QString)));

    connect(mainFrame, SIGNAL(iconChanged()),

            this, SIGNAL(iconChanged()));

    connect(mainFrame, SIGNAL(urlChanged(QUrl)),

            this, SIGNAL(urlChanged(QUrl)));

    connect(d->page, SIGNAL(loadStarted()),

            this, SIGNAL(loadStarted()));

    connect(d->page, SIGNAL(loadProgress(int)),

            this, SIGNAL(loadProgress(int)));

    connect(d->page, SIGNAL(loadFinished(bool)),

            this, SLOT(_q_doLoadFinished(bool)));

    connect(d->page, SIGNAL(statusBarMessage(QString)),

            this, SIGNAL(statusBarMessage(QString)));

    connect(d->page, SIGNAL(linkClicked(QUrl)),

            this, SIGNAL(linkClicked(QUrl)));

}

/*!

    \property QGraphicsWebView::url

    \brief the url of the web page currently viewed

    Setting this property clears the view and loads the URL.

    By default, this property contains an empty, invalid URL.

    \sa load(), urlChanged()

*/

void QGraphicsWebView::setUrl(const QUrl &url)

{

    page()->mainFrame()->setUrl(url);

}

QUrl QGraphicsWebView::url() const

{

    if (d->page)

        return d->page->mainFrame()->url();

    return QUrl();

}

/*!

    \property QGraphicsWebView::title

    \brief the title of the web page currently viewed

    By default, this property contains an empty string.

    \sa titleChanged()

*/

QString QGraphicsWebView::title() const

{

    if (d->page)

        return d->page->mainFrame()->title();

    return QString();

}

/*!

    \property QGraphicsWebView::icon

    \brief the icon associated with the web page currently viewed

    By default, this property contains a null icon.

    \sa iconChanged(), QWebSettings::iconForUrl()

*/