123 lines
3.6 KiB
C++
123 lines
3.6 KiB
C++
|
/* This file is part of Clementine.
|
||
|
Copyright 2012, David Sansome <me@davidsansome.com>
|
||
|
|
||
|
Clementine 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.
|
||
|
|
||
|
Clementine 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 Clementine. If not, see <http://www.gnu.org/licenses/>.
|
||
|
*/
|
||
|
|
||
|
#include "musicbrainzcoverprovider.h"
|
||
|
|
||
|
#include <boost/bind.hpp>
|
||
|
|
||
|
#include <QXmlStreamReader>
|
||
|
|
||
|
#include "core/closure.h"
|
||
|
#include "core/network.h"
|
||
|
|
||
|
namespace {
|
||
|
|
||
|
static const char* kReleaseSearchUrl =
|
||
|
"http://musicbrainz.org/ws/2/release/";
|
||
|
static const char* kAlbumCoverUrl =
|
||
|
"http://coverartarchive.org/release/%1/front";
|
||
|
} // namespace
|
||
|
|
||
|
|
||
|
MusicbrainzCoverProvider::MusicbrainzCoverProvider(QObject* parent)
|
||
|
: CoverProvider("MusicBrainz", parent),
|
||
|
network_(new NetworkAccessManager(this)) {
|
||
|
}
|
||
|
|
||
|
bool MusicbrainzCoverProvider::StartSearch(
|
||
|
const QString& artist, const QString& album, int id) {
|
||
|
// Find release information.
|
||
|
QUrl url(kReleaseSearchUrl);
|
||
|
QString query = QString("release:\"%1\" AND artist:\"%2\"")
|
||
|
.arg(album.trimmed().replace('"', "\\\""))
|
||
|
.arg(artist.trimmed().replace('"', "\\\""));
|
||
|
url.addQueryItem("query", query);
|
||
|
url.addQueryItem("limit", "5");
|
||
|
QNetworkRequest request(url);
|
||
|
|
||
|
QNetworkReply* reply = network_->get(request);
|
||
|
NewClosure(
|
||
|
reply,
|
||
|
SIGNAL(finished()),
|
||
|
this,
|
||
|
SLOT(ReleaseSearchFinished(QNetworkReply*, int)),
|
||
|
reply,
|
||
|
id);
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
void MusicbrainzCoverProvider::ReleaseSearchFinished(
|
||
|
QNetworkReply* reply, int id) {
|
||
|
reply->deleteLater();
|
||
|
|
||
|
QList<QString> releases;
|
||
|
|
||
|
QXmlStreamReader reader(reply);
|
||
|
while (!reader.atEnd()) {
|
||
|
QXmlStreamReader::TokenType type = reader.readNext();
|
||
|
if (type == QXmlStreamReader::StartElement &&
|
||
|
reader.name() == "release") {
|
||
|
QStringRef release_id = reader.attributes().value("id");
|
||
|
if (!release_id.isEmpty()) {
|
||
|
releases.append(release_id.toString());
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
foreach (const QString& release_id, releases) {
|
||
|
QUrl url(QString(kAlbumCoverUrl).arg(release_id));
|
||
|
QNetworkReply* reply = network_->head(QNetworkRequest(url));
|
||
|
image_checks_.insert(id, reply);
|
||
|
NewClosure(
|
||
|
reply,
|
||
|
SIGNAL(finished()),
|
||
|
this,
|
||
|
SLOT(ImageCheckFinished(int)),
|
||
|
id);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void MusicbrainzCoverProvider::ImageCheckFinished(int id) {
|
||
|
QList<QNetworkReply*> replies = image_checks_.values(id);
|
||
|
int finished_count = std::count_if(
|
||
|
replies.constBegin(), replies.constEnd(),
|
||
|
boost::bind(&QNetworkReply::isFinished, _1));
|
||
|
if (finished_count == replies.size()) {
|
||
|
QList<CoverSearchResult> results;
|
||
|
foreach (QNetworkReply* reply, replies) {
|
||
|
reply->deleteLater();
|
||
|
if (reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt() < 400) {
|
||
|
CoverSearchResult result;
|
||
|
result.description = "foobar";
|
||
|
result.image_url = reply->url();
|
||
|
results.append(result);
|
||
|
}
|
||
|
}
|
||
|
image_checks_.remove(id);
|
||
|
emit SearchFinished(id, results);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void MusicbrainzCoverProvider::CancelSearch(int id) {
|
||
|
QList<QNetworkReply*> replies = image_checks_.values(id);
|
||
|
foreach (QNetworkReply* reply, replies) {
|
||
|
reply->abort();
|
||
|
reply->deleteLater();
|
||
|
}
|
||
|
image_checks_.remove(id);
|
||
|
}
|