plugin: add sleep notifier

Lock before sleep

4 files changed, 58 insertions, 1 deletions

diff --git a/plugin/src/Caelestia/CMakeLists.txt b/plugin/src/Caelestia/CMakeLists.txt
index d344ef7..18c7315 100644
--- a/plugin/src/Caelestia/CMakeLists.txt
+++ b/plugin/src/Caelestia/CMakeLists.txt
@@ -1,4 +1,4 @@
-find_package(Qt6 REQUIRED COMPONENTS Core Qml Gui Quick Concurrent Sql Network)
+find_package(Qt6 REQUIRED COMPONENTS Core Qml Gui Quick Concurrent Sql Network DBus)
 find_package(PkgConfig REQUIRED)
 pkg_check_modules(Qalculate IMPORTED_TARGET libqalculate REQUIRED)
 pkg_check_modules(Pipewire IMPORTED_TARGET libpipewire-0.3 REQUIRED)
diff --git a/plugin/src/Caelestia/Internal/CMakeLists.txt b/plugin/src/Caelestia/Internal/CMakeLists.txt
index 96538b8..3a49d5a 100644
--- a/plugin/src/Caelestia/Internal/CMakeLists.txt
+++ b/plugin/src/Caelestia/Internal/CMakeLists.txt
@@ -5,9 +5,11 @@ qml_module(caelestia-internal
         circularindicatormanager.hpp circularindicatormanager.cpp
         hyprdevices.hpp hyprdevices.cpp
         hyprextras.hpp hyprextras.cpp
+        sleepnotifier.hpp sleepnotifier.cpp
     LIBRARIES
         Qt::Gui
         Qt::Quick
         Qt::Concurrent
         Qt::Network
+        Qt::DBus
 )
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
diff --git a/plugin/src/Caelestia/Internal/sleepnotifier.hpp b/plugin/src/Caelestia/Internal/sleepnotifier.hpp
new file mode 100644
index 0000000..aaa6c68
--- /dev/null
+++ b/plugin/src/Caelestia/Internal/sleepnotifier.hpp
@@ -0,0 +1,23 @@
+#pragma once
+
+#include <qobject.h>
+#include <qqmlintegration.h>
+
+namespace caelestia::internal {
+
+class SleepNotifier : public QObject {
+    Q_OBJECT
+    QML_ELEMENT
+
+public:
+    explicit SleepNotifier(QObject* parent = nullptr);
+
+signals:
+    void aboutToSleep();
+    void resumed();
+
+private slots:
+    void handlePrepareForSleep(bool sleep);
+};
+
+} // namespace caelestia::internal