Store sway_outputs so that they can be reenabled

7 files changed, 85 insertions, 13 deletions

diff --git a/sway/commands/output.c b/sway/commands/output.c
index bc12310e..6c789cc9 100644
--- a/sway/commands/output.c
+++ b/sway/commands/output.c
@@ -1,5 +1,7 @@
 #include "sway/commands.h"
 #include "sway/config.h"
+#include "sway/output.h"
+#include "sway/tree/layout.h"
 #include "list.h"
 #include "log.h"
 
@@ -80,16 +82,25 @@ struct cmd_results *cmd_output(int argc, char **argv) {
 	// will be applied during normal "new output" event from wlroots.
 	char identifier[128];
 	bool all = strcmp(output->name, "*") == 0;
-	for (int i = 0; i < root_container.children->length; ++i) {
-		struct sway_container *cont = root_container.children->items[i];
-		if (cont->type != C_OUTPUT) {
-			continue;
-		}
+	list_t *sway_outputs = root_container.sway_root->outputs;
+	for (int i = 0; i < sway_outputs->length; ++i) {
+		struct sway_output *sway_output = sway_outputs->items[i];
+		output_get_identifier(identifier, sizeof(identifier), sway_output);
+		wlr_log(L_DEBUG, "Checking identifier %s", identifier);
+		if (all || strcmp(sway_output->wlr_output->name, output->name) == 0
+				|| strcmp(identifier, output->name) == 0) {
+			if (!sway_output->swayc) {
+				if (!output->enabled) {
+					if (!all) {
+						break;
+					}
+					continue;
+				}
+
+				output_enable(sway_output);
+			}
 
-		output_get_identifier(identifier, sizeof(identifier), cont->sway_output);
-		if (all || strcmp(cont->name, output->name) == 0 ||
-				strcmp(identifier, output->name) == 0) {
-			apply_output_config(output, cont);
+			apply_output_config(output, sway_output->swayc);
 
 			if (!all) {
 				// Stop looking if the output config isn't applicable to all
diff --git a/sway/config/output.c b/sway/config/output.c
index ee2440ea..648ded27 100644
--- a/sway/config/output.c
+++ b/sway/config/output.c
@@ -131,11 +131,13 @@ void apply_output_config(struct output_config *oc, struct sway_container *output
 	struct wlr_output *wlr_output = output->sway_output->wlr_output;
 
 	if (oc && oc->enabled == 0) {
+		struct sway_output *sway_output = output->sway_output;
 		if (output->sway_output->bg_pid != 0) {
 			terminate_swaybg(output->sway_output->bg_pid);
 			output->sway_output->bg_pid = 0;
 		}
 		container_destroy(output);
+		sway_output->swayc = NULL;
 		wlr_output_layout_remove(root_container.sway_root->output_layout,
 			wlr_output);
 		return;
diff --git a/sway/desktop/output.c b/sway/desktop/output.c
index acc9caae..5d02f5eb 100644
--- a/sway/desktop/output.c
+++ b/sway/desktop/output.c
@@ -1157,6 +1157,10 @@ void output_damage_whole_container(struct sway_output *output,
 	wlr_output_damage_add_box(output->damage, &box);
 }
 
+static int find_output(const void *output1, const void *output2) {
+	return output1 == output2 ? 0 : 1;
+}
+
 static void damage_handle_destroy(struct wl_listener *listener, void *data) {
 	struct sway_output *output =
 		wl_container_of(listener, output, damage_destroy);
@@ -1165,7 +1169,19 @@ static void damage_handle_destroy(struct wl_listener *listener, void *data) {
 
 static void handle_destroy(struct wl_listener *listener, void *data) {
 	struct sway_output *output = wl_container_of(listener, output, destroy);
-	container_destroy(output->swayc);
+	if (output->swayc) {
+		container_destroy(output->swayc);
+	}
+	int index = list_seq_find(root_container.sway_root->outputs, find_output,
+			output);
+	if (index >= 0 && index < root_container.sway_root->outputs->length) {
+		wlr_log(L_DEBUG, "Removing %s from outputs list",
+				output->wlr_output->name);
+		list_del(root_container.sway_root->outputs, index);
+		wl_list_remove(&output->destroy.link);
+		output->wlr_output = NULL;
+		free(output);
+	}
 }
 
 static void handle_mode(struct wl_listener *listener, void *data) {
@@ -1203,6 +1219,7 @@ void handle_new_output(struct wl_listener *listener, void *data) {
 	output->wlr_output = wlr_output;
 	wlr_output->data = output;
 	output->server = server;
+	list_add(root_container.sway_root->outputs, output);
 
 	if (!wl_list_empty(&wlr_output->modes)) {
 		struct wlr_output_mode *mode =
@@ -1210,11 +1227,23 @@ void handle_new_output(struct wl_listener *listener, void *data) {
 		wlr_output_set_mode(wlr_output, mode);
 	}
 
-	output->damage = wlr_output_damage_create(wlr_output);
+	output_enable(output);
+}
+
+void output_enable(struct sway_output *output) {
+	struct wlr_output *wlr_output = output->wlr_output;
+
+	if (!wlr_output->data) {
+		wlr_output->data = output;
+	}
+
+	if (!output->damage) {
+		output->damage = wlr_output_damage_create(wlr_output);
+	}
 
 	output->swayc = output_create(output);
 	if (!output->swayc) {
-		free(output);
+		// Output is disabled
 		return;
 	}
 
diff --git a/sway/ipc-json.c b/sway/ipc-json.c
index 6d185449..5d402d1b 100644
--- a/sway/ipc-json.c
+++ b/sway/ipc-json.c
@@ -139,6 +139,26 @@ static void ipc_json_describe_output(struct sway_container *container, json_obje
 	json_object_object_add(object, "layout", json_object_new_string("output"));
 }
 
+json_object *ipc_json_describe_disabled_output(struct sway_output *output) {
+	struct wlr_output *wlr_output = output->wlr_output;
+	
+	json_object *object = json_object_new_object();
+
+	json_object_object_add(object, "type", json_object_new_string("output"));
+	json_object_object_add(object, "name",
+			wlr_output->name ? json_object_new_string(wlr_output->name) : NULL);
+	json_object_object_add(object, "active", json_object_new_boolean(false));
+	json_object_object_add(object, "make",
+			json_object_new_string(wlr_output->make));
+	json_object_object_add(object, "model",
+			json_object_new_string(wlr_output->model));
+	json_object_object_add(object, "serial",
+			json_object_new_string(wlr_output->serial));
+	json_object_object_add(object, "modes", json_object_new_array());
+
+	return object;
+}
+
 static void ipc_json_describe_workspace(struct sway_container *workspace,
 		json_object *object) {
 	int num = isdigit(workspace->name[0]) ? atoi(workspace->name) : -1;
diff --git a/sway/ipc-server.c b/sway/ipc-server.c
index 8d9ab06a..2a2346b4 100644
--- a/sway/ipc-server.c
+++ b/sway/ipc-server.c
@@ -19,6 +19,7 @@
 #include "sway/commands.h"
 #include "sway/ipc-json.h"
 #include "sway/ipc-server.h"
+#include "sway/output.h"
 #include "sway/server.h"
 #include "sway/input/input-manager.h"
 #include "sway/input/seat.h"
@@ -488,6 +489,14 @@ void ipc_client_handle_command(struct ipc_client *client) {
 					ipc_json_describe_container(container));
 			}
 		}
+		for (int i = 0; i < root_container.sway_root->outputs->length; ++i) {
+			struct sway_output *output =
+				root_container.sway_root->outputs->items[i];
+			if (!output->swayc) {
+				json_object_array_add(outputs,
+						ipc_json_describe_disabled_output(output));
+			}
+		}
 		const char *json_string = json_object_to_json_string(outputs);
 		ipc_send_reply(client, json_string, (uint32_t) strlen(json_string));
 		json_object_put(outputs); // free
diff --git a/sway/tree/container.c b/sway/tree/container.c
index d0d26631..af965857 100644
--- a/sway/tree/container.c
+++ b/sway/tree/container.c
@@ -255,7 +255,6 @@ static struct sway_container *container_output_destroy(
 		}
 	}
 
-	wl_list_remove(&output->sway_output->destroy.link);
 	wl_list_remove(&output->sway_output->mode.link);
 	wl_list_remove(&output->sway_output->transform.link);
 	wl_list_remove(&output->sway_output->scale.link);
@@ -265,6 +264,7 @@ static struct sway_container *container_output_destroy(
 
 	// clear the wlr_output reference to this container
 	output->sway_output->wlr_output->data = NULL;
+	output->sway_output->swayc = NULL;
 
 	wlr_log(L_DEBUG, "OUTPUT: Destroying output '%s'", output->name);
 	_container_destroy(output);
diff --git a/sway/tree/layout.c b/sway/tree/layout.c
index 82502e1b..ce53a515 100644
--- a/sway/tree/layout.c
+++ b/sway/tree/layout.c
@@ -35,6 +35,7 @@ void layout_init(void) {
 
 	root_container.sway_root = calloc(1, sizeof(*root_container.sway_root));
 	root_container.sway_root->output_layout = wlr_output_layout_create();
+	root_container.sway_root->outputs = create_list();
 	wl_list_init(&root_container.sway_root->xwayland_unmanaged);
 	wl_signal_init(&root_container.sway_root->events.new_container);