Implement inhibit_idle command

This implements the following command to set/unset a user idle
inhibitor for a view:
`inhibit_idle focus|fullscreen|open|none|visible`

The modes are as follows:
- focus: inhibited when the view is focused by any seat
- fullscreen: inhibited when the view is fullscreen (or a descendant of
  a fullscreen container) and is visible on any output
- open: inhibited until the view is closed or the inhibitor is unset or
  changed
- none: unsets any user set idle inhibitors for the view
- visible: inhibited when the view is visible on any output

This should have no effect on idle inhibitors set by the applications
themselves and those should still work as intended.

Since this operates on the view in the handler context, it is possible
to set it on the currently focused view, on any existing view with
criteria, or for any future view with for_window.

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;
                 }
         }