Add support for niri workspaces (#39)

1 files changed, 7 insertions, 0 deletions

diff --git a/src/wm_info_provider.rs b/src/wm_info_provider.rs
index a8ee3be..b3c6618 100644
--- a/src/wm_info_provider.rs
+++ b/src/wm_info_provider.rs
@@ -19,6 +19,9 @@ pub use river::*;
 mod hyprland;
 pub use hyprland::*;
 
+mod niri;
+pub use niri::*;
+
 pub trait WmInfoProvider {
     fn register(&self, _: &mut EventLoop) {}
 
@@ -62,6 +65,10 @@ pub fn bind(
         return Box::new(hyprland);
     }
 
+    if let Some(niri) = NiriInfoProvider::new() {
+        return Box::new(niri);
+    }
+
     Box::new(DummyInfoProvider)
 }