File indexing completed on 2024-04-28 15:39:06

 // SPDX-FileCopyrightText: 2020-2022 Tobias Leupold <tl at stonemx dot de>
 //
 // SPDX-License-Identifier: GPL-3.0-only OR LicenseRef-KDE-Accepted-GPL
 
 // Local includes
 #include "ImagesListFilter.h"
 #include "Coordinates.h"
 #include "ImagesModel.h"
 #include "MimeHelper.h"
 
 // Qt includes
 #include <QDebug>
 #include <QMimeData>
 #include <QUrl>
 
 ImagesListFilter::ImagesListFilter(QObject *parent, KGeoTag::ImagesListType type)
     : QSortFilterProxyModel(parent),
       m_listType(type)
 {
 }
 
 void ImagesListFilter::setListType(KGeoTag::ImagesListType type)
 {
     m_listType = type;
     invalidateFilter();
 }
 
 void ImagesListFilter::setSourceModel(QAbstractItemModel *sourceModel)
 {
     QSortFilterProxyModel::setSourceModel(sourceModel);
     m_imagesModel = qobject_cast<ImagesModel *>(sourceModel);
 }
 
 bool ImagesListFilter::filterAcceptsRow(int sourceRow, const QModelIndex &) const
 {
     if (m_listType == KGeoTag::AllImages) {
         return true;
     }
 
     const bool coordinatesSet = sourceModel()->index(sourceRow, 0).data(
                                     KGeoTag::CoordinatesRole).value<Coordinates>().isSet();
 
     return    (m_listType == KGeoTag::AssignedImages   &&   coordinatesSet)
            || (m_listType == KGeoTag::UnAssignedImages && ! coordinatesSet);
 }
 
 Qt::DropActions ImagesListFilter::supportedDropActions() const
 {
     return Qt::CopyAction | Qt::MoveAction;
 }
 
 Qt::ItemFlags ImagesListFilter::flags(const QModelIndex &index) const
 {
     auto defaultFlags = QSortFilterProxyModel::flags(index);
 
     if (index.isValid()) {
         return defaultFlags;
     } else {
         return Qt::ItemIsDropEnabled | defaultFlags;
     }
 }
 
 bool ImagesListFilter::canDropMimeData(const QMimeData *data, Qt::DropAction action, int, int,
                                        const QModelIndex &) const
 {
     if (! (action & (Qt::CopyAction | Qt::MoveAction)) || ! data->hasUrls()) {
         return false;
     }
 
     // Check if the drag's origin is an image list
     const auto source = data->data(KGeoTag::SourceImagesListMimeType);
     if (! source.isEmpty()) {
         // Don't allow drops on the "assigned" images list
         if (m_listType == KGeoTag::AssignedImages) {
             return false;
         }
         // Don't allow drops on the origin list
         if (source == KGeoTag::SourceImagesList.value(m_listType)) {
             return false;
         }
 
     } else {
         // The user wants to drop files from an external source
         if (MimeHelper::getUsablePaths(KGeoTag::DroppedOnImageList, data).isEmpty()) {
             return false;
         }
     }
 
     return true;
 }
 
 bool ImagesListFilter::dropMimeData(const QMimeData *data, Qt::DropAction action, int, int,
                                     const QModelIndex &)
 {
     if (! (action & (Qt::CopyAction | Qt::MoveAction)) || ! data->hasUrls()) {
         return false;
     }
 
     QVector<QString> paths;
     QVector<QString> removeCoordinates;
 
     const auto usablePaths = MimeHelper::getUsablePaths(KGeoTag::DroppedOnImageList, data);
     for (const auto &path : usablePaths) {
         if (! m_imagesModel->contains(path)) {
             paths.append(path);
         } else {
             if (m_listType == KGeoTag::UnAssignedImages
                 && m_imagesModel->coordinates(path).isSet()) {
 
                 removeCoordinates.append(path);
             }
         }
     }
 
     if (! removeCoordinates.isEmpty()) {
         Q_EMIT requestRemoveCoordinates(removeCoordinates);
     }
 
     if (! paths.isEmpty()) {
         Q_EMIT requestAddingImages(paths);
     }
 
     return true;
 }