1
mirror of https://invent.kde.org/network/falkon.git synced 2024-09-22 18:22:10 +02:00
falkonOfficial/src/lib/plugins/speeddial.cpp
nowrep 528f398cbe Using QLatin1String and QLatin1Char now everywhere
- it saves few unnecessary allocations
2012-09-04 12:42:45 +02:00

408 lines
9.9 KiB
C++

/* ============================================================
* QupZilla - WebKit based browser
* Copyright (C) 2010-2012 David Rosca <nowrep@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 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 "speeddial.h"
#include "mainapplication.h"
#include "pagethumbnailer.h"
#include "settings.h"
#include <QDir>
#include <QCryptographicHash>
#include <QFileDialog>
#include <QDebug>
#include <QWebFrame>
#include <QWebPage>
#include <QImage>
#define ENSURE_LOADED if (!m_loaded) loadSettings();
SpeedDial::SpeedDial(QObject* parent)
: QObject(parent)
, m_maxPagesInRow(4)
, m_sizeOfSpeedDials(231)
, m_loaded(false)
, m_regenerateScript(true)
{
}
void SpeedDial::loadSettings()
{
m_loaded = true;
Settings settings;
settings.beginGroup("SpeedDial");
QString allPages = settings.value("pages", QString()).toString();
m_backgroundImage = settings.value("background", QString()).toString();
m_backgroundImageSize = settings.value("backsize", "auto").toString();
m_maxPagesInRow = settings.value("pagesrow", 4).toInt();
m_sizeOfSpeedDials = settings.value("sdsize", 231).toInt();
m_sdcentered = settings.value("sdcenter", 0).toInt();
settings.endGroup();
if (allPages.isEmpty()) {
allPages = "url:\"http://www.google.com\"|title:\"Google\";"
"url:\"http://www.qupzilla.com\"|title:\"QupZilla\";"
"url:\"http://blog.qupzilla.com\"|title:\"QupZilla Blog\";"
"url:\"https://github.com/QupZilla/qupzilla\"|title:\"QupZilla GitHub\";"
"url:\"https://facebook.com\"|title:\"Facebook\";";
}
changed(allPages);
m_thumbnailsDir = mApp->currentProfilePath() + "thumbnails/";
// If needed, create thumbnails directory
if (!QDir(m_thumbnailsDir).exists()) {
QDir(mApp->currentProfilePath()).mkdir("thumbnails");
}
}
void SpeedDial::saveSettings()
{
ENSURE_LOADED;
if (m_webPages.isEmpty()) {
return;
}
Settings settings;
settings.beginGroup("SpeedDial");
settings.setValue("pages", generateAllPages());
settings.setValue("background", m_backgroundImage);
settings.setValue("backsize", m_backgroundImageSize);
settings.setValue("pagesrow", m_maxPagesInRow);
settings.setValue("sdsize", m_sizeOfSpeedDials);
settings.setValue("sdcenter", m_sdcentered);
settings.endGroup();
}
SpeedDial::Page SpeedDial::pageForUrl(const QUrl &url)
{
ENSURE_LOADED;
const QString &urlString = url.toString();
foreach(const Page & page, m_webPages) {
if (page.url == urlString) {
return page;
}
}
return Page();
}
QUrl SpeedDial::urlForShortcut(int key)
{
ENSURE_LOADED;
if (key < 0 || m_webPages.count() <= key) {
return QUrl();
}
return QUrl::fromEncoded(m_webPages.at(key).url.toUtf8());
}
void SpeedDial::addWebFrame(QWebFrame* frame)
{
if (!m_webFrames.contains(frame)) {
m_webFrames.append(frame);
}
}
void SpeedDial::addPage(const QUrl &url, const QString &title)
{
ENSURE_LOADED;
if (url.isEmpty()) {
return;
}
Page page;
page.title = escapeTitle(title);
page.url = escapeUrl(url.toString());
m_webPages.append(page);
m_regenerateScript = true;
foreach(QWebFrame * frame, cleanFrames()) {
frame->page()->triggerAction(QWebPage::Reload);
}
emit pagesChanged();
}
void SpeedDial::removePage(const Page &page)
{
ENSURE_LOADED;
if (page.url.isEmpty()) {
return;
}
removeImageForUrl(page.url);
m_webPages.removeAll(page);
m_regenerateScript = true;
foreach(QWebFrame * frame, cleanFrames()) {
frame->page()->triggerAction(QWebPage::Reload);
}
emit pagesChanged();
}
int SpeedDial::pagesInRow()
{
ENSURE_LOADED;
return m_maxPagesInRow;
}
int SpeedDial::sdSize()
{
ENSURE_LOADED;
return m_sizeOfSpeedDials;
}
int SpeedDial::sdCntr()
{
ENSURE_LOADED;
return m_sdcentered;
}
QString SpeedDial::backgroundImage()
{
ENSURE_LOADED;
return m_backgroundImage;
}
QString SpeedDial::backgroundImageSize()
{
ENSURE_LOADED;
return m_backgroundImageSize;
}
QString SpeedDial::initialScript()
{
ENSURE_LOADED;
if (!m_regenerateScript) {
return m_initialScript;
}
m_regenerateScript = false;
m_initialScript.clear();
foreach(const Page & page, m_webPages) {
QString imgSource = m_thumbnailsDir + QCryptographicHash::hash(page.url.toUtf8(), QCryptographicHash::Md4).toHex() + ".png";
if (!QFile(imgSource).exists()) {
loadThumbnail(page.url);
imgSource = "qrc:html/loading.gif";
if (page.url.isEmpty()) {
imgSource.clear();
}
}
else {
imgSource = QUrl::fromLocalFile(imgSource).toString();
}
m_initialScript.append(QString("addBox('%1', '%2', '%3');\n").arg(page.url, page.title, imgSource));
}
return m_initialScript;
}
void SpeedDial::changed(const QString &allPages)
{
if (allPages.isEmpty()) {
return;
}
const QStringList &entries = allPages.split(QLatin1String("\";"));
m_webPages.clear();
foreach(const QString & entry, entries) {
if (entry.isEmpty()) {
continue;
}
const QStringList &tmp = entry.split(QLatin1String("\"|"));
if (tmp.count() != 2) {
continue;
}
Page page;
page.url = tmp.at(0).mid(5);
page.title = tmp.at(1).mid(7);
m_webPages.append(page);
}
m_regenerateScript = true;
emit pagesChanged();
}
void SpeedDial::loadThumbnail(const QString &url, bool loadTitle)
{
if (url.isEmpty()) {
return;
}
PageThumbnailer* thumbnailer = new PageThumbnailer(this);
thumbnailer->setUrl(QUrl::fromEncoded(url.toUtf8()));
thumbnailer->setLoadTitle(loadTitle);
connect(thumbnailer, SIGNAL(thumbnailCreated(const QPixmap &)), this, SLOT(thumbnailCreated(const QPixmap &)));
thumbnailer->start();
}
void SpeedDial::removeImageForUrl(const QString &url)
{
QString fileName = m_thumbnailsDir + QCryptographicHash::hash(url.toUtf8(), QCryptographicHash::Md4).toHex() + ".png";
if (QFile(fileName).exists()) {
QFile(fileName).remove();
}
}
QString SpeedDial::getOpenFileName()
{
const QString &fileTypes = QString("%3(*.png *.jpg *.jpeg *.bmp *.gif *.svg *.tiff)").arg(tr("Image files"));
const QString &image = QFileDialog::getOpenFileName(0, tr("Select image..."), QDir::homePath(), fileTypes);
if (image.isEmpty()) {
return image;
}
return QUrl::fromLocalFile(image).toEncoded();
}
QString SpeedDial::urlFromUserInput(const QString &url)
{
return QUrl::fromUserInput(url).toString();
}
void SpeedDial::setBackgroundImage(const QString &image)
{
m_backgroundImage = image;
}
void SpeedDial::setBackgroundImageSize(const QString &size)
{
m_backgroundImageSize = size;
}
void SpeedDial::setPagesInRow(int count)
{
m_maxPagesInRow = count;
}
void SpeedDial::setSdSize(int count)
{
m_sizeOfSpeedDials = count;
}
void SpeedDial::setSdCentered(int cntr)
{
m_sdcentered = cntr;
}
void SpeedDial::thumbnailCreated(const QPixmap &pixmap)
{
PageThumbnailer* thumbnailer = qobject_cast<PageThumbnailer*>(sender());
if (!thumbnailer) {
return;
}
bool loadTitle = thumbnailer->loadTitle();
QString title = thumbnailer->title();
QString url = thumbnailer->url().toString();
QString fileName = m_thumbnailsDir + QCryptographicHash::hash(url.toUtf8(), QCryptographicHash::Md4).toHex() + ".png";
if (pixmap.isNull()) {
fileName = "qrc:/html/broken-page.png";
title = tr("Unable to load");
loadTitle = true;
}
else {
if (!pixmap.save(fileName, "PNG")) {
qWarning() << "SpeedDial::thumbnailCreated Cannot save thumbnail to " << fileName;
}
fileName = QUrl::fromLocalFile(fileName).toString();
}
m_regenerateScript = true;
cleanFrames();
foreach(QWebFrame * frame, cleanFrames()) {
frame->evaluateJavaScript(QString("setImageToUrl('%1', '%2');").arg(escapeUrl(url), escapeTitle(fileName)));
if (loadTitle) {
frame->evaluateJavaScript(QString("setTitleToUrl('%1', '%2');").arg(escapeUrl(url), escapeTitle(title)));
}
}
thumbnailer->deleteLater();
}
QString SpeedDial::escapeTitle(QString title) const
{
title.replace(QLatin1Char('"'), QLatin1String("&quot;"));
title.replace(QLatin1Char('\''), QLatin1String("&apos;"));
return title;
}
QString SpeedDial::escapeUrl(QString url) const
{
url.remove(QLatin1Char('"'));
url.remove(QLatin1Char('\''));
return url;
}
QList<QWebFrame*> SpeedDial::cleanFrames()
{
QList<QWebFrame*> list;
for (int i = 0; i < m_webFrames.count(); i++) {
QWebFrame* frame = m_webFrames.at(i).data();
if (!frame || frame->url().toString() != "qupzilla:speeddial") {
m_webFrames.removeAt(i);
i--;
continue;
}
list.append(frame);
}
return list;
}
QString SpeedDial::generateAllPages()
{
QString allPages;
foreach(const Page & page, m_webPages) {
const QString &string = QString("url:\"%1\"|title:\"%2\";").arg(page.url, page.title);
allPages.append(string);
}
return allPages;
}