File indexing completed on 2024-05-05 04:50:33

 /*
    SPDX-FileCopyrightText: 2018 (c) Alexander Stippich <a.stippich@gmx.net>
 
    SPDX-License-Identifier: LGPL-3.0-or-later
  */
 
 #include "filebrowsermodel.h"
 
 #include <QUrl>
 #include <QString>
 #include <QMimeDatabase>
 #include <KDirLister>
 
 #include "models/modelLogging.h"
 
 FileBrowserModel::FileBrowserModel(QObject *parent) : KDirModel(parent)
 {
     QMimeDatabase db;
     const QList<QMimeType> mimeList = db.allMimeTypes();
     QStringList mimeTypes = { QStringLiteral("inode/directory") };
     for (const QMimeType &mime : mimeList) {
         if (mime.name().startsWith(QLatin1String("audio/"))) {
             mimeTypes << mime.name();
         }
     }
 
     dirLister()->setMimeFilter(mimeTypes);
 }
 
 FileBrowserModel::~FileBrowserModel()
 = default;
 
 void FileBrowserModel::setUrl(const QUrl &url)
 {
     if (dirLister()->url() == url) {
         dirLister()->updateDirectory(url);
         return;
     }
 
     beginResetModel();
     dirLister()->openUrl(url);
 
     endResetModel();
 }
 
 bool FileBrowserModel::isBusy() const
 {
     return false;
 }
 
 QHash<int, QByteArray> FileBrowserModel::roleNames() const
 {
     auto roles = KDirModel::roleNames();
 
     roles[static_cast<int>(DataTypes::ColumnsRoles::TitleRole)] = "title";
     roles[static_cast<int>(DataTypes::ColumnsRoles::SecondaryTextRole)] = "secondaryText";
     roles[static_cast<int>(DataTypes::ColumnsRoles::ImageUrlRole)] = "imageUrl";
     roles[static_cast<int>(DataTypes::ColumnsRoles::DatabaseIdRole)] = "databaseId";
     roles[static_cast<int>(DataTypes::ColumnsRoles::ElementTypeRole)] = "dataType";
     roles[static_cast<int>(DataTypes::ColumnsRoles::ResourceRole)] = "url";
 
     roles[static_cast<int>(DataTypes::ColumnsRoles::ArtistRole)] = "artist";
     roles[static_cast<int>(DataTypes::ColumnsRoles::AllArtistsRole)] = "allArtists";
     roles[static_cast<int>(DataTypes::ColumnsRoles::HighestTrackRating)] = "highestTrackRating";
     roles[static_cast<int>(DataTypes::ColumnsRoles::GenreRole)] = "genre";
 
     roles[static_cast<int>(DataTypes::ColumnsRoles::AlbumRole)] = "album";
     roles[static_cast<int>(DataTypes::ColumnsRoles::AlbumArtistRole)] = "albumArtist";
     roles[static_cast<int>(DataTypes::ColumnsRoles::DurationRole)] = "duration";
     roles[static_cast<int>(DataTypes::ColumnsRoles::TrackNumberRole)] = "trackNumber";
     roles[static_cast<int>(DataTypes::ColumnsRoles::DiscNumberRole)] = "discNumber";
     roles[static_cast<int>(DataTypes::ColumnsRoles::RatingRole)] = "rating";
     roles[static_cast<int>(DataTypes::ColumnsRoles::IsSingleDiscAlbumRole)] = "isSingleDiscAlbum";
     roles[static_cast<int>(DataTypes::ColumnsRoles::FullDataRole)] = "fullData";
     roles[static_cast<int>(DataTypes::ColumnsRoles::HasChildrenRole)] = "hasChildren";
 
     roles[static_cast<int>(DataTypes::ColumnsRoles::IsDirectoryRole)] = "isDirectory";
     roles[static_cast<int>(DataTypes::ColumnsRoles::IsPlayListRole)] = "isPlaylist";
 
     return roles;
 }
 
 QVariant FileBrowserModel::data(const QModelIndex &index, int role) const
 {
     auto result = QVariant();
 
     switch(role)
     {
     case DataTypes::ColumnsRoles::ResourceRole:
     {
         KFileItem item = itemForIndex(index);
         result = item.url();
         break;
     }
     case DataTypes::ColumnsRoles::ImageUrlRole:
     {
         KFileItem item = itemForIndex(index);
         if (item.isDir()) {
             result = QUrl(QStringLiteral("image://icon/folder"));
         } else {
             result = QUrl(QStringLiteral("image://icon/audio-x-generic"));
         }
         break;
     }
     case DataTypes::ColumnsRoles::IsDirectoryRole:
     case DataTypes::ColumnsRoles::HasChildrenRole:
     {
         KFileItem item = itemForIndex(index);
         result = item.isDir();
         break;
     }
     case DataTypes::ColumnsRoles::IsPlayListRole:
     {
         KFileItem item = itemForIndex(index);
         result = (ElisaUtils::isPlayList(item.currentMimeType()));
         break;
     }
     case DataTypes::ColumnsRoles::ElementTypeRole:
     {
         KFileItem item = itemForIndex(index);
         result = (item.isDir() ? ElisaUtils::Container : ElisaUtils::FileName);
         break;
     }
     case DataTypes::ColumnsRoles::FullDataRole:
     {
         KFileItem item = itemForIndex(index);
         if (item.isDir()) {
             result = QVariant::fromValue(DataTypes::MusicDataType{{DataTypes::ColumnsRoles::FilePathRole, item.url()},
                                                                   {DataTypes::ColumnsRoles::ElementTypeRole, ElisaUtils::Container},
                                                                   {DataTypes::TitleRole, item.name()},
                                                                   {DataTypes::ImageUrlRole, QUrl(QStringLiteral("image://icon/folder"))}});
         } else {
             auto mimeType = item.currentMimeType();
             if (ElisaUtils::isPlayList(mimeType)) {
                 result = QVariant::fromValue(DataTypes::MusicDataType{{DataTypes::ColumnsRoles::ResourceRole, item.url()},
                                                                       {DataTypes::ColumnsRoles::ElementTypeRole, ElisaUtils::PlayList},
                                                                       {DataTypes::TitleRole, item.name()},
                                                                       {DataTypes::ImageUrlRole, QUrl(QStringLiteral("image://icon/audio-x-generic"))}});
             } else {
                 result = QVariant::fromValue(DataTypes::MusicDataType{{DataTypes::ColumnsRoles::ResourceRole, item.url()},
                                                                       {DataTypes::ColumnsRoles::ElementTypeRole, ElisaUtils::Track},
                                                                       {DataTypes::TitleRole, item.name()},
                                                                       {DataTypes::ImageUrlRole, QUrl(QStringLiteral("image://icon/audio-x-generic"))}});
             }
         }
         break;
     }
     case DataTypes::ColumnsRoles::FilePathRole:
     {
         KFileItem item = itemForIndex(index);
         result = item.url();
         break;
     }
     default:
         result = KDirModel::data(index,role);
     }
 
     return result;
 }
 
 void FileBrowserModel::initialize(MusicListenersManager *manager, DatabaseInterface *database,
                                   ElisaUtils::PlayListEntryType modelType, ElisaUtils::FilterType filter,
                                   const QString &genre, const QString &artist, qulonglong databaseId,
                                   const QUrl &pathFilter)
 {
     Q_UNUSED(manager)
     Q_UNUSED(database)
     Q_UNUSED(modelType)
     Q_UNUSED(filter)
     Q_UNUSED(genre)
     Q_UNUSED(artist)
     Q_UNUSED(databaseId)
 
     setUrl(pathFilter);
 }
 
 void FileBrowserModel::initializeByData(MusicListenersManager *manager, DatabaseInterface *database,
                                         ElisaUtils::PlayListEntryType modelType, ElisaUtils::FilterType filter,
                                         const DataTypes::DataType &dataFilter)
 {
     Q_UNUSED(manager)
     Q_UNUSED(database)
     Q_UNUSED(modelType)
     Q_UNUSED(filter)
 
     setUrl(dataFilter[DataTypes::FilePathRole].toUrl());
 }
 
 
 #include "moc_filebrowsermodel.cpp"