From a5f2587cb16c0f03e2db77a499be490b6f1e8e1b Mon Sep 17 00:00:00 2001
From: Rutherther <rutherther@proton.me>
Date: Sat, 4 May 2024 20:59:51 +0200
Subject: [PATCH] fix: make sure selmon doesn't get set to disabled mon

---
 dwl.c | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/dwl.c b/dwl.c
index bf763df..752e66d 100644
--- a/dwl.c
+++ b/dwl.c
@@ -716,6 +716,10 @@ closemon(Monitor *m)
 		do /* don't switch to disabled mons */
 			selmon = wl_container_of(mons.next, selmon, link);
 		while (!selmon->wlr_output->enabled && i++ < nmons);
+
+		if (!selmon->wlr_output->enabled) {
+			selmon = NULL;
+		}
 	}
 
 	wl_list_for_each(c, &clients, link) {
@@ -866,6 +870,7 @@ createmon(struct wl_listener *listener, void *data)
 	 * monitor) becomes available. */
 	struct wlr_output *wlr_output = data;
 	const MonitorRule *r;
+	Client *c;
 	size_t i;
 	struct wlr_output_state state;
 	Monitor *m;
@@ -934,10 +939,25 @@ createmon(struct wl_listener *listener, void *data)
 	 * output (such as DPI, scale factor, manufacturer, etc).
 	 */
 	m->scene_output = wlr_scene_output_create(scene, wlr_output);
+
 	if (m->m.x < 0 || m->m.y < 0)
 		wlr_output_layout_add_auto(output_layout, wlr_output);
 	else
 		wlr_output_layout_add(output_layout, wlr_output, m->m.x, m->m.y);
+
+	/* Recovering from state with no monitors
+	 * Add all clients to the only monitor.
+	 */
+	if (selmon == NULL) {
+		selmon = m;
+
+		wl_list_for_each(c, &clients, link) {
+			if (c->isfloating && c->geom.x > m->m.width)
+				resize(c, (struct wlr_box){.x = c->geom.x - m->w.width, .y = c->geom.y,
+						.width = c->geom.width, .height = c->geom.height}, 0);
+			setmon(c, selmon, c->tags);
+		}
+	}
 }
 
 void
@@ -2773,6 +2793,10 @@ updatemons(struct wl_listener *listener, void *data)
 
 		config_head->state.x = m->m.x;
 		config_head->state.y = m->m.y;
+
+		if (!selmon) {
+			selmon = m;
+		}
 	}
 
 	if (selmon && selmon->wlr_output->enabled) {
-- 
2.49.0