File indexing completed on 2025-04-27 04:08:15

 /* This file is part of the KDE project
  *
  * SPDX-FileCopyrightText: 2012 Arjen Hiemstra <ahiemstra@heimr.nl>
  *
  * SPDX-License-Identifier: LGPL-2.0-or-later
  */
 
 #include "FileSystemModel.h"
 
 #include <QDateTime>
 #include <KisMimeDatabase.h>
 #include <QFileInfo>
 #include <QDir>
 #include <QStandardPaths>
 #include <QDebug>
 
 class FileSystemModel::Private
 {
 public:
     QDir dir;
     QFileInfoList list;
 
     static const QString drivesPath;
 };
 
 const QString FileSystemModel::Private::drivesPath("special://drives");
 
 FileSystemModel::FileSystemModel(QObject* parent)
     : QAbstractListModel(parent), d(new Private)
 {
     d->dir.setFilter(QDir::AllDirs | QDir::Files | QDir::NoDotAndDotDot);
     d->dir.setSorting(QDir::DirsFirst | QDir::Name | QDir::IgnoreCase);
 }
 
 FileSystemModel::~FileSystemModel()
 {
     delete d;
 }
 
 QVariant FileSystemModel::data(const QModelIndex& index, int role) const
 {
     if (index.isValid()) {
         const QFileInfo &fileInfo = d->list.at(index.row());
 
         switch(role) {
         case FileNameRole:
             return fileInfo.fileName();
             break;
         case FilePathRole:
             return fileInfo.absoluteFilePath();
             break;
         case FileIconRole:
             return fileInfo.isDir() ? "inode/directory" : QString("image://recentimage/%1").arg(fileInfo.absoluteFilePath());
             break;
         case FileDateRole:
             return fileInfo.lastModified().toString(Qt::SystemLocaleShortDate);
         }
     }
     return QVariant();
 }
 
 int FileSystemModel::rowCount(const QModelIndex&) const
 {
     return d->list.count();
 }
 
 void FileSystemModel::classBegin()
 {
 
 }
 
 void FileSystemModel::componentComplete()
 {
     setPath(QStandardPaths::writableLocation(QStandardPaths::PicturesLocation));
 }
 
 QString FileSystemModel::path()
 {
     QString path = d->dir.absolutePath();
     if (path.isEmpty()) {
         return Private::drivesPath;
     } else {
         return d->dir.absolutePath();
     }
 }
 
 void FileSystemModel::setPath(const QString& path)
 {
     if (path != d->dir.path()) {
         if (d->list.count() > 0) {
             beginRemoveRows(QModelIndex(), 0, d->list.count() - 1);
             endRemoveRows();
         }
 
         if (path != Private::drivesPath) {
             d->dir.setPath(path);
             d->dir.refresh();
             d->list = d->dir.entryInfoList();
             if (d->list.count() > 0) {
                 beginInsertRows(QModelIndex(), 0, d->list.count() - 1);
                 endInsertRows();
             }
         } else {
             d->dir.setPath("");
             d->dir.refresh();
             d->list = QDir::drives();
 
             beginInsertRows(QModelIndex(), 0, d->list.count() - 1);
             endInsertRows();
         }
         emit pathChanged();
     }
 }
 
 QString FileSystemModel::parentFolder()
 {
     if (path() != Private::drivesPath) {
         if (QRegExp("^[A-Z]{1,3}:/$").exactMatch(path())) {
             return Private::drivesPath;
         } else {
             QDir root(path());
             root.cdUp();
             return root.path();
         }
     } else {
         return Private::drivesPath;
     }
 }
 
 QString FileSystemModel::filter()
 {
     return d->dir.nameFilters().join(" ");
 }
 
 void FileSystemModel::setFilter(const QString& filter)
 {
     d->dir.setNameFilters(filter.split(" "));
 }
 
 QHash<int, QByteArray> FileSystemModel::roleNames() const
 {
     QHash<int, QByteArray> roles;
     roles.insert(FileNameRole, "fileName");
     roles.insert(FilePathRole, "path");
     roles.insert(FileIconRole, "icon");
     roles.insert(FileDateRole, "date");
     return roles;
 }