/* ============================================================
* 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/>.
* ============================================================ */
#include "iconprovider.h"
#include "webview.h"

IconProvider::IconProvider(QObject* parent) :
    QObject(parent)
{
    m_timer = new QTimer(this);
    m_timer->setInterval(10*1000);
    m_timer->start();
    connect(m_timer, SIGNAL(timeout()), this, SLOT(saveIconsToDatabase()));
}

void IconProvider::saveIcon(WebView* view)
{
    Icon item;
    item.icon = view->icon();
    item.url = view->url();

    if (item.icon.isNull())
        return;

    QPixmap iconPixmap = item.icon.pixmap(16,16);
    foreach (Icon ic, m_iconBuffer) {
        if (ic.url == item.url && ic.icon.pixmap(16,16).toImage() == iconPixmap.toImage())
            return;
    }

    item.icon = QIcon(iconPixmap);
    m_iconBuffer.append(item);
}

QIcon IconProvider::iconForUrl(const QUrl &url)
{
    foreach (Icon ic, m_iconBuffer) {
        if (ic.url == url)
            return ic.icon;
    }

    QSqlQuery query;
    query.prepare("SELECT icon FROM icons WHERE url=?");
    query.bindValue(0, url.toEncoded(QUrl::RemoveFragment));
    query.exec();
    if (query.next())
        return iconFromBase64(query.value(0).toByteArray());

    return QWebSettings::webGraphic(QWebSettings::DefaultFrameIconGraphic);
}

void IconProvider::saveIconsToDatabase()
{
    foreach (Icon ic, m_iconBuffer) {
        QSqlQuery query;
        query.prepare("SELECT id FROM icons WHERE url = ?");
        query.bindValue(0, ic.url.toEncoded(QUrl::RemoveFragment));
        query.exec();

        if (query.next())
            query.prepare("UPDATE icons SET icon = ? WHERE url = ?");
        else
            query.prepare("INSERT INTO icons (icon, url) VALUES (?,?)");

        query.bindValue(0, iconToBase64(ic.icon));
        query.bindValue(1, ic.url.toEncoded(QUrl::RemoveFragment));
        query.exec();
    }

    m_iconBuffer.clear();
}

void IconProvider::clearIconDatabase()
{
    QSqlQuery query;
    query.exec("DELETE FROM icons");
    query.exec("VACUUM");

    m_iconBuffer.clear();
}

QIcon IconProvider::iconFromBase64(const QByteArray &data)
{
    QIcon image;
    QByteArray bArray = QByteArray::fromBase64(data);
    QBuffer buffer(&bArray);
    buffer.open(QIODevice::ReadOnly);
    QDataStream in(&buffer);
    in >> image;
    buffer.close();

    if (!image.isNull())
        return image;
    return QWebSettings::webGraphic(QWebSettings::DefaultFrameIconGraphic);
}

QByteArray IconProvider::iconToBase64(const QIcon &icon)
{
    QByteArray bArray;
    QBuffer buffer(&bArray);
    buffer.open(QIODevice::WriteOnly);
    QDataStream out(&buffer);
    out << icon;
    buffer.close();
    return bArray.toBase64();
}