workspace/plasma/applets/kickoff/core/recentlyusedmodel.cpp

   1 /*
   2     Copyright 2007 Robert Knight <robertknight@gmail.com>
   3
   4     This library is free software; you can redistribute it and/or
   5     modify it under the terms of the GNU Library General Public
   6     License as published by the Free Software Foundation; either
   7     version 2 of the License, or (at your option) any later version.
   8
   9     This library is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12     Library General Public License for more details.
  13
  14     You should have received a copy of the GNU Library General Public License
  15     along with this library; see the file COPYING.LIB.  If not, write to
  16     the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  17     Boston, MA 02110-1301, USA.
  18 */
  19
  20 // Own
  21 #include "core/recentlyusedmodel.h"
  22
  23 // Qt
  24 #include <QFileInfo>
  25
  26 // KDE
  27 #include <KDesktopFile>
  28 #include <KDirWatch>
  29 #include <KIcon>
  30 #include <KLocalizedString>
  31 #include <KRecentDocument>
  32 #include <KUrl>
  33 #include <KDebug>
  34
  35 // Local
  36 #include "core/models.h"
  37 #include "core/recentapplications.h"
  38 #include "recentadaptor.h"
  39
  40 using namespace Kickoff;
  41
  42 class RecentlyUsedModel::Private
  43 {
  44 public:
  45     Private(RecentlyUsedModel *parent, RecentType recenttype, int maxRecentApps)
  46             : q(parent)
  47             , recenttype(recenttype)
  48             , maxRecentApps(maxRecentApps >= 0 ? maxRecentApps : Kickoff::RecentApplications::self()->defaultMaximum())
  49             , recentDocumentItem(0)
  50             , recentAppItem(0)
  51     {
  52     }
  53     void removeExistingItem(const QString& path) {
  54         if (!itemsByPath.contains(path)) {
  55             return;
  56         }
  57
  58         QStandardItem *existingItem = itemsByPath[path];
  59         //kDebug() << "Removing existing item" << existingItem;
  60         Q_ASSERT(existingItem->parent());
  61         existingItem->parent()->removeRow(existingItem->row());
  62         itemsByPath.remove(path);
  63     }
  64     void addRecentApplication(KService::Ptr service, bool append) {
  65         // remove existing item if any
  66         removeExistingItem(service->entryPath());
  67
  68         QStandardItem *appItem = StandardItemFactory::createItemForService(service);
  69         itemsByPath.insert(service->entryPath(), appItem);
  70
  71         if (append) {
  72             recentAppItem->appendRow(appItem);
  73         } else {
  74             recentAppItem->insertRow(0, appItem);
  75         }
  76
  77         while (recentAppItem->rowCount() > maxRecentApps) {
  78             recentAppItem->removeRow(recentAppItem->rowCount() - 1);
  79         }
  80     }
  81     void addRecentDocument(const QString& desktopPath, bool append) {
  82         // remove existing item if any
  83         KDesktopFile desktopFile(desktopPath);
  84         KUrl documentUrl = desktopFile.readUrl();
  85
  86         removeExistingItem(documentUrl.url());
  87
  88         QStandardItem *documentItem = StandardItemFactory::createItemForUrl(desktopPath);
  89         documentItem->setData(true, Kickoff::SubTitleMandatoryRole);
  90         itemsByPath.insert(desktopPath, documentItem);
  91
  92         //kDebug() << "Document item" << documentItem << "text" << documentItem->text() << "url" << documentUrl.url();
  93         if (append) {
  94             recentDocumentItem->appendRow(documentItem);
  95         } else {
  96             recentDocumentItem->insertRow(0, documentItem);
  97         }
  98     }
  99     void loadRecentDocuments() {
 100         // create branch for documents and add existing items
 101         recentDocumentItem = new QStandardItem(i18n("Documents"));
 102         QStringList documents = KRecentDocument::recentDocuments();
 103         foreach(const QString& document, documents) {
 104             addRecentDocument(document, true);
 105         }
 106         q->appendRow(recentDocumentItem);
 107     }
 108     void loadRecentApplications() {
 109         recentAppItem = new QStandardItem(i18n("Applications"));
 110         QList<KService::Ptr> services = RecentApplications::self()->recentApplications();
 111         for(int i = 0; i < maxRecentApps && i < services.count(); ++i) {
 112             addRecentApplication(services[i], true);
 113         }
 114         q->appendRow(recentAppItem);
 115     }
 116
 117     RecentlyUsedModel * const q;
 118     RecentType recenttype;
 119     int maxRecentApps;
 120
 121     QStandardItem *recentDocumentItem;
 122     QStandardItem *recentAppItem;
 123     QHash<QString, QStandardItem*> itemsByPath;
 124 };
 125
 126 RecentlyUsedModel::RecentlyUsedModel(QObject *parent, RecentType recenttype, int maxRecentApps)
 127         : KickoffModel(parent)
 128         , d(new Private(this, recenttype, maxRecentApps))
 129 {
 130     QDBusConnection dbus = QDBusConnection::sessionBus();
 131     (void)new RecentAdaptor(this);
 132     QDBusConnection::sessionBus().registerObject("/kickoff/RecentAppDoc", this);
 133     dbus.connect(QString(), "/kickoff/RecentAppDoc", "org.kde.plasma", "clearRecentDocumentsAndApplications", this, SLOT(clearRecentDocumentsAndApplications()));
 134
 135     if(recenttype != DocumentsOnly) {
 136         d->loadRecentApplications();
 137
 138         // listen for changes to the list of recent applications
 139         connect(RecentApplications::self(), SIGNAL(applicationAdded(KService::Ptr, int)),
 140                 this, SLOT(recentApplicationAdded(KService::Ptr, int)));
 141         connect(RecentApplications::self(), SIGNAL(applicationRemoved(KService::Ptr)),
 142                 this, SLOT(recentApplicationRemoved(KService::Ptr)));
 143         connect(RecentApplications::self(), SIGNAL(cleared()),
 144                 this, SLOT(recentApplicationsCleared()));
 145     }
 146     if(recenttype != ApplicationsOnly) {
 147         d->loadRecentDocuments();
 148
 149         // listen for changes to the list of recent documents
 150         KDirWatch *recentDocWatch = new KDirWatch(this);
 151         recentDocWatch->addDir(KRecentDocument::recentDocumentDirectory(), KDirWatch::WatchFiles);
 152         connect(recentDocWatch, SIGNAL(created(QString)), this, SLOT(recentDocumentAdded(QString)));
 153         connect(recentDocWatch, SIGNAL(deleted(QString)), this, SLOT(recentDocumentRemoved(QString)));
 154     }
 155 }
 156 RecentlyUsedModel::~RecentlyUsedModel()
 157 {
 158     delete d;
 159 }
 160
 161 void RecentlyUsedModel::recentDocumentAdded(const QString& path)
 162 {
 163     kDebug() << "Recent document added" << path;
 164     d->addRecentDocument(path, false);
 165 }
 166 void RecentlyUsedModel::recentDocumentRemoved(const QString& path)
 167 {
 168     kDebug() << "Recent document removed" << path;
 169     d->removeExistingItem(path);
 170 }
 171
 172 void RecentlyUsedModel::recentApplicationAdded(KService::Ptr service, int)
 173 {
 174     if (service) {
 175         d->addRecentApplication(service, false);
 176     }
 177 }
 178
 179 void RecentlyUsedModel::recentApplicationRemoved(KService::Ptr service)
 180 {
 181     if (service) {
 182         d->removeExistingItem(service->entryPath());
 183     }
 184 }
 185
 186 void RecentlyUsedModel::recentApplicationsCleared()
 187 {
 188     QSet<QStandardItem*> appItems;
 189     const int rows = d->recentAppItem->rowCount();
 190     for (int i = 0;i < rows;i++) {
 191         appItems << d->recentAppItem->child(i);
 192     }
 193     QMutableHashIterator<QString, QStandardItem*> iter(d->itemsByPath);
 194     while (iter.hasNext()) {
 195         iter.next();
 196         if (appItems.contains(iter.value())) {
 197             iter.remove();
 198         }
 199     }
 200
 201     d->recentAppItem->removeRows(0, d->recentAppItem->rowCount());
 202 }
 203 void RecentlyUsedModel::clearRecentApplications()
 204 {
 205     RecentApplications::self()->clear();
 206 }
 207 void RecentlyUsedModel::clearRecentDocuments()
 208 {
 209     KRecentDocument::clear();
 210 }
 211
 212 void RecentlyUsedModel::clearRecentDocumentsAndApplications()
 213 {
 214     clearRecentDocuments();
 215     clearRecentApplications();
 216 }
 217
 218
 219 #include "recentlyusedmodel.moc"
 220