use { crate::{ client::{Client, ClientError}, ifs::ipc::{ break_offer_loops, destroy_data_offer, receive_data_offer, wl_data_device::{ClipboardIpc, WlDataDevice}, wl_data_device_manager::DND_ALL, OfferData, Role, OFFER_STATE_ACCEPTED, OFFER_STATE_DROPPED, OFFER_STATE_FINISHED, SOURCE_STATE_FINISHED, }, leaks::Tracker, object::Object, utils::{ bitflags::BitflagsExt, buffd::{MsgParser, MsgParserError}, }, wire::{wl_data_offer::*, WlDataOfferId}, xwayland::XWaylandEvent, }, std::rc::Rc, thiserror::Error, }; #[allow(dead_code)] const INVALID_FINISH: u32 = 0; #[allow(dead_code)] const INVALID_ACTION_MASK: u32 = 1; #[allow(dead_code)] const INVALID_ACTION: u32 = 2; #[allow(dead_code)] const INVALID_OFFER: u32 = 3; pub struct WlDataOffer { pub id: WlDataOfferId, pub u64_id: u64, pub client: Rc, pub device: Rc, pub data: OfferData, pub tracker: Tracker, } impl WlDataOffer { pub fn send_offer(self: &Rc, mime_type: &str) { if self.data.is_xwm { if let Some(src) = self.data.source.get() { if !src.data.is_xwm { self.client.state.xwayland.queue.push( XWaylandEvent::ClipboardAddOfferMimeType( self.clone(), mime_type.to_string(), ), ); } } } else { self.client.event(Offer { self_id: self.id, mime_type, }) } } pub fn send_source_actions(&self) { if !self.data.is_xwm { if let Some(src) = self.data.source.get() { if let Some(source_actions) = src.data.actions.get() { self.client.event(SourceActions { self_id: self.id, source_actions, }) } } } } pub fn send_action(&self, dnd_action: u32) { if !self.data.is_xwm { self.client.event(Action { self_id: self.id, dnd_action, }) } } fn accept(&self, parser: MsgParser<'_, '_>) -> Result<(), WlDataOfferError> { let req: Accept = self.client.parse(self, parser)?; let _ = req.serial; // unused let mut state = self.data.shared.state.get(); if state.contains(OFFER_STATE_FINISHED) { return Err(WlDataOfferError::AlreadyFinished); } if req.mime_type.is_some() { state |= OFFER_STATE_ACCEPTED; } else { state &= !OFFER_STATE_ACCEPTED; } self.data.shared.state.set(state); if let Some(src) = self.data.source.get() { src.send_target(req.mime_type); } Ok(()) } fn receive(&self, parser: MsgParser<'_, '_>) -> Result<(), WlDataOfferError> { let req: Receive = self.client.parse(self, parser)?; if self.data.shared.state.get().contains(OFFER_STATE_FINISHED) { return Err(WlDataOfferError::AlreadyFinished); } receive_data_offer::(self, req.mime_type, req.fd); Ok(()) } fn destroy(&self, parser: MsgParser<'_, '_>) -> Result<(), WlDataOfferError> { let _req: Destroy = self.client.parse(self, parser)?; destroy_data_offer::(self); self.client.remove_obj(self)?; Ok(()) } fn finish(&self, parser: MsgParser<'_, '_>) -> Result<(), WlDataOfferError> { let _req: Finish = self.client.parse(self, parser)?; if self.data.shared.role.get() != Role::Dnd { return Err(WlDataOfferError::NotDnd); } let mut state = self.data.shared.state.get(); if state.contains(OFFER_STATE_FINISHED) { return Err(WlDataOfferError::AlreadyFinished); } if !state.contains(OFFER_STATE_DROPPED) { return Err(WlDataOfferError::StillDragging); } if !state.contains(OFFER_STATE_ACCEPTED) { return Err(WlDataOfferError::NoMimeTypeAccepted); } state |= OFFER_STATE_FINISHED; if let Some(src) = self.data.source.get() { src.data.state.or_assign(SOURCE_STATE_FINISHED); src.send_dnd_finished(); } else { log::error!("no source"); } self.data.shared.state.set(state); Ok(()) } fn set_actions(&self, parser: MsgParser<'_, '_>) -> Result<(), WlDataOfferError> { let req: SetActions = self.client.parse(self, parser)?; let state = self.data.shared.state.get(); if state.contains(OFFER_STATE_FINISHED) { return Err(WlDataOfferError::AlreadyFinished); } if (req.dnd_actions & !DND_ALL, req.preferred_action & !DND_ALL) != (0, 0) { return Err(WlDataOfferError::InvalidActions); } if req.preferred_action.count_ones() > 1 { return Err(WlDataOfferError::MultiplePreferred); } self.data.shared.receiver_actions.set(req.dnd_actions); self.data .shared .receiver_preferred_action .set(req.preferred_action); if let Some(src) = self.data.source.get() { src.update_selected_action(); } Ok(()) } } object_base! { self = WlDataOffer; ACCEPT => accept, RECEIVE => receive, DESTROY => destroy, FINISH => finish if self.device.version >= 3, SET_ACTIONS => set_actions if self.device.version >= 3, } impl Object for WlDataOffer { fn break_loops(&self) { break_offer_loops::(self); } } simple_add_obj!(WlDataOffer); #[derive(Debug, Error)] pub enum WlDataOfferError { #[error(transparent)] ClientError(Box), #[error("Parsing failed")] MsgParserError(#[source] Box), #[error("`finish` was already called")] AlreadyFinished, #[error("The drag operation is still ongoing")] StillDragging, #[error("Client did not accept a mime type")] NoMimeTypeAccepted, #[error("This is not a drag-and-drop offer")] NotDnd, #[error("The set of actions is invalid")] InvalidActions, #[error("Multiple preferred actions were specified")] MultiplePreferred, } efrom!(WlDataOfferError, ClientError); efrom!(WlDataOfferError, MsgParserError);