wayland: add jay_output

src/client/objects.rs

@@ -6,6 +6,7 @@ use {
                 wl_data_source::WlDataSource,
                 zwp_primary_selection_source_v1::ZwpPrimarySelectionSourceV1,
             },
+            jay_output::JayOutput,
             wl_buffer::WlBuffer,
             wl_display::WlDisplay,
             wl_output::WlOutput,
@@ -25,9 +26,9 @@ use {
             copyhashmap::{CopyHashMap, Locked},
         },
         wire::{
-            WlBufferId, WlDataSourceId, WlOutputId, WlPointerId, WlRegionId, WlRegistryId,
-            WlSeatId, WlSurfaceId, XdgPositionerId, XdgSurfaceId, XdgToplevelId, XdgWmBaseId,
-            ZwpPrimarySelectionSourceV1Id,
+            JayOutputId, WlBufferId, WlDataSourceId, WlOutputId, WlPointerId, WlRegionId,
+            WlRegistryId, WlSeatId, WlSurfaceId, XdgPositionerId, XdgSurfaceId, XdgToplevelId,
+            XdgWmBaseId, ZwpPrimarySelectionSourceV1Id,
         },
     },
     std::{cell::RefCell, mem, rc::Rc},
@@ -47,6 +48,7 @@ pub struct Objects {
     pub xdg_positioners: CopyHashMap<XdgPositionerId, Rc<XdgPositioner>>,
     pub regions: CopyHashMap<WlRegionId, Rc<WlRegion>>,
     pub buffers: CopyHashMap<WlBufferId, Rc<WlBuffer>>,
+    pub jay_outputs: CopyHashMap<JayOutputId, Rc<JayOutput>>,
     pub pointers: CopyHashMap<WlPointerId, Rc<WlPointer>>,
     pub xdg_wm_bases: CopyHashMap<XdgWmBaseId, Rc<XdgWmBase>>,
     pub seats: CopyHashMap<WlSeatId, Rc<WlSeat>>,
@@ -71,6 +73,7 @@ impl Objects {
             xdg_positioners: Default::default(),
             regions: Default::default(),
             buffers: Default::default(),
+            jay_outputs: Default::default(),
             pointers: Default::default(),
             xdg_wm_bases: Default::default(),
             seats: Default::default(),
@@ -99,6 +102,7 @@ impl Objects {
         self.xdg_positioners.clear();
         self.regions.clear();
         self.buffers.clear();
+        self.jay_outputs.clear();
         self.xdg_wm_bases.clear();
         self.seats.clear();
         self.pointers.clear();

src/compositor.rs

@@ -374,6 +374,7 @@ fn create_dummy_output(state: &Rc<State>) {
             0,
             0,
         )),
+        jay_outputs: Default::default(),
         workspaces: Default::default(),
         workspace: Default::default(),
         seat_state: Default::default(),

src/ifs.rs

@@ -4,6 +4,7 @@ pub mod ipc;
 pub mod jay_compositor;
 pub mod jay_idle;
 pub mod jay_log_file;
+pub mod jay_output;
 pub mod jay_screenshot;
 pub mod jay_seat_events;
 pub mod org_kde_kwin_server_decoration;

src/ifs/jay_compositor.rs

@@ -4,14 +4,15 @@ use {
         client::{Client, ClientError},
         globals::{Global, GlobalName},
         ifs::{
-            jay_idle::JayIdle, jay_log_file::JayLogFile, jay_screenshot::JayScreenshot,
-            jay_seat_events::JaySeatEvents,
+            jay_idle::JayIdle, jay_log_file::JayLogFile, jay_output::JayOutput,
+            jay_screenshot::JayScreenshot, jay_seat_events::JaySeatEvents,
         },
         leaks::Tracker,
         object::Object,
         screenshoter::take_screenshot,
         utils::{
             buffd::{MsgParser, MsgParserError},
+            clonecell::CloneCell,
             errorfmt::ErrorFmt,
         },
         wire::{jay_compositor::*, JayCompositorId},
@@ -225,6 +226,26 @@ impl JayCompositor {
             .insert((self.client.id, req.id), se);
         Ok(())
     }
+
+    fn get_output(&self, parser: MsgParser<'_, '_>) -> Result<(), JayCompositorError> {
+        let req: GetOutput = self.client.parse(self, parser)?;
+        let output = self.client.lookup(req.output)?;
+        let jo = Rc::new(JayOutput {
+            id: req.id,
+            client: self.client.clone(),
+            output: CloneCell::new(output.global.node.get()),
+            tracker: Default::default(),
+        });
+        track!(self.client, jo);
+        self.client.add_client_obj(&jo)?;
+        if let Some(node) = jo.output.get() {
+            node.jay_outputs.set((self.client.id, req.id), jo.clone());
+            jo.send_linear_id();
+        } else {
+            jo.send_destroyed();
+        }
+        Ok(())
+    }
 }
 
 object_base! {
@@ -241,11 +262,12 @@ object_base! {
     UNLOCK => unlock,
     GET_SEATS => get_seats,
     SEAT_EVENTS => seat_events,
+    GET_OUTPUT => get_output,
 }
 
 impl Object for JayCompositor {
     fn num_requests(&self) -> u32 {
-        SEAT_EVENTS + 1
+        GET_OUTPUT + 1
     }
 }
 

src/ifs/jay_output.rs

@@ -0,0 +1,78 @@
+use {
+    crate::{
+        client::{Client, ClientError},
+        leaks::Tracker,
+        object::Object,
+        tree::OutputNode,
+        utils::{
+            buffd::{MsgParser, MsgParserError},
+            clonecell::CloneCell,
+        },
+        wire::{jay_output::*, JayOutputId},
+    },
+    std::rc::Rc,
+    thiserror::Error,
+};
+
+pub struct JayOutput {
+    pub id: JayOutputId,
+    pub client: Rc<Client>,
+    pub output: CloneCell<Option<Rc<OutputNode>>>,
+    pub tracker: Tracker<Self>,
+}
+
+impl JayOutput {
+    pub fn send_destroyed(&self) {
+        self.client.event(Destroyed { self_id: self.id });
+    }
+
+    pub fn send_linear_id(&self) {
+        if let Some(output) = self.output.get() {
+            self.client.event(LinearId {
+                self_id: self.id,
+                linear_id: output.id.raw(),
+            });
+        }
+    }
+
+    fn destroy(&self, parser: MsgParser<'_, '_>) -> Result<(), JayOutputError> {
+        let _req: Destroy = self.client.parse(self, parser)?;
+        self.remove_from_node();
+        self.client.remove_obj(self)?;
+        Ok(())
+    }
+
+    fn remove_from_node(&self) {
+        if let Some(output) = self.output.get() {
+            output.jay_outputs.remove(&(self.client.id, self.id));
+        }
+    }
+}
+
+object_base! {
+    JayOutput;
+
+    DESTROY => destroy,
+}
+
+impl Object for JayOutput {
+    fn num_requests(&self) -> u32 {
+        1
+    }
+
+    fn break_loops(&self) {
+        self.remove_from_node();
+    }
+}
+
+dedicated_add_obj!(JayOutput, JayOutputId, jay_outputs);
+
+#[derive(Debug, Error)]
+pub enum JayOutputError {
+    #[error("Parsing failed")]
+    MsgParserError(Box<MsgParserError>),
+    #[error(transparent)]
+    ClientError(Box<ClientError>),
+}
+efrom!(JayOutputError, MsgParserError);
+efrom!(JayOutputError, ClientError);

src/tasks/connector.rs

@@ -124,6 +124,7 @@ impl ConnectorHandler {
             lock_surface: Default::default(),
             preferred_scale: Cell::new(Fixed::from_int(1)),
             hardware_cursor: Default::default(),
+            jay_outputs: Default::default(),
         });
         self.state.add_output_scale(on.preferred_scale.get());
         let mode = info.initial_mode;
@@ -209,6 +210,10 @@ impl ConnectorHandler {
             config.connector_disconnected(self.id);
         }
         global.node.set(None);
+        for (_, jo) in on.jay_outputs.lock().drain() {
+            jo.send_destroyed();
+            jo.output.take();
+        }
         global.destroyed.set(true);
         self.state.root.outputs.remove(&self.id);
         self.data.connected.set(false);

src/tree/output.rs

@@ -1,9 +1,11 @@
 use {
     crate::{
         backend::{HardwareCursor, KeyState, Mode},
+        client::ClientId,
         cursor::KnownCursor,
         fixed::Fixed,
         ifs::{
+            jay_output::JayOutput,
             wl_output::WlOutputGlobal,
             wl_seat::{
                 collect_kb_foci2, wl_pointer::PendingScroll, NodeSeatState, SeatId, WlSeatGlobal,
@@ -26,6 +28,7 @@ use {
             clonecell::CloneCell, copyhashmap::CopyHashMap, errorfmt::ErrorFmt,
             linkedlist::LinkedList, scroller::Scroller,
         },
+        wire::JayOutputId,
     },
     smallvec::SmallVec,
     std::{
@@ -40,6 +43,7 @@ tree_id!(OutputNodeId);
 pub struct OutputNode {
     pub id: OutputNodeId,
     pub global: Rc<WlOutputGlobal>,
+    pub jay_outputs: CopyHashMap<(ClientId, JayOutputId), Rc<JayOutput>>,
     pub workspaces: LinkedList<Rc<WorkspaceNode>>,
     pub workspace: CloneCell<Option<Rc<WorkspaceNode>>>,
     pub seat_state: NodeSeatState,
@@ -65,6 +69,7 @@ impl OutputNode {
         }
         self.render_data.borrow_mut().titles.clear();
         self.lock_surface.take();
+        self.jay_outputs.clear();
     }
 
     pub fn on_spaces_changed(self: &Rc<Self>) {