Merge branch 'handleUnplug' into output-management

1 files changed, 12 insertions, 3 deletions

diff --git a/dwl.c b/dwl.c
index 8fc52d5..dda407a 100644
--- a/dwl.c
+++ b/dwl.c
@@ -205,6 +205,7 @@ static void chvt(const Arg *arg);
 static void cleanup(void);
 static void cleanupkeyboard(struct wl_listener *listener, void *data);
 static void cleanupmon(struct wl_listener *listener, void *data);
+static void closemon(Monitor *m);
 static void commitlayersurfacenotify(struct wl_listener *listener, void *data);
 static void commitnotify(struct wl_listener *listener, void *data);
 static void createkeyboard(struct wlr_input_device *device);
@@ -694,8 +695,7 @@ void
 cleanupmon(struct wl_listener *listener, void *data)
 {
 	struct wlr_output *wlr_output = data;
-	Monitor *m = wlr_output->data, *newmon;
-	Client *c;
+	Monitor *m = wlr_output->data;
 
 	wl_list_remove(&m->destroy.link);
 	wl_list_remove(&m->frame.link);
@@ -703,6 +703,16 @@ cleanupmon(struct wl_listener *listener, void *data)
 	wlr_output_layout_remove(output_layout, m->wlr_output);
 
 	updatemons();
+	closemon(m);
+	free(m);
+}
+
+void
+closemon(Monitor *m)
+{
+	// move all the clients on a closed monitor to another one
+	Monitor *newmon;
+	Client *c;
 
 	wl_list_for_each(newmon, &mons, link) {
 		wl_list_for_each(c, &clients, link) {
@@ -715,7 +725,6 @@ cleanupmon(struct wl_listener *listener, void *data)
 		}
 		break;
 	}
-	free(m);
 }
 
 void