Split up wayland code into more files

src/wayland/screencopy.rs

@@ -0,0 +1,101 @@
+use cctk::{
+    cosmic_protocols::screencopy::v1::client::zcosmic_screencopy_session_v1,
+    screencopy::{BufferInfo, ScreencopyHandler, ScreencopyState},
+    wayland_client::{protocol::wl_shm, Connection, QueueHandle, WEnum},
+};
+
+use std::sync::atomic::Ordering;
+
+use super::{AppData, Buffer, Capture, CaptureSource, Event};
+
+impl ScreencopyHandler for AppData {
+    fn screencopy_state(&mut self) -> &mut ScreencopyState {
+        &mut self.screencopy_state
+    }
+
+    fn init_done(
+        &mut self,
+        conn: &Connection,
+        qh: &QueueHandle<Self>,
+        session: &zcosmic_screencopy_session_v1::ZcosmicScreencopySessionV1,
+        buffer_infos: &[BufferInfo],
+    ) {
+        let capture = Capture::for_session(session).unwrap();
+        if capture.cancelled.load(Ordering::SeqCst) {
+            session.destroy();
+            return;
+        }
+
+        let buffer_info = buffer_infos
+            .iter()
+            .find(|x| {
+                x.type_ == WEnum::Value(zcosmic_screencopy_session_v1::BufferType::WlShm)
+                    && x.format == wl_shm::Format::Abgr8888.into()
+            })
+            .unwrap();
+
+        // XXX fix in compositor
+        if buffer_info.width == 0 || buffer_info.height == 0 || buffer_info.stride == 0 {
+            session.destroy();
+            return;
+        }
+
+        let mut buffer = capture.buffer.lock().unwrap();
+        // Create new buffer if none, or different format
+        if !buffer
+            .as_ref()
+            .map_or(false, |x| &x.buffer_info == buffer_info)
+        {
+            *buffer = Some(Buffer::new(buffer_info.clone(), &self.shm_state, qh));
+        }
+        let buffer = buffer.as_ref().unwrap();
+
+        session.attach_buffer(&buffer.buffer, None, 0); // XXX age?
+        if capture.first_frame.load(Ordering::SeqCst) {
+            session.commit(zcosmic_screencopy_session_v1::Options::empty());
+        } else {
+            session.commit(zcosmic_screencopy_session_v1::Options::OnDamage);
+        }
+        conn.flush().unwrap();
+    }
+
+    fn ready(
+        &mut self,
+        _conn: &Connection,
+        _qh: &QueueHandle<Self>,
+        session: &zcosmic_screencopy_session_v1::ZcosmicScreencopySessionV1,
+    ) {
+        let capture = Capture::for_session(session).unwrap();
+        if capture.cancelled.load(Ordering::SeqCst) {
+            session.destroy();
+            return;
+        }
+
+        let mut buffer = capture.buffer.lock().unwrap();
+        let image = unsafe { buffer.as_mut().unwrap().to_image() };
+        let event = match &capture.source {
+            CaptureSource::Toplevel(toplevel) => Event::ToplevelCapture(toplevel.clone(), image),
+            CaptureSource::Workspace(workspace, _) => {
+                Event::WorkspaceCapture(workspace.clone(), image)
+            }
+        };
+        self.send_event(event);
+        session.destroy();
+
+        // Capture again on damage
+        capture.first_frame.store(false, Ordering::SeqCst);
+        capture.capture(&self.screencopy_state.screencopy_manager, &self.qh);
+    }
+
+    fn failed(
+        &mut self,
+        _conn: &Connection,
+        _qh: &QueueHandle<Self>,
+        session: &zcosmic_screencopy_session_v1::ZcosmicScreencopySessionV1,
+        _reason: WEnum<zcosmic_screencopy_session_v1::FailureReason>,
+    ) {
+        // TODO
+        println!("Failed");
+        session.destroy();
+    }
+}