diff --git a/statusnotifierwatcher/statusnotifierwatcher.cpp b/statusnotifierwatcher/statusnotifierwatcher.cpp index b65973188..9cbea3e51 100644 --- a/statusnotifierwatcher/statusnotifierwatcher.cpp +++ b/statusnotifierwatcher/statusnotifierwatcher.cpp @@ -1,135 +1,139 @@ /*************************************************************************** * Copyright 2009 by Marco Martin * * * * This program is free software; you can redistribute it and/or modify * * it under the terms of the GNU General Public License as published by * * the Free Software Foundation; either version 2 of the License, or * * (at your option) any later version. * * * * This program is distributed in the hope that it will be useful, * * but WITHOUT ANY WARRANTY; without even the implied warranty of * * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * * GNU General Public License for more details. * * * * You should have received a copy of the GNU General Public License * * along with this program; if not, write to the * * Free Software Foundation, Inc., * * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA . * ***************************************************************************/ #include "statusnotifierwatcher.h" #include #include #include #include #include "statusnotifierwatcheradaptor.h" #include "statusnotifieritem_interface.h" K_PLUGIN_CLASS_WITH_JSON(StatusNotifierWatcher, "statusnotifierwatcher.json") StatusNotifierWatcher::StatusNotifierWatcher(QObject *parent, const QList&) : KDEDModule(parent) { setModuleName(QStringLiteral("StatusNotifierWatcher")); new StatusNotifierWatcherAdaptor(this); QDBusConnection dbus = QDBusConnection::sessionBus(); dbus.registerObject(QStringLiteral("/StatusNotifierWatcher"), this); dbus.registerService(QStringLiteral("org.kde.StatusNotifierWatcher")); m_serviceWatcher = new QDBusServiceWatcher(this); m_serviceWatcher->setConnection(dbus); m_serviceWatcher->setWatchMode(QDBusServiceWatcher::WatchForUnregistration); connect(m_serviceWatcher, &QDBusServiceWatcher::serviceUnregistered, this, &StatusNotifierWatcher::serviceUnregistered); } StatusNotifierWatcher::~StatusNotifierWatcher() { QDBusConnection dbus = QDBusConnection::sessionBus(); dbus.unregisterService(QStringLiteral("org.kde.StatusNotifierWatcher")); } void StatusNotifierWatcher::RegisterStatusNotifierItem(const QString &serviceOrPath) { QString service; QString path; if (serviceOrPath.startsWith(QLatin1Char('/'))) { service = message().service(); path = serviceOrPath; } else { service = serviceOrPath; path = QStringLiteral("/StatusNotifierItem"); } QString notifierItemId = service + path; - if (QDBusConnection::sessionBus().interface()->isServiceRegistered(service).value() && - !m_registeredServices.contains(notifierItemId)) { - qDebug()<<"Registering" << notifierItemId << "to system tray"; - + if (m_registeredServices.contains(notifierItemId)) { + return; + } + m_serviceWatcher->addWatchedService(service); + if (QDBusConnection::sessionBus().interface()->isServiceRegistered(service).value()) { //check if the service has registered a SystemTray object - org::kde::StatusNotifierItem trayclient(service, path, - QDBusConnection::sessionBus()); + org::kde::StatusNotifierItem trayclient(service, path, QDBusConnection::sessionBus()); if (trayclient.isValid()) { + qDebug() << "Registering" << notifierItemId << "to system tray"; m_registeredServices.append(notifierItemId); - m_serviceWatcher->addWatchedService(service); emit StatusNotifierItemRegistered(notifierItemId); + } else { + m_serviceWatcher->removeWatchedService(service); } + } else { + m_serviceWatcher->removeWatchedService(service); } } QStringList StatusNotifierWatcher::RegisteredStatusNotifierItems() const { return m_registeredServices; } void StatusNotifierWatcher::serviceUnregistered(const QString& name) { qDebug()<<"Service "<< name << "unregistered"; m_serviceWatcher->removeWatchedService(name); QString match = name + QLatin1Char('/'); QStringList::Iterator it = m_registeredServices.begin(); while (it != m_registeredServices.end()) { if (it->startsWith(match)) { QString name = *it; it = m_registeredServices.erase(it); emit StatusNotifierItemUnregistered(name); } else { ++it; } } if (m_statusNotifierHostServices.contains(name)) { m_statusNotifierHostServices.remove(name); emit StatusNotifierHostUnregistered(); } } void StatusNotifierWatcher::RegisterStatusNotifierHost(const QString &service) { if (service.contains(QLatin1String("org.kde.StatusNotifierHost-")) && QDBusConnection::sessionBus().interface()->isServiceRegistered(service).value() && !m_statusNotifierHostServices.contains(service)) { qDebug()<<"Registering"<addWatchedService(service); emit StatusNotifierHostRegistered(); } } bool StatusNotifierWatcher::IsStatusNotifierHostRegistered() const { return !m_statusNotifierHostServices.isEmpty(); } int StatusNotifierWatcher::ProtocolVersion() const { return 0; } #include "statusnotifierwatcher.moc"