wayland: add jay_render_ctx

src/ifs/jay_compositor.rs

@@ -5,7 +5,8 @@ use {
         globals::{Global, GlobalName},
         ifs::{
             jay_idle::JayIdle, jay_log_file::JayLogFile, jay_output::JayOutput,
-            jay_pointer::JayPointer, jay_screenshot::JayScreenshot, jay_seat_events::JaySeatEvents,
+            jay_pointer::JayPointer, jay_render_ctx::JayRenderCtx, jay_screenshot::JayScreenshot,
+            jay_seat_events::JaySeatEvents,
         },
         leaks::Tracker,
         object::Object,
@@ -260,6 +261,24 @@ impl JayCompositor {
         self.client.add_client_obj(&ctx)?;
         Ok(())
     }
+
+    fn get_render_ctx(&self, parser: MsgParser<'_, '_>) -> Result<(), JayCompositorError> {
+        let req: GetRenderCtx = self.client.parse(self, parser)?;
+        let ctx = Rc::new(JayRenderCtx {
+            id: req.id,
+            client: self.client.clone(),
+            tracker: Default::default(),
+        });
+        track!(self.client, ctx);
+        self.client.add_client_obj(&ctx)?;
+        self.client
+            .state
+            .render_ctx_watchers
+            .set((self.client.id, req.id), ctx.clone());
+        let rctx = self.client.state.render_ctx.get();
+        ctx.send_render_ctx(rctx.as_ref());
+        Ok(())
+    }
 }
 
 object_base! {
@@ -278,11 +297,12 @@ object_base! {
     SEAT_EVENTS => seat_events,
     GET_OUTPUT => get_output,
     GET_POINTER => get_pointer,
+    GET_RENDER_CTX => get_render_ctx,
 }
 
 impl Object for JayCompositor {
     fn num_requests(&self) -> u32 {
-        GET_POINTER + 1
+        GET_RENDER_CTX + 1
     }
 }
 

src/ifs/jay_render_ctx.rs

@@ -0,0 +1,86 @@
+use {
+    crate::{
+        client::{Client, ClientError},
+        leaks::Tracker,
+        object::Object,
+        render::RenderContext,
+        utils::{
+            buffd::{MsgParser, MsgParserError},
+            errorfmt::ErrorFmt,
+        },
+        wire::{jay_render_ctx::*, JayRenderCtxId},
+    },
+    std::rc::Rc,
+    thiserror::Error,
+};
+
+pub struct JayRenderCtx {
+    pub id: JayRenderCtxId,
+    pub client: Rc<Client>,
+    pub tracker: Tracker<Self>,
+}
+
+impl JayRenderCtx {
+    pub fn send_render_ctx(&self, ctx: Option<&Rc<RenderContext>>) {
+        let mut fd = None;
+        if let Some(ctx) = ctx {
+            match ctx.gbm.drm.dup_render() {
+                Ok(d) => fd = Some(d.fd().clone()),
+                Err(e) => {
+                    log::error!("Could not dup drm fd: {}", ErrorFmt(e));
+                }
+            }
+        } else {
+            self.client.event(NoDevice { self_id: self.id });
+        }
+        match fd {
+            Some(fd) => self.client.event(Device {
+                self_id: self.id,
+                fd,
+            }),
+            _ => self.client.event(NoDevice { self_id: self.id }),
+        }
+    }
+
+    fn destroy(&self, parser: MsgParser<'_, '_>) -> Result<(), JayRenderCtxError> {
+        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
+            .render_ctx_watchers
+            .remove(&(self.client.id, self.id));
+    }
+}
+
+object_base! {
+    JayRenderCtx;
+
+    DESTROY => destroy,
+}
+
+impl Object for JayRenderCtx {
+    fn num_requests(&self) -> u32 {
+        DESTROY + 1
+    }
+
+    fn break_loops(&self) {
+        self.remove_from_state();
+    }
+}
+
+simple_add_obj!(JayRenderCtx);
+
+#[derive(Debug, Error)]
+pub enum JayRenderCtxError {
+    #[error("Parsing failed")]
+    MsgParserError(Box<MsgParserError>),
+    #[error(transparent)]
+    ClientError(Box<ClientError>),
+}
+efrom!(JayRenderCtxError, MsgParserError);
+efrom!(JayRenderCtxError, ClientError);