1 files changed, 93 insertions, 0 deletions
diff --git a/src/keep-screen-on.c b/src/keep-screen-on.c
new file mode 100644
index 0000000..c9979ac
--- /dev/null
+++ b/src/keep-screen-on.c
@@ -0,0 +1,93 @@
+/*
+ * Copyright 2025 UBports Foundation
+ *
+ * This program is free software: you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 3, as published
+ * by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranties of
+ * MERCHANTABILITY, SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR
+ * PURPOSE.  See the GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ * Authors:
+ *   Lorenzo Torracchi <lorenzotorracchi@mail.com>
+ */
+
+#include "keep-screen-on.h"
+
+#include <ayatana/common/utils.h>
+#include <gio/gio.h>
+
+#define UNITY_SCREEN_BUS_NAME "com.canonical.Unity.Screen"
+#define UNITY_SCREEN_OBJECT_PATH "/com/canonical/Unity/Screen"
+#define UNITY_SCREEN_INTERFACE "com.canonical.Unity.Screen"
+
+static guint request_id = 0;
+
+void
+toggle_keep_screen_on_action (GAction *action,
+                              GVariant *parameter G_GNUC_UNUSED,
+                              gpointer data G_GNUC_UNUSED)
+{
+  GVariant *state = g_action_get_state (action);
+  gboolean enabled = g_variant_get_boolean (state);
+  GError *error = NULL;
+  GDBusProxy *proxy = g_dbus_proxy_new_for_bus_sync (
+      G_BUS_TYPE_SYSTEM, G_DBUS_PROXY_FLAGS_NONE, NULL, UNITY_SCREEN_BUS_NAME,
+      UNITY_SCREEN_OBJECT_PATH, UNITY_SCREEN_INTERFACE, NULL, &error);
+
+  if (!proxy)
+    {
+      g_warning ("Failed to create D-Bus proxy: %s", error->message);
+      g_clear_error (&error);
+      return;
+    }
+  if (!enabled)
+    {
+      GVariant *result
+          = g_dbus_proxy_call_sync (proxy, "keepDisplayOn", NULL,
+                                    G_DBUS_CALL_FLAGS_NONE, -1, NULL, &error);
+      if (result)
+        {
+          g_variant_get (result, "(i)", &request_id);
+          g_variant_unref (result);
+          g_simple_action_set_state (G_SIMPLE_ACTION (action),
+                                     g_variant_new_boolean (!enabled));
+        }
+      else
+        {
+          g_warning ("D-Bus call keepDisplayOn failed: %s", error->message);
+          g_clear_error (&error);
+        }
+    }
+  else if (request_id != 0)
+    {
+      GVariant *params = g_variant_new ("(i)", request_id);
+      g_dbus_proxy_call_sync (proxy, "removeDisplayOnRequest", params,
+                              G_DBUS_CALL_FLAGS_NONE, -1, NULL, &error);
+      if (error)
+        {
+          g_warning ("D-Bus call removeDisplayOnRequest failed: %s",
+                     error->message);
+          g_clear_error (&error);
+        }
+      else
+        {
+          g_simple_action_set_state (G_SIMPLE_ACTION (action),
+                                     g_variant_new_boolean (!enabled));
+          request_id = 0;
+        }
+    }
+
+  g_object_unref (proxy);
+}
+
+int
+keep_screen_on_supported ()
+{
+  return ayatana_common_utils_is_lomiri ();
+}