use std::convert::Infallible;
use std::error::Error;
use std::marker::PhantomData;
use std::sync::Arc;
use std::mem::MaybeUninit;
use crate::TraitObject;
use super::null_ptr_error;
use super::{FromForeign, InputType, ReturnType, ToForeign};
pub struct ArcMarshaler<T: ?Sized>(PhantomData<T>);
impl<T: ?Sized> InputType for ArcMarshaler<T> {
type Foreign = *const T;
type ForeignTraitObject = ();
}
impl<T: ?Sized> ReturnType for ArcMarshaler<T> {
type Foreign = *const T;
type ForeignTraitObject = TraitObject<T>;
fn foreign_default() -> Self::Foreign {
unsafe { MaybeUninit::zeroed().assume_init() }
}
fn foreign_default_trait_object() -> Self::ForeignTraitObject {
TraitObject {
data: std::ptr::null_mut(),
vtable: std::ptr::null_mut(),
ty: PhantomData
}
}
}
impl<T: ?Sized> ToForeign<Arc<T>, *const T> for ArcMarshaler<T> {
type Error = Infallible;
#[inline(always)]
fn to_foreign(local: Arc<T>) -> Result<*const T, Self::Error> {
log::debug!(
"<ArcMarshaler<{ty}> as ToForeign<{ty}, {o}>>::to_foreign",
ty = std::any::type_name::<T>(),
o = "*const c_void"
);
Ok(Arc::into_raw(local))
}
}
impl<T: ?Sized> FromForeign<*const T, Arc<T>> for ArcMarshaler<T> {
type Error = Box<dyn Error>;
#[inline(always)]
unsafe fn from_foreign(foreign: *const T) -> Result<Arc<T>, Self::Error> {
log::debug!(
"<ArcMarshaler<{ty}> as FromForeign<*const std::ffi::c_void, T>>::from_foreign({:?})",
foreign,
ty = std::any::type_name::<T>()
);
if foreign.is_null() {
return Err(null_ptr_error());
}
Ok(Arc::from_raw(foreign as *const _))
}
}
impl<T: ?Sized> ToForeign<Result<Arc<T>, Box<dyn Error>>, *const T> for ArcMarshaler<T> {
type Error = Box<dyn Error>;
#[inline(always)]
fn to_foreign(local: Result<Arc<T>, Box<dyn Error>>) -> Result<*const T, Self::Error> {
local.and_then(|x| Ok(Arc::into_raw(x) as *const _))
}
}