feat: add wayland impl for custom mime types

This commit is contained in:
Ashley Wulber 2024-02-29 17:15:30 -05:00
parent 8e7827ebbe
commit 6c41143f5c
No known key found for this signature in database
GPG key ID: 5216D4F46A90A820
5 changed files with 225 additions and 1 deletions

View file

@ -1,10 +1,17 @@
use crate::ClipboardProvider;
use crate::{
mime::{ClipboardLoadData, ClipboardStoreData},
ClipboardProvider,
};
use raw_window_handle::{HasDisplayHandle, RawDisplayHandle};
use std::error::Error;
use wayland::MimeType;
pub use clipboard_wayland as wayland;
pub use clipboard_x11 as x11;
pub use wayland::{
AllowedMimeTypes as InnerAllowedMimeTypes, AsMimeTypes as InnerAsMimeTypes,
};
pub enum Clipboard {
Wayland(wayland::Clipboard),
@ -44,6 +51,62 @@ impl ClipboardProvider for Clipboard {
}
}
}
fn read<T: 'static>(&self) -> Option<Result<T, Box<dyn Error>>>
where
ClipboardLoadData<T>: InnerAllowedMimeTypes,
{
match self {
Clipboard::Wayland(c) => {
let ret = c.read::<ClipboardLoadData<T>>();
Some(ret.map(|ret| ret.0))
}
Clipboard::X11(_) => None,
}
}
fn write<T: Send + Sync + 'static>(
&mut self,
contents: ClipboardStoreData<T>,
) -> Option<Result<(), Box<dyn Error>>>
where
ClipboardStoreData<T>: InnerAsMimeTypes,
{
match self {
Clipboard::Wayland(c) => {
Some(c.write::<ClipboardStoreData<T>>(contents))
}
Clipboard::X11(_) => None,
}
}
fn read_primary<T: 'static>(&self) -> Option<Result<T, Box<dyn Error>>>
where
ClipboardLoadData<T>: InnerAllowedMimeTypes,
{
match self {
Clipboard::Wayland(c) => {
let ret = c.read_primary::<ClipboardLoadData<T>>();
Some(ret.map(|ret| ret.0))
}
Clipboard::X11(_) => None,
}
}
fn write_primary<T: Send + Sync + 'static>(
&mut self,
contents: ClipboardStoreData<T>,
) -> Option<Result<(), Box<dyn Error>>>
where
ClipboardStoreData<T>: InnerAsMimeTypes,
{
match self {
Clipboard::Wayland(c) => {
Some(c.write_primary::<ClipboardStoreData<T>>(contents))
}
Clipboard::X11(_) => None,
}
}
}
pub unsafe fn connect<W: HasDisplayHandle>(
@ -58,3 +121,51 @@ pub unsafe fn connect<W: HasDisplayHandle>(
Ok(clipboard)
}
impl<T: crate::mime::AsMimeTypes> InnerAsMimeTypes for ClipboardLoadData<T> {
fn available(&self) -> std::borrow::Cow<'static, [MimeType]> {
self.0
.available()
.into_iter()
.map(|m| MimeType::Other(m.clone().into()))
.collect()
}
fn as_bytes(
&self,
mime_type: &MimeType,
) -> Option<std::borrow::Cow<'static, [u8]>> {
self.0.as_bytes(mime_type.as_ref())
}
}
impl<T: crate::mime::AllowedMimeTypes> InnerAllowedMimeTypes
for ClipboardLoadData<T>
where
ClipboardLoadData<T>: TryFrom<(Vec<u8>, MimeType)>,
{
// TODO select text variants if string matches...
fn allowed() -> std::borrow::Cow<'static, [wayland::MimeType]> {
T::allowed()
.into_iter()
.map(|s| MimeType::Other(s.clone().into()))
.collect()
}
}
impl<T> TryFrom<(Vec<u8>, MimeType)> for ClipboardLoadData<T>
where
T: for<'b> TryFrom<(Vec<u8>, String)>,
T: 'static,
{
type Error = crate::mime::Error;
fn try_from(
(value, mime): (Vec<u8>, MimeType),
) -> Result<Self, Self::Error> {
let mime = mime.to_string();
Ok(ClipboardLoadData(
T::try_from((value, mime)).map_err(|_| crate::mime::Error)?,
))
}
}