release/src/router/transmission/qt/favicon.cc

   1 /*
   2  * This file Copyright (C) 2012-2014 Mnemosyne LLC
   3  *
   4  * It may be used under the GNU Public License v2 or v3 licenses,
   5  * or any future license endorsed by Mnemosyne LLC.
   6  *
   7  * $Id: favicon.cc 14225 2014-01-19 01:09:44Z jordan $
   8  */
   9
  10 #include <QDir>
  11 #include <QNetworkAccessManager>
  12 #include <QNetworkReply>
  13 #include <QNetworkRequest>
  14
  15 #if QT_VERSION < QT_VERSION_CHECK(5, 0, 0)
  16  #include <QDesktopServices>
  17 #else
  18  #include <QStandardPaths>
  19 #endif
  20
  21 #include "favicon.h"
  22
  23 /***
  24 ****
  25 ***/
  26
  27 Favicons :: Favicons ()
  28 {
  29   myNAM = new QNetworkAccessManager ();
  30   connect (myNAM, SIGNAL(finished(QNetworkReply*)), this, SLOT(onRequestFinished(QNetworkReply*)));
  31 }
  32
  33 Favicons :: ~Favicons ()
  34 {
  35   delete myNAM;
  36 }
  37
  38 /***
  39 ****
  40 ***/
  41
  42 QString
  43 Favicons :: getCacheDir ()
  44 {
  45   const QString base =
  46 #if QT_VERSION < QT_VERSION_CHECK(5, 0, 0)
  47     QDesktopServices::storageLocation (QDesktopServices::CacheLocation);
  48 #else
  49     QStandardPaths::writableLocation (QStandardPaths::CacheLocation);
  50 #endif
  51
  52   return QDir(base).absoluteFilePath ("favicons");
  53 }
  54
  55 void
  56 Favicons :: ensureCacheDirHasBeenScanned ()
  57 {
  58   static bool hasBeenScanned = false;
  59
  60   if (!hasBeenScanned)
  61     {
  62       hasBeenScanned = true;
  63
  64       QDir cacheDir (getCacheDir ());
  65       cacheDir.mkpath (cacheDir.absolutePath ());
  66
  67       QStringList files = cacheDir.entryList (QDir::Files|QDir::Readable);
  68       foreach (QString file, files)
  69         {
  70           QPixmap pixmap;
  71           pixmap.load (cacheDir.absoluteFilePath (file));
  72           if (!pixmap.isNull ())
  73             myPixmaps.insert (file, pixmap);
  74         }
  75     }
  76 }
  77
  78 QString
  79 Favicons :: getHost (const QUrl& url)
  80 {
  81   QString host = url.host ();
  82   const int first_dot = host.indexOf ('.');
  83   const int last_dot = host.lastIndexOf ('.');
  84
  85   if ((first_dot != -1) && (last_dot != -1) &&  (first_dot != last_dot))
  86     host.remove (0, first_dot + 1);
  87
  88   return host;
  89 }
  90
  91 QPixmap
  92 Favicons :: find (const QUrl& url)
  93 {
  94   return findFromHost (getHost (url));
  95 }
  96
  97 namespace
  98 {
  99   const QSize rightSize (16, 16);
 100 };
 101
 102 QPixmap
 103 Favicons :: findFromHost (const QString& host)
 104 {
 105   ensureCacheDirHasBeenScanned ();
 106
 107   const QPixmap pixmap = myPixmaps[ host ];
 108   return pixmap.size()==rightSize ? pixmap : pixmap.scaled(rightSize);
 109 }
 110
 111 void
 112 Favicons :: add (const QUrl& url)
 113 {
 114   ensureCacheDirHasBeenScanned ();
 115
 116   const QString host = getHost (url);
 117
 118   if (!myPixmaps.contains (host))
 119     {
 120       // add a placholder s.t. we only ping the server once per session
 121       QPixmap tmp (rightSize);
 122       tmp.fill (Qt::transparent);
 123       myPixmaps.insert (host, tmp);
 124
 125       // try to download the favicon
 126       const QString path = "http://" + host + "/favicon.";
 127       QStringList suffixes;
 128       suffixes << "ico" << "png" << "gif" << "jpg";
 129       foreach (QString suffix, suffixes)
 130         myNAM->get (QNetworkRequest (path + suffix));
 131     }
 132 }
 133
 134 void
 135 Favicons :: onRequestFinished (QNetworkReply * reply)
 136 {
 137   const QString host = reply->url().host();
 138
 139   QPixmap pixmap;
 140
 141   const QByteArray content = reply->readAll ();
 142   if (!reply->error ())
 143     pixmap.loadFromData (content);
 144
 145   if (!pixmap.isNull ())
 146     {
 147       // save it in memory...
 148       myPixmaps.insert (host, pixmap);
 149
 150       // save it on disk...
 151       QDir cacheDir (getCacheDir ());
 152       cacheDir.mkpath (cacheDir.absolutePath ());
 153       QFile file (cacheDir.absoluteFilePath (host));
 154       file.open (QIODevice::WriteOnly);
 155       file.write (content);
 156       file.close ();
 157
 158       // notify listeners
 159       emit pixmapReady (host);
 160     }
 161 }