Make separate gaps functions per container type

In preparation for using type safety.

1 files changed, 35 insertions, 0 deletions

diff --git a/sway/tree/workspace.c b/sway/tree/workspace.c
index 93c4b3d3..d930826e 100644
--- a/sway/tree/workspace.c
+++ b/sway/tree/workspace.c
@@ -665,3 +665,38 @@ void workspace_add_floating(struct sway_container *workspace,
         container_set_dirty(workspace);
         container_set_dirty(con);
 }
+
+void workspace_remove_gaps(struct sway_container *ws) {
+        if (!sway_assert(ws->type == C_WORKSPACE, "Expected a workspace")) {
+                return;
+        }
+        if (ws->current_gaps == 0) {
+                return;
+        }
+
+        ws->width += ws->current_gaps * 2;
+        ws->height += ws->current_gaps * 2;
+        ws->x -= ws->current_gaps;
+        ws->y -= ws->current_gaps;
+        ws->current_gaps = 0;
+}
+
+void workspace_add_gaps(struct sway_container *ws) {
+        if (!sway_assert(ws->type == C_WORKSPACE, "Expected a workspace")) {
+                return;
+        }
+        if (ws->current_gaps > 0) {
+                return;
+        }
+        bool should_apply =
+                config->edge_gaps || (config->smart_gaps && ws->children->length > 1);
+        if (!should_apply) {
+                return;
+        }
+
+        ws->current_gaps = ws->has_gaps ? ws->gaps_inner : config->gaps_inner;
+        ws->x += ws->current_gaps;
+        ws->y += ws->current_gaps;
+        ws->width -= 2 * ws->current_gaps;
+        ws->height -= 2 * ws->current_gaps;
+}