use crate::{CallbackResult, ClipboardHandler};
use std::io;
use std::sync::mpsc::{self, SyncSender, Receiver, sync_channel};
pub struct Shutdown {
sender: SyncSender<()>,
}
impl Drop for Shutdown {
#[inline(always)]
fn drop(&mut self) {
let _ = self.sender.send(());
}
}
pub struct Master<H> {
handler: H,
sender: SyncSender<()>,
recv: Receiver<()>
}
impl<H: ClipboardHandler> Master<H> {
#[inline(always)]
pub fn new(handler: H) -> io::Result<Self> {
let (sender, recv) = sync_channel(0);
Ok(Self {
handler,
sender,
recv,
})
}
#[inline(always)]
pub fn shutdown_channel(&self) -> Shutdown {
Shutdown {
sender: self.sender.clone()
}
}
pub fn run(&mut self) -> io::Result<()> {
let clipboard = x11_clipboard::Clipboard::new();
let clipboard = match clipboard {
Ok(clipboard) => clipboard,
Err(error) => {
return Err(io::Error::new(
io::ErrorKind::Other,
format!("Failed to initialize clipboard: {:?}", error),
))
}
};
loop {
let res = clipboard.load(
clipboard.getter.atoms.clipboard,
clipboard.getter.atoms.incr,
clipboard.getter.atoms.property,
self.handler.sleep_interval(),
);
match res {
Ok(_) => {
match self.handler.on_clipboard_change() {
CallbackResult::Next => (),
CallbackResult::Stop => break,
CallbackResult::StopWithError(error) => {
return Err(error);
}
}
},
Err(x11_clipboard::error::Error::Timeout) => (),
Err(error) => {
let error = io::Error::new(
io::ErrorKind::Other,
format!("Failed to load clipboard: {:?}", error),
);
match self.handler.on_clipboard_error(error) {
CallbackResult::Next => (),
CallbackResult::Stop => break,
CallbackResult::StopWithError(error) => {
return Err(error);
}
}
}
}
match self.recv.try_recv() {
Ok(()) => break,
Err(mpsc::TryRecvError::Empty) => continue,
Err(mpsc::TryRecvError::Disconnected) => break,
}
}
Ok(())
}
}