kfile/kfileplacesselector.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2006 by Peter Penz (peter.penz@gmx.at)                  *
   3  *   Copyright (C) 2007 by Kevin Ottens (ervin@kde.org)                    *
   4  *                                                                         *
   5  *   This library is free software; you can redistribute it and/or         *
   6  *   modify it under the terms of the GNU Lesser General Public            *
   7  *   License as published by the Free Software Foundation; either          *
   8  *   version 2 of the License, or (at your option) any later version.      *
   9  *                                                                         *
  10  *   This library is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU     *
  13  *   Lesser General Public License for more details.                       *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU Lesser General Public      *
  16  *   License along with this library; if not, write to the                 *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA          *
  19  ***************************************************************************/
  20
  21 #include "kfileplacesselector_p.h"
  22
  23 #include "kurlnavigator.h"
  24
  25 #include <kiconloader.h>
  26 #include <kglobalsettings.h>
  27 #include <kfileplacesmodel.h>
  28 #include <kmenu.h>
  29 #include <kmimetype.h>
  30 #include <kdebug.h>
  31
  32 #include <QtGui/QDragEnterEvent>
  33 #include <QtGui/QDragLeaveEvent>
  34 #include <QtGui/QDropEvent>
  35 #include <QtGui/QPainter>
  36 #include <QtGui/QPixmap>
  37 #include <kicon.h>
  38
  39 KFilePlacesSelector::KFilePlacesSelector(KUrlNavigator* parent, KFilePlacesModel* placesModel) :
  40     KUrlButton(parent),
  41     m_selectedItem(-1),
  42     m_placesModel(placesModel)
  43 {
  44     setFocusPolicy(Qt::NoFocus);
  45
  46     m_placesMenu = new KMenu(this);
  47
  48     updateMenu();
  49
  50     connect(m_placesModel, SIGNAL(rowsInserted(const QModelIndex&, int, int)),
  51             this, SLOT(updateMenu()));
  52     connect(m_placesModel, SIGNAL(rowsRemoved(const QModelIndex&, int, int)),
  53             this, SLOT(updateMenu()));
  54     connect(m_placesModel, SIGNAL(dataChanged(const QModelIndex&, const QModelIndex&)),
  55             this, SLOT(updateMenu()));
  56     connect(m_placesMenu, SIGNAL(triggered(QAction*)),
  57             this, SLOT(activatePlace(QAction*)));
  58
  59     setMenu(m_placesMenu);
  60
  61     setAcceptDrops(true);
  62 }
  63
  64 KFilePlacesSelector::~KFilePlacesSelector()
  65 {
  66 }
  67
  68 void KFilePlacesSelector::updateMenu()
  69 {
  70     m_placesMenu->clear();
  71
  72     updateSelection(m_selectedUrl);
  73     const int rowCount = m_placesModel->rowCount();
  74     for (int i = 0; i < rowCount; ++i) {
  75         QModelIndex index = m_placesModel->index(i, 0);
  76         QAction* action = new QAction(m_placesModel->icon(index),
  77                                       m_placesModel->text(index),
  78                                       m_placesMenu);
  79         m_placesMenu->addAction(action);
  80         action->setData(i);
  81         if (i == m_selectedItem) {
  82             setIcon(m_placesModel->icon(index));
  83         }
  84     }
  85
  86     updateTeardownAction();
  87 }
  88
  89 void KFilePlacesSelector::updateTeardownAction()
  90 {
  91     const int rowCount = m_placesModel->rowCount();
  92     if (m_placesMenu->actions().size()==rowCount+2) {
  93         // remove teardown action
  94         QAction *action = m_placesMenu->actions().at(rowCount+1);
  95         m_placesMenu->removeAction(action);
  96         delete action;
  97
  98         // remove separator
  99         action = m_placesMenu->actions().at(rowCount);
 100         m_placesMenu->removeAction(action);
 101         delete action;
 102     }
 103
 104     const QModelIndex index = m_placesModel->index(m_selectedItem, 0);
 105     QAction *teardown = m_placesModel->teardownActionForIndex(index);
 106     if (teardown!=0) {
 107         teardown->setParent(m_placesMenu);
 108         teardown->setData("teardownAction");
 109
 110         m_placesMenu->addSeparator();
 111         m_placesMenu->addAction(teardown);
 112     }
 113 }
 114
 115 void KFilePlacesSelector::updateSelection(const KUrl& url)
 116 {
 117     const QModelIndex index = m_placesModel->closestItem(url);
 118     if (index.isValid()) {
 119         m_selectedItem = index.row();
 120         m_selectedUrl = url;
 121         setIcon(m_placesModel->icon(index));
 122     }
 123     else {
 124         m_selectedItem = -1;
 125         // No bookmark has been found which matches to the given Url. Show
 126         // a generic folder icon as pixmap for indication:
 127         setIcon(KIcon("folder"));
 128     }
 129     updateTeardownAction();
 130 }
 131
 132 KUrl KFilePlacesSelector::selectedPlaceUrl() const
 133 {
 134     const QModelIndex index = m_placesModel->index(m_selectedItem, 0);
 135     return index.isValid() ? m_placesModel->url(index) : KUrl();
 136 }
 137
 138 QString KFilePlacesSelector::selectedPlaceText() const
 139 {
 140     const QModelIndex index = m_placesModel->index(m_selectedItem, 0);
 141     return index.isValid() ? m_placesModel->text(index) : QString();
 142 }
 143
 144 QSize KFilePlacesSelector::sizeHint() const
 145 {
 146     const int height = KUrlButton::sizeHint().height();
 147     return QSize(height, height);
 148 }
 149
 150 void KFilePlacesSelector::paintEvent(QPaintEvent* event)
 151 {
 152     Q_UNUSED(event);
 153     QPainter painter(this);
 154     drawHoverBackground(&painter);
 155
 156     // draw icon
 157     const QPixmap pixmap = icon().pixmap(QSize(22, 22), QIcon::Normal);
 158     const int x = (width() -  pixmap.width()) / 2;
 159     const int y = (height() - pixmap.height()) / 2;
 160     painter.drawPixmap(x, y, pixmap);
 161 }
 162
 163 void KFilePlacesSelector::dragEnterEvent(QDragEnterEvent* event)
 164 {
 165     if (event->mimeData()->hasUrls()) {
 166         setDisplayHintEnabled(DraggedHint, true);
 167         event->acceptProposedAction();
 168
 169         update();
 170     }
 171 }
 172
 173 void KFilePlacesSelector::dragLeaveEvent(QDragLeaveEvent* event)
 174 {
 175     KUrlButton::dragLeaveEvent(event);
 176
 177     setDisplayHintEnabled(DraggedHint, false);
 178     update();
 179 }
 180
 181 void KFilePlacesSelector::dropEvent(QDropEvent* event)
 182 {
 183     setDisplayHintEnabled(DraggedHint, false);
 184     update();
 185
 186     const KUrl::List urlList = KUrl::List::fromMimeData(event->mimeData());
 187     if (urlList.isEmpty()) {
 188         return;
 189     }
 190     foreach(const KUrl &url, urlList) {
 191         KMimeType::Ptr mimetype = KMimeType::findByUrl(url);
 192         if (mimetype->is("inode/directory")) {
 193             m_placesModel->addPlace(url.fileName(), url);
 194         }
 195     }
 196 }
 197
 198 void KFilePlacesSelector::activatePlace(QAction* action)
 199 {
 200     Q_ASSERT(action != 0);
 201     if (action->data().toString()=="teardownAction") {
 202         QModelIndex index = m_placesModel->index(m_selectedItem, 0);
 203         m_placesModel->requestTeardown(index);
 204         return;
 205     }
 206
 207     QModelIndex index = m_placesModel->index(action->data().toInt(), 0);
 208
 209     m_lastClickedIndex = QPersistentModelIndex();
 210
 211     if (m_placesModel->setupNeeded(index)) {
 212         connect(m_placesModel, SIGNAL(setupDone(const QModelIndex &, bool)),
 213                 this, SLOT(onStorageSetupDone(const QModelIndex &, bool)));
 214
 215         m_lastClickedIndex = index;
 216         m_placesModel->requestSetup(index);
 217         return;
 218     }
 219     else if (index.isValid()) {
 220         m_selectedItem = index.row();
 221         setIcon(m_placesModel->icon(index));
 222         updateTeardownAction();
 223         emit placeActivated(m_placesModel->url(index));
 224     }
 225 }
 226
 227 void KFilePlacesSelector::onStorageSetupDone(const QModelIndex &index, bool success)
 228 {
 229     if (m_lastClickedIndex==index)  {
 230         if (success) {
 231             m_selectedItem = index.row();
 232             setIcon(m_placesModel->icon(index));
 233             updateTeardownAction();
 234             emit placeActivated(m_placesModel->url(index));
 235         }
 236         m_lastClickedIndex = QPersistentModelIndex();
 237     }
 238 }
 239
 240 #include "kfileplacesselector_p.moc"
 241