wry/src/ifs/zwlr_screencopy_manager_v1.rs

use {
    crate::{
        client::{CAP_SCREENCOPY_MANAGER, Client, ClientCaps, ClientError},
        globals::{Global, GlobalName},
        ifs::zwlr_screencopy_frame_v1::ZwlrScreencopyFrameV1,
        leaks::Tracker,
        object::{Object, Version},
        rect::Rect,
        wire::{
            WlOutputId, ZwlrScreencopyFrameV1Id, ZwlrScreencopyManagerV1Id,
            zwlr_screencopy_manager_v1::*,
        },
    },
    std::{cell::Cell, rc::Rc},
    thiserror::Error,
};

pub struct ZwlrScreencopyManagerV1Global {
    pub name: GlobalName,
}

impl ZwlrScreencopyManagerV1Global {
    pub fn new(name: GlobalName) -> Self {
        Self { name }
    }

    fn bind_(
        self: Rc<Self>,
        id: ZwlrScreencopyManagerV1Id,
        client: &Rc<Client>,
        version: Version,
    ) -> Result<(), ZwlrScreencopyManagerV1Error> {
        let mgr = Rc::new(ZwlrScreencopyManagerV1 {
            id,
            client: client.clone(),
            tracker: Default::default(),
            version,
        });
        track!(client, mgr);
        client.add_client_obj(&mgr)?;
        Ok(())
    }
}

global_base!(
    ZwlrScreencopyManagerV1Global,
    ZwlrScreencopyManagerV1,
    ZwlrScreencopyManagerV1Error
);

simple_add_global!(ZwlrScreencopyManagerV1Global);

impl Global for ZwlrScreencopyManagerV1Global {
    fn version(&self) -> u32 {
        3
    }

    fn required_caps(&self) -> ClientCaps {
        CAP_SCREENCOPY_MANAGER
    }
}

pub struct ZwlrScreencopyManagerV1 {
    pub id: ZwlrScreencopyManagerV1Id,
    pub client: Rc<Client>,
    pub tracker: Tracker<Self>,
    pub version: Version,
}

impl ZwlrScreencopyManagerV1RequestHandler for ZwlrScreencopyManagerV1 {
    type Error = ZwlrScreencopyManagerV1Error;

    fn capture_output(&self, req: CaptureOutput, _slf: &Rc<Self>) -> Result<(), Self::Error> {
        self.do_capture_output(req.output, req.overlay_cursor != 0, req.frame, None)
    }

    fn capture_output_region(
        &self,
        req: CaptureOutputRegion,
        _slf: &Rc<Self>,
    ) -> Result<(), Self::Error> {
        let region = match Rect::new_sized(req.x, req.y, req.width, req.height) {
            Some(r) => r,
            _ => return Err(ZwlrScreencopyManagerV1Error::InvalidRegion),
        };
        self.do_capture_output(req.output, req.overlay_cursor != 0, req.frame, Some(region))
    }

    fn destroy(&self, _req: Destroy, _slf: &Rc<Self>) -> Result<(), Self::Error> {
        self.client.remove_obj(self)?;
        Ok(())
    }
}

impl ZwlrScreencopyManagerV1 {
    fn do_capture_output(
        &self,
        output: WlOutputId,
        overlay_cursor: bool,
        frame: ZwlrScreencopyFrameV1Id,
        region: Option<Rect>,
    ) -> Result<(), ZwlrScreencopyManagerV1Error> {
        let output = self.client.lookup(output)?;
        let Some(global) = output.global.get() else {
            return Ok(());
        };
        let mode = global.mode.get();
        let mut rect = Rect::new_sized_saturating(0, 0, mode.width, mode.height);
        if let Some(region) = region {
            let scale = global.persistent.scale.get().to_f64();
            let x1 = (region.x1() as f64 * scale).round() as i32;
            let y1 = (region.y1() as f64 * scale).round() as i32;
            let x2 = (region.x2() as f64 * scale).round() as i32;
            let y2 = (region.y2() as f64 * scale).round() as i32;
            let region = Rect::new_saturating(x1, y1, x2, y2);
            rect = rect.intersect(region);
        }
        let frame = Rc::new(ZwlrScreencopyFrameV1 {
            id: frame,
            client: self.client.clone(),
            tracker: Default::default(),
            output: output.global.clone(),
            rect,
            _overlay_cursor: overlay_cursor,
            used: Cell::new(false),
            with_damage: Cell::new(false),
            buffer: Cell::new(None),
            version: self.version,
            pending: Default::default(),
        });
        track!(self.client, frame);
        self.client.add_client_obj(&frame)?;
        frame.send_buffer();
        if self.version >= 3 {
            frame.send_linux_dmabuf();
            frame.send_buffer_done();
        }
        Ok(())
    }
}

object_base! {
    self = ZwlrScreencopyManagerV1;
    version = self.version;
}

impl Object for ZwlrScreencopyManagerV1 {}

simple_add_obj!(ZwlrScreencopyManagerV1);

#[derive(Debug, Error)]
pub enum ZwlrScreencopyManagerV1Error {
    #[error(transparent)]
    ClientError(Box<ClientError>),
    #[error("The passed region is invalid")]
    InvalidRegion,
}
efrom!(ZwlrScreencopyManagerV1Error, ClientError);