abstract over tag and layout name provider

1 files changed, 58 insertions, 0 deletions

diff --git a/src/wm_info_provider.rs b/src/wm_info_provider.rs
new file mode 100644
index 0000000..b7854b1
--- /dev/null
+++ b/src/wm_info_provider.rs
@@ -0,0 +1,58 @@
+mod river;
+
+use std::any::Any;
+
+use wayrs_client::connection::Connection;
+use wayrs_client::global::*;
+
+use crate::protocol::*;
+use crate::state::State;
+
+pub type WmInfoCallback = fn(&mut Connection<State>, &mut State, WlOutput, WmInfo);
+
+pub trait WmInfoProvider {
+    fn as_any(&mut self) -> &mut dyn Any;
+
+    fn new_outut(&mut self, conn: &mut Connection<State>, output: WlOutput);
+
+    fn output_removed(&mut self, conn: &mut Connection<State>, output: WlOutput);
+
+    fn left_click_on_tag(
+        &mut self,
+        conn: &mut Connection<State>,
+        output: WlOutput,
+        seat: WlSeat,
+        tag: &str,
+    );
+
+    fn middle_click_on_tag(
+        &mut self,
+        conn: &mut Connection<State>,
+        output: WlOutput,
+        seat: WlSeat,
+        tag: &str,
+    );
+}
+
+#[derive(Default)]
+pub struct WmInfo {
+    pub layout_name: Option<String>,
+    pub tags: Vec<Tag>,
+}
+
+pub struct Tag {
+    pub name: String,
+    pub is_focused: bool,
+    pub is_active: bool,
+    pub is_urgent: bool,
+}
+
+pub fn bind_wayland(
+    conn: &mut Connection<State>,
+    globals: &Globals,
+    callback: WmInfoCallback,
+) -> Option<Box<dyn WmInfoProvider>> {
+    // TODO: add more providers
+    let river = river::RiverInfoProvider::bind(conn, globals, callback)?;
+    Some(Box::new(river))
+}