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);