use std::io::Read;
use std::sync::mpsc;
use smithay_client_toolkit::data_device_manager::data_device::{ DataDeviceData, DataDeviceHandler };
use smithay_client_toolkit::data_device_manager::data_offer::{ DataOfferHandler, DragOffer, SelectionOffer };
use smithay_client_toolkit::data_device_manager::data_source::DataSourceHandler;
use smithay_client_toolkit::reexports::client::
{
protocol::
{
wl_data_device::WlDataDevice,
wl_data_device_manager::DndAction,
wl_data_source::WlDataSource,
wl_surface::WlSurface,
},
Connection, Proxy, QueueHandle,
};
use crate::app::App;
use super::app_data::AppData;
pub( super ) const CLIPBOARD_MIMES: &[ &str ] = &[ "text/plain;charset=utf-8", "text/plain", "UTF8_STRING" ];
pub( super ) const DND_MIMES: &[ &str ] = &[ "text/uri-list", "text/plain;charset=utf-8", "text/plain", "UTF8_STRING" ];
pub( crate ) struct DropPayload
{
pub mime: String,
pub text: String,
pub x: f64,
pub y: f64,
}
pub( crate ) fn drop_inbox() -> ( std::sync::mpsc::Sender<DropPayload>, std::sync::mpsc::Receiver<DropPayload> )
{
std::sync::mpsc::channel()
}
impl<A: App> AppData<A>
{
pub( crate ) fn publish_clipboard_selection( &mut self )
{
let ( Some( ref ddm ), Some( ref dd ) ) = ( self.data_device_manager.as_ref(), self.data_device.as_ref() ) else { return };
let source = ddm.create_copy_paste_source( &self.qh, CLIPBOARD_MIMES.iter().copied() );
source.set_selection( dd, self.last_input_serial );
self.clipboard_source = Some( source );
}
}
impl<A: App> DataDeviceHandler for AppData<A>
{
fn enter(
&mut self,
_c: &Connection,
_qh: &QueueHandle<Self>,
d: &WlDataDevice,
x: f64,
y: f64,
_s: &WlSurface,
)
{
let Some( data ) = d.data::<DataDeviceData>() else { return };
let Some( offer ) = data.drag_offer() else { return };
let mimes = offer.with_mime_types( |m| m.to_vec() );
let accepted = DND_MIMES.iter().copied()
.find( |c| mimes.iter().any( |m| m == *c ) )
.map( |s| s.to_string() );
offer.accept_mime_type( self.last_input_serial, accepted.clone() );
offer.set_actions( smithay_client_toolkit::reexports::client::protocol::wl_data_device_manager::DndAction::Copy,
smithay_client_toolkit::reexports::client::protocol::wl_data_device_manager::DndAction::Copy );
self.drop_position = Some( ( x, y ) );
self.drop_mime = accepted;
self.app.on_drop_motion( x as f32, y as f32 );
}
fn leave( &mut self, _c: &Connection, _qh: &QueueHandle<Self>, _d: &WlDataDevice )
{
if self.drop_position.is_some() { self.app.on_drop_leave(); }
self.drop_position = None;
self.drop_mime = None;
}
fn motion( &mut self, _c: &Connection, _qh: &QueueHandle<Self>, _d: &WlDataDevice, x: f64, y: f64 )
{
self.drop_position = Some( ( x, y ) );
self.app.on_drop_motion( x as f32, y as f32 );
}
fn drop_performed( &mut self, _c: &Connection, _qh: &QueueHandle<Self>, d: &WlDataDevice )
{
let Some( data ) = d.data::<DataDeviceData>() else { return };
let Some( offer ) = data.drag_offer() else { return };
let Some( mime ) = self.drop_mime.clone() else { offer.finish(); return };
let ( x, y ) = self.drop_position.unwrap_or( ( 0.0, 0.0 ) );
let Ok( pipe ) = offer.receive( mime.clone() ) else { offer.finish(); return };
offer.finish();
let tx = self.drop_inbox_tx.clone();
std::thread::spawn( move ||
{
use std::io::Read;
let mut reader = std::fs::File::from( std::os::fd::OwnedFd::from( pipe ) );
let mut buf = String::new();
let mut bounded = ( &mut reader ).take( 16 * 1024 * 1024 );
let _ = bounded.read_to_string( &mut buf );
let _ = tx.send( DropPayload { mime, text: buf, x, y } );
} );
self.drop_position = None;
self.drop_mime = None;
}
fn selection( &mut self, _conn: &Connection, _qh: &QueueHandle<Self>, data_device: &WlDataDevice )
{
let Some( data ) = data_device.data::<DataDeviceData>() else { return };
let Some( offer ) = data.selection_offer() else { return };
let Some( mime ) = pick_mime( &offer ) else { return };
let Ok( pipe ) = offer.receive( mime ) else { return };
let tx = self.clipboard_inbox_tx.clone();
std::thread::spawn( move ||
{
let mut reader = std::fs::File::from( std::os::fd::OwnedFd::from( pipe ) );
let mut buf = String::new();
let mut bounded = (&mut reader).take( 16 * 1024 * 1024 );
let _ = bounded.read_to_string( &mut buf );
let _ = tx.send( buf );
} );
}
}
impl<A: App> DataOfferHandler for AppData<A>
{
fn source_actions( &mut self, _c: &Connection, _qh: &QueueHandle<Self>, _offer: &mut DragOffer, _actions: DndAction ) {}
fn selected_action( &mut self, _c: &Connection, _qh: &QueueHandle<Self>, _offer: &mut DragOffer, _actions: DndAction ) {}
}
impl<A: App> DataSourceHandler for AppData<A>
{
fn accept_mime( &mut self, _c: &Connection, _qh: &QueueHandle<Self>, _source: &WlDataSource, _mime: Option<String> ) {}
fn send_request( &mut self, _conn: &Connection, _qh: &QueueHandle<Self>, source: &WlDataSource, _mime: String, fd: smithay_client_toolkit::data_device_manager::WritePipe )
{
let is_active = self.clipboard_source.as_ref()
.map( |s| s.inner() == source )
.unwrap_or( false );
if !is_active { return; }
use std::io::Write;
let mut writer = std::fs::File::from( std::os::fd::OwnedFd::from( fd ) );
let _ = writer.write_all( self.clipboard.as_bytes() );
}
fn cancelled( &mut self, _c: &Connection, _qh: &QueueHandle<Self>, source: &WlDataSource )
{
if self.clipboard_source.as_ref().map( |s| s.inner() == source ).unwrap_or( false )
{
self.clipboard_source = None;
}
}
fn dnd_dropped( &mut self, _c: &Connection, _qh: &QueueHandle<Self>, _source: &WlDataSource ) {}
fn dnd_finished( &mut self, _c: &Connection, _qh: &QueueHandle<Self>, _source: &WlDataSource ) {}
fn action( &mut self, _c: &Connection, _qh: &QueueHandle<Self>, _source: &WlDataSource, _action: DndAction ) {}
}
fn pick_mime( offer: &SelectionOffer ) -> Option<String>
{
let mimes = offer.with_mime_types( |m| m.to_vec() );
CLIPBOARD_MIMES.iter()
.find( |c| mimes.iter().any( |m| m == *c ) )
.map( |s| s.to_string() )
}
pub( crate ) fn clipboard_inbox() -> ( mpsc::Sender<String>, mpsc::Receiver<String> )
{
mpsc::channel()
}