1
mirror of https://invent.kde.org/network/falkon.git synced 2024-11-11 17:42:10 +01:00
falkonOfficial/src/tools/iconprovider.cpp

116 lines
3.2 KiB
C++

/* ============================================================
* 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()) {
QIcon image;
QByteArray bArray = QByteArray::fromBase64(query.value(0).toByteArray());
QBuffer buffer(&bArray);
buffer.open(QIODevice::ReadOnly);
QDataStream in(&buffer);
in >> image;
buffer.close();
if (!image.isNull())
return image;
}
#ifdef Q_WS_X11
return QIcon::fromTheme("text-plain");
#else
return QIcon(":icons/locationbar/unknownpage.png");
#endif
}
void IconProvider::saveIconsToDatabase()
{
foreach (Icon ic, m_iconBuffer) {
QByteArray bArray;
QBuffer buffer(&bArray);
buffer.open(QIODevice::WriteOnly);
QDataStream out(&buffer);
out << ic.icon;
buffer.close();
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, bArray.toBase64());
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();
}