From b84d12c92a4afc584e54533ee0f3c6165bb4a619 Mon Sep 17 00:00:00 2001
From: Tassilo Horn <tsdh@gnu.org>
Date: Sat, 20 Nov 2021 19:43:57 +0100
Subject: [PATCH] Handle shutdown event

---
 src/demon.rs | 36 +++++++++++++++++++++++++++---------
 1 file changed, 27 insertions(+), 9 deletions(-)

diff --git a/src/demon.rs b/src/demon.rs
index dc313e8..10fb7a0 100644
--- a/src/demon.rs
+++ b/src/demon.rs
@@ -42,31 +42,41 @@ pub fn run_demon() {
 }
 
 fn connect_and_subscribe() -> s::Fallible<s::EventStream> {
-    s::Connection::new()?
-        .subscribe(&[s::EventType::Window, s::EventType::Workspace])
+    s::Connection::new()?.subscribe(&[
+        s::EventType::Window,
+        s::EventType::Workspace,
+        s::EventType::Shutdown,
+    ])
 }
 
 pub fn monitor_sway_events(
     extra_props: Arc<RwLock<HashMap<i64, t::ExtraProps>>>,
 ) {
     let config = config::load_config();
+    let mut resets = 0;
+    let max_resets = 10;
 
     'reset: loop {
+        if resets >= max_resets {
+            break;
+        }
+        resets += 1;
+
         println!("Connecting to sway for subscribing to events...");
         match connect_and_subscribe() {
             Err(err) => {
                 eprintln!("Could not connect and subscribe: {}", err);
                 std::thread::sleep(std::time::Duration::from_secs(3));
-                break 'reset;
             }
             Ok(iter) => {
                 for ev_result in iter {
-                    let handled;
+                    let show_extra_props_state;
+                    resets = 0;
                     match ev_result {
                         Ok(ev) => match ev {
                             s::Event::Window(win_ev) => {
                                 let extra_props_clone = extra_props.clone();
-                                handled = handle_window_event(
+                                show_extra_props_state = handle_window_event(
                                     win_ev,
                                     extra_props_clone,
                                     &config,
@@ -74,23 +84,30 @@ pub fn monitor_sway_events(
                             }
                             s::Event::Workspace(ws_ev) => {
                                 let extra_props_clone = extra_props.clone();
-                                handled = handle_workspace_event(
+                                show_extra_props_state = handle_workspace_event(
                                     ws_ev,
                                     extra_props_clone,
                                 );
                             }
-                            _ => handled = false,
+                            s::Event::Shutdown(sd_ev) => {
+                                println!(
+                                    "Sway shuts down with reason '{:?}'.",
+                                    sd_ev.change
+                                );
+                                break 'reset;
+                            }
+                            _ => show_extra_props_state = false,
                         },
                         Err(e) => {
                             eprintln!("Error while receiving events: {}", e);
                             std::thread::sleep(std::time::Duration::from_secs(
                                 3,
                             ));
+                            show_extra_props_state = false;
                             eprintln!("Resetting!");
-                            break 'reset;
                         }
                     }
-                    if handled {
+                    if show_extra_props_state {
                         println!(
                             "New extra_props state:\n{:#?}",
                             *extra_props.read().unwrap()
@@ -100,6 +117,7 @@ pub fn monitor_sway_events(
             }
         }
     }
+    println!("Swayr demon shutting down.")
 }
 
 fn handle_window_event(