1 files changed, 173 insertions, 0 deletions

diff --git a/sway/scratchpad.c b/sway/scratchpad.c
new file mode 100644
index 00000000..e1f931a4
--- /dev/null
+++ b/sway/scratchpad.c
@@ -0,0 +1,173 @@
+#define _XOPEN_SOURCE 700
+#include <stdlib.h>
+#include <stdio.h>
+#include <stdbool.h>
+#include "sway/scratchpad.h"
+#include "sway/input/seat.h"
+#include "sway/tree/arrange.h"
+#include "sway/tree/container.h"
+#include "sway/tree/view.h"
+#include "sway/tree/workspace.h"
+#include "list.h"
+#include "log.h"
+
+void scratchpad_add_container(struct sway_container *con) {
+        if (!sway_assert(!con->scratchpad, "Container is already in scratchpad")) {
+                return;
+        }
+        con->scratchpad = true;
+        list_add(server.scratchpad, con);
+
+        struct sway_container *parent = con->parent;
+        container_set_floating(con, true);
+        container_remove_child(con);
+        arrange_windows(parent);
+
+        struct sway_seat *seat = input_manager_current_seat(input_manager);
+        seat_set_focus(seat, seat_get_focus_inactive(seat, parent));
+}
+
+void scratchpad_remove_container(struct sway_container *con) {
+        if (!sway_assert(con->scratchpad, "Container is not in scratchpad")) {
+                return;
+        }
+        con->scratchpad = false;
+        for (int i = 0; i < server.scratchpad->length; ++i) {
+                if (server.scratchpad->items[i] == con) {
+                        list_del(server.scratchpad, i);
+                        break;
+                }
+        }
+}
+
+/**
+ * Show a single scratchpad container.
+ * The container might be visible on another workspace already.
+ */
+static void scratchpad_show(struct sway_container *con) {
+        struct sway_seat *seat = input_manager_current_seat(input_manager);
+        struct sway_container *ws = seat_get_focus(seat);
+        if (ws->type != C_WORKSPACE) {
+                ws = container_parent(ws, C_WORKSPACE);
+        }
+
+    // If the current con or any of its parents are in fullscreen mode, we
+    // first need to disable it before showing the scratchpad con.
+        if (ws->sway_workspace->fullscreen) {
+                view_set_fullscreen(ws->sway_workspace->fullscreen, false);
+        }
+
+        // Show the container
+        if (con->parent) {
+                container_remove_child(con);
+        }
+        container_add_child(ws->sway_workspace->floating, con);
+
+        // Make sure the container's center point overlaps this workspace
+        double center_lx = con->x + con->width / 2;
+        double center_ly = con->y + con->height / 2;
+
+        struct wlr_box workspace_box;
+        container_get_box(ws, &workspace_box);
+        if (!wlr_box_contains_point(&workspace_box, center_lx, center_ly)) {
+                // Maybe resize it
+                if (con->width > ws->width || con->height > ws->height) {
+                        // TODO: Do this properly once we can float C_CONTAINERs
+                        if (con->type == C_VIEW) {
+                                view_init_floating(con->sway_view);
+                                arrange_windows(con);
+                        }
+                }
+
+                // Center it
+                double new_lx = ws->x + (ws->width - con->width) / 2;
+                double new_ly = ws->y + (ws->height - con->height) / 2;
+                container_floating_move_to(con, new_lx, new_ly);
+        }
+
+        seat_set_focus(seat, con);
+
+        container_set_dirty(con->parent);
+}
+
+/**
+ * Hide a single scratchpad container.
+ * The container might not be the focused container (eg. when using criteria).
+ */
+static void scratchpad_hide(struct sway_container *con) {
+        struct sway_seat *seat = input_manager_current_seat(input_manager);
+        struct sway_container *focus = seat_get_focus(seat);
+        struct sway_container *ws = container_parent(con, C_WORKSPACE);
+
+        container_remove_child(con);
+        arrange_windows(ws);
+        if (con == focus) {
+                seat_set_focus(seat, seat_get_focus_inactive(seat, ws));
+        }
+        list_move_to_end(server.scratchpad, con);
+}
+
+void scratchpad_toggle_auto(void) {
+        struct sway_seat *seat = input_manager_current_seat(input_manager);
+        struct sway_container *focus = seat_get_focus(seat);
+        struct sway_container *ws = focus->type == C_WORKSPACE ?
+                focus : container_parent(focus, C_WORKSPACE);
+
+    // Check if the currently focused window is a scratchpad window and should
+    // be hidden again.
+        if (focus->scratchpad) {
+                wlr_log(WLR_DEBUG, "Focus is a scratchpad window - hiding %s",
+                                focus->name);
+                scratchpad_hide(focus);
+                return;
+        }
+
+    // Check if there is an unfocused scratchpad window on the current workspace
+    // and focus it.
+        for (int i = 0; i < ws->sway_workspace->floating->children->length; ++i) {
+                struct sway_container *floater =
+                        ws->sway_workspace->floating->children->items[i];
+                if (floater->scratchpad && focus != floater) {
+                        wlr_log(WLR_DEBUG,
+                                        "Focusing other scratchpad window (%s) in this workspace",
+                                        floater->name);
+                        scratchpad_show(floater);
+                        return;
+                }
+        }
+
+    // Check if there is a visible scratchpad window on another workspace.
+    // In this case we move it to the current workspace.
+        for (int i = 0; i < server.scratchpad->length; ++i) {
+                struct sway_container *con = server.scratchpad->items[i];
+                if (con->parent) {
+                        wlr_log(WLR_DEBUG,
+                                        "Moving a visible scratchpad window (%s) to this workspace",
+                                        con->name);
+                        scratchpad_show(con);
+                        return;
+                }
+        }
+
+        // Take the container at the bottom of the scratchpad list
+        if (!sway_assert(server.scratchpad->length, "Scratchpad is empty")) {
+                return;
+        }
+        struct sway_container *con = server.scratchpad->items[0];
+        wlr_log(WLR_DEBUG, "Showing %s from list", con->name);
+        scratchpad_show(con);
+}
+
+void scratchpad_toggle_container(struct sway_container *con) {
+        if (!sway_assert(con->scratchpad, "Container isn't in the scratchpad")) {
+                return;
+        }
+
+    // Check if it matches a currently visible scratchpad window and hide it.
+        if (con->parent) {
+                scratchpad_hide(con);
+                return;
+        }
+
+        scratchpad_show(con);
+}