/* ============================================================
* QupZilla - WebKit based browser
* Copyright (C) 2010-2011  nowrep
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* 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, see <http://www.gnu.org/licenses/>.
* ============================================================ */
/* ============================================================
*
* Copyright (C) 2009 by Benjamin C. Meyer <ben@meyerhome.net>
* Copyright (C) 2010 by Matthieu Gicquel <matgic78@gmail.com>
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License as
* published by the Free Software Foundation; either version 2 of
* the License or (at your option) version 3 or any later version
* accepted by the membership of KDE e.V. (or its successor approved
* by the membership of KDE e.V.), which shall act as a proxy
* defined in Section 14 of version 3 of the license.
*
* 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, see <http://www.gnu.org/licenses/>.
*
* ============================================================ */

#include "clicktoflash.h"
#include "clickablelabel.h"
#include "mainapplication.h"
#include "pluginproxy.h"
#include "adblockmanager.h"
#include "adblocksubscription.h"

ClickToFlash::ClickToFlash(const QUrl &pluginUrl, const QStringList &argumentNames, const QStringList &argumentValues, QWidget* parent)
        : QWidget(parent)
        , m_argumentNames(argumentNames)
        , m_argumentValues(argumentValues)
        , m_url(pluginUrl)
{
    //AdBlock
    AdBlockManager* manager = AdBlockManager::instance();
    if (manager->isEnabled()) {
        QString urlString = pluginUrl.toEncoded();
        AdBlockSubscription* subscription = manager->subscription();
        if (!subscription->allow(urlString) && subscription->block(urlString)) {
            QTimer::singleShot(200, this, SLOT(hideAdBlocked()));
            return;
        }
    }
    QHBoxLayout* horizontalLayout;
    QFrame* frame;
    QHBoxLayout* horizontalLayout_2;
    QToolButton* toolButton;

    horizontalLayout = new QHBoxLayout(this);
    frame = new QFrame(this);
    frame->setStyleSheet("QFrame { border: 1px solid #e8e8e8; }");
    frame->setContentsMargins(0,0,0,0);
    horizontalLayout_2 = new QHBoxLayout(frame);
    toolButton = new QToolButton(frame);
    toolButton->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);
    toolButton->setStyleSheet("QToolButton { background: url(:/icons/other/flash.png) no-repeat;\n"
        "background-position: center; border: none;}\n"
        "QToolButton:hover { background: url(:/icons/other/flashstart.png) no-repeat; \n"
        "background-position: center;  border:none;}");
    toolButton->setCursor(Qt::PointingHandCursor);
    horizontalLayout_2->addWidget(toolButton);
    horizontalLayout->addWidget(frame);
    horizontalLayout->setContentsMargins(0,0,0,0);
    horizontalLayout_2->setContentsMargins(0,0,0,0);

    connect(toolButton, SIGNAL(clicked(bool)), this, SLOT(load()));
    setMinimumSize(27,27);

    setContextMenuPolicy(Qt::CustomContextMenu);
    connect(this, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(customContextMenuRequested(QPoint)));
}

void ClickToFlash::customContextMenuRequested(const QPoint &pos)
{
    QMenu menu;
    menu.addAction(tr("Flash blocked by ClickToFlash"));
    menu.addSeparator();
    menu.addAction(tr("Add %1 to whitelist").arg(m_url.host()), this, SLOT(toWhitelist()));
    menu.actions().at(0)->setEnabled(false);
    menu.exec(mapToGlobal(pos));
}

void ClickToFlash::toWhitelist()
{
    mApp->plugins()->c2f_addWhitelist(m_url.host());
    load();
}

void ClickToFlash::hideAdBlocked()
{
    findElement();
    if (!m_element.isNull()) {
        m_element.setAttribute("style", "display:none;");
        //deleteLater(); //Well, it should be there, but therefore it sometimes crashes
    }
}

void ClickToFlash::findElement()
{
    QWidget* parent = parentWidget();
    QWebView* view = 0;
    while (parent) {
        if (QWebView* aView = qobject_cast<QWebView*>(parent)) {
            view = aView;
            break;
        }
        parent = parent->parentWidget();
    }
    if (!view)
        return;

    QList<QWebFrame*> frames;
    m_mainFrame = view->page()->mainFrame();
    frames.append(m_mainFrame);

    while (!frames.isEmpty()) {
        QWebFrame* frame = frames.takeFirst();
        QWebElement docElement = frame->documentElement();

        QWebElementCollection elements;
        elements.append(docElement.findAll(QLatin1String("embed")));
        elements.append(docElement.findAll(QLatin1String("object")));

        QWebElement element;
        foreach (element, elements) {
            if (!checkElement(element) && !checkUrlOnElement(element))
                continue;
            m_element = element;
            return;
        }
        frames += frame->childFrames();
    }
}

void ClickToFlash::load()
{
    findElement();
    if (m_element.isNull()) {
        qWarning("Click2Flash: Cannot find Flash object.");
    } else {
        QWebElement substitute = m_element.clone();
        substitute.setAttribute(QLatin1String("type"), "application/futuresplash");
        m_element.replace(substitute);
    }
}

bool ClickToFlash::checkUrlOnElement(QWebElement el)
{
    QString checkString = QUrl(el.attribute("src")).toString(QUrl::RemoveQuery);
    if (checkString.isEmpty())
        checkString = QUrl(el.attribute("data")).toString(QUrl::RemoveQuery);
    if (checkString.isEmpty())
        checkString = QUrl(el.attribute("value")).toString(QUrl::RemoveQuery);

    if (m_url.toEncoded().contains(checkString.toAscii()))
        return true;
    return false;
}

bool ClickToFlash::checkElement(QWebElement el)
{
    if (m_argumentNames == el.attributeNames()) {
        foreach (QString name, m_argumentNames) {
            if (m_argumentValues.indexOf(el.attribute(name)) == -1)
                return false;
        }
        return true;
    }
    return false;
}