Merge pull request #2115 from RedSoxFan/restore-workspaces

Restore workspaces to output when re-enabled

1 files changed, 59 insertions, 0 deletions

diff --git a/sway/tree/workspace.c b/sway/tree/workspace.c
index 3c42e259..9ba210fd 100644
--- a/sway/tree/workspace.c
+++ b/sway/tree/workspace.c
@@ -68,7 +68,9 @@ struct sway_container *workspace_create(struct sway_container *output,
 	swayws->floating = container_create(C_CONTAINER);
 	swayws->floating->parent = swayws->swayc;
 	swayws->floating->layout = L_FLOATING;
+	swayws->output_priority = create_list();
 	workspace->sway_workspace = swayws;
+	workspace_output_add_priority(workspace, output);
 
 	container_add_child(output, workspace);
 	container_sort_workspaces(output);
@@ -454,3 +456,60 @@ bool workspace_is_empty(struct sway_container *ws) {
 	}
 	return true;
 }
+
+static int find_output(const void *id1, const void *id2) {
+	return strcmp(id1, id2) ? 0 : 1;
+}
+
+void workspace_output_raise_priority(struct sway_container *workspace,
+		struct sway_container *old_output, struct sway_container *output) {
+	struct sway_workspace *ws = workspace->sway_workspace;
+
+	int old_index = list_seq_find(ws->output_priority, find_output,
+			old_output->name);
+	if (old_index < 0) {
+		return;
+	}
+
+	int new_index = list_seq_find(ws->output_priority, find_output,
+			output->name);
+	if (new_index < 0) {
+		list_insert(ws->output_priority, old_index, strdup(output->name));
+	} else if (new_index > old_index) {
+		char *name = ws->output_priority->items[new_index];
+		list_del(ws->output_priority, new_index);
+		list_insert(ws->output_priority, old_index, name);
+	}
+}
+
+void workspace_output_add_priority(struct sway_container *workspace,
+		struct sway_container *output) {
+	int index = list_seq_find(workspace->sway_workspace->output_priority,
+			find_output, output->name);
+	if (index < 0) {
+		list_add(workspace->sway_workspace->output_priority,
+				strdup(output->name));
+	}
+}
+
+static bool _output_by_name(struct sway_container *output, void *data) {
+	return output->type == C_OUTPUT && strcasecmp(output->name, data) == 0;
+}
+
+struct sway_container *workspace_output_get_highest_available(
+		struct sway_container *ws, struct sway_container *exclude) {
+	for (int i = 0; i < ws->sway_workspace->output_priority->length; i++) {
+		char *name = ws->sway_workspace->output_priority->items[i];
+		if (exclude && strcasecmp(name, exclude->name) == 0) {
+			continue;
+		}
+
+		struct sway_container *output = container_find(&root_container,
+				_output_by_name, name);
+		if (output) {
+			return output;
+		}
+	}
+
+	return NULL;
+}