autocommit 2022-02-05 02:07:07 CET

src/ifs/wl_data_offer/mod.rs

@@ -1,8 +1,12 @@
 mod types;
 
-use crate::client::Client;
+use crate::client::{Client, DynEventFormatter};
+use crate::ifs::wl_data_source::{WlDataSource};
+use crate::ifs::wl_seat::WlSeatGlobal;
 use crate::object::{Interface, Object, ObjectId};
 use crate::utils::buffd::MsgParser;
+use crate::utils::clonecell::CloneCell;
+use std::ops::Deref;
 use std::rc::Rc;
 pub use types::*;
 
@@ -27,24 +31,83 @@ const INVALID_OFFER: u32 = 3;
 
 id!(WlDataOfferId);
 
+#[derive(Debug, Copy, Clone, Eq, PartialEq)]
+pub enum DataOfferRole {
+    Selection,
+}
+
 pub struct WlDataOffer {
-    id: WlDataOfferId,
-    client: Rc<Client>,
+    pub id: WlDataOfferId,
+    pub client: Rc<Client>,
+    pub role: DataOfferRole,
+    pub source: CloneCell<Option<Rc<WlDataSource>>>,
 }
 
 impl WlDataOffer {
+    pub fn create(
+        client: &Rc<Client>,
+        role: DataOfferRole,
+        src: &Rc<WlDataSource>,
+        seat: &Rc<WlSeatGlobal>,
+    ) -> Option<Rc<Self>> {
+        let id = match client.new_id() {
+            Ok(id) => id,
+            Err(e) => {
+                client.error(e);
+                return None;
+            }
+        };
+        let slf = Rc::new(Self {
+            id,
+            client: client.clone(),
+            role,
+            source: CloneCell::new(Some(src.clone())),
+        });
+        let mt = src.mime_types.borrow_mut();
+        seat.for_each_data_device(0, client.id, |device| {
+            client.event(device.data_offer(slf.id));
+            for mt in mt.deref() {
+                client.event(slf.offer(mt));
+            }
+            let ev = match role {
+                DataOfferRole::Selection => device.selection(id),
+            };
+            client.event(ev);
+        });
+        client.add_server_obj(&slf);
+        Some(slf)
+    }
+
+    pub fn offer(self: &Rc<Self>, mime_type: &str) -> DynEventFormatter {
+        Box::new(Offer {
+            obj: self.clone(),
+            mime_type: mime_type.to_string(),
+        })
+    }
+
     fn accept(&self, parser: MsgParser<'_, '_>) -> Result<(), AcceptError> {
         let _req: Accept = self.client.parse(self, parser)?;
         Ok(())
     }
 
     fn receive(&self, parser: MsgParser<'_, '_>) -> Result<(), ReceiveError> {
-        let _req: Receive = self.client.parse(self, parser)?;
+        let req: Receive = self.client.parse(self, parser)?;
+        if let Some(src) = self.source.get() {
+            src.client.event(src.send(req.mime_type, req.fd));
+            src.client.flush();
+        }
         Ok(())
     }
 
+    fn disconnect(&self) {
+        if let Some(src) = self.source.set(None) {
+            src.destroy_offer();
+        }
+    }
+
     fn destroy(&self, parser: MsgParser<'_, '_>) -> Result<(), DestroyError> {
         let _req: Destroy = self.client.parse(self, parser)?;
+        self.disconnect();
         self.client.remove_obj(self)?;
         Ok(())
     }
@@ -90,4 +153,8 @@ impl Object for WlDataOffer {
     fn num_requests(&self) -> u32 {
         SET_ACTIONS + 1
     }
+
+    fn break_loops(&self) {
+        self.disconnect();
+    }
 }