2018-02-27 18:06:05 +01:00
|
|
|
/*
|
|
|
|
* Strawberry Music Player
|
|
|
|
* This file was part of Clementine.
|
|
|
|
* Copyright 2012, Martin Björklund <mbj4668@gmail.com>
|
|
|
|
* Copyright 2016, Jonas Kvinge <jonas@jkvinge.net>
|
|
|
|
*
|
|
|
|
* Strawberry 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 3 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
|
|
|
* Strawberry 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 Strawberry. If not, see <http://www.gnu.org/licenses/>.
|
2018-08-09 18:39:44 +02:00
|
|
|
*
|
2018-02-27 18:06:05 +01:00
|
|
|
*/
|
|
|
|
|
|
|
|
#include "config.h"
|
|
|
|
|
2018-05-01 00:41:33 +02:00
|
|
|
#include <QObject>
|
2018-03-17 14:28:45 +01:00
|
|
|
#include <QByteArray>
|
|
|
|
#include <QList>
|
2018-05-01 00:41:33 +02:00
|
|
|
#include <QPair>
|
|
|
|
#include <QMap>
|
|
|
|
#include <QSet>
|
|
|
|
#include <QVariant>
|
2018-03-17 14:28:45 +01:00
|
|
|
#include <QString>
|
2018-05-01 00:41:33 +02:00
|
|
|
#include <QStringBuilder>
|
2018-02-27 18:06:05 +01:00
|
|
|
#include <QStringList>
|
2018-05-01 00:41:33 +02:00
|
|
|
#include <QUrl>
|
2018-02-27 18:06:05 +01:00
|
|
|
#include <QUrlQuery>
|
2018-05-01 00:41:33 +02:00
|
|
|
#include <QNetworkAccessManager>
|
|
|
|
#include <QNetworkRequest>
|
|
|
|
#include <QNetworkReply>
|
2018-03-02 22:51:42 +01:00
|
|
|
#include <QJsonDocument>
|
2018-03-17 14:28:45 +01:00
|
|
|
#include <QJsonObject>
|
2018-05-01 00:41:33 +02:00
|
|
|
#include <QVector>
|
2018-02-27 18:06:05 +01:00
|
|
|
|
|
|
|
#include "core/closure.h"
|
|
|
|
#include "core/logging.h"
|
|
|
|
#include "core/network.h"
|
|
|
|
#include "core/utilities.h"
|
2018-05-01 00:41:33 +02:00
|
|
|
#include "coverprovider.h"
|
|
|
|
#include "albumcoverfetcher.h"
|
|
|
|
#include "discogscoverprovider.h"
|
2018-02-27 18:06:05 +01:00
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
const char *DiscogsCoverProvider::kUrlSearch = "https://api.discogs.com/database/search";
|
|
|
|
const char *DiscogsCoverProvider::kUrlReleases = "https://api.discogs.com/releases";
|
2018-02-27 18:06:05 +01:00
|
|
|
|
|
|
|
const char *DiscogsCoverProvider::kAccessKeyB64 = "dGh6ZnljUGJlZ1NEeXBuSFFxSVk=";
|
2018-03-17 14:28:45 +01:00
|
|
|
const char *DiscogsCoverProvider::kSecretKeyB64 = "ZkFIcmlaSER4aHhRSlF2U3d0bm5ZVmdxeXFLWUl0UXI=";
|
2018-02-27 18:06:05 +01:00
|
|
|
|
2018-03-05 21:05:30 +01:00
|
|
|
DiscogsCoverProvider::DiscogsCoverProvider(QObject *parent) : CoverProvider("Discogs", false, parent), network_(new NetworkAccessManager(this)) {}
|
2018-02-27 18:06:05 +01:00
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
bool DiscogsCoverProvider::StartSearch(const QString &artist, const QString &album, int s_id) {
|
|
|
|
|
|
|
|
DiscogsCoverSearchContext *s_ctx = new DiscogsCoverSearchContext;
|
2018-03-17 14:28:45 +01:00
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
s_ctx->id = s_id;
|
|
|
|
s_ctx->artist = artist;
|
|
|
|
s_ctx->album = album;
|
|
|
|
s_ctx->r_count = 0;
|
|
|
|
requests_search_.insert(s_id, s_ctx);
|
|
|
|
SendSearchRequest(s_ctx);
|
2018-02-27 18:06:05 +01:00
|
|
|
|
|
|
|
return true;
|
2018-03-02 22:51:42 +01:00
|
|
|
|
2018-02-27 18:06:05 +01:00
|
|
|
}
|
2018-03-02 22:51:42 +01:00
|
|
|
|
2018-03-17 14:28:45 +01:00
|
|
|
void DiscogsCoverProvider::CancelSearch(int id) {
|
|
|
|
delete requests_search_.take(id);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
bool DiscogsCoverProvider::StartRelease(DiscogsCoverSearchContext *s_ctx, int r_id, QString resource_url) {
|
|
|
|
|
|
|
|
DiscogsCoverReleaseContext *r_ctx = new DiscogsCoverReleaseContext;
|
|
|
|
|
|
|
|
s_ctx->r_count++;
|
|
|
|
|
|
|
|
r_ctx->id = r_id;
|
|
|
|
r_ctx->resource_url = resource_url;
|
|
|
|
|
|
|
|
r_ctx->s_id = s_ctx->id;
|
|
|
|
|
|
|
|
requests_release_.insert(r_id, r_ctx);
|
|
|
|
SendReleaseRequest(s_ctx, r_ctx);
|
|
|
|
|
|
|
|
return true;
|
2018-10-02 00:38:52 +02:00
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void DiscogsCoverProvider::SendSearchRequest(DiscogsCoverSearchContext *s_ctx) {
|
2018-10-02 00:38:52 +02:00
|
|
|
|
2018-02-27 18:06:05 +01:00
|
|
|
typedef QPair<QString, QString> Arg;
|
|
|
|
typedef QList<Arg> ArgList;
|
|
|
|
|
|
|
|
typedef QPair<QByteArray, QByteArray> EncodedArg;
|
|
|
|
typedef QList<EncodedArg> EncodedArgList;
|
|
|
|
|
|
|
|
ArgList args = ArgList()
|
|
|
|
<< Arg("key", QByteArray::fromBase64(kAccessKeyB64))
|
2018-03-17 14:28:45 +01:00
|
|
|
<< Arg("secret", QByteArray::fromBase64(kSecretKeyB64));
|
2018-03-02 22:51:42 +01:00
|
|
|
|
|
|
|
args.append(Arg("type", "release"));
|
|
|
|
if (!s_ctx->artist.isEmpty()) {
|
|
|
|
args.append(Arg("artist", s_ctx->artist.toLower()));
|
2018-02-27 18:06:05 +01:00
|
|
|
}
|
2018-03-02 22:51:42 +01:00
|
|
|
if (!s_ctx->album.isEmpty()) {
|
|
|
|
args.append(Arg("release_title", s_ctx->album.toLower()));
|
2018-02-27 18:06:05 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
QUrlQuery url_query;
|
2018-03-02 22:51:42 +01:00
|
|
|
QUrl url(kUrlSearch);
|
2018-02-27 18:06:05 +01:00
|
|
|
QStringList query_items;
|
|
|
|
|
|
|
|
// Encode the arguments
|
|
|
|
for (const Arg &arg : args) {
|
|
|
|
EncodedArg encoded_arg(QUrl::toPercentEncoding(arg.first), QUrl::toPercentEncoding(arg.second));
|
|
|
|
query_items << QString(encoded_arg.first + "=" + encoded_arg.second);
|
|
|
|
url_query.addQueryItem(encoded_arg.first, encoded_arg.second);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Sign the request
|
2018-04-05 21:40:05 +02:00
|
|
|
const QByteArray data_to_sign = QString("GET\n%1\n%2\n%3").arg(url.host(), url.path(), query_items.join("&")).toUtf8();
|
2018-03-17 14:28:45 +01:00
|
|
|
const QByteArray signature(Utilities::HmacSha256(QByteArray::fromBase64(kSecretKeyB64), data_to_sign));
|
2018-02-27 18:06:05 +01:00
|
|
|
|
|
|
|
// Add the signature to the request
|
|
|
|
url_query.addQueryItem("Signature", QUrl::toPercentEncoding(signature.toBase64()));
|
|
|
|
|
|
|
|
url.setQuery(url_query);
|
|
|
|
QNetworkReply *reply = network_->get(QNetworkRequest(url));
|
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
NewClosure(reply, SIGNAL(error(QNetworkReply::NetworkError)), this, SLOT(SearchRequestError(QNetworkReply::NetworkError, QNetworkReply*, int)), reply, s_ctx->id);
|
|
|
|
NewClosure(reply, SIGNAL(finished()), this, SLOT(HandleSearchReply(QNetworkReply*, int)), reply, s_ctx->id);
|
2018-02-27 18:06:05 +01:00
|
|
|
|
|
|
|
}
|
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
void DiscogsCoverProvider::SendReleaseRequest(DiscogsCoverSearchContext *s_ctx, DiscogsCoverReleaseContext *r_ctx) {
|
2018-02-27 18:06:05 +01:00
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
typedef QPair<QString, QString> Arg;
|
|
|
|
typedef QList<Arg> ArgList;
|
|
|
|
|
|
|
|
typedef QPair<QByteArray, QByteArray> EncodedArg;
|
|
|
|
typedef QList<EncodedArg> EncodedArgList;
|
|
|
|
|
|
|
|
QUrlQuery url_query;
|
|
|
|
QStringList query_items;
|
|
|
|
|
|
|
|
ArgList args = ArgList()
|
|
|
|
<< Arg("key", QByteArray::fromBase64(kAccessKeyB64))
|
2018-03-17 14:28:45 +01:00
|
|
|
<< Arg("secret", QByteArray::fromBase64(kSecretKeyB64));
|
2018-03-02 22:51:42 +01:00
|
|
|
// Encode the arguments
|
|
|
|
for (const Arg &arg : args) {
|
|
|
|
EncodedArg encoded_arg(QUrl::toPercentEncoding(arg.first), QUrl::toPercentEncoding(arg.second));
|
|
|
|
query_items << QString(encoded_arg.first + "=" + encoded_arg.second);
|
|
|
|
url_query.addQueryItem(encoded_arg.first, encoded_arg.second);
|
|
|
|
}
|
|
|
|
|
|
|
|
QUrl url(r_ctx->resource_url);
|
|
|
|
|
|
|
|
// Sign the request
|
2018-03-17 14:28:45 +01:00
|
|
|
const QByteArray data_to_sign = QString("GET\n%1\n%2\n%3").arg(url.host(), url.path(), query_items.join("&")).toUtf8();
|
|
|
|
const QByteArray signature(Utilities::HmacSha256(QByteArray::fromBase64(kSecretKeyB64), data_to_sign));
|
2018-03-02 22:51:42 +01:00
|
|
|
|
|
|
|
// Add the signature to the request
|
|
|
|
url_query.addQueryItem("Signature", QUrl::toPercentEncoding(signature.toBase64()));
|
|
|
|
|
|
|
|
url.setQuery(url_query);
|
|
|
|
QNetworkReply *reply = network_->get(QNetworkRequest(url));
|
|
|
|
|
|
|
|
NewClosure(reply, SIGNAL(error(QNetworkReply::NetworkError)), this, SLOT(ReleaseRequestError(QNetworkReply::NetworkError, QNetworkReply*, int, int)), reply, s_ctx->id, r_ctx->id);
|
|
|
|
NewClosure(reply, SIGNAL(finished()), this, SLOT(HandleReleaseReply(QNetworkReply*, int, int)), reply, s_ctx->id, r_ctx->id);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
void DiscogsCoverProvider::HandleSearchReply(QNetworkReply *reply, int s_id) {
|
|
|
|
|
2018-02-27 18:06:05 +01:00
|
|
|
reply->deleteLater();
|
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
if (!requests_search_.contains(s_id)) {
|
2018-03-17 14:28:45 +01:00
|
|
|
//qLog(Error) << "Discogs: Got reply for cancelled request: " << s_id;
|
2018-03-02 22:51:42 +01:00
|
|
|
return;
|
|
|
|
}
|
2018-03-17 14:28:45 +01:00
|
|
|
DiscogsCoverSearchContext *s_ctx = requests_search_.value(s_id);
|
2018-10-02 00:38:52 +02:00
|
|
|
|
2018-03-17 14:28:45 +01:00
|
|
|
QJsonDocument json_doc = QJsonDocument::fromJson(reply->readAll());
|
|
|
|
if ((json_doc.isNull()) || (!json_doc.isObject())) {
|
2018-03-02 22:51:42 +01:00
|
|
|
qLog(Error) << "Discogs: Failed to create JSON doc.";
|
|
|
|
EndSearch(s_ctx);
|
|
|
|
return;
|
|
|
|
}
|
2018-02-27 18:06:05 +01:00
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
QJsonObject json_obj = json_doc.object();
|
|
|
|
if (json_obj.isEmpty()) {
|
|
|
|
qLog(Error) << "Discogs: JSON object is empty.";
|
|
|
|
EndSearch(s_ctx);
|
2018-02-27 18:06:05 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
QVariantMap reply_map = json_obj.toVariantMap();
|
|
|
|
if (!reply_map.contains("results")) {
|
2018-03-17 14:28:45 +01:00
|
|
|
//qLog(Error) << "Discogs: Search reply from server is missing JSON results.";
|
2018-03-02 22:51:42 +01:00
|
|
|
//qLog(Error) << "Discogs: Map dump:";
|
|
|
|
//qLog(Error) << reply_map;
|
|
|
|
EndSearch(s_ctx);
|
|
|
|
return;
|
|
|
|
}
|
2018-10-02 00:38:52 +02:00
|
|
|
|
2018-02-27 18:06:05 +01:00
|
|
|
QVariantList results = reply_map["results"].toList();
|
2018-03-02 22:51:42 +01:00
|
|
|
int i = 0;
|
2018-02-27 18:06:05 +01:00
|
|
|
|
|
|
|
for (const QVariant &result : results) {
|
|
|
|
QVariantMap result_map = result.toMap();
|
2018-03-17 14:28:45 +01:00
|
|
|
if ((result_map.contains("id")) && (result_map.contains("resource_url"))) {
|
|
|
|
int r_id = result_map["id"].toInt();
|
|
|
|
QString title = result_map["title"].toString();
|
|
|
|
QString resource_url = result_map["resource_url"].toString();
|
|
|
|
if (resource_url.isEmpty()) continue;
|
|
|
|
StartRelease(s_ctx, r_id, resource_url);
|
|
|
|
i++;
|
2018-03-02 22:51:42 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
if (i <= 0) EndSearch(s_ctx);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
void DiscogsCoverProvider::HandleReleaseReply(QNetworkReply *reply, int s_id, int r_id) {
|
|
|
|
|
|
|
|
reply->deleteLater();
|
|
|
|
|
|
|
|
if (!requests_release_.contains(r_id)) {
|
2018-03-04 14:10:50 +01:00
|
|
|
//qLog(Error) << "Discogs: Got reply for cancelled request: " << r_id;
|
2018-03-02 22:51:42 +01:00
|
|
|
return;
|
2018-02-27 18:06:05 +01:00
|
|
|
}
|
2018-03-17 14:28:45 +01:00
|
|
|
DiscogsCoverReleaseContext *r_ctx = requests_release_.value(r_id);
|
2018-03-02 22:51:42 +01:00
|
|
|
|
|
|
|
if (!requests_search_.contains(s_id)) {
|
2018-03-17 14:28:45 +01:00
|
|
|
//qLog(Error) << "Discogs: Got reply for cancelled request: " << s_id << " " << r_id;
|
|
|
|
EndSearch(r_ctx);
|
2018-03-02 22:51:42 +01:00
|
|
|
return;
|
|
|
|
}
|
2018-03-17 14:28:45 +01:00
|
|
|
DiscogsCoverSearchContext *s_ctx = requests_search_.value(s_id);
|
2018-10-02 00:38:52 +02:00
|
|
|
|
2018-03-17 14:28:45 +01:00
|
|
|
QJsonDocument json_doc = QJsonDocument::fromJson(reply->readAll());
|
|
|
|
if ((json_doc.isNull()) || (!json_doc.isObject())) {
|
2018-03-02 22:51:42 +01:00
|
|
|
qLog(Error) << "Discogs: Failed to create JSON doc.";
|
|
|
|
EndSearch(s_ctx, r_ctx);
|
|
|
|
return;
|
|
|
|
}
|
2018-02-27 18:06:05 +01:00
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
QJsonObject json_obj = json_doc.object();
|
|
|
|
if (json_obj.isEmpty()) {
|
|
|
|
qLog(Error) << "Discogs: JSON object is empty.";
|
|
|
|
EndSearch(s_ctx, r_ctx);
|
|
|
|
return;
|
2018-02-27 18:06:05 +01:00
|
|
|
}
|
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
QVariantMap reply_map = json_obj.toVariantMap();
|
|
|
|
if (!reply_map.contains("images")) {
|
|
|
|
//qLog(Error) << "Discogs: Search reply from server is missing JSON images.";
|
|
|
|
//qLog(Error) << "Discogs: Map dump:";
|
|
|
|
//qLog(Error) << reply_map;
|
|
|
|
EndSearch(s_ctx, r_ctx);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
QVariantList results = reply_map["images"].toList();
|
|
|
|
|
|
|
|
for (const QVariant &result : results) {
|
|
|
|
QVariantMap result_map = result.toMap();
|
|
|
|
CoverSearchResult cover_result;
|
|
|
|
cover_result.description = s_ctx->title;
|
2018-10-02 00:38:52 +02:00
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
if (result_map.contains("type")) {
|
|
|
|
QString type = result_map["type"].toString();
|
|
|
|
if (type != "primary") continue;
|
|
|
|
}
|
|
|
|
if (result_map.contains("resource_url")) {
|
|
|
|
cover_result.image_url = QUrl(result_map["resource_url"].toString());
|
|
|
|
}
|
2018-03-17 14:28:45 +01:00
|
|
|
if (cover_result.image_url.isEmpty()) continue;
|
2018-03-02 22:51:42 +01:00
|
|
|
s_ctx->results.append(cover_result);
|
|
|
|
}
|
|
|
|
|
|
|
|
EndSearch(s_ctx, r_ctx);
|
|
|
|
|
2018-02-27 18:06:05 +01:00
|
|
|
}
|
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
void DiscogsCoverProvider::SearchRequestError(QNetworkReply::NetworkError error, QNetworkReply *reply, int s_id) {
|
2018-02-27 18:06:05 +01:00
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
if (!requests_search_.contains(s_id)) {
|
2018-03-17 14:28:45 +01:00
|
|
|
//qLog(Error) << "Discogs: got reply for cancelled request: " << s_id;
|
2018-03-02 22:51:42 +01:00
|
|
|
return;
|
|
|
|
}
|
2018-03-17 14:28:45 +01:00
|
|
|
DiscogsCoverSearchContext *s_ctx = requests_search_.value(s_id);
|
2018-03-02 22:51:42 +01:00
|
|
|
EndSearch(s_ctx);
|
|
|
|
|
2018-02-27 18:06:05 +01:00
|
|
|
}
|
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
void DiscogsCoverProvider::ReleaseRequestError(QNetworkReply::NetworkError error, QNetworkReply *reply, int s_id, int r_id) {
|
|
|
|
|
2018-03-17 14:28:45 +01:00
|
|
|
if (!requests_release_.contains(r_id)) {
|
|
|
|
//qLog(Error) << "Discogs: got reply for cancelled request: " << s_id << r_id;
|
2018-03-02 22:51:42 +01:00
|
|
|
return;
|
|
|
|
}
|
2018-03-17 14:28:45 +01:00
|
|
|
DiscogsCoverReleaseContext *r_ctx = requests_release_.value(r_id);
|
2018-03-02 22:51:42 +01:00
|
|
|
|
2018-03-17 14:28:45 +01:00
|
|
|
if (!requests_search_.contains(s_id)) {
|
|
|
|
EndSearch(r_ctx);
|
|
|
|
//qLog(Error) << "Discogs: got reply for cancelled request: " << s_id << r_id;
|
2018-03-02 22:51:42 +01:00
|
|
|
return;
|
|
|
|
}
|
2018-03-17 14:28:45 +01:00
|
|
|
DiscogsCoverSearchContext *s_ctx = requests_search_.value(s_id);
|
2018-03-02 22:51:42 +01:00
|
|
|
|
|
|
|
EndSearch(s_ctx, r_ctx);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
void DiscogsCoverProvider::EndSearch(DiscogsCoverSearchContext *s_ctx, DiscogsCoverReleaseContext *r_ctx) {
|
|
|
|
|
2018-03-17 14:28:45 +01:00
|
|
|
delete requests_release_.take(r_ctx->id);
|
2018-10-02 00:38:52 +02:00
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
s_ctx->r_count--;
|
2018-10-02 00:38:52 +02:00
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
//qLog(Debug) << "r_count: " << s_ctx->r_count;
|
2018-10-02 00:38:52 +02:00
|
|
|
|
2018-03-02 22:51:42 +01:00
|
|
|
if (s_ctx->r_count <= 0) EndSearch(s_ctx);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
void DiscogsCoverProvider::EndSearch(DiscogsCoverSearchContext *s_ctx) {
|
|
|
|
|
2018-03-04 14:10:50 +01:00
|
|
|
//qLog(Debug) << "Discogs: Finished." << s_ctx->id;
|
2018-02-27 18:06:05 +01:00
|
|
|
|
2018-03-17 14:28:45 +01:00
|
|
|
requests_search_.remove(s_ctx->id);
|
2018-03-02 22:51:42 +01:00
|
|
|
emit SearchFinished(s_ctx->id, s_ctx->results);
|
|
|
|
delete s_ctx;
|
2018-02-27 18:06:05 +01:00
|
|
|
|
|
|
|
}
|
2018-03-17 14:28:45 +01:00
|
|
|
|
|
|
|
void DiscogsCoverProvider::EndSearch(DiscogsCoverReleaseContext* r_ctx) {
|
|
|
|
delete requests_release_.take(r_ctx->id);
|
|
|
|
}
|