workspace/plasma/dataengines/places/placesengine.cpp

   1 /*
   2  *   Copyright (C) 2007 Alex Merry <huntedhacker@tiscali.co.uk>
   3  *
   4  *   This program is free software; you can redistribute it and/or modify
   5  *   it under the terms of the GNU Library General Public License version 2 as
   6  *   published by the Free Software Foundation
   7  *
   8  *   This program is distributed in the hope that it will be useful,
   9  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11  *   GNU General Public License for more details
  12  *
  13  *   You should have received a copy of the GNU Library General Public
  14  *   License along with this program; if not, write to the
  15  *   Free Software Foundation, Inc.,
  16  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  17  */
  18
  19 // Local includes
  20 #include "placesengine.h"
  21
  22 // Qt includes
  23 #include <QtCore/QString>
  24 #include <QtCore/QVariantList>
  25
  26
  27 // KDE includes
  28 #include <KDiskFreeSpace>
  29
  30 PlacesEngine::PlacesEngine(QObject *parent, const QVariantList &args)
  31     : Plasma::DataEngine(parent, args)
  32 {
  33     // dataChanged(), rowsRemoved() and setupDone() signals from
  34     // KFilePlacesModel are not propagated between applications.
  35     // layoutChanged() is not emitted at all.
  36     connect(&m_placesModel, SIGNAL(modelReset()),
  37             this, SLOT(modelReset()));
  38     connect(&m_placesModel, SIGNAL(rowsInserted(QModelIndex,int,int)),
  39             this, SLOT(placesAdded(QModelIndex,int,int)));
  40
  41     sendData();
  42 }
  43
  44 PlacesEngine::~PlacesEngine()
  45 {
  46 }
  47
  48 void PlacesEngine::modelReset()
  49 {
  50     kDebug() << "Model reset";
  51
  52     clearSources();
  53 }
  54
  55 void PlacesEngine::placesAdded(const QModelIndex &parent, int start, int end)
  56 {
  57     kDebug() << "Places added:" << parent << "from" << start << "to" << end;
  58     sendData();
  59 }
  60
  61 void PlacesEngine::diskFreeSpaceFound(const QString &mountPoint,
  62                                       quint64 kBSize,
  63                                       quint64 kBUsed,
  64                                       quint64 kBAvailable)
  65 {
  66     kDebug() << "Sir! We got one!" << mountPoint + ": "
  67         << "size =" << kBSize
  68         << "used =" << kBUsed
  69         << "avail =" << kBAvailable;
  70     QString source;
  71
  72     foreach (QString testsource, sources()) {
  73         kDebug() << "Testing" << query(testsource)["url"];
  74         KUrl url(query(testsource)["url"].toString());
  75         if (url.isLocalFile() && url.path() == mountPoint) {
  76             source = testsource;
  77             break;
  78         }
  79     }
  80
  81     kDebug() << "Source:" << source;
  82     if (!source.isEmpty()) {
  83         setData(source, "kBSize", kBSize);
  84         setData(source, "kBUsed", kBUsed);
  85         setData(source, "kBAvailable", kBAvailable);
  86     }
  87 }
  88
  89 void PlacesEngine::tryGetFreeSpace(const KUrl &url)
  90 {
  91     if (!url.isLocalFile()) {
  92         return;
  93     }
  94
  95     kDebug() << "Requesting free space on" << url;
  96
  97     // suicidal object: don't need to worry about cleanup
  98     KDiskFreeSpace *diskFreeSpace = new KDiskFreeSpace(this);
  99     connect(diskFreeSpace,
 100             SIGNAL(foundMountPoint(QString,quint64,quint64,quint64)),
 101             this,
 102             SLOT(diskFreeSpaceFound(QString,quint64,quint64,quint64)));
 103     diskFreeSpace->readDF(url.path());
 104 }
 105
 106 void PlacesEngine::sendData()
 107 {
 108     int rowCount = m_placesModel.rowCount();
 109     for (int i = 0; i < rowCount; ++i) {
 110         QModelIndex index = m_placesModel.index(i,0);
 111
 112         Data map;
 113
 114         QString source = QString::number(i);
 115
 116         setData(source, "name", m_placesModel.text(index));
 117         setData(source, "url", m_placesModel.url(index).url());
 118         setData(source, "icon", m_placesModel.icon(index));
 119         setData(source, "hidden",
 120                 m_placesModel.data(index, KFilePlacesModel::HiddenRole));
 121         setData(source, "setupNeeded",
 122                 m_placesModel.data(index, KFilePlacesModel::SetupNeededRole));
 123
 124         if (m_placesModel.deviceForIndex(index).isValid()) {
 125             setData(source, "isDevice", true);
 126             tryGetFreeSpace(m_placesModel.url(index));
 127         } else {
 128             setData(source, "isDevice", false);
 129         }
 130     }
 131 }
 132
 133 K_EXPORT_PLASMA_DATAENGINE(places, PlacesEngine)
 134
 135