1 files changed, 85 insertions, 14 deletions

diff --git a/sway/desktop/idle_inhibit_v1.c b/sway/desktop/idle_inhibit_v1.c
index 87b4ef43..b981e5e5 100644
--- a/sway/desktop/idle_inhibit_v1.c
+++ b/sway/desktop/idle_inhibit_v1.c
@@ -2,18 +2,24 @@
 #include <wlr/types/wlr_idle.h>
 #include "log.h"
 #include "sway/desktop/idle_inhibit_v1.h"
+#include "sway/input/seat.h"
+#include "sway/tree/container.h"
 #include "sway/tree/view.h"
 #include "sway/server.h"
 
 
+static void destroy_inhibitor(struct sway_idle_inhibitor_v1 *inhibitor) {
+        wl_list_remove(&inhibitor->link);
+        wl_list_remove(&inhibitor->destroy.link);
+        sway_idle_inhibit_v1_check_active(inhibitor->manager);
+        free(inhibitor);
+}
+
 static void handle_destroy(struct wl_listener *listener, void *data) {
         struct sway_idle_inhibitor_v1 *inhibitor =
                 wl_container_of(listener, inhibitor, destroy);
         sway_log(SWAY_DEBUG, "Sway idle inhibitor destroyed");
-        wl_list_remove(&inhibitor->link);
-        wl_list_remove(&inhibitor->destroy.link);
-        idle_inhibit_v1_check_active(inhibitor->manager);
-        free(inhibitor);
+        destroy_inhibitor(inhibitor);
 }
 
 void handle_idle_inhibitor_v1(struct wl_listener *listener, void *data) {
@@ -29,28 +35,93 @@ void handle_idle_inhibitor_v1(struct wl_listener *listener, void *data) {
         }
 
         inhibitor->manager = manager;
+        inhibitor->mode = INHIBIT_IDLE_APPLICATION;
         inhibitor->view = view_from_wlr_surface(wlr_inhibitor->surface);
         wl_list_insert(&manager->inhibitors, &inhibitor->link);
 
-
         inhibitor->destroy.notify = handle_destroy;
         wl_signal_add(&wlr_inhibitor->events.destroy, &inhibitor->destroy);
 
-        idle_inhibit_v1_check_active(manager);
+        sway_idle_inhibit_v1_check_active(manager);
+}
+
+void sway_idle_inhibit_v1_user_inhibitor_register(struct sway_view *view,
+                enum sway_idle_inhibit_mode mode) {
+        struct sway_idle_inhibitor_v1 *inhibitor =
+                calloc(1, sizeof(struct sway_idle_inhibitor_v1));
+        if (!inhibitor) {
+                return;
+        }
+
+        inhibitor->manager = server.idle_inhibit_manager_v1;
+        inhibitor->mode = mode;
+        inhibitor->view = view;
+        wl_list_insert(&inhibitor->manager->inhibitors, &inhibitor->link);
+
+        inhibitor->destroy.notify = handle_destroy;
+        wl_signal_add(&view->events.unmap, &inhibitor->destroy);
+
+        sway_idle_inhibit_v1_check_active(inhibitor->manager);
+}
+
+struct sway_idle_inhibitor_v1 *sway_idle_inhibit_v1_user_inhibitor_for_view(
+                struct sway_view *view) {
+        struct sway_idle_inhibitor_v1 *inhibitor;
+        wl_list_for_each(inhibitor, &server.idle_inhibit_manager_v1->inhibitors,
+                        link) {
+                if (inhibitor->view == view &&
+                                inhibitor->mode != INHIBIT_IDLE_APPLICATION) {
+                        return inhibitor;
+                }
+        }
+        return NULL;
 }
 
-void idle_inhibit_v1_check_active(
+void sway_idle_inhibit_v1_user_inhibitor_destroy(
+                struct sway_idle_inhibitor_v1 *inhibitor) {
+        if (!inhibitor) {
+                return;
+        }
+        if (!sway_assert(inhibitor->mode != INHIBIT_IDLE_APPLICATION,
+                                "User should not be able to destroy application inhibitor")) {
+                return;
+        }
+        destroy_inhibitor(inhibitor);
+}
+
+static bool check_active(struct sway_idle_inhibitor_v1 *inhibitor) {
+        switch (inhibitor->mode) {
+        case INHIBIT_IDLE_APPLICATION:
+                // If there is no view associated with the inhibitor, assume visible
+                return !inhibitor->view || view_is_visible(inhibitor->view);
+        case INHIBIT_IDLE_FOCUS:;
+                struct sway_seat *seat = NULL;
+                wl_list_for_each(seat, &server.input->seats, link) {
+                        struct sway_container *con = seat_get_focused_container(seat);
+                        if (con && con->view && con->view == inhibitor->view) {
+                                return true;
+                        }
+                }
+                return false;
+        case INHIBIT_IDLE_FULLSCREEN:
+                return inhibitor->view->container &&
+                        container_is_fullscreen_or_child(inhibitor->view->container) &&
+                        view_is_visible(inhibitor->view);
+        case INHIBIT_IDLE_OPEN:
+                // Inhibitor is destroyed on unmap so it must be open/mapped
+                return true;
+        case INHIBIT_IDLE_VISIBLE:
+                return view_is_visible(inhibitor->view);
+        }
+        return false;
+}
+
+void sway_idle_inhibit_v1_check_active(
                 struct sway_idle_inhibit_manager_v1 *manager) {
         struct sway_idle_inhibitor_v1 *inhibitor;
         bool inhibited = false;
         wl_list_for_each(inhibitor, &manager->inhibitors, link) {
-                if (!inhibitor->view || !inhibitor->view->container) {
-                        /* Cannot guess if view is visible so assume it is */
-                        inhibited = true;
-                        break;
-                }
-                if (view_is_visible(inhibitor->view)) {
-                        inhibited = true;
+                if ((inhibited = check_active(inhibitor))) {
                         break;
                 }
         }