mirror of https://github.com/cutefishos/statusbar
Add StatusNotifierItemHost
reionwong
4 years ago
parent
f415f331c7
commit
017515d5d5
@ -0,0 +1,189 @@
|
||||
/*
|
||||
|
||||
SPDX-FileCopyrightText: 2009 Marco Martin <notmart@gmail.com>
|
||||
SPDX-FileCopyrightText: 2009 Matthieu Gallien <matthieu_gallien@yahoo.fr>
|
||||
|
||||
SPDX-License-Identifier: GPL-2.0-or-later
|
||||
*/
|
||||
|
||||
#include "statusnotifieritemhost.h"
|
||||
#include "statusnotifieritemsource.h"
|
||||
#include <QStringList>
|
||||
#include <QDebug>
|
||||
|
||||
#include "dbusproperties.h"
|
||||
|
||||
#include <iostream>
|
||||
|
||||
class StatusNotifierItemHostSingleton
|
||||
{
|
||||
public:
|
||||
StatusNotifierItemHost self;
|
||||
};
|
||||
|
||||
Q_GLOBAL_STATIC(StatusNotifierItemHostSingleton, privateStatusNotifierItemHostSelf)
|
||||
|
||||
static const QString s_watcherServiceName(QStringLiteral("org.kde.StatusNotifierWatcher"));
|
||||
|
||||
StatusNotifierItemHost::StatusNotifierItemHost()
|
||||
: QObject()
|
||||
, m_statusNotifierWatcher(nullptr)
|
||||
{
|
||||
init();
|
||||
}
|
||||
|
||||
StatusNotifierItemHost::~StatusNotifierItemHost()
|
||||
{
|
||||
QDBusConnection::sessionBus().unregisterService(m_serviceName);
|
||||
}
|
||||
|
||||
StatusNotifierItemHost *StatusNotifierItemHost::self()
|
||||
{
|
||||
return &privateStatusNotifierItemHostSelf()->self;
|
||||
}
|
||||
|
||||
const QList<QString> StatusNotifierItemHost::services() const
|
||||
{
|
||||
return m_sniServices.keys();
|
||||
}
|
||||
|
||||
StatusNotifierItemSource *StatusNotifierItemHost::itemForService(const QString service)
|
||||
{
|
||||
return m_sniServices.value(service);
|
||||
}
|
||||
|
||||
void StatusNotifierItemHost::init()
|
||||
{
|
||||
if (QDBusConnection::sessionBus().isConnected()) {
|
||||
m_serviceName = "org.kde.StatusNotifierHost-" + QString::number(QCoreApplication::applicationPid());
|
||||
QDBusConnection::sessionBus().registerService(m_serviceName);
|
||||
|
||||
QDBusServiceWatcher *watcher =
|
||||
new QDBusServiceWatcher(s_watcherServiceName, QDBusConnection::sessionBus(), QDBusServiceWatcher::WatchForOwnerChange, this);
|
||||
connect(watcher, &QDBusServiceWatcher::serviceOwnerChanged, this, &StatusNotifierItemHost::serviceChange);
|
||||
|
||||
registerWatcher(s_watcherServiceName);
|
||||
}
|
||||
}
|
||||
|
||||
void StatusNotifierItemHost::serviceChange(const QString &name, const QString &oldOwner, const QString &newOwner)
|
||||
{
|
||||
qDebug() << "Service" << name << "status change, old owner:" << oldOwner << "new:" << newOwner;
|
||||
|
||||
if (newOwner.isEmpty()) {
|
||||
// unregistered
|
||||
unregisterWatcher(name);
|
||||
} else if (oldOwner.isEmpty()) {
|
||||
// registered
|
||||
registerWatcher(name);
|
||||
}
|
||||
}
|
||||
|
||||
void StatusNotifierItemHost::registerWatcher(const QString &service)
|
||||
{
|
||||
if (service == s_watcherServiceName) {
|
||||
delete m_statusNotifierWatcher;
|
||||
|
||||
m_statusNotifierWatcher =
|
||||
new org::kde::StatusNotifierWatcher(s_watcherServiceName, QStringLiteral("/StatusNotifierWatcher"), QDBusConnection::sessionBus());
|
||||
if (m_statusNotifierWatcher->isValid()) {
|
||||
m_statusNotifierWatcher->call(QDBus::NoBlock, QStringLiteral("RegisterStatusNotifierHost"), m_serviceName);
|
||||
|
||||
OrgFreedesktopDBusPropertiesInterface propetriesIface(m_statusNotifierWatcher->service(),
|
||||
m_statusNotifierWatcher->path(),
|
||||
m_statusNotifierWatcher->connection());
|
||||
|
||||
connect(m_statusNotifierWatcher,
|
||||
&OrgKdeStatusNotifierWatcherInterface::StatusNotifierItemRegistered,
|
||||
this,
|
||||
&StatusNotifierItemHost::serviceRegistered);
|
||||
connect(m_statusNotifierWatcher,
|
||||
&OrgKdeStatusNotifierWatcherInterface::StatusNotifierItemUnregistered,
|
||||
this,
|
||||
&StatusNotifierItemHost::serviceUnregistered);
|
||||
|
||||
QDBusPendingReply<QDBusVariant> pendingItems = propetriesIface.Get(m_statusNotifierWatcher->interface(), "RegisteredStatusNotifierItems");
|
||||
|
||||
QDBusPendingCallWatcher *watcher = new QDBusPendingCallWatcher(pendingItems, this);
|
||||
connect(watcher, &QDBusPendingCallWatcher::finished, this, [=]() {
|
||||
watcher->deleteLater();
|
||||
QDBusReply<QDBusVariant> reply = *watcher;
|
||||
QStringList registeredItems = reply.value().variant().toStringList();
|
||||
foreach (const QString &service, registeredItems) {
|
||||
if (!m_sniServices.contains(service)) { // due to async nature of this call, service may be already there
|
||||
addSNIService(service);
|
||||
}
|
||||
}
|
||||
});
|
||||
} else {
|
||||
delete m_statusNotifierWatcher;
|
||||
m_statusNotifierWatcher = nullptr;
|
||||
qDebug() << "System tray daemon not reachable";
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void StatusNotifierItemHost::unregisterWatcher(const QString &service)
|
||||
{
|
||||
if (service == s_watcherServiceName) {
|
||||
qDebug() << s_watcherServiceName << "disappeared";
|
||||
|
||||
disconnect(m_statusNotifierWatcher,
|
||||
&OrgKdeStatusNotifierWatcherInterface::StatusNotifierItemRegistered,
|
||||
this,
|
||||
&StatusNotifierItemHost::serviceRegistered);
|
||||
disconnect(m_statusNotifierWatcher,
|
||||
&OrgKdeStatusNotifierWatcherInterface::StatusNotifierItemUnregistered,
|
||||
this,
|
||||
&StatusNotifierItemHost::serviceUnregistered);
|
||||
|
||||
removeAllSNIServices();
|
||||
|
||||
delete m_statusNotifierWatcher;
|
||||
m_statusNotifierWatcher = nullptr;
|
||||
}
|
||||
}
|
||||
|
||||
void StatusNotifierItemHost::serviceRegistered(const QString &service)
|
||||
{
|
||||
qDebug() << "Registering" << service;
|
||||
addSNIService(service);
|
||||
}
|
||||
|
||||
void StatusNotifierItemHost::serviceUnregistered(const QString &service)
|
||||
{
|
||||
removeSNIService(service);
|
||||
}
|
||||
|
||||
void StatusNotifierItemHost::removeAllSNIServices()
|
||||
{
|
||||
QHashIterator<QString, StatusNotifierItemSource *> it(m_sniServices);
|
||||
while (it.hasNext()) {
|
||||
it.next();
|
||||
|
||||
StatusNotifierItemSource *item = it.value();
|
||||
item->disconnect();
|
||||
item->deleteLater();
|
||||
Q_EMIT itemRemoved(it.key());
|
||||
}
|
||||
m_sniServices.clear();
|
||||
}
|
||||
|
||||
void StatusNotifierItemHost::addSNIService(const QString &service)
|
||||
{
|
||||
StatusNotifierItemSource *item = new StatusNotifierItemSource(service, this);
|
||||
m_sniServices.insert(service, item);
|
||||
Q_EMIT itemAdded(service);
|
||||
}
|
||||
|
||||
void StatusNotifierItemHost::removeSNIService(const QString &service)
|
||||
{
|
||||
if (m_sniServices.contains(service)) {
|
||||
auto item = m_sniServices.value(service);
|
||||
item->disconnect();
|
||||
item->deleteLater();
|
||||
m_sniServices.remove(service);
|
||||
Q_EMIT itemRemoved(service);
|
||||
}
|
||||
}
|
||||
|
||||
@ -0,0 +1,51 @@
|
||||
/*
|
||||
SPDX-FileCopyrightText: 2009 Marco Martin <notmart@gmail.com>
|
||||
SPDX-FileCopyrightText: 2009 Matthieu Gallien <matthieu_gallien@yahoo.fr>
|
||||
|
||||
SPDX-License-Identifier: GPL-2.0-or-later
|
||||
*/
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "statusnotifierwatcher_interface.h"
|
||||
#include <QDBusConnection>
|
||||
|
||||
class StatusNotifierItemSource;
|
||||
|
||||
// Define our plasma Runner
|
||||
class StatusNotifierItemHost : public QObject
|
||||
{
|
||||
Q_OBJECT
|
||||
|
||||
public:
|
||||
StatusNotifierItemHost();
|
||||
virtual ~StatusNotifierItemHost();
|
||||
|
||||
static StatusNotifierItemHost *self();
|
||||
|
||||
const QList<QString> services() const;
|
||||
StatusNotifierItemSource *itemForService(const QString service);
|
||||
|
||||
Q_SIGNALS:
|
||||
void itemAdded(const QString &service);
|
||||
void itemRemoved(const QString &service);
|
||||
|
||||
private Q_SLOTS:
|
||||
void serviceChange(const QString &name, const QString &oldOwner, const QString &newOwner);
|
||||
void registerWatcher(const QString &service);
|
||||
void unregisterWatcher(const QString &service);
|
||||
void serviceRegistered(const QString &service);
|
||||
void serviceUnregistered(const QString &service);
|
||||
|
||||
private:
|
||||
void init();
|
||||
void removeAllSNIServices();
|
||||
void addSNIService(const QString &service);
|
||||
void removeSNIService(const QString &service);
|
||||
int indexOfItem(const QString &service) const;
|
||||
|
||||
org::kde::StatusNotifierWatcher *m_statusNotifierWatcher;
|
||||
QString m_serviceName;
|
||||
static const int s_protocolVersion = 0;
|
||||
QHash<QString, StatusNotifierItemSource *> m_sniServices;
|
||||
};
|
||||
Loading…
Reference in New Issue