File indexing completed on 2024-10-06 04:31:44

 /***************************************************************************
  *   Copyright (C) 2009 Matthias Fuchs <mat69@gmx.net>                     *
  *                                                                         *
  *   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, write to the                         *
  *   Free Software Foundation, Inc.,                                       *
  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA .        *
  ***************************************************************************/
 
 #include "filemodel.h"
 
 #include "signature.h"
 #include "verifier.h"
 
 #include <KIO/Global>
 #include <KLocalizedString>
 
 FileItem::FileItem(const QString &name, FileItem *parent)
     : m_name(name)
     , m_state(Qt::Checked)
     , m_status(Job::Stopped)
     , m_totalSize(0)
     , m_checkusmVerified(0)
     , m_signatureVerified(0)
     , m_parent(parent)
 {
 }
 
 FileItem::~FileItem()
 {
     qDeleteAll(m_childItems);
 }
 
 void FileItem::appendChild(FileItem *child)
 {
     m_childItems.append(child);
 }
 
 FileItem *FileItem::child(int row)
 {
     return m_childItems.value(row);
 }
 
 int FileItem::childCount() const
 {
     return m_childItems.count();
 }
 
 bool FileItem::isFile() const
 {
     return m_childItems.isEmpty();
 }
 
 int FileItem::columnCount() const
 {
     return 5;
 }
 
 QVariant FileItem::data(int column, int role) const
 {
     if (column == FileItem::File) {
         if (role == Qt::CheckStateRole) {
             return m_state;
         } else if (role == Qt::DisplayRole) {
             return m_name;
         } else if (role == Qt::DecorationRole) {
             if (m_mimeType.isNull()) {
                 if (isFile()) {
                     m_mimeType = QIcon::fromTheme(KIO::iconNameForUrl(QUrl(m_name)));
                 } else {
                     m_mimeType = QIcon::fromTheme("folder");
                 }
             }
 
             return m_mimeType;
         }
     } else if (column == FileItem::Status) {
         if ((role == Qt::DisplayRole) || (role == Qt::DecorationRole)) {
             if (isFile()) {
                 return m_status;
             }
         }
     } else if (column == FileItem::Size) {
         if (role == Qt::DisplayRole) {
             return KIO::convertSize(m_totalSize);
         }
     } else if (column == FileItem::ChecksumVerified) {
         if (role == Qt::DecorationRole) {
             switch (m_checkusmVerified) {
             case Verifier::Verified:
                 return QIcon::fromTheme("dialog-ok");
             case Verifier::NotVerified:
                 return QIcon::fromTheme("dialog-error");
             case Verifier::NoResult:
             default:
                 return QIcon::fromTheme(QString());
             }
         }
     } else if (column == FileItem::SignatureVerified) { // TODO implement all cases
         if (role == Qt::DecorationRole) {
             switch (m_signatureVerified) {
             case Signature::Verified:
                 return QIcon::fromTheme("dialog-ok");
             case Signature::VerifiedInformation:
                 return QIcon::fromTheme("dialog-information");
             case Signature::VerifiedWarning:
                 return QIcon::fromTheme("dialog-warning");
             case Signature::NotVerified:
                 return QIcon::fromTheme("dialog-error");
             case Signature::NoResult:
             default:
                 return QIcon::fromTheme(QString());
             }
         }
     }
 
     return QVariant();
 }
 
 bool FileItem::setData(int column, const QVariant &value, FileModel *model, int role)
 {
     if (value.isNull()) {
         return false;
     }
 
     if (column == FileItem::File) {
         if (role == Qt::CheckStateRole) {
             m_state = static_cast<Qt::CheckState>(value.toInt());
             model->changeData(this->row(), column, this);
             checkParents(m_state, model);
             checkChildren(m_state, model);
             return true;
         } else if (role == Qt::EditRole) {
             m_name = value.toString();
             model->changeData(this->row(), column, this);
             return true;
         }
     } else if (column == FileItem::Status) {
         if (role == Qt::EditRole) {
             if (isFile()) {
                 m_status = static_cast<Job::Status>(value.toInt());
                 bool finished = (m_status == Job::Finished);
                 model->changeData(this->row(), column, this, finished);
 
                 return true;
             }
         }
     } else if (column == FileItem::Size) {
         if (role == Qt::EditRole) {
             KIO::fileoffset_t newSize = value.toLongLong();
             if (m_parent) {
                 m_parent->addSize(newSize - m_totalSize, model);
             }
             m_totalSize = newSize;
             model->changeData(this->row(), column, this);
             return true;
         }
     } else if (column == FileItem::ChecksumVerified) {
         m_checkusmVerified = value.toInt();
         model->changeData(this->row(), column, this);
         return true;
     } else if (column == FileItem::SignatureVerified) {
         m_signatureVerified = value.toInt();
         model->changeData(this->row(), column, this);
         return true;
     }
 
     return false;
 }
 
 void FileItem::checkParents(Qt::CheckState state, FileModel *model)
 {
     if (!model) {
         return;
     }
 
     if (!m_parent) {
         return;
     }
 
     foreach (FileItem *child, m_parent->m_childItems) {
         if (child->m_state != state) {
             state = Qt::Unchecked;
             break;
         }
     }
 
     m_parent->m_state = state;
     model->changeData(m_parent->row(), FileItem::File, m_parent);
     m_parent->checkParents(state, model);
 }
 
 void FileItem::checkChildren(Qt::CheckState state, FileModel *model)
 {
     if (!model) {
         return;
     }
 
     m_state = state;
     model->changeData(row(), FileItem::File, this);
 
     foreach (FileItem *child, m_childItems) {
         child->checkChildren(state, model);
     }
 }
 
 FileItem *FileItem::parent()
 {
     return m_parent;
 }
 
 int FileItem::row() const
 {
     if (m_parent) {
         return m_parent->m_childItems.indexOf(const_cast<FileItem *>(this));
     }
 
     return 0;
 }
 
 void FileItem::addSize(KIO::fileoffset_t size, FileModel *model)
 {
     if (!isFile()) {
         m_totalSize += size;
         model->changeData(this->row(), FileItem::Size, this);
         if (m_parent) {
             m_parent->addSize(size, model);
         }
     }
 }
 
 FileModel::FileModel(const QList<QUrl> &files, const QUrl &destDirectory, QObject *parent)
     : QAbstractItemModel(parent)
     , m_destDirectory(destDirectory)
     , m_checkStateChanged(false)
 {
     m_rootItem = new FileItem("root");
     m_header << i18nc("file in a filesystem", "File") << i18nc("status of the download", "Status") << i18nc("size of the download", "Size")
              << i18nc("checksum of a file", "Checksum") << i18nc("signature of a file", "Signature");
 
     setupModelData(files);
 }
 
 FileModel::~FileModel()
 {
     delete m_rootItem;
 }
 
 void FileModel::setupModelData(const QList<QUrl> &files)
 {
     QString destDirectory = m_destDirectory.toLocalFile();
 
     foreach (const QUrl &file, files) {
         FileItem *parent = m_rootItem;
         QStringList directories = file.toLocalFile().remove(destDirectory).split('/', Qt::SkipEmptyParts);
         FileItem *child = nullptr;
         while (directories.count()) {
             QString part = directories.takeFirst();
             for (int i = 0; i < parent->childCount(); ++i) {
                 // folder already exists
                 if (parent->child(i)->data(0, Qt::DisplayRole).toString() == part) {
                     parent = parent->child(i);
                     // file already exists
                     if (!directories.count()) {
                         break;
                     }
                     part = directories.takeFirst();
                     i = -1;
                     continue;
                 }
             }
             child = new FileItem(part, parent);
             parent->appendChild(child);
             parent = parent->child(parent->childCount() - 1);
         }
         if (child) {
             m_files.append(child);
         }
     }
 }
 
 int FileModel::columnCount(const QModelIndex &parent) const
 {
     if (parent.isValid()) {
         return static_cast<FileItem *>(parent.internalPointer())->columnCount();
     } else {
         return m_rootItem->columnCount();
     }
 }
 
 QVariant FileModel::data(const QModelIndex &index, int role) const
 {
     if (!index.isValid()) {
         return QVariant();
     }
 
     auto *item = static_cast<FileItem *>(index.internalPointer());
     const QVariant data = item->data(index.column(), role);
 
     // get the status icon as well as status text
     if (index.column() == FileItem::Status) {
         const auto status = static_cast<Job::Status>(data.toInt());
         if (item->isFile()) {
             if (role == Qt::DisplayRole) {
                 if (m_customStatusTexts.contains(status)) {
                     return m_customStatusTexts[status];
                 } else {
                     return Transfer::statusText(status);
                 }
             } else if (role == Qt::DecorationRole) {
                 if (m_customStatusIcons.contains(status)) {
                     return m_customStatusIcons[status];
                 } else {
                     return QIcon::fromTheme(Transfer::statusIconName(status));
                 }
             }
         } else {
             return QVariant();
         }
     }
 
     return data;
 }
 
 bool FileModel::setData(const QModelIndex &index, const QVariant &value, int role)
 {
     if (!index.isValid()) {
         return false;
     }
 
     auto *item = static_cast<FileItem *>(index.internalPointer());
 
     if ((index.column() == FileItem::File) && (role == Qt::CheckStateRole)) {
         const bool worked = item->setData(index.column(), value, this, role);
         if (worked) {
             m_checkStateChanged = true;
         }
 
         return worked;
     }
 
     return item->setData(index.column(), value, this, role);
 }
 
 Qt::ItemFlags FileModel::flags(const QModelIndex &index) const
 {
     if (!index.isValid()) {
         return Qt::NoItemFlags;
     }
 
     if (index.column() == FileItem::File) {
         return QAbstractItemModel::flags(index) | Qt::ItemIsUserCheckable;
     }
 
     return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
 }
 
 QVariant FileModel::headerData(int section, Qt::Orientation orientation, int role) const
 {
     if ((orientation == Qt::Horizontal) && (role == Qt::DisplayRole)) {
         return m_header.value(section);
     }
 
     return QVariant();
 }
 
 QModelIndex FileModel::index(int row, int column, const QModelIndex &parent) const
 {
     if (!hasIndex(row, column, parent)) {
         return QModelIndex();
     }
 
     FileItem *parentItem;
     if (parent.isValid()) {
         parentItem = static_cast<FileItem *>(parent.internalPointer());
     } else {
         parentItem = m_rootItem;
     }
 
     FileItem *childItem = parentItem->child(row);
     if (childItem) {
         return createIndex(row, column, childItem);
     } else {
         return QModelIndex();
     }
 }
 
 QModelIndex FileModel::index(const QUrl &file, int column)
 {
     FileItem *item = getItem(file);
     if (!item) {
         return QModelIndex();
     }
 
     return createIndex(item->row(), column, item);
 }
 
 QModelIndexList FileModel::fileIndexes(int column) const
 {
     QModelIndexList indexList;
     foreach (FileItem *item, m_files) {
         int row = item->row();
         indexList.append(createIndex(row, column, item));
     }
 
     return indexList;
 }
 
 QModelIndex FileModel::parent(const QModelIndex &index) const
 {
     if (!index.isValid()) {
         return QModelIndex();
     }
 
     auto *childItem = static_cast<FileItem *>(index.internalPointer());
     FileItem *parentItem = childItem->parent();
     if ((parentItem == m_rootItem) || (!parentItem)) {
         return QModelIndex();
     } else {
         return createIndex(parentItem->row(), 0, parentItem);
     }
 }
 
 int FileModel::rowCount(const QModelIndex &parent) const
 {
     if (parent.column() > 0) {
         return 0;
     }
 
     FileItem *parentItem;
     if (parent.isValid()) {
         parentItem = static_cast<FileItem *>(parent.internalPointer());
     } else {
         parentItem = m_rootItem;
     }
 
     return parentItem->childCount();
 }
 
 void FileModel::changeData(int row, int column, FileItem *item, bool finished)
 {
     QModelIndex index = createIndex(row, column, item);
     Q_EMIT dataChanged(index, index);
 
     if (finished) {
         const QUrl file = getUrl(index);
         Q_EMIT fileFinished(file);
     }
 }
 
 void FileModel::setDirectory(const QUrl &newDirectory)
 {
     m_destDirectory = newDirectory;
     m_itemCache.clear();
 }
 
 QUrl FileModel::getUrl(const QModelIndex &index)
 {
     if (!index.isValid()) {
         return QUrl();
     }
 
     const QModelIndex file = index.sibling(index.row(), FileItem::File);
 
     return getUrl(static_cast<FileItem *>(file.internalPointer()));
 }
 
 QUrl FileModel::getUrl(FileItem *item)
 {
     const QString path = getPath(item);
     const QString name = item->data(FileItem::File, Qt::DisplayRole).toString();
     QUrl url = m_destDirectory;
     url.setPath(m_destDirectory.path() + path + name);
 
     return url;
 }
 
 QString FileModel::getPath(FileItem *item)
 {
     FileItem *parent = item->parent();
     QString path;
     while (parent && parent->parent()) {
         path = parent->data(FileItem::File, Qt::DisplayRole).toString() + '/' + path;
         parent = parent->parent();
     }
 
     return path;
 }
 
 FileItem *FileModel::getItem(const QUrl &file)
 {
     if (m_itemCache.contains(file)) {
         return m_itemCache[file];
     }
 
     QString destDirectory = m_destDirectory.toLocalFile();
 
     FileItem *item = m_rootItem;
     QStringList directories = file.toLocalFile().remove(destDirectory).split('/', Qt::SkipEmptyParts);
     while (directories.count()) {
         QString part = directories.takeFirst();
         for (int i = 0; i < item->childCount(); ++i) {
             // folder already exists
             if (item->child(i)->data(FileItem::File, Qt::DisplayRole).toString() == part) {
                 item = item->child(i);
                 // file already exists
                 if (!directories.count()) {
                     break;
                 }
                 part = directories.takeFirst();
                 i = -1;
                 continue;
             }
         }
     }
 
     if (item == m_rootItem) {
         item = nullptr;
     } else {
         m_itemCache[file] = item;
     }
 
     return item;
 }
 
 bool FileModel::downloadFinished(const QUrl &file)
 {
     FileItem *item = getItem(file);
     if (item) {
         const Job::Status status = static_cast<Job::Status>(item->data(FileItem::Status, Qt::DisplayRole).toInt());
         if (status == Job::Finished) {
             return true;
         }
     }
 
     return false;
 }
 
 bool FileModel::isFile(const QModelIndex &index) const
 {
     if (!index.isValid()) {
         return false;
     }
 
     auto *item = static_cast<FileItem *>(index.internalPointer());
 
     // only files can be renamed, no folders
     return item->isFile();
 }
 
 void FileModel::rename(const QModelIndex &file, const QString &newName)
 {
     if (!file.isValid() || (file.column() != FileItem::File)) {
         return;
     }
 
     auto *item = static_cast<FileItem *>(file.internalPointer());
     // only files can be renamed, no folders
     if (!item->isFile()) {
         return;
     }
 
     // Find out the old and the new QUrl
     QString oldName = file.data(Qt::DisplayRole).toString();
     QString path = getPath(item);
 
     QUrl oldUrl = m_destDirectory;
     oldUrl.setPath(m_destDirectory.path() + path + oldName);
     QUrl newUrl = m_destDirectory;
     newUrl.setPath(m_destDirectory.path() + path + newName);
 
     m_itemCache.remove(oldUrl);
 
     setData(file, newName);
 
     Q_EMIT rename(oldUrl, newUrl);
 }
 
 void FileModel::renameFailed(const QUrl &beforeRename, const QUrl &afterRename)
 {
     Q_UNUSED(beforeRename)
     Q_UNUSED(afterRename)
 }
 
 void FileModel::watchCheckState()
 {
     m_checkStateChanged = false;
 }
 
 void FileModel::stopWatchCheckState()
 {
     if (m_checkStateChanged) {
         Q_EMIT checkStateChanged();
     }
 
     m_checkStateChanged = false;
 }
 
 #include "moc_filemodel.cpp"