wayland: add jay_workspace_watcher

src/compositor.rs

@@ -196,6 +196,7 @@ fn start_compositor2(
         hardware_tick_cursor: Default::default(),
         testers: Default::default(),
         render_ctx_watchers: Default::default(),
+        workspace_watchers: Default::default(),
     });
     state.tracker.register(ClientId::from_raw(0));
     create_dummy_output(&state);

src/ifs.rs

@@ -10,6 +10,7 @@ pub mod jay_render_ctx;
 pub mod jay_screenshot;
 pub mod jay_seat_events;
 pub mod jay_workspace;
+pub mod jay_workspace_watcher;
 pub mod org_kde_kwin_server_decoration;
 pub mod org_kde_kwin_server_decoration_manager;
 pub mod wl_buffer;

src/ifs/jay_compositor.rs

@@ -6,7 +6,7 @@ use {
         ifs::{
             jay_idle::JayIdle, jay_log_file::JayLogFile, jay_output::JayOutput,
             jay_pointer::JayPointer, jay_render_ctx::JayRenderCtx, jay_screenshot::JayScreenshot,
-            jay_seat_events::JaySeatEvents,
+            jay_seat_events::JaySeatEvents, jay_workspace_watcher::JayWorkspaceWatcher,
         },
         leaks::Tracker,
         object::Object,
@@ -279,6 +279,25 @@ impl JayCompositor {
         ctx.send_render_ctx(rctx.as_ref());
         Ok(())
     }
+
+    fn watch_workspaces(&self, parser: MsgParser<'_, '_>) -> Result<(), JayCompositorError> {
+        let req: WatchWorkspaces = self.client.parse(self, parser)?;
+        let watcher = Rc::new(JayWorkspaceWatcher {
+            id: req.id,
+            client: self.client.clone(),
+            tracker: Default::default(),
+        });
+        track!(self.client, watcher);
+        self.client.add_client_obj(&watcher)?;
+        self.client
+            .state
+            .workspace_watchers
+            .set((self.client.id, req.id), watcher.clone());
+        for ws in self.client.state.workspaces.lock().values() {
+            watcher.send_workspace(ws)?;
+        }
+        Ok(())
+    }
 }
 
 object_base! {
@@ -298,11 +317,12 @@ object_base! {
     GET_OUTPUT => get_output,
     GET_POINTER => get_pointer,
     GET_RENDER_CTX => get_render_ctx,
+    WATCH_WORKSPACES => watch_workspaces,
 }
 
 impl Object for JayCompositor {
     fn num_requests(&self) -> u32 {
-        GET_RENDER_CTX + 1
+        WATCH_WORKSPACES + 1
     }
 }
 

src/ifs/jay_workspace.rs

@@ -21,7 +21,6 @@ pub struct JayWorkspace {
     pub tracker: Tracker<Self>,
 }
 
-#[allow(dead_code)]
 impl JayWorkspace {
     pub fn send_linear_id(&self, ws: &WorkspaceNode) {
         self.client.event(LinearId {

src/ifs/jay_workspace_watcher.rs

@@ -0,0 +1,91 @@
+use {
+    crate::{
+        client::{Client, ClientError},
+        ifs::jay_workspace::JayWorkspace,
+        leaks::Tracker,
+        object::Object,
+        tree::WorkspaceNode,
+        utils::{
+            buffd::{MsgParser, MsgParserError},
+            clonecell::CloneCell,
+        },
+        wire::{jay_workspace_watcher::*, JayWorkspaceWatcherId},
+    },
+    std::rc::Rc,
+    thiserror::Error,
+};
+
+pub struct JayWorkspaceWatcher {
+    pub id: JayWorkspaceWatcherId,
+    pub client: Rc<Client>,
+    pub tracker: Tracker<Self>,
+}
+
+impl JayWorkspaceWatcher {
+    pub fn send_workspace(&self, workspace: &Rc<WorkspaceNode>) -> Result<(), ClientError> {
+        let jw = Rc::new(JayWorkspace {
+            id: self.client.new_id()?,
+            client: self.client.clone(),
+            workspace: CloneCell::new(Some(workspace.clone())),
+            tracker: Default::default(),
+        });
+        track!(self.client, jw);
+        self.client.add_server_obj(&jw);
+        workspace
+            .jay_workspaces
+            .set((self.client.id, jw.id), jw.clone());
+        self.client.event(New {
+            self_id: self.id,
+            id: jw.id,
+            linear_id: workspace.id.raw(),
+        });
+        jw.send_linear_id(workspace);
+        jw.send_name(workspace);
+        jw.send_output(&workspace.output.get());
+        jw.send_visible(workspace.visible.get());
+        jw.send_done();
+        Ok(())
+    }
+
+    fn destroy(&self, parser: MsgParser<'_, '_>) -> Result<(), JayWorkspaceWatcherError> {
+        let _req: Destroy = self.client.parse(self, parser)?;
+        self.remove_from_state();
+        self.client.remove_obj(self)?;
+        Ok(())
+    }
+
+    fn remove_from_state(&self) {
+        self.client
+            .state
+            .workspace_watchers
+            .remove(&(self.client.id, self.id));
+    }
+}
+
+object_base! {
+    JayWorkspaceWatcher;
+
+    DESTROY => destroy,
+}
+
+impl Object for JayWorkspaceWatcher {
+    fn num_requests(&self) -> u32 {
+        DESTROY + 1
+    }
+
+    fn break_loops(&self) {
+        self.remove_from_state();
+    }
+}
+
+simple_add_obj!(JayWorkspaceWatcher);
+
+#[derive(Debug, Error)]
+pub enum JayWorkspaceWatcherError {
+    #[error("Parsing failed")]
+    MsgParserError(Box<MsgParserError>),
+    #[error(transparent)]
+    ClientError(Box<ClientError>),
+}
+efrom!(JayWorkspaceWatcherError, MsgParserError);
+efrom!(JayWorkspaceWatcherError, ClientError);

src/state.rs

@@ -19,6 +19,7 @@ use {
             ext_session_lock_v1::ExtSessionLockV1,
             jay_render_ctx::JayRenderCtx,
             jay_seat_events::JaySeatEvents,
+            jay_workspace_watcher::JayWorkspaceWatcher,
             wl_drm::WlDrmGlobal,
             wl_seat::{SeatIds, WlSeatGlobal},
             wl_surface::{
@@ -43,7 +44,7 @@ use {
             queue::AsyncQueue, refcounted::RefCounted, run_toplevel::RunToplevel,
         },
         wheel::Wheel,
-        wire::{JayRenderCtxId, JaySeatEventsId},
+        wire::{JayRenderCtxId, JaySeatEventsId, JayWorkspaceWatcherId},
         xkbcommon::{XkbContext, XkbKeymap},
         xwayland::{self, XWaylandEvent},
     },
@@ -119,6 +120,7 @@ pub struct State {
     pub hardware_tick_cursor: AsyncQueue<Option<Rc<dyn Cursor>>>,
     pub testers: RefCell<AHashMap<(ClientId, JaySeatEventsId), Rc<JaySeatEvents>>>,
     pub render_ctx_watchers: CopyHashMap<(ClientId, JayRenderCtxId), Rc<JayRenderCtx>>,
+    pub workspace_watchers: CopyHashMap<(ClientId, JayWorkspaceWatcherId), Rc<JayWorkspaceWatcher>>,
 }
 
 // impl Drop for State {
@@ -567,6 +569,7 @@ impl State {
         self.pending_float_layout.clear();
         self.pending_float_titles.clear();
         self.render_ctx_watchers.clear();
+        self.workspace_watchers.clear();
         self.slow_clients.clear();
         for (_, h) in self.input_device_handlers.borrow_mut().drain() {
             h.async_event.clear();

src/tree/output.rs

@@ -30,6 +30,7 @@ use {
         },
         wire::JayOutputId,
     },
+    ahash::AHashMap,
     smallvec::SmallVec,
     std::{
         cell::{Cell, RefCell},
@@ -301,6 +302,15 @@ impl OutputNode {
         if self.workspace.get().is_none() {
             self.show_workspace(&ws);
         }
+        let mut clients_to_kill = AHashMap::new();
+        for watcher in self.state.workspace_watchers.lock().values() {
+            if let Err(e) = watcher.send_workspace(&ws) {
+                clients_to_kill.insert(watcher.client.id, (watcher.client.clone(), e));
+            }
+        }
+        for (client, e) in clients_to_kill.values() {
+            client.error(e);
+        }
         self.update_render_data();
         ws
     }