7 files changed, 104 insertions, 5 deletions

diff --git a/include/sway/container.h b/include/sway/container.h
index 35d1c146..2a96165f 100644
--- a/include/sway/container.h
+++ b/include/sway/container.h
@@ -126,5 +126,6 @@ struct sway_container {
 };
 
 swayc_t *new_output(struct sway_output *sway_output);
+swayc_t *new_workspace(swayc_t *output, const char *name);
 
 #endif
diff --git a/include/sway/layout.h b/include/sway/layout.h
index 7e7a9c35..6356ad00 100644
--- a/include/sway/layout.h
+++ b/include/sway/layout.h
@@ -5,5 +5,7 @@ struct sway_container;
 
 void init_layout(void);
 void add_child(struct sway_container *parent, struct sway_container *child);
+enum swayc_layouts default_layout(struct sway_container *output);
+void sort_workspaces(struct sway_container *output);
 
 #endif
diff --git a/include/sway/workspace.h b/include/sway/workspace.h
new file mode 100644
index 00000000..04b2ea4e
--- /dev/null
+++ b/include/sway/workspace.h
@@ -0,0 +1,6 @@
+#ifndef _SWAY_WORKSPACE_H
+#define _SWAY_WORKSPACE_H
+
+char *workspace_next_name(const char *output_name);
+
+#endif
diff --git a/sway/CMakeLists.txt b/sway/CMakeLists.txt
index 9a92466c..6d520e76 100644
--- a/sway/CMakeLists.txt
+++ b/sway/CMakeLists.txt
@@ -14,8 +14,9 @@ add_executable(sway
     desktop/output.c
     desktop/xdg_shell_v6.c
 
-    tree/layout.c
     tree/container.c
+    tree/layout.c
+    tree/workspace.c
 
         base64.c
         main.c
diff --git a/sway/tree/container.c b/sway/tree/container.c
index 54bcf478..ac79356a 100644
--- a/sway/tree/container.c
+++ b/sway/tree/container.c
@@ -5,6 +5,8 @@
 #include "sway/container.h"
 #include "sway/layout.h"
 #include "sway/output.h"
+#include "sway/workspace.h"
+#include "log.h"
 
 static swayc_t *new_swayc(enum swayc_types type) {
         // next id starts at 1 because 0 is assigned to root_container in layout.c
@@ -27,8 +29,8 @@ static swayc_t *new_swayc(enum swayc_types type) {
 
 swayc_t *new_output(struct sway_output *sway_output) {
         struct wlr_box size;
-        wlr_output_effective_resolution(sway_output->wlr_output,
-                        &size.width, &size.height);
+        wlr_output_effective_resolution(
+                        sway_output->wlr_output, &size.width, &size.height);
         const char *name = sway_output->wlr_output->name;
 
         swayc_t *output = new_swayc(C_OUTPUT);
@@ -39,7 +41,31 @@ swayc_t *new_output(struct sway_output *sway_output) {
 
         add_child(&root_container, output);
 
-        // TODO: Create workspace
-
+        // Create workspace
+        char *ws_name = workspace_next_name(output->name);
+        sway_log(L_DEBUG, "Creating default workspace %s", ws_name);
+        new_workspace(output, ws_name);
+        free(ws_name);
         return output;
 }
+
+swayc_t *new_workspace(swayc_t *output, const char *name) {
+        if (!sway_assert(output, "new_workspace called with null output")) {
+                return NULL;
+        }
+        sway_log(L_DEBUG, "Added workspace %s for output %s", name, output->name);
+        swayc_t *workspace = new_swayc(C_WORKSPACE);
+
+        workspace->x = output->x;
+        workspace->y = output->y;
+        workspace->width = output->width;
+        workspace->height = output->height;
+        workspace->name = !name ? NULL : strdup(name);
+        workspace->prev_layout = L_NONE;
+        workspace->layout = default_layout(output);
+        workspace->workspace_layout = default_layout(output);
+
+        add_child(output, workspace);
+        sort_workspaces(output);
+        return workspace;
+}
diff --git a/sway/tree/layout.c b/sway/tree/layout.c
index 06200bbf..5a70c570 100644
--- a/sway/tree/layout.c
+++ b/sway/tree/layout.c
@@ -1,5 +1,7 @@
 #define _POSIX_C_SOURCE 200809L
+#include <ctype.h>
 #include <stdbool.h>
+#include <stdlib.h>
 #include <string.h>
 #include <wlr/types/wlr_output_layout.h>
 #include "sway/container.h"
@@ -33,3 +35,38 @@ void add_child(swayc_t *parent, swayc_t *child) {
         }
         */
 }
+
+enum swayc_layouts default_layout(swayc_t *output) {
+        /* TODO WLR
+        if (config->default_layout != L_NONE) {
+                //return config->default_layout;
+        } else if (config->default_orientation != L_NONE) {
+                return config->default_orientation;
+        } else */if (output->width >= output->height) {
+                return L_HORIZ;
+        } else {
+                return L_VERT;
+        }
+}
+
+static int sort_workspace_cmp_qsort(const void *_a, const void *_b) {
+        swayc_t *a = *(void **)_a;
+        swayc_t *b = *(void **)_b;
+        int retval = 0;
+
+        if (isdigit(a->name[0]) && isdigit(b->name[0])) {
+                int a_num = strtol(a->name, NULL, 10);
+                int b_num = strtol(b->name, NULL, 10);
+                retval = (a_num < b_num) ? -1 : (a_num > b_num);
+        } else if (isdigit(a->name[0])) {
+                retval = -1;
+        } else if (isdigit(b->name[0])) {
+                retval = 1;
+        }
+
+        return retval;
+}
+
+void sort_workspaces(swayc_t *output) {
+        list_stable_sort(output->children, sort_workspace_cmp_qsort);
+}
diff --git a/sway/tree/workspace.c b/sway/tree/workspace.c
new file mode 100644
index 00000000..e8ed4102
--- /dev/null
+++ b/sway/tree/workspace.c
@@ -0,0 +1,26 @@
+#define _XOPEN_SOURCE 500
+#include <stdbool.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include "sway/container.h"
+#include "log.h"
+
+void next_name_map(swayc_t *ws, void *data) {
+        int *count = data;
+        ++count;
+}
+
+char *workspace_next_name(const char *output_name) {
+        sway_log(L_DEBUG, "Workspace: Generating new workspace name for output %s",
+                        output_name);
+        int count = 0;
+        next_name_map(&root_container, &count);
+        ++count;
+        int len = snprintf(NULL, 0, "%d", count);
+        char *name = malloc(len + 1);
+        if (!sway_assert(name, "Failed to allocate workspace name")) {
+                return NULL;
+        }
+        snprintf(name, len + 1, "%d", count);
+        return name;
+}