workspace/plasma/applets/devicenotifier/devicenotifier.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2007 by Alexis Ménard <darktears31@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 //own
  21 #include "devicenotifier.h"
  22 #include "notifierview.h"
  23 #include "notifierdialog.h"
  24
  25 //Qt
  26 #include <QGraphicsLinearLayout>
  27 #include <QGraphicsProxyWidget>
  28 #include <QTimer>
  29
  30 //KDE
  31 #include <KIcon>
  32 #include <KConfigDialog>
  33 #include <KStandardDirs>
  34 #include <KDesktopFile>
  35 #include <kdesktopfileactions.h>
  36 #include <KIconLoader>
  37
  38 //plasma
  39 #include <Plasma/Dialog>
  40 //use for desktop view
  41 #include <Plasma/IconWidget>
  42 #include <Plasma/Theme>
  43
  44 //solid
  45 #include <solid/device.h>
  46 #include <solid/storagedrive.h>
  47
  48
  49 using namespace Plasma;
  50 using namespace Notifier;
  51
  52 K_EXPORT_PLASMA_APPLET(devicenotifier, DeviceNotifier)
  53
  54 DeviceNotifier::DeviceNotifier(QObject *parent, const QVariantList &args)
  55     : Plasma::PopupApplet(parent, args),
  56       m_solidEngine(0),
  57       m_solidDeviceEngine(0),
  58       m_icon(0),
  59       m_iconName(""),
  60       m_dialog(0),
  61       m_numberItems(0),
  62       m_itemsValidity(0)
  63 {
  64     setBackgroundHints(StandardBackground);
  65     setAspectRatioMode(IgnoreAspectRatio);
  66
  67     // let's initialize the widget
  68     (void)widget();
  69
  70     resize(widget()->sizeHint());
  71 }
  72
  73 DeviceNotifier::~DeviceNotifier()
  74 {
  75     delete m_icon;
  76     delete m_dialog;
  77 }
  78
  79 void DeviceNotifier::init()
  80 {
  81     KConfigGroup cg = config();
  82     m_numberItems = cg.readEntry("NumberItems", 4);
  83     m_itemsValidity = cg.readEntry("ItemsValidity", 5);
  84
  85     m_solidEngine = dataEngine("hotplug");
  86     m_solidDeviceEngine = dataEngine("soliddevice");
  87
  88     m_icon = new Plasma::IconWidget(KIcon("device-notifier",NULL), QString());
  89     m_iconName = QString("device-notifier");
  90
  91     Plasma::ToolTipManager::self()->registerWidget(this);
  92
  93     setPopupIcon(m_icon->icon());
  94
  95     //feed the list with what is already reported by the engine
  96
  97     //connect to engine when a device is plug
  98     connect(m_solidEngine, SIGNAL(sourceAdded(const QString&)),
  99             this, SLOT(onSourceAdded(const QString&)));
 100     connect(m_solidEngine, SIGNAL(sourceRemoved(const QString&)),
 101             this, SLOT(onSourceRemoved(const QString&)));
 102
 103     fillPreviousDevices();
 104 }
 105
 106 QWidget *DeviceNotifier::widget()
 107 {
 108     if (!m_dialog) {
 109         m_dialog = new NotifierDialog(this);
 110     }
 111
 112     return m_dialog->dialog();
 113 }
 114
 115 void DeviceNotifier::fillPreviousDevices()
 116 {
 117     m_fillingPreviousDevices = true;
 118     foreach (const QString &source, m_solidEngine->sources()) {
 119             Solid::Device device = Solid::Device(source);
 120             Solid::Device parentDevice = device.parent();
 121             Solid::StorageDrive *drive = parentDevice.as<Solid::StorageDrive>();
 122             if (drive && (drive->isHotpluggable() || drive->isRemovable())) {
 123                 onSourceAdded(source);
 124             }
 125     }
 126     m_fillingPreviousDevices = false;
 127 }
 128
 129 void DeviceNotifier::changeNotifierIcon(const QString& name)
 130 {
 131     if (m_icon && name.isNull()) {
 132         m_icon->setIcon(m_iconName);
 133     } else if (m_icon) {
 134         m_icon->setIcon(name);
 135     }
 136
 137     setPopupIcon(m_icon->icon());
 138 }
 139
 140 void DeviceNotifier::popupEvent(bool show)
 141 {
 142     if (show) {
 143         Plasma::ToolTipManager::self()->clearContent(this);
 144     }
 145 }
 146
 147 void DeviceNotifier::dataUpdated(const QString &source, Plasma::DataEngine::Data data)
 148 {
 149     if (data.size() > 0) {
 150         //data from hotplug engine
 151         if (!data["predicateFiles"].isNull()) {
 152             int nb_actions = 0;
 153             QString lastActionLabel;
 154             foreach (const QString &desktop, data["predicateFiles"].toStringList()) {
 155                 QString filePath = KStandardDirs::locate("data", "solid/actions/" + desktop);
 156                 QList<KServiceAction> services = KDesktopFileActions::userDefinedServices(filePath, true);
 157                 nb_actions += services.size();
 158                 if (services.size() > 0) {
 159                     lastActionLabel = QString(services[0].text());
 160                 }
 161             }
 162             m_dialog->setDeviceData(source,data["predicateFiles"],NotifierDialog::PredicateFilesRole);
 163             m_dialog->setDeviceData(source,data["text"], Qt::DisplayRole);
 164
 165             //icon name
 166             m_dialog->setDeviceData(source,data["icon"], NotifierDialog::IconNameRole);
 167             //icon data
 168             m_dialog->setDeviceData(source,KIcon(data["icon"].toString()), Qt::DecorationRole);
 169
 170             if (nb_actions > 1) {
 171                 QString s = i18np("1 action for this device",
 172                                   "%1 actions for this device",
 173                                   nb_actions);
 174                 m_dialog->setDeviceData(source, s, NotifierDialog::ActionRole);
 175             } else {
 176                 m_dialog->setDeviceData(source, lastActionLabel, NotifierDialog::ActionRole);
 177             }
 178
 179         //data from soliddevice engine
 180         } else {
 181             kDebug() << "DeviceNotifier::solidDeviceEngine updated" << source;
 182             if (data["Device Types"].toStringList().contains("Storage Access")) {
 183                 if (data["Accessible"].toBool() == true) {
 184                     m_dialog->setUnMount(true,source);
 185
 186                     //set icon to mounted device
 187                     QStringList overlays;
 188                     overlays << "emblem-mounted";
 189                     m_dialog->setDeviceData(source, KIcon(m_dialog->getDeviceData(source,NotifierDialog::IconNameRole).toString(), NULL, overlays), Qt::DecorationRole);
 190                 } else if (data["Device Types"].toStringList().contains("OpticalDisc")) {
 191                     //Unmounted optical drive
 192                     m_dialog->setDeviceData(source, KIcon("media-eject"), Qt::DecorationRole);
 193                     //set icon to unmounted device
 194                     m_dialog->setUnMount(true,source);
 195                     m_dialog->setDeviceData(source, KIcon(m_dialog->getDeviceData(source,NotifierDialog::IconNameRole).toString()), Qt::DecorationRole);
 196                 } else {
 197                     m_dialog->setUnMount(false,source);
 198
 199                     //set icon to unmounted device
 200                     m_dialog->setDeviceData(source, KIcon(m_dialog->getDeviceData(source,NotifierDialog::IconNameRole).toString()), Qt::DecorationRole);
 201                 }
 202             }
 203             // actions specific for other types of devices will go here
 204         }
 205    }
 206 }
 207
 208 void DeviceNotifier::notifyDevice(const QString &name)
 209 {
 210     m_lastPlugged<<name;
 211
 212     if (!m_fillingPreviousDevices) {
 213         showPopup();
 214     }
 215 }
 216
 217 void DeviceNotifier::toolTipAboutToShow()
 218 {
 219     Plasma::ToolTipContent toolTip;
 220     if (!m_lastPlugged.isEmpty()) {
 221         Solid::Device device(m_lastPlugged.last());
 222
 223         toolTip.setSubText(i18n("Last plugged in device: %1", device.product()));
 224         toolTip.setImage(KIcon(device.icon()));
 225     } else {
 226         toolTip.setSubText(i18n("No devices plugged in"));
 227         toolTip.setImage(KIcon("device-notifier"));
 228     }
 229
 230     Plasma::ToolTipManager::self()->setContent(this, toolTip);
 231 }
 232
 233 void DeviceNotifier::toolTipHidden()
 234 {
 235     Plasma::ToolTipManager::self()->clearContent(this);
 236 }
 237
 238 void DeviceNotifier::removeLastDeviceNotification(const QString &name)
 239 {
 240     m_lastPlugged.removeAll(name);
 241 }
 242
 243 void DeviceNotifier::onSourceAdded(const QString &name)
 244 {
 245     kDebug() << "DeviceNotifier:: source added" << name;
 246     if (m_dialog->countDevices() == m_numberItems && m_numberItems != 0) {
 247         QString itemUdi = m_dialog->getDeviceUdi(m_dialog->countDevices() - 1);
 248         //disconnect sources and after (important) remove the row
 249         m_solidDeviceEngine->disconnectSource(itemUdi, this);
 250         m_solidEngine->disconnectSource(itemUdi, this);
 251
 252         m_dialog->removeDevice(m_dialog->countDevices() - 1);
 253     }
 254
 255     m_dialog->insertDevice(name);
 256     notifyDevice(name);
 257
 258     m_solidEngine->connectSource(name, this);
 259     m_solidDeviceEngine->connectSource(name, this);
 260 }
 261
 262 void DeviceNotifier::onSourceRemoved(const QString &name)
 263 {
 264     m_solidEngine->disconnectSource(name, this);
 265     m_solidDeviceEngine->disconnectSource(name, this);
 266
 267     m_dialog->removeDevice(name);
 268     removeLastDeviceNotification(name);
 269 }
 270
 271 #include "devicenotifier.moc"