#include <QNetworkReply>
#include <QNetworkRequest>
#include <QJsonDocument>
#include <QJsonObject>
#include <QJsonArray>

#include "webdownloader.h"

WebDownloader::WebDownloader(QObject *parent) : QObject(parent) {
    mAccessMgr = new QNetworkAccessManager(this);
    connect(mAccessMgr, &QNetworkAccessManager::finished, this, &WebDownloader::dlFinished);
}

void WebDownloader::fetchData(QString artist, QString album){
    mArtist = artist;
    mAlbum = album;
    QString artistQ = QString("artist:\"%1\"").arg(artist);
    QByteArray peq = QUrl::toPercentEncoding(artistQ);
    QString url = QString("https://musicbrainz.org/ws/2/release/?query=%1&fmt=json&limit=100").arg(QString(peq));
    QUrl aUrl(url);
    QNetworkRequest req(aUrl);
    req.setHeader(QNetworkRequest::UserAgentHeader, "Beetplayer/1.0.0 (beetplayer@d-tor.org");
    mAccessMgr->get(req);
}

void WebDownloader::dlFinished(QNetworkReply *reply){
    mData.clear();
    mOtherData.clear();
    QByteArray res = reply->readAll();
    QJsonDocument jDoc = QJsonDocument::fromJson(res);
    QJsonObject docObj = jDoc.object();
    QJsonArray relArr = docObj.value("releases").toArray();
    QJsonArray::const_iterator it = std::find_if(relArr.constBegin(), relArr.constEnd(), [this](auto c) {
        QString title = c.toObject().value("title").toString().toLower();
        return title == mAlbum;
    });
    if(it != relArr.constEnd()){
        QJsonValue titleRes = *it;
        QString curId = titleRes.toObject().value("release-group").toObject().value("id").toString();
        if(!mData.contains(curId)){
            mData << curId;
        }
        QString aId = titleRes.toObject().value("artist-credit").toArray().at(0).toObject().value("artist").toObject().value("id").toString();
        mArtistId = aId;
        std::for_each(relArr.constBegin(), relArr.constEnd(), [this, aId](auto c){
            QString id = c.toObject().value("artist-credit").toArray().at(0).toObject().value("artist").toObject().value("id").toString();
            if(id == aId){
                QString primary = c.toObject().value("release-group").toObject().value("primary-type").toString().toLower();
                QString status = c.toObject().value("status").toString().toLower();
                if(primary == "album" && status == "official"){
                    QString linkId = c.toObject().value("release-group").toObject().value("id").toString();
                    QString title = c.toObject().value("title").toString().toLower();
                    mOtherData.insert(linkId, title);
                }
            }
        });
    }
    reply->deleteLater();
    emit done();
}