show hide with pkill -SIGUSR1 i3bar-river

1 files changed, 11 insertions, 0 deletions

diff --git a/src/main.rs b/src/main.rs
index 241cd19..d3457b5 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -20,6 +20,11 @@ mod text;
 mod utils;
 mod wm_info_provider;
 
+use signal_hook::consts::*;
+use signal_hook_tokio::Signals;
+
+use futures::stream::StreamExt;
+
 use wayrs_client::connection::Connection;
 
 use state::State;
@@ -28,6 +33,8 @@ use state::State;
 async fn main() -> anyhow::Result<()> {
     env_logger::init();
 
+    let mut signals = Signals::new([SIGUSR1])?;
+
     let mut conn = Connection::connect()?;
     let globals = conn.async_collect_initial_globals().await?;
     let mut state = State::new(&mut conn, &globals);
@@ -49,6 +56,10 @@ async fn main() -> anyhow::Result<()> {
                 }
                 conn.async_flush().await?;
             }
+            Some(signal) = signals.next() => match signal {
+                SIGUSR1 => state.toggle_visibility(&mut conn),
+                _ => unreachable!(),
+            }
         }
     }
 }