File indexing completed on 2024-06-16 04:40:27

 /*
  *   SPDX-FileCopyrightText: 2012-2016 Ivan Cukic <ivan.cukic@kde.org>
  *
  *   SPDX-License-Identifier: GPL-2.0-only OR GPL-3.0-only OR LicenseRef-KDE-Accepted-GPL
  */
 
 #include "FileItemLinkingPluginActionLoader.h"
 
 #include <KFileItemListProperties>
 #include <utils/d_ptr_implementation.h>
 #include <utils/qsqlquery_iterator.h>
 
 #include <QCursor>
 #include <QDBusPendingCall>
 #include <QDebug>
 #include <QFileInfo>
 #include <QMenu>
 #include <QSqlDatabase>
 #include <QSqlDriver>
 #include <QSqlError>
 #include <QSqlField>
 #include <QSqlQuery>
 #include <QStandardPaths>
 
 #include <KLocalizedString>
 #include <KPluginFactory>
 
 #include "common/dbus/common.h"
 
 FileItemLinkingPluginActionLoader *FileItemLinkingPluginActionLoader::create(const KFileItemListProperties &items)
 {
     auto l = new FileItemLinkingPluginActionLoader(items);
     connect(l, &QThread::finished, l, &QObject::deleteLater);
     return l;
 }
 
 FileItemLinkingPluginActionLoader::FileItemLinkingPluginActionLoader(const KFileItemListProperties &items)
     : items(items)
 {
 }
 
 void FileItemLinkingPluginActionLoader::run()
 {
     ActionList actions;
 
     const auto activitiesList = activities.activities();
     const auto itemsSize = items.urlList().size();
 
     if (itemsSize >= 10) {
         // we are not going to check for this large number of files
         actions << createAction(QString(), true, i18n("Link to the current activity"), "list-add");
         actions << createAction(QString(), false, i18n("Unlink from the current activity"), "list-remove");
 
         actions << createSeparator(i18n("Link to:"));
         for (const auto &activity : activitiesList) {
             actions << createAction(activity, true);
         }
 
         actions << createSeparator(i18n("Unlink from:"));
         for (const auto &activity : activitiesList) {
             actions << createAction(activity, false);
         }
 
     } else {
         const QString connectionName = QStringLiteral("kactivities_db_resources_") + QString::number((quintptr)this);
 
         {
             auto database = QSqlDatabase::addDatabase(QStringLiteral("QSQLITE"), connectionName);
 
             database.setDatabaseName(QStandardPaths::writableLocation(QStandardPaths::GenericDataLocation)
                                      + QStringLiteral("/kactivitymanagerd/resources/database"));
 
             if (database.open()) {
                 static const auto queryString = QStringLiteral(
                     "SELECT usedActivity, COUNT(targettedResource) "
                     "FROM ResourceLink "
                     "WHERE targettedResource IN (%1) "
                     "AND initiatingAgent = \":global\" "
                     "AND usedActivity != \":global\" "
                     "GROUP BY usedActivity");
 
                 QStringList escapedFiles;
                 QSqlField field;
                 field.setType(QVariant::String);
 
                 const auto urlList = items.urlList();
                 for (const auto &item : urlList) {
                     field.setValue(QFileInfo(item.toLocalFile()).canonicalFilePath());
                     escapedFiles << database.driver()->formatValue(field);
                 }
 
                 QSqlQuery query(queryString.arg(escapedFiles.join(",")), database);
 
                 QStringList activitiesForLinking;
                 QStringList activitiesForUnlinking;
 
                 for (const auto &result : query) {
                     const auto linkedFileCount = result[1].toInt();
                     const auto activity = result[0].toString();
                     if (linkedFileCount < itemsSize) {
                         activitiesForLinking << activity;
                     }
 
                     if (linkedFileCount > 0) {
                         activitiesForUnlinking << activity;
                     }
                 }
 
                 if (activitiesForLinking.contains(activities.currentActivity()) || !activitiesForUnlinking.contains(activities.currentActivity())) {
                     actions << createAction(QString(), true, i18n("Link to the current activity"), "list-add");
                 }
                 if (activitiesForUnlinking.contains(activities.currentActivity())) {
                     actions << createAction(QString(), false, i18n("Unlink from the current activity"), "list-remove");
                 }
 
                 actions << createSeparator(i18n("Link to:"));
                 for (const auto &activity : activitiesList) {
                     if (activitiesForLinking.contains(activity) || !activitiesForUnlinking.contains(activity)) {
                         actions << createAction(activity, true);
                     }
                 }
 
                 actions << createSeparator(i18n("Unlink from:"));
                 for (const auto &activity : activitiesList) {
                     if (activitiesForUnlinking.contains(activity)) {
                         actions << createAction(activity, false);
                     }
                 }
 
                 database.close();
             }
         }
 
         QSqlDatabase::removeDatabase(connectionName);
     }
 
     Q_EMIT result(actions);
 }
 
 Action FileItemLinkingPluginActionLoader::createAction(const QString &activity, bool link, const QString &title, const QString &icon) const
 {
     Action action = {};
     action.link = link;
 
     if (title.isEmpty()) {
         KActivities::Info activityInfo(activity);
         action.title = activityInfo.name();
         action.icon = activityInfo.icon().isEmpty() ? "activities" : activityInfo.icon();
 
     } else {
         action.title = title;
     }
 
     if (!icon.isEmpty()) {
         action.icon = icon;
     }
 
     action.activity = activity.isEmpty() ? activities.currentActivity() : activity;
 
     return action;
 }
 
 Action FileItemLinkingPluginActionLoader::createSeparator(const QString &title) const
 {
     Action action = {};
     action.icon = "-";
     action.title = title;
     return action;
 }
 
 #include "moc_FileItemLinkingPluginActionLoader.cpp"