diff options
Diffstat (limited to 'src/private/castable_factory/blocking.rs')
-rw-r--r-- | src/private/castable_factory/blocking.rs | 153 |
1 files changed, 153 insertions, 0 deletions
diff --git a/src/private/castable_factory/blocking.rs b/src/private/castable_factory/blocking.rs new file mode 100644 index 0000000..abc3e26 --- /dev/null +++ b/src/private/castable_factory/blocking.rs @@ -0,0 +1,153 @@ +use std::any::type_name; +use std::fmt::Debug; +use std::marker::Tuple; + +use crate::private::any_factory::AnyFactory; +use crate::private::factory::IFactory; +use crate::ptr::TransientPtr; + +pub struct CastableFactory<Args, ReturnInterface> +where + Args: Tuple + 'static, + ReturnInterface: 'static + ?Sized, +{ + func: &'static dyn Fn<Args, Output = TransientPtr<ReturnInterface>>, +} + +impl<Args, ReturnInterface> CastableFactory<Args, ReturnInterface> +where + Args: Tuple + 'static, + ReturnInterface: 'static + ?Sized, +{ + pub fn new( + func: &'static dyn Fn<Args, Output = TransientPtr<ReturnInterface>>, + ) -> Self + { + Self { func } + } +} + +impl<Args, ReturnInterface> IFactory<Args, ReturnInterface> + for CastableFactory<Args, ReturnInterface> +where + Args: Tuple + 'static, + ReturnInterface: 'static + ?Sized, +{ +} + +impl<Args, ReturnInterface> Fn<Args> for CastableFactory<Args, ReturnInterface> +where + Args: Tuple + 'static, + ReturnInterface: 'static + ?Sized, +{ + extern "rust-call" fn call(&self, args: Args) -> Self::Output + { + self.func.call(args) + } +} + +impl<Args, ReturnInterface> FnMut<Args> for CastableFactory<Args, ReturnInterface> +where + Args: Tuple + 'static, + ReturnInterface: 'static + ?Sized, +{ + extern "rust-call" fn call_mut(&mut self, args: Args) -> Self::Output + { + self.call(args) + } +} + +impl<Args, ReturnInterface> FnOnce<Args> for CastableFactory<Args, ReturnInterface> +where + Args: Tuple + 'static, + ReturnInterface: 'static + ?Sized, +{ + type Output = TransientPtr<ReturnInterface>; + + extern "rust-call" fn call_once(self, args: Args) -> Self::Output + { + self.call(args) + } +} + +impl<Args, ReturnInterface> AnyFactory for CastableFactory<Args, ReturnInterface> +where + Args: Tuple + 'static, + ReturnInterface: 'static + ?Sized, +{ +} + +impl<Args, ReturnInterface> Debug for CastableFactory<Args, ReturnInterface> +where + Args: Tuple + 'static, + ReturnInterface: 'static + ?Sized, +{ + #[cfg(not(tarpaulin_include))] + fn fmt(&self, formatter: &mut std::fmt::Formatter<'_>) -> std::fmt::Result + { + let mut args = type_name::<Args>(); + + if args.len() < 2 { + return Err(std::fmt::Error::default()); + } + + args = args + .get(1..args.len() - 1) + .map_or_else(|| Err(std::fmt::Error::default()), Ok)?; + + if args.ends_with(',') { + args = args + .get(..args.len() - 1) + .map_or_else(|| Err(std::fmt::Error), Ok)?; + } + + let ret = type_name::<TransientPtr<ReturnInterface>>(); + + formatter.write_fmt(format_args!("CastableFactory ({}) -> {}", args, ret)) + } +} + +#[cfg(test)] +mod tests +{ + use super::*; + + #[derive(Debug, PartialEq, Eq)] + struct Bacon + { + heal_amount: u32, + } + + #[test] + fn can_call() + { + let castable_factory = + CastableFactory::new(&|heal_amount| TransientPtr::new(Bacon { heal_amount })); + + let output = castable_factory.call((27,)); + + assert_eq!(output, TransientPtr::new(Bacon { heal_amount: 27 })); + } + + #[test] + fn can_call_mut() + { + let mut castable_factory = + CastableFactory::new(&|heal_amount| TransientPtr::new(Bacon { heal_amount })); + + let output = castable_factory.call_mut((103,)); + + assert_eq!(output, TransientPtr::new(Bacon { heal_amount: 103 })); + } + + #[test] + fn can_call_once() + { + let castable_factory = + CastableFactory::new(&|heal_amount| TransientPtr::new(Bacon { heal_amount })); + + let output = castable_factory.call_once((19,)); + + assert_eq!(output, TransientPtr::new(Bacon { heal_amount: 19 })); + } +} |