/*
 * Bittorrent Client using Qt and libtorrent.
 * Copyright (C) 2015  Vladimir Golovnev <glassez@yandex.ru>
 * Copyright (C) 2006  Christophe Dumez <chris@qbittorrent.org>
 *
 * 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 2
 * 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, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 *
 * In addition, as a special exception, the copyright holders give permission to
 * link this program with the OpenSSL project's "OpenSSL" library (or with
 * modified versions of it that use the same license as the "OpenSSL" library),
 * and distribute the linked executables. You must obey the GNU General Public
 * License in all respects for all of the code used other than "OpenSSL".  If you
 * modify file(s), you may extend this exception to your version of the file(s),
 * but you are not obligated to do so. If you do not wish to do so, delete this
 * exception statement from your version.
 */

#ifndef NET_DOWNLOADHANDLER_H
#define NET_DOWNLOADHANDLER_H

#include <QObject>

QT_BEGIN_NAMESPACE
class QNetworkAccessManager;
class QNetworkReply;
class QUrl;
QT_END_NAMESPACE

namespace Net
{
    class DownloadManager;

    class DownloadHandler : public QObject
    {
        Q_OBJECT

    public:
        DownloadHandler(QNetworkReply *reply, DownloadManager *manager, bool saveToFile = false, qint64 limit = 0, bool handleRedirectToMagnet = false);
        ~DownloadHandler();

        QString url() const;

    signals:
        void downloadFinished(const QString &url, const QByteArray &data);
        void downloadFinished(const QString &url, const QString &filePath);
        void downloadFailed(const QString &url, const QString &reason);
        void redirectedToMagnet(const QString &url, const QString &magnetUri);

    private slots:
        void processFinishedDownload();
        void checkDownloadSize(qint64 bytesReceived, qint64 bytesTotal);

    private:
        void init();
        bool saveToFile(const QByteArray &replyData, QString &filePath);
        void handleRedirection(QUrl newUrl);

        QNetworkReply *m_reply;
        DownloadManager *m_manager;
        bool m_saveToFile;
        qint64 m_sizeLimit;
        bool m_handleRedirectToMagnet;
        QString m_url;
    };
}

#endif // NET_DOWNLOADHANDLER_H
