File indexing completed on 2024-05-12 16:21:15

 // SPDX-FileCopyrightText: 2021 Jonah Brüchert <jbb@kaidan.im>
 //
 // SPDX-License-Identifier: GPL-2.0-only OR GPL-3.0-only OR LicenseRef-KDE-Accepted-GPL
 
 #include "albummodel.h"
 #include "playlistutils.h"
 
 #include <QStringBuilder>
 
 AlbumModel::AlbumModel(QObject *parent)
     : AbstractYTMusicModel(parent)
 {
     connect(this, &AlbumModel::browseIdChanged, this, [this] {
         setLoading(true);
         auto future = YTMusicThread::instance()->fetchAlbum(m_browseId);
         QCoro::connect(std::move(future), this, [=, this](album::Album &&album) {
             setLoading(false);
 
             beginResetModel();
             m_album = std::move(album);
             endResetModel();
             std::sort(m_album.thumbnails.begin(), m_album.thumbnails.end());
 
             Q_EMIT titleChanged();
             Q_EMIT artistsChanged();
             Q_EMIT thumbnailUrlChanged();
             Q_EMIT playlistIdChanged();
         });
     });
 
     connect(&YTMusicThread::instance().get(), &AsyncYTMusic::errorOccurred, this, [=, this] {
         setLoading(false);
     });
 }
 
 int AlbumModel::rowCount(const QModelIndex &parent) const
 {
     return parent.isValid() ? 0 : int(m_album.tracks.size());
 }
 
 QVariant AlbumModel::data(const QModelIndex &index, int role) const
 {
     switch (role) {
     case Title:
         return QString::fromStdString(m_album.tracks[index.row()].title);
     case VideoId:
         return QString::fromStdString(m_album.tracks[index.row()].video_id.value_or(std::string()));
     case Artists:
         return QVariant::fromValue(m_album.tracks[index.row()].artists);
     case ThumbnailUrl:
         if (!m_album.thumbnails.empty()) {
             return QString::fromStdString(m_album.thumbnails.front().url);
         } else {
             return {};
         }
     case ArtistsDisplayString:
         return PlaylistUtils::artistsToString(m_album.tracks[index.row()].artists);
     }
 
     Q_UNREACHABLE();
 
     return {};
 }
 
 QHash<int, QByteArray> AlbumModel::roleNames() const
 {
     return {
         {Title, "title"},
         {VideoId, "videoId"},
         {Artists, "artists"},
         {ThumbnailUrl, "thumbnailUrl"},
         {ArtistsDisplayString, "artistsDisplayString"}
     };
 }
 
 QString AlbumModel::browseId() const
 {
     return m_browseId;
 }
 
 void AlbumModel::setBrowseId(const QString &value)
 {
     m_browseId = value;
     Q_EMIT browseIdChanged();
 }
 
 QString AlbumModel::title() const
 {
     return QString::fromStdString(m_album.title);
 }
 
 QString AlbumModel::artists() const
 {
     return PlaylistUtils::artistsToString(m_album.artists);
 }
 
 
 QUrl AlbumModel::thumbnailUrl() const
 {
     if (m_album.thumbnails.empty()) {
         return QUrl();
     }
 
     return QUrl(QString::fromStdString(m_album.thumbnails.back().url));
 }
 
 QString AlbumModel::playlistId() const
 {
     return QString::fromStdString(m_album.audio_playlist_id);
 }
 
 QUrl AlbumModel::webUrl() const
 {
     return QUrl(YTMUSIC_WEB_BASE_URL % "/playlist?list=" % QString::fromStdString(m_album.audio_playlist_id));
 }
 
 const album::Album &AlbumModel::album() const
 {
     return m_album;
 }