Added in better exit handling

1 files changed, 11 insertions, 3 deletions

diff --git a/sway/commands.c b/sway/commands.c
index 02fdcf6c..5cb661eb 100644
--- a/sway/commands.c
+++ b/sway/commands.c
@@ -12,6 +12,7 @@
 #include "log.h"
 #include "workspace.h"
 #include "commands.h"
+#include "container.h"
 
 struct modifier_key {
         char *name;
@@ -152,11 +153,18 @@ static bool cmd_exec(struct sway_config *config, int argc, char **argv) {
         return cmd_exec_always(config, argc, argv);
 }
 
+static void kill_views(swayc_t *container, void *data) {
+        if (container->type == C_VIEW) {
+                wlc_view_close(container->handle);
+        }
+}
+
 static bool cmd_exit(struct sway_config *config, int argc, char **argv) {
         if (!checkarg(argc, "exit", EXPECTED_EQUAL_TO, 0)) {
                 return false;
         }
-        // TODO: Some kind of clean up is probably in order
+        // Close all views
+        container_map(&root_container, kill_views, NULL);
         exit(0);
         return true;
 }
@@ -190,8 +198,8 @@ static bool cmd_focus_follows_mouse(struct sway_config *config, int argc, char *
 
 static bool cmd_kill(struct sway_config *config, int argc, char **argv) {
         swayc_t *view = get_focused_container(&root_container);
-    wlc_view_close(view->handle);
-    return true;
+        wlc_view_close(view->handle);
+        return true;
 }
 
 static bool cmd_layout(struct sway_config *config, int argc, char **argv) {