#include <Network/Downloader.h>
#include <memory>

#include <QNetworkRequest>
#include <QNetworkReply>
#include <QNetworkAccessManager>
#include <QAuthenticator>
#include <QVariant>
#include <QHash>
#include <QPair>
#include <QCoreApplication>
#include <QReadWriteLock>
#include <QThread>

class Downloader::p_Downloader
{
    using login_pair=QPair<QString,QString>;
    QNetworkAccessManager manager;
    QHash<QString,login_pair> auth;
    QReadWriteLock pending_requests_lock;
    QHash<QUuid,QNetworkReply*> pending_requests;

    QNetworkRequest buildRequest(const QString& url, const QString &user="", const QString &passwd="")
    {
        QNetworkRequest request;
        request.setUrl(QUrl(url));
        request.setRawHeader("User-Agent", "SciQLop 1.0");
        if(user!="" and passwd!="")
        {
            //might grow quickly since we can have tons of URLs for the same host
            auth[url]=login_pair(user,passwd);
            QString login = "Basic "+user+":"+passwd;
            request.setRawHeader("Authorization",login.toLocal8Bit());
        }
        return request;
    }

public:
    explicit p_Downloader()
    {

        auto login_bambda = [this](QNetworkReply * reply, QAuthenticator * authenticator)
        {
            if(auth.contains(reply->url().toString()))
            {
                auto login = auth[reply->url().toString()];
                authenticator->setUser(login.first);
                authenticator->setPassword(login.second);
            }
        };

        QObject::connect(&manager, &QNetworkAccessManager::authenticationRequired, login_bambda);
    }

    Response get(const QString& url, const QString &user="", const QString &passwd="")
    {
        QNetworkRequest request = buildRequest(url, user, passwd);
        QNetworkReply *reply = manager.get(request);
        while (!reply->isFinished())
        {
            QCoreApplication::processEvents();
            QThread::usleep(10000);
        }
        QVariant status_code = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);
        Response resp = Response(reply->readAll(), status_code.toInt());
        delete reply;
        if(user!="" and passwd!="")
            auth.remove(url);
        return resp;
    }

    QUuid getAsync(const QString &url, std::function<void (QUuid ,Response)> callback, const QString &user, const QString &passwd)
    {
        auto uuid = QUuid::createUuid();
        QNetworkRequest request = buildRequest(url, user, passwd);
        QNetworkReply *reply = manager.get(request);
        auto callback_wrapper = [url, uuid, callback, this](){
            QNetworkReply* reply;
            {
                QWriteLocker locker(&pending_requests_lock);
                reply = pending_requests.take(uuid);
            }
            QVariant status_code = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);
            Response resp = Response(reply->readAll(), status_code.toInt());
            auth.remove(url);
            delete reply;
            callback(uuid, resp);
        };
        QObject::connect(reply, &QNetworkReply::finished, callback_wrapper);
        {
            QWriteLocker locker(&pending_requests_lock);
            pending_requests[uuid] = reply;
        }
        return uuid;
    }
    bool downloadFinished(QUuid uuid)
    {
        QReadLocker locker(&pending_requests_lock);
        if(pending_requests.contains(uuid))
        {
            auto req = pending_requests[uuid];
            return req->isFinished();
        }
        return true;
    }
};

Response Downloader::get(const QString &url, const QString &user, const QString &passwd)
{
    return  Downloader::instance().impl->get(url, user, passwd);
}

QUuid Downloader::getAsync(const QString &url, std::function<void (QUuid ,Response)> callback, const QString &user, const QString &passwd)
{
    return  Downloader::instance().impl->getAsync(url, callback, user, passwd);
}

bool Downloader::downloadFinished(QUuid uuid)
{
    return Downloader::instance().impl->downloadFinished(uuid);
}

Downloader::Downloader()
    :impl(spimpl::make_unique_impl<p_Downloader>())
{
}