strawberry-audio-player-windows-mac-linux/src/covermanager/currentalbumcoverloader.cpp

/*
 * Strawberry Music Player
 * This file was part of Clementine.
 * Copyright 2010, David Sansome <me@davidsansome.com>
 * Copyright 2019-2021, 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/>.
 *
 */

#include "config.h"

#include <memory>

#include <QtGlobal>
#include <QObject>
#include <QDir>
#include <QImage>
#include <QTemporaryFile>

#include "core/application.h"
#include "core/song.h"
#include "playlist/playlistmanager.h"
#include "albumcoverloader.h"
#include "albumcoverloaderresult.h"
#include "currentalbumcoverloader.h"

using std::make_unique;

CurrentAlbumCoverLoader::CurrentAlbumCoverLoader(Application *app, QObject *parent)
    : QObject(parent),
      app_(app),
      temp_file_pattern_(QDir::tempPath() + QStringLiteral("/strawberry-cover-XXXXXX.jpg")),
      id_(0) {

  options_.options = AlbumCoverLoaderOptions::Option::RawImageData | AlbumCoverLoaderOptions::Option::OriginalImage | AlbumCoverLoaderOptions::Option::ScaledImage;
  options_.desired_scaled_size = QSize(120, 120);
  options_.default_cover = QStringLiteral(":/pictures/cdcase.png");

  QObject::connect(&*app_->playlist_manager(), &PlaylistManager::CurrentSongChanged, this, &CurrentAlbumCoverLoader::LoadAlbumCover);
  QObject::connect(&*app_->album_cover_loader(), &AlbumCoverLoader::AlbumCoverLoaded, this, &CurrentAlbumCoverLoader::AlbumCoverReady);

  ReloadSettingsAsync();

}

CurrentAlbumCoverLoader::~CurrentAlbumCoverLoader() {

  if (temp_cover_) temp_cover_->remove();
  if (temp_cover_thumbnail_) temp_cover_thumbnail_->remove();

}

void CurrentAlbumCoverLoader::ReloadSettingsAsync() {

  QMetaObject::invokeMethod(this, &CurrentAlbumCoverLoader::ReloadSettings);

}

void CurrentAlbumCoverLoader::ReloadSettings() {

  options_.types = AlbumCoverLoaderOptions::LoadTypes();

}

void CurrentAlbumCoverLoader::LoadAlbumCover(const Song &song) {

  last_song_ = song;
  id_ = app_->album_cover_loader()->LoadImageAsync(options_, last_song_);

}

void CurrentAlbumCoverLoader::AlbumCoverReady(const quint64 id, AlbumCoverLoaderResult result) {

  if (id != id_) return;
  id_ = 0;

  if (!result.album_cover.image.isNull()) {
    temp_cover_ = make_unique<QTemporaryFile>(temp_file_pattern_);
    temp_cover_->setAutoRemove(true);
    if (temp_cover_->open()) {
      if (result.album_cover.image.save(temp_cover_->fileName(), "JPEG")) {
        result.temp_cover_url = QUrl::fromLocalFile(temp_cover_->fileName());
      }
      else {
        qLog(Error) << "Failed to save cover image to" << temp_cover_->fileName() << temp_cover_->errorString();
      }
    }
    else {
      qLog(Error) << "Failed to open" << temp_cover_->fileName() << temp_cover_->errorString();
    }
  }

  QUrl thumbnail_url;
  if (!result.image_scaled.isNull()) {
    temp_cover_thumbnail_ = make_unique<QTemporaryFile>(temp_file_pattern_);
    temp_cover_thumbnail_->setAutoRemove(true);
    if (temp_cover_thumbnail_->open()) {
      if (result.image_scaled.save(temp_cover_thumbnail_->fileName(), "JPEG")) {
        thumbnail_url = QUrl::fromLocalFile(temp_cover_thumbnail_->fileName());
      }
      else {
        qLog(Error) << "Unable to save cover thumbnail image to" << temp_cover_thumbnail_->fileName();
      }
    }
    else {
      qLog(Error) << "Unable to open" << temp_cover_thumbnail_->fileName();
    }
  }

  if (result.art_manual_updated.isValid()) {
    last_song_.set_art_manual(result.art_manual_updated);
  }
  if (result.art_automatic_updated.isValid()) {
    last_song_.set_art_automatic(result.art_automatic_updated);
  }

  emit AlbumCoverLoaded(last_song_, result);
  emit ThumbnailLoaded(last_song_, thumbnail_url, result.image_scaled);

}