#include "searchablewebview.h"

#include "webview.h"

#include <QLineEdit>
#include <QShortcut>
#include <QStyle>
#include <QResizeEvent>

#ifdef USE_WEBENGINE
    #include <QWebEngineHistory>
    #include <QWebEnginePage>
#else
    #include <QWebFrame>
    #include <QWebHistory>
    #include <QWebPage>
#endif

SearchableWebView::SearchableWebView(QWidget *parent) :
    QWidget(parent),
    m_lineEdit(new QLineEdit(this)),
    m_webView(new WebView(this))
{
    m_webView->setAttribute(Qt::WA_AcceptTouchEvents, false);
    m_lineEdit->hide();
    connect(m_lineEdit, &QLineEdit::textChanged, [&](const QString &text) {
        // clear selection:
#ifdef USE_WEBENGINE
        m_webView->findText(text);
#else
        m_webView->findText(QString());
        m_webView->findText(QString(), QWebPage::HighlightAllOccurrences);
        if (!text.isEmpty()) {
            // select&scroll to one occurence:
            m_webView->findText(text, QWebPage::FindWrapsAroundDocument);
            // highlight other occurences:
            m_webView->findText(text, QWebPage::HighlightAllOccurrences);
        }
#endif

        // store text for later searches
        m_searchText = text;
    });

    QShortcut *shortcut = new QShortcut(QKeySequence::Find, this);
    connect(shortcut, &QShortcut::activated, [&] {
        m_lineEdit->show();
        m_lineEdit->setFocus();
    });

    connect(m_webView, &QWebView::loadFinished, [&](bool ok) {
        Q_UNUSED(ok)
        moveLineEdit();
    });

    connect(m_webView, &QWebView::urlChanged, this, &SearchableWebView::urlChanged);
    connect(m_webView, &QWebView::titleChanged, this, &SearchableWebView::titleChanged);
#ifdef USE_WEBENGINE
    // not implemented?
    // connect(m_webView->page(), &QWebPage::linkClicked, this, &SearchableWebView::linkClicked);
#else
    connect(m_webView, &QWebView::linkClicked, this, &SearchableWebView::linkClicked);
#endif

    connect(m_webView, &QWebView::loadStarted, [&]() {
        m_lineEdit->clear();
    });

    // Display tooltip showing link location when hovered over.
#ifdef USE_WEBENGINE
    connect(m_webView->page(), &QWebPage::linkHovered, [&](const QString &link) {
#else
    connect(m_webView->page(), &QWebPage::linkHovered,
            [&](const QString &link, const QString &title, const QString &textContent) {
        Q_UNUSED(title)
        Q_UNUSED(textContent)
#endif
        if (!link.startsWith(QLatin1String("file:///")))
            setToolTip(link);
    });
}

void SearchableWebView::setPage(QWebPage *page)
{
    m_webView->setPage(page);
}

int SearchableWebView::zealZoomFactor() const
{
    return m_webView->zealZoomFactor();
}

void SearchableWebView::setZealZoomFactor(int zf)
{
    m_webView->setZealZoomFactor(zf);
}

void SearchableWebView::load(const QUrl &url)
{
    m_webView->load(url);
}

void SearchableWebView::focus()
{
    m_webView->setFocus();
}

QWebPage *SearchableWebView::page() const
{
    return m_webView->page();
}

QSize SearchableWebView::sizeHint() const
{
    return m_webView->sizeHint();
}

void SearchableWebView::back()
{
    m_webView->back();
}

void SearchableWebView::forward()
{
    m_webView->forward();
}

bool SearchableWebView::canGoBack() const
{
    return m_webView->history()->canGoBack();
}

bool SearchableWebView::canGoForward() const
{
    return m_webView->history()->canGoForward();
}

void SearchableWebView::keyPressEvent(QKeyEvent *event)
{
    if (event->key() == Qt::Key_Return || event->key() == Qt::Key_Enter) {
#ifdef USE_WEBENGINE
        QWebPage::FindFlags flags = 0;
#else
        QWebPage::FindFlags flags = QWebPage::FindWrapsAroundDocument;
#endif
        if (event->modifiers() & Qt::ShiftModifier)
            flags |= QWebPage::FindBackward;
        m_webView->findText(m_searchText, flags);
    }

    if (event->key() == Qt::Key_Slash) {
        m_lineEdit->show();
        m_lineEdit->setFocus();
    }

    // Ignore all other events and pass them to the parent widget.
    event->ignore();
}

void SearchableWebView::resizeEvent(QResizeEvent *event)
{
    QWidget::resizeEvent(event);
    m_webView->resize(event->size().width(), event->size().height());
    moveLineEdit();
}

void SearchableWebView::moveLineEdit()
{
    int frameWidth = style()->pixelMetric(QStyle::PM_DefaultFrameWidth);
#ifdef USE_WEBENGINE
    /// FIXME: scrollbar width
#else
    frameWidth += m_webView->page()->currentFrame()->scrollBarGeometry(Qt::Vertical).width();
#endif
    m_lineEdit->move(rect().right() - frameWidth - m_lineEdit->sizeHint().width(), rect().top());
    m_lineEdit->raise();
}