124 lines
4.1 KiB
C++
124 lines
4.1 KiB
C++
/*
|
|
* Copyright (C) 2019 Tianjin KYLIN Information Technology Co., Ltd.
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU Lesser General Public License as published by
|
|
* the Free Software Foundation; either version 2.1, 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, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
#include "snidaemon.h"
|
|
#include "statusnotifieritem_interface.h"
|
|
SniDaemon::SniDaemon()
|
|
{
|
|
QDBusConnection dbus = QDBusConnection::sessionBus();
|
|
dbus.registerObject(QStringLiteral("/StatusNotifierWatcher"),"org.kde.StatusNotifierWatcher",this,QDBusConnection::ExportAllContents);
|
|
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, &SniDaemon::serviceUnregistered);
|
|
|
|
}
|
|
|
|
SniDaemon::~SniDaemon()
|
|
{
|
|
QDBusConnection dbus = QDBusConnection::sessionBus();
|
|
dbus.unregisterService(QStringLiteral("org.kde.StatusNotifierWatcher"));
|
|
}
|
|
|
|
|
|
void SniDaemon::serviceUnregistered(const QString& name)
|
|
{
|
|
qDebug()<<"Service "<< name << "unregistered";
|
|
m_serviceWatcher->removeWatchedService(name);
|
|
|
|
QString match = name + QLatin1Char('/');
|
|
QStringList::Iterator it = m_registeredServices.begin();
|
|
qDebug()<<m_registeredServices;
|
|
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 SniDaemon::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 (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());
|
|
if (trayclient.isValid()) {
|
|
qDebug() << "Registering" << notifierItemId << "to system tray";
|
|
m_registeredServices.append(notifierItemId);
|
|
emit StatusNotifierItemRegistered(notifierItemId);
|
|
} else {
|
|
m_serviceWatcher->removeWatchedService(service);
|
|
}
|
|
} else {
|
|
m_serviceWatcher->removeWatchedService(service);
|
|
}
|
|
}
|
|
|
|
void SniDaemon::RegisterStatusNotifierHost(const QString &service)
|
|
{
|
|
if (service.contains(QLatin1String("org.kde.StatusNotifierHost-")) &&
|
|
QDBusConnection::sessionBus().interface()->isServiceRegistered(service).value() &&
|
|
!m_statusNotifierHostServices.contains(service)) {
|
|
qDebug()<<"Registering"<<service<<"as system tray";
|
|
|
|
m_statusNotifierHostServices.insert(service);
|
|
m_serviceWatcher->addWatchedService(service);
|
|
emit StatusNotifierHostRegistered();
|
|
}
|
|
}
|
|
|
|
QStringList SniDaemon::RegisteredStatusNotifierItems() const
|
|
{
|
|
return m_registeredServices;
|
|
}
|
|
|
|
bool SniDaemon::IsStatusNotifierHostRegistered() const
|
|
{
|
|
return !m_statusNotifierHostServices.isEmpty();
|
|
}
|
|
|
|
int SniDaemon::ProtocolVersion() const
|
|
{
|
|
return 0;
|
|
}
|