166 lines
4.0 KiB
C++
166 lines
4.0 KiB
C++
/*
|
|
* Copyright (C) 2014 Emeric Poupon
|
|
*
|
|
* This file is part of LMS.
|
|
*
|
|
* LMS 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.
|
|
*
|
|
* LMS 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 LMS. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <boost/foreach.hpp>
|
|
|
|
#include <Wt/WApplication>
|
|
#include <Wt/Http/Response>
|
|
|
|
#include "logger/Logger.hpp"
|
|
|
|
#include "cover/CoverArtGrabber.hpp"
|
|
|
|
#include "CoverResource.hpp"
|
|
|
|
namespace UserInterface {
|
|
|
|
const std::string CoverResource::unknownCoverPath = "/images/unknown-cover.jpg";
|
|
|
|
CoverResource::CoverResource(Database::Handler& db, Wt::WObject *parent)
|
|
: Wt::WResource(parent),
|
|
_db(db)
|
|
{
|
|
}
|
|
|
|
CoverResource:: ~CoverResource()
|
|
{
|
|
beingDeleted();
|
|
}
|
|
|
|
const CoverArt::CoverArt&
|
|
CoverResource::getDefaultCover(std::size_t size)
|
|
{
|
|
auto itCover = _defaultCovers.find(size);
|
|
if (itCover == _defaultCovers.end())
|
|
{
|
|
// Load default cover art for this size
|
|
|
|
CoverArt::CoverArt defaultCover;
|
|
|
|
std::vector<unsigned char> data;
|
|
{
|
|
std::ifstream ist(Wt::WApplication::instance()->docRoot() + unknownCoverPath);
|
|
char c;
|
|
while(ist.get(c))
|
|
data.push_back(c);
|
|
}
|
|
|
|
defaultCover.setData(data);
|
|
defaultCover.setMimeType("image/jpeg");
|
|
defaultCover.scale(size);
|
|
|
|
auto res = _defaultCovers.insert(std::make_pair(size, defaultCover));
|
|
itCover = res.first;
|
|
}
|
|
|
|
return itCover->second;
|
|
}
|
|
|
|
std::string
|
|
CoverResource::getReleaseUrl(std::string releaseName, std::size_t size) const
|
|
{
|
|
return url() + "&release=" + releaseName + "&size=" + std::to_string(size);
|
|
}
|
|
|
|
std::string
|
|
CoverResource::getTrackUrl(Database::Track::id_type trackId, std::size_t size) const
|
|
{
|
|
return url()+ "&trackid=" + std::to_string(trackId) + "&size=" + std::to_string(size);
|
|
}
|
|
|
|
std::string
|
|
CoverResource::getUnkownTrackUrl(size_t size) const
|
|
{
|
|
return Wt::WApplication::instance()->docRoot() + unknownCoverPath + "&size=" + std::to_string(size);
|
|
}
|
|
|
|
void
|
|
CoverResource::putCover(Wt::Http::Response& response, const CoverArt::CoverArt& cover)
|
|
{
|
|
response.setMimeType( cover.getMimeType() );
|
|
BOOST_FOREACH(unsigned char c, cover.getData())
|
|
response.out().put( c );
|
|
}
|
|
|
|
void
|
|
CoverResource::handleRequest(const Wt::Http::Request& request, Wt::Http::Response& response)
|
|
{
|
|
|
|
// Get the id of the track
|
|
const std::string *trackIdStr = request.getParameter("trackid");
|
|
const std::string *sizeStr = request.getParameter("size");
|
|
const std::string *releaseStr = request.getParameter("release");
|
|
|
|
std::vector<CoverArt::CoverArt> covers;
|
|
|
|
// Mandatory parameter size
|
|
if (!sizeStr)
|
|
return;
|
|
|
|
std::size_t size = std::stol(*sizeStr);
|
|
if (size > maxSize)
|
|
return;
|
|
|
|
if (trackIdStr)
|
|
{
|
|
Database::Track::id_type trackId = std::stol(*trackIdStr);
|
|
std::string path;
|
|
bool hasCover = false;
|
|
|
|
{
|
|
// transactions are not thread safe
|
|
std::unique_lock<std::mutex> lock(_mutex);
|
|
|
|
Wt::Dbo::Transaction transaction(_db.getSession());
|
|
|
|
Database::Track::pointer track = Database::Track::getById(_db.getSession(), trackId);
|
|
if (track)
|
|
{
|
|
hasCover = track->hasCover();
|
|
path = track->getPath();
|
|
}
|
|
}
|
|
|
|
if (hasCover)
|
|
covers = CoverArt::Grabber::getFromTrack(path);
|
|
}
|
|
else if (releaseStr)
|
|
{
|
|
// transactions are not thread safe
|
|
std::unique_lock<std::mutex> lock(_mutex);
|
|
Wt::Dbo::Transaction transaction(_db.getSession());
|
|
|
|
covers = CoverArt::Grabber::getFromRelease(_db.getSession(), *releaseStr);
|
|
}
|
|
|
|
BOOST_FOREACH(CoverArt::CoverArt& cover, covers)
|
|
{
|
|
if (cover.scale(size))
|
|
{
|
|
putCover(response, cover);
|
|
return;
|
|
}
|
|
}
|
|
|
|
// If no cover found, just send default one
|
|
putCover(response, getDefaultCover(size));
|
|
}
|
|
|
|
} // namespace UserInterface
|