File indexing completed on 2024-05-19 04:50:16

 /****************************************************************************************
  * Copyright (c) 2009 Casey Link <unnamedrambler@gmail.com>                             *
  * Copyright (c) 2009 Nikolaj Hald Nielsen <nhn@kde.org>                                *
  * Copyright (c) 2009 Mark Kretschmann <kretschmann@kde.org>                            *
  *                                                                                      *
  * This program 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 2 of the License, or (at your option) any later           *
  * version.                                                                             *
  *                                                                                      *
  * This program 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         *
  * this program.  If not, see <http://www.gnu.org/licenses/>.                           *
  ****************************************************************************************/
 
 #ifndef LASTFMTREEMODEL_H
 #define LASTFMTREEMODEL_H
 
 #include "core/meta/forward_declarations.h"
 
 #include <QUrl>
 
 #include <QAbstractItemModel>
 #include <QPixmap>
 
 #include <User.h>
 
 
 namespace LastFm
 {
 enum Type
 {
     Root = 0,
     MyRecommendations,
     PersonalRadio,
     MixRadio,
     TopArtists,
     MyTags,
     Friends,
     RowCount,
     MyTagsChild,
     FriendsChild,
     ArtistsChild,
     RecentlyBannedTrack,
     RecentlyPlayedTrack,
     RecentlyLovedTrack,
     HistoryStation,
     UserChildPersonal,
     TypeUnknown
 };
 
 enum Role
 {
     StationUrlRole = Qt::UserRole,
     UrlRole,
     TrackRole,
     TypeRole
 };
 
 enum SortOrder
 {
     MostWeightOrder,
     AscendingOrder,
     DescendingOrder
 };
 
 
 }
 
 class LastFmTreeItem;
 class QUrl;
 
 
 class LastFmTreeModel : public QAbstractItemModel
 {
     Q_OBJECT
 
 public:
     explicit LastFmTreeModel( QObject *parent = nullptr );
     ~LastFmTreeModel() override;
 
     QVariant data( const QModelIndex &index, int role ) const override;
     Qt::ItemFlags flags( const QModelIndex &index ) const override;
     QModelIndex index( int row, int column,
                         const QModelIndex &parent = QModelIndex() ) const override;
     QModelIndex parent( const QModelIndex &index ) const override;
     int rowCount( const QModelIndex &parent = QModelIndex() ) const override;
     int columnCount( const QModelIndex &parent = QModelIndex() ) const override;
     static int avatarSize();
     void prepareAvatar( QPixmap& avatar, int size );
 
     QMimeData *mimeData( const QModelIndexList &indices ) const override;
 
 private Q_SLOTS:
     void onAvatarDownloaded( const QString& username, QPixmap );
     void slotAddFriends();
     void slotAddTags();
     void slotAddTopArtists();
 
 private:
     void setupModelData( LastFmTreeItem *parent );
 
     QIcon avatar( const QString &username, const QUrl &avatarUrl ) const;
     QString mapTypeToUrl( LastFm::Type type, const QString &key = "" );
 
     void appendUserStations( LastFmTreeItem* item, const QString& user );
 
     lastfm::User m_user;
 
     LastFmTreeItem *m_rootItem;
     LastFmTreeItem *m_myTags;
     LastFmTreeItem *m_myFriends;
     LastFmTreeItem *m_myTopArtists;
     QHash<QString, QIcon> m_avatars;
 };
 
 class LastFmTreeItem
 {
 public:
     LastFmTreeItem ( const LastFm::Type &type, const QVariant &data, LastFmTreeItem *parent = nullptr );
     LastFmTreeItem ( const QString &url, const LastFm::Type &type, const QVariant &data, LastFmTreeItem *parent = nullptr );
     explicit LastFmTreeItem ( const LastFm::Type &type, LastFmTreeItem *parent = nullptr );
     LastFmTreeItem ( const QString &url, const LastFm::Type &type, LastFmTreeItem *parent = nullptr );
     ~LastFmTreeItem();
 
     void appendChild ( LastFmTreeItem *child );
 
     LastFmTreeItem *child ( int row );
     int childCount() const;
     QVariant data () const;
     int row() const;
     LastFmTreeItem *parent();
     Meta::TrackPtr track() const;
     LastFm::Type type() const
     {
         return mType;
     }
 
     QUrl avatarUrl() const
     {
         return avatar;
     }
     void setAvatarUrl( const QUrl &url )
     {
         avatar = url;
     }
 
 private:
     QList<LastFmTreeItem*> childItems;
     LastFm::Type mType;
     LastFmTreeItem *parentItem;
     QVariant itemData;
     QString mUrl;
     QUrl avatar;
 };
 
 #endif