From cdccb9e72673c90e329aaa9d737b98b9ecb1a9ba Mon Sep 17 00:00:00 2001
From: S4NKALP <sankalptharu50028@gmail.com>
Date: Wed, 15 Jul 2026 09:03:39 +0545
Subject: [PATCH] feat(src/window/controlcenter): implement caffeine inhibitor
 toggling
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Add support for managing systemd‑inhibit processes via a PID file; introduce _caffeine_is_active helper; modify _check_initial_states and toggle_caffeine to start/stop inhibition and update UI icons and labels.
---
 src/window/controlcenter/main.py | 65 ++++++++++++++++++++++++++++++--
 1 file changed, 62 insertions(+), 3 deletions(-)

diff --git a/src/window/controlcenter/main.py b/src/window/controlcenter/main.py
index 4cf9cd7..ef9ffa2 100644
--- a/src/window/controlcenter/main.py
+++ b/src/window/controlcenter/main.py
@@ -1,3 +1,7 @@
+import os
+import signal
+import subprocess
+
 from fabric.utils import Gdk, GLib, idle_add, logger
 from fabric.widgets.box import Box
 from fabric.widgets.button import Button
@@ -7,7 +11,6 @@ from fabric.widgets.label import Label
 from fabric.widgets.overlay import Overlay
 
 from services.brightness import Brightness
-from services.inhibit import get_inhibit_service
 from services.network import NetworkClient
 from shared.widgets.flat_scale import FlatScale
 from shared.window.applet_window import AppletWindow
@@ -33,6 +36,26 @@ def get_brightness_service() -> "Brightness":
     return _brightness_instance
 
 
+_CAFFEINE_PID_FILE = os.path.join(
+    os.environ.get("XDG_RUNTIME_DIR") or "/tmp", "modus-caffeine.pid"
+)
+_CAFFEINE_INHIBIT_WHY = "Modus Caffeine"
+
+
+def _caffeine_is_active() -> bool:
+    try:
+        with open(_CAFFEINE_PID_FILE) as f:
+            pid = int(f.read().strip())
+    except (OSError, ValueError):
+        return False
+    try:
+        with open(f"/proc/{pid}/cmdline", "rb") as f:
+            data = f.read().decode(errors="ignore")
+        return "systemd-inhibit" in data and _CAFFEINE_INHIBIT_WHY in data
+    except OSError:
+        return False
+
+
 class ModusControlCenter(AppletWindow):
     def __init__(self, parent=None, pointing_to=None, **kwargs):
         super().__init__(
@@ -719,7 +742,13 @@ class ModusControlCenter(AppletWindow):
             self.per_app_volume_center_box.set_size_request(300, -1)
 
     def _check_initial_states(self):
-        self.caffeine_mode = get_inhibit_service().active
+        self.caffeine_mode = _caffeine_is_active()
+        self.caffeine_icon.dynamic_file(
+            "applets/caffeine-on.svg"
+            if self.caffeine_mode
+            else "applets/caffeine-off.svg"
+        )
+        self.caffeine_status_label.set_label("On" if self.caffeine_mode else "Off")
         self.flight_mode = False
 
     def set_dont_disturb(self, *_):
@@ -763,7 +792,37 @@ class ModusControlCenter(AppletWindow):
 
     def toggle_caffeine(self, *_):
         try:
-            self.caffeine_mode = get_inhibit_service().toggle()
+            if _caffeine_is_active():
+                try:
+                    with open(_CAFFEINE_PID_FILE) as f:
+                        pid = int(f.read().strip())
+                    with open(f"/proc/{pid}/cmdline", "rb") as f:
+                        data = f.read().decode(errors="ignore")
+                    if "systemd-inhibit" in data and _CAFFEINE_INHIBIT_WHY in data:
+                        os.kill(pid, signal.SIGTERM)
+                except (OSError, ValueError):
+                    pass
+                try:
+                    os.unlink(_CAFFEINE_PID_FILE)
+                except OSError:
+                    pass
+                self.caffeine_mode = False
+            else:
+                cmd = [
+                    "systemd-inhibit",
+                    "--what=idle:sleep",
+                    f"--why={_CAFFEINE_INHIBIT_WHY}",
+                    "--mode=block",
+                    "sleep",
+                    str(100_000_000),
+                ]
+                proc = subprocess.Popen(cmd)
+                try:
+                    with open(_CAFFEINE_PID_FILE, "w") as f:
+                        f.write(str(proc.pid))
+                except OSError:
+                    pass
+                self.caffeine_mode = True
             self.caffeine_icon.dynamic_file(
                 "applets/caffeine-on.svg"
                 if self.caffeine_mode