1 files changed, 54 insertions, 0 deletions
diff --git a/modules/BatteryMonitor.qml b/modules/BatteryMonitor.qml
new file mode 100644
index 0000000..f407edf
--- /dev/null
+++ b/modules/BatteryMonitor.qml
@@ -0,0 +1,54 @@
+import qs.config
+import Caelestia
+import Quickshell
+import Quickshell.Services.UPower
+import QtQuick
+
+Scope {
+    id: root
+
+    readonly property list<var> warnLevels: [...Config.general.battery.warnLevels].sort((a, b) => b.level - a.level)
+
+    Connections {
+        target: UPower
+
+        function onOnBatteryChanged(): void {
+            if (UPower.onBattery) {
+                Toaster.toast(qsTr("Charger unplugged"), qsTr("Battery is now on AC"), "power_off");
+            } else {
+                Toaster.toast(qsTr("Charger plugged in"), qsTr("Battery is charging"), "battery_android_frame_bolt");
+                for (const level of root.warnLevels)
+                    level.warned = false;
+            }
+        }
+    }
+
+    Connections {
+        target: UPower.displayDevice
+
+        function onPercentageChanged(): void {
+            if (!UPower.onBattery)
+                return;
+
+            const p = UPower.displayDevice.percentage;
+            for (const level of root.warnLevels) {
+                if (p <= level.level && !level.warned) {
+                    level.warned = true;
+                    Toaster.toast(level.title ?? qsTr("Battery warning"), level.message ?? qsTr("Battery level is low"), level.icon ?? "battery_android_alert", level.critical ? Toast.Error : Toast.Warning);
+                }
+            }
+
+            if (!hibernateTimer.running && p <= Config.general.battery.criticalLevel) {
+                Toaster.toast(qsTr("Hibernating in 5 seconds"), qsTr("Hibernating to prevent data loss"), "battery_android_alert", Toast.Error);
+                hibernateTimer.start();
+            }
+        }
+    }
+
+    Timer {
+        id: hibernateTimer
+
+        interval: 5000
+        onTriggered: Quickshell.execDetached(["systemctl", "hibernate"])
+    }
+}