File indexing completed on 2024-05-19 05:39:09

 /*
     SPDX-FileCopyrightText: 2017 Valerio Pilo <vpilo@coldshock.net>
 
     SPDX-License-Identifier: LGPL-2.0-only
 */
 
 #include <QCommandLineParser>
 #include <QDebug>
 
 #include <QDBusConnection>
 #include <QDBusConnectionInterface>
 #include <QDBusServiceWatcher>
 
 #include <unistd.h>
 
 #include "debug_p.h"
 #include "waiter.h"
 
 constexpr static const char dbusServiceName[] = "org.freedesktop.Notifications";
 
 Waiter::Waiter(int argc, char **argv)
     : QCoreApplication(argc, argv)
     , mService(dbusServiceName)
 {
     setApplicationName(QStringLiteral("plasma_waitforname"));
     setApplicationVersion(QStringLiteral("1.0"));
 
     QCommandLineParser parser;
     parser.setApplicationDescription(
         QStringLiteral("Waits for D-Bus registration of the Notifications service.\n"
                        "Prevents notifications from being processed before the desktop is ready."));
     parser.addHelpOption();
     parser.addVersionOption();
     parser.addPositionalArgument(QStringLiteral("service"),
                                  QStringLiteral("Optionally listen for a service different than '%1'").arg(mService),
                                  QStringLiteral("[service]"));
     parser.process(*this);
 
     const QStringList args = parser.positionalArguments();
     if (!args.isEmpty()) {
         mService = args.at(0);
     }
 }
 
 bool Waiter::waitForService()
 {
     QDBusConnection sessionBus = QDBusConnection::sessionBus();
 
     if (sessionBus.interface()->isServiceRegistered(mService)) {
         qCDebug(LOG_PLASMA) << "WaitForName: Service" << mService << "is already registered";
         return false;
     }
 
     qCDebug(LOG_PLASMA) << "WaitForName: Waiting for appearance of service" << mService << "for" << dbusTimeoutSec << "seconds";
 
     QDBusServiceWatcher *watcher = new QDBusServiceWatcher(this);
     watcher->setConnection(sessionBus);
     watcher->setWatchMode(QDBusServiceWatcher::WatchForRegistration);
     watcher->addWatchedService(mService);
     connect(watcher, &QDBusServiceWatcher::serviceRegistered, this, &Waiter::registered);
 
     mTimeoutTimer.setSingleShot(true);
     mTimeoutTimer.setInterval(dbusTimeoutSec * 1000);
     connect(&mTimeoutTimer, &QTimer::timeout, this, &Waiter::timeout);
     mTimeoutTimer.start();
 
     return true;
 }
 
 void Waiter::registered()
 {
     qCDebug(LOG_PLASMA) << "WaitForName: Service was registered after" << (dbusTimeoutSec - (mTimeoutTimer.remainingTime() / 1000)) << "seconds";
     mTimeoutTimer.stop();
     exit(0);
 }
 
 void Waiter::timeout()
 {
     qCInfo(LOG_PLASMA) << "WaitForName: Service was not registered within timeout";
     exit(1);
 }