Add treeland output manager support

- add treeland_output_manager_v1 protocol dependency
- track treeland output manager state and primary output
- expose treeland output manager data in JSON and CLI output

Author: wineee

Cargo.lock

@@ -13,5 +13,6 @@ documentation = "https://github.com/dwapp/wayland-info-rs/blob/main/USAGE.md"
 colored = "3.0.0"
 wayland-client = "0.31.7"
 wayland-protocols = { version = "0.32.10", features = ["staging", "unstable", "client"] }
+wayland-protocols-treeland = { version = "0.1.0", features = ["client"] }
 serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0"

Cargo.toml

@@ -17,6 +17,9 @@ use wayland_protocols::xdg::xdg_output::zv1::client::{
     zxdg_output_manager_v1::{self, ZxdgOutputManagerV1},
     zxdg_output_v1::{self, ZxdgOutputV1},
 };
+use wayland_protocols_treeland::output_manager::v1::client::treeland_output_manager_v1::{
+    self, TreelandOutputManagerV1,
+};
 
 // 全局信息结构
 #[derive(Debug, Clone, Serialize)]
@@ -113,6 +116,15 @@ struct PresentationInfo {
     clock_id: Option<u32>,
 }
 
+// Treeland Output Manager 信息结构
+#[derive(Debug, Clone, Serialize)]
+#[serde(rename_all = "camelCase")]
+struct TreelandOutputManagerInfo {
+    #[serde(skip_serializing)]
+    name: u32,
+    primary_output: Option<String>,
+}
+
 // XDG Output Manager 信息结构
 #[derive(Debug, Clone, Serialize)]
 #[serde(rename_all = "camelCase")]
@@ -157,11 +169,13 @@ struct AppData {
     shm_info: Vec<ShmInfo>,
     drm_lease_devices: Vec<DrmLeaseDeviceInfo>,
     presentation_info: Vec<PresentationInfo>,
+    treeland_output_managers: Vec<TreelandOutputManagerInfo>,
     xdg_output_managers: Vec<XdgOutputManagerInfo>,
     seat_objects: Vec<WlSeat>,
     output_objects: Vec<WlOutput>,
     shm_objects: Vec<WlShm>,
     presentation_objects: Vec<WpPresentation>,
+    treeland_output_manager_objects: Vec<TreelandOutputManagerV1>,
     xdg_output_manager_objects: Vec<ZxdgOutputManagerV1>,
     xdg_output_objects: Vec<ZxdgOutputV1>,
 }
@@ -176,6 +190,7 @@ struct JsonOutput {
     shm_info: Vec<ShmInfo>,
     drm_lease_devices: Vec<DrmLeaseDeviceInfo>,
     presentation_info: Vec<PresentationInfo>,
+    treeland_output_managers: Vec<TreelandOutputManagerInfo>,
     xdg_output_managers: Vec<XdgOutputManagerInfo>,
 }
 
@@ -202,11 +217,13 @@ impl AppData {
             shm_info: Vec::new(),
             drm_lease_devices: Vec::new(),
             presentation_info: Vec::new(),
+            treeland_output_managers: Vec::new(),
             xdg_output_managers: Vec::new(),
             seat_objects: Vec::new(),
             output_objects: Vec::new(),
             shm_objects: Vec::new(),
             presentation_objects: Vec::new(),
+            treeland_output_manager_objects: Vec::new(),
             xdg_output_manager_objects: Vec::new(),
             xdg_output_objects: Vec::new(),
         }
@@ -307,6 +324,19 @@ impl AppData {
         }
     }
 
+    fn add_treeland_output_manager(&mut self, name: u32) {
+        self.treeland_output_managers.push(TreelandOutputManagerInfo {
+            name,
+            primary_output: None,
+        });
+    }
+
+    fn update_treeland_primary_output(&mut self, manager_index: usize, output_name: String) {
+        if let Some(manager) = self.treeland_output_managers.get_mut(manager_index) {
+            manager.primary_output = Some(output_name);
+        }
+    }
+
     fn update_seat_capabilities(&mut self, seat_index: usize, capabilities: Vec<String>) {
         if let Some(seat) = self.seats.get_mut(seat_index) {
             seat.capabilities = capabilities;
@@ -491,6 +521,7 @@ impl AppData {
             shm_info: self.shm_info.clone(),
             drm_lease_devices: self.drm_lease_devices.clone(),
             presentation_info: self.presentation_info.clone(),
+            treeland_output_managers: self.treeland_output_managers.clone(),
             xdg_output_managers: self.xdg_output_managers.clone(),
         }
     }
@@ -657,6 +688,21 @@ impl AppData {
                 }
             }
 
+            // 打印 treeland_output_manager_v1 信息
+            if global.interface == "treeland_output_manager_v1" {
+                if let Some(manager) = self
+                    .treeland_output_managers
+                    .iter()
+                    .find(|m| m.name == global.name)
+                {
+                    if let Some(primary) = &manager.primary_output {
+                        println!("        Primary output: {}", primary.yellow());
+                    } else {
+                        println!("{}", "        Primary output: <unknown>".red());
+                    }
+                }
+            }
+
             // 打印 zxdg_output_manager_v1 信息
             if global.interface == "zxdg_output_manager_v1" {
                 if let Some(manager) = self
@@ -724,6 +770,9 @@ enum UserData {
     Presentation {
         presentation_index: usize,
     },
+    TreelandOutputManager {
+        manager_index: usize,
+    },
     XdgOutputManager {
         manager_index: usize,
     },
@@ -800,6 +849,16 @@ impl Dispatch<wl_registry::WlRegistry, ()> for AppData {
                     UserData::Presentation { presentation_index },
                 );
                 state.presentation_objects.push(presentation);
+            } else if interface == "treeland_output_manager_v1" {
+                state.add_treeland_output_manager(name);
+                let manager_index = state.treeland_output_managers.len() - 1;
+                let manager = registry.bind::<TreelandOutputManagerV1, _, _>(
+                    name,
+                    version,
+                    qh,
+                    UserData::TreelandOutputManager { manager_index },
+                );
+                state.treeland_output_manager_objects.push(manager);
             } else if interface == "zxdg_output_manager_v1" {
                 state.add_xdg_output_manager(name);
                 let manager_index = state.xdg_output_managers.len() - 1;
@@ -816,6 +875,24 @@ impl Dispatch<wl_registry::WlRegistry, ()> for AppData {
     }
 }
 
+// 实现 treeland_output_manager_v1 的事件处理
+impl Dispatch<TreelandOutputManagerV1, UserData> for AppData {
+    fn event(
+        state: &mut Self,
+        _manager: &TreelandOutputManagerV1,
+        event: treeland_output_manager_v1::Event,
+        data: &UserData,
+        _conn: &Connection,
+        _qh: &QueueHandle<AppData>,
+    ) {
+        if let UserData::TreelandOutputManager { manager_index } = data {
+            if let treeland_output_manager_v1::Event::PrimaryOutput { output_name } = event {
+                state.update_treeland_primary_output(*manager_index, output_name);
+            }
+        }
+    }
+}
+
 // 实现 wl_seat 的事件处理
 impl Dispatch<WlSeat, UserData> for AppData {
     fn event(