src/filedownloader.h

   1 /*
   2   Copyright 2013-2015 Mats Sjöberg
   3
   4   This file is part of the Pumpa programme.
   5
   6   Pumpa is free software: you can redistribute it and/or modify it
   7   under the terms of the GNU General Public License as published by
   8   the Free Software Foundation, either version 3 of the License, or
   9   (at your option) any later version.
  10
  11   Pumpa is distributed in the hope that it will be useful, but WITHOUT
  12   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  13   or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public
  14   License for more details.
  15
  16   You should have received a copy of the GNU General Public License
  17   along with Pumpa.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19
  20 #ifndef FILEDOWNLOADER_H
  21 #define FILEDOWNLOADER_H
  22
  23 #include <QtCore>
  24 #include <QtNetwork>
  25 #include <QPixmap>
  26 #include <QMovie>
  27
  28 #include "QtKOAuth"
  29
  30 //------------------------------------------------------------------------------
  31
  32 class FileDownloader;  // forward declaration
  33
  34 class FileDownloadManager : public QObject {
  35   Q_OBJECT
  36
  37   FileDownloadManager(QObject*);
  38  public:
  39
  40   static FileDownloadManager* getManager(QObject* = 0);
  41
  42   bool hasFile(QString url);
  43
  44   QString fileName(QString url);
  45
  46   QPixmap pixmap(QString url, QString brokenImage = "");
  47   QMovie* movie(QString url);
  48   bool supportsAnimation(QString url);
  49
  50   FileDownloader* download(QString url);
  51
  52   void dumpStats();
  53
  54  private slots:
  55   void onSslErrors(QNetworkReply*, QList<QSslError>);
  56   void onAuthorizedRequestReady(QByteArray response, int id);
  57   void onFileReady(QString = "");
  58
  59  private:
  60   void executeAuthorizedRequest(KQOAuthRequest*, FileDownloader*);
  61   QString urlToPath(QString url);
  62
  63   QNetworkAccessManager* m_nam;
  64   KQOAuthManager* m_oam;
  65
  66   QMap<QString, FileDownloader*> m_inProgress;
  67   QMap<QString, QString> m_urlMap;
  68   QString m_cacheDir;
  69
  70   int m_nextRequestId;
  71
  72   typedef QPair<KQOAuthRequest*, FileDownloader*> requestData_t;
  73   QMap<int, requestData_t> m_requestMap;
  74
  75   static FileDownloadManager* s_instance;
  76
  77   friend class FileDownloader;
  78 };
  79
  80 //------------------------------------------------------------------------------
  81
  82 class FileDownloader : public QObject {
  83   Q_OBJECT
  84
  85  public:
  86   FileDownloader(QString url, FileDownloadManager* fdm);
  87
  88   QString url() const { return m_url; }
  89
  90  signals:
  91   void networkError(QString);
  92   void fileReady();
  93
  94  private slots:
  95     void replyFinished();
  96
  97  private:
  98   void requestReady(QByteArray response, KQOAuthRequest* oar);
  99   static void resizeImage(QPixmap pix, QString fn);
 100
 101   QString m_url;
 102   KQOAuthRequest* m_oar;
 103   int m_redirs;
 104
 105   FileDownloadManager* m_fdm;
 106
 107   friend class FileDownloadManager;
 108 };
 109
 110 #endif