1 files changed, 32 insertions, 0 deletions
diff --git a/plugin/src/Caelestia/Internal/sleepnotifier.cpp b/plugin/src/Caelestia/Internal/sleepnotifier.cpp
new file mode 100644
index 0000000..b730f38
--- /dev/null
+++ b/plugin/src/Caelestia/Internal/sleepnotifier.cpp
@@ -0,0 +1,32 @@
+#include "sleepnotifier.hpp"
+
+#include <QtDBus/qdbusconnection.h>
+#include <QtDBus/qdbuserror.h>
+
+namespace caelestia::internal {
+
+SleepNotifier::SleepNotifier(QObject* parent)
+    : QObject(parent) {
+    auto bus = QDBusConnection::systemBus();
+    if (!bus.isConnected()) {
+        qWarning() << "SleepNotifier::SleepNotifier: failed to connect to system bus:" << bus.lastError().message();
+        return;
+    }
+
+    const bool ok = bus.connect("org.freedesktop.login1", "/org/freedesktop/login1", "org.freedesktop.login1.Manager",
+        "PrepareForSleep", this, SLOT(handlePrepareForSleep(bool)));
+
+    if (!ok) {
+        qWarning() << "SleepNotifier::SleepNotifier: failed to connect to dbus:" << bus.lastError().message();
+    }
+}
+
+void SleepNotifier::handlePrepareForSleep(bool sleep) {
+    if (sleep) {
+        emit aboutToSleep();
+    } else {
+        emit resumed();
+    }
+}
+
+} // namespace caelestia::internal