src/panels/places/placesitemmodel.h

   1 /***************************************************************************
   2  *   Copyright (C) 2012 by Peter Penz <peter.penz19@gmail.com>             *
   3  *                                                                         *
   4  *   This program is free software; you can redistribute it and/or modify  *
   5  *   it under the terms of the GNU General Public License as published by  *
   6  *   the Free Software Foundation; either version 2 of the License, or     *
   7  *   (at your option) any later version.                                   *
   8  *                                                                         *
   9  *   This program 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         *
  12  *   GNU General Public License for more details.                          *
  13  *                                                                         *
  14  *   You should have received a copy of the GNU General Public License     *
  15  *   along with this program; if not, write to the                         *
  16  *   Free Software Foundation, Inc.,                                       *
  17  *   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA            *
  18  ***************************************************************************/
  19
  20 #ifndef PLACESITEMMODEL_H
  21 #define PLACESITEMMODEL_H
  22
  23 #include <config-baloo.h>
  24
  25 #include <kitemviews/kstandarditemmodel.h>
  26
  27 #include <QUrl>
  28 #include <QHash>
  29 #include <QList>
  30 #include <QSet>
  31 #include <Solid/Predicate>
  32 #include <Solid/StorageAccess>
  33
  34 class KBookmark;
  35 class KBookmarkManager;
  36 class PlacesItem;
  37 class QAction;
  38 class QTimer;
  39
  40 // #define PLACESITEMMODEL_DEBUG
  41
  42 /**
  43  * @brief Model for maintaining the bookmarks of the places panel.
  44  *
  45  * It is compatible to the KFilePlacesModel from kdelibs but adds
  46  * the ability to have groups for places.
  47  */
  48 class PlacesItemModel: public KStandardItemModel
  49 {
  50     Q_OBJECT
  51
  52 public:
  53     explicit PlacesItemModel(QObject* parent = 0);
  54     virtual ~PlacesItemModel();
  55
  56     /**
  57      * @return A new instance of a places item with the given
  58      *         attributes.
  59      */
  60     PlacesItem* createPlacesItem(const QString& text,
  61                                  const QUrl& url,
  62                                  const QString& iconName = QString());
  63
  64     PlacesItem* placesItem(int index) const;
  65
  66     /**
  67      * If set to true, all items that are marked as hidden
  68      * will be shown in the view. The items will
  69      * stay marked as hidden, which is visually indicated
  70      * by the view by desaturating the icon and the text.
  71      */
  72     void setHiddenItemsShown(bool show);
  73     bool hiddenItemsShown() const;
  74
  75     /**
  76      * @return Number of items that are marked as hidden.
  77      *         Note that this does not mean that the items
  78      *         are really hidden
  79      *         (see PlacesItemModel::setHiddenItemsShown()).
  80      */
  81     int hiddenCount() const;
  82
  83     /**
  84      * Search the item which is equal to the URL or at least
  85      * is a parent URL. If there are more than one possible
  86      * candidates, return the item which covers the biggest
  87      * range of the URL. -1 is returned if no closest item
  88      * could be found.
  89      */
  90     int closestItem(const QUrl& url) const;
  91
  92     /**
  93      * Appends the item \a item as last element of the group
  94      * the item belongs to. If no item with the same group is
  95      * present, the item gets appended as last element of the
  96      * model. PlacesItemModel takes the ownership
  97      * of the item.
  98      */
  99     void appendItemToGroup(PlacesItem* item);
 100
 101     QAction* ejectAction(int index) const;
 102     QAction* teardownAction(int index) const;
 103
 104     void requestEject(int index);
 105     void requestTeardown(int index);
 106
 107     bool storageSetupNeeded(int index) const;
 108     void requestStorageSetup(int index);
 109
 110     virtual QMimeData* createMimeData(const KItemSet& indexes) const Q_DECL_OVERRIDE;
 111
 112     virtual bool supportsDropping(int index) const Q_DECL_OVERRIDE;
 113
 114     void dropMimeDataBefore(int index, const QMimeData* mimeData);
 115
 116     /**
 117      * @return Converts the URL, which contains "virtual" URLs for system-items like
 118      *         "search:/documents" into a Query-URL that will be handled by
 119      *         the corresponding IO-slave. Virtual URLs for bookmarks are used to
 120      *         be independent from internal format changes.
 121      */
 122     static QUrl convertedUrl(const QUrl& url);
 123
 124     virtual void clear() Q_DECL_OVERRIDE;
 125 signals:
 126     void errorMessage(const QString& message);
 127     void storageSetupDone(int index, bool success);
 128
 129 protected:
 130     virtual void onItemInserted(int index) Q_DECL_OVERRIDE;
 131     virtual void onItemRemoved(int index, KStandardItem* removedItem) Q_DECL_OVERRIDE;
 132     virtual void onItemChanged(int index, const QSet<QByteArray>& changedRoles) Q_DECL_OVERRIDE;
 133
 134 private slots:
 135     void slotDeviceAdded(const QString& udi);
 136     void slotDeviceRemoved(const QString& udi);
 137     void slotStorageTeardownDone(Solid::ErrorType error, const QVariant& errorData);
 138     void slotStorageSetupDone(Solid::ErrorType error, const QVariant& errorData, const QString& udi);
 139     void hideItem();
 140
 141     /**
 142      * Updates the bookmarks from the model corresponding to the changed
 143      * bookmarks stored by the bookmark-manager. Is called whenever the bookmarks
 144      * have been changed by another application.
 145      */
 146     void updateBookmarks();
 147
 148     /**
 149      * Saves the bookmarks and indicates to other applications that the
 150      * state of the bookmarks has been changed. Is only called by the
 151      * timeout of m_saveBookmarksTimer to prevent unnecessary savings.
 152      */
 153     void saveBookmarks();
 154 private:
 155     struct SystemBookmarkData;
 156
 157     /**
 158      * Loads the bookmarks from the bookmark-manager and creates items for
 159      * the model or moves hidden items to m_bookmarkedItems.
 160      */
 161     void loadBookmarks();
 162
 163     /**
 164      * @return True, if the bookmark can be accepted in the context of the
 165      *         current application (e.g. bookmarks from other applications
 166      *         will be ignored).
 167      */
 168     bool acceptBookmark(const KBookmark& bookmark,
 169                         const QSet<QString>& availableDevices) const;
 170
 171     /**
 172      * Creates a PlacesItem for a system-bookmark:
 173      * - PlacesItem::isSystemItem() will return true
 174      * - Default view-properties will be created for "Search For" items
 175      * The item is not inserted to the model yet.
 176      */
 177     PlacesItem* createSystemPlacesItem(const SystemBookmarkData& data);
 178
 179     /**
 180      * Creates system bookmarks that are shown per default and can
 181      * only be hidden but not removed. The result will be stored
 182      * in m_systemBookmarks.
 183      */
 184     void createSystemBookmarks();
 185
 186     void initializeAvailableDevices();
 187
 188     /**
 189      * @param index Item index related to the model.
 190      * @return      Corresponding index related to m_bookmarkedItems.
 191      */
 192     int bookmarkIndex(int index) const;
 193
 194     /**
 195      * Marks the item with the index \a index as hidden and
 196      * removes it from the model so that it gets invisible.
 197      */
 198     void hideItem(int index);
 199
 200     /**
 201      * Triggers a delayed saving of bookmarks by starting
 202      * m_saveBookmarksTimer.
 203      */
 204     void triggerBookmarksSaving();
 205
 206     QString internalMimeType() const;
 207
 208     /**
 209      * @return Adjusted drop index which assures that the item is aligned
 210      *         into the same group as specified by PlacesItem::groupType().
 211      */
 212     int groupedDropIndex(int index, const PlacesItem* item) const;
 213
 214     /**
 215      * @return True if the bookmarks have the same identifiers. The identifier
 216      *         is the unique "ID"-property in case if no UDI is set, otherwise
 217      *         the UDI is used as identifier.
 218      */
 219     static bool equalBookmarkIdentifiers(const KBookmark& b1, const KBookmark& b2);
 220
 221     /**
 222      * @return URL using the timeline-protocol for searching (see convertedUrl()).
 223      */
 224     static QUrl createTimelineUrl(const QUrl& url);
 225
 226     /**
 227      * Helper method for createTimelineUrl().
 228      * @return String that represents a date-path in the format that
 229      *         the timeline-protocol expects.
 230      */
 231     static QString timelineDateString(int year, int month, int day = 0);
 232
 233     /**
 234      * @return URL that can be listed by KIO and results in searching
 235      *         for a given term. The URL \a url represents a places-internal
 236      *         URL like e.g. "search:/documents" (see convertedUrl()).
 237      */
 238     static QUrl createSearchUrl(const QUrl& url);
 239
 240 #ifdef HAVE_BALOO
 241     /**
 242      * Helper method for createSearchUrl()
 243      * @return URL that can be listed by KIO and results in searching
 244      *         for the given type
 245      */
 246     static QUrl searchUrlForType(const QString& type);
 247 #endif
 248
 249 #ifdef PLACESITEMMODEL_DEBUG
 250     void showModelState();
 251 #endif
 252
 253 private:
 254     bool m_fileIndexingEnabled;
 255     bool m_hiddenItemsShown;
 256
 257     QSet<QString> m_availableDevices;
 258     Solid::Predicate m_predicate;
 259     KBookmarkManager* m_bookmarkManager;
 260
 261     struct SystemBookmarkData
 262     {
 263         SystemBookmarkData(const QUrl& url,
 264                            const QString& icon,
 265                            const QString& text) :
 266             url(url), icon(icon), text(text) {}
 267         QUrl url;
 268         QString icon;
 269         QString text;
 270     };
 271
 272     QList<SystemBookmarkData> m_systemBookmarks;
 273     QHash<QUrl, int> m_systemBookmarksIndexes;
 274
 275     // Contains hidden and unhidden items that are stored as
 276     // bookmark (the model itself only contains items that
 277     // are shown in the view). If an entry is 0, then the
 278     // places-item is part of the model. If an entry is not
 279     // 0, the item is hidden and not part of the model.
 280     QList<PlacesItem*> m_bookmarkedItems;
 281
 282     // Index of the hidden item that should be removed in
 283     // removeHiddenItem(). The removing must be done
 284     // asynchronously as in the scope of onItemChanged()
 285     // removing an item is not allowed.
 286     int m_hiddenItemToRemove;
 287
 288     QTimer* m_saveBookmarksTimer;
 289     QTimer* m_updateBookmarksTimer;
 290
 291     QHash<QObject*, int> m_storageSetupInProgress;
 292 };
 293
 294 #endif
 295
 296