use std::marker::PhantomData; use std::rc::Rc; use crate::castable_function::AnyCastableFunction; use crate::errors::injectable::InjectableError; use crate::interfaces::injectable::Injectable; use crate::ptr::{SingletonPtr, TransientPtr}; use crate::util::use_double; use_double!(crate::dependency_history::DependencyHistory); #[derive(strum_macros::Display, Debug)] pub enum Providable { Transient(TransientPtr>), Singleton(SingletonPtr>), Function(Rc, ProvidableFunctionKind), } #[derive(Debug, Clone, Copy)] pub enum ProvidableFunctionKind { #[cfg(feature = "factory")] UserCalled, Instant, } #[cfg_attr(test, mockall::automock)] pub trait IProvider { fn provide( &self, di_container: &DIContainerType, dependency_history: DependencyHistory, ) -> Result, InjectableError>; } pub struct TransientTypeProvider where InjectableType: Injectable, { injectable_phantom: PhantomData, di_container_phantom: PhantomData, } impl TransientTypeProvider where InjectableType: Injectable, { pub fn new() -> Self { Self { injectable_phantom: PhantomData, di_container_phantom: PhantomData, } } } impl IProvider for TransientTypeProvider where InjectableType: Injectable, { fn provide( &self, di_container: &DIContainerType, dependency_history: DependencyHistory, ) -> Result, InjectableError> { Ok(Providable::Transient(InjectableType::resolve( di_container, dependency_history, )?)) } } pub struct SingletonProvider where InjectableType: Injectable, { singleton: SingletonPtr, di_container_phantom: PhantomData, } impl SingletonProvider where InjectableType: Injectable, { pub fn new(singleton: SingletonPtr) -> Self { Self { singleton, di_container_phantom: PhantomData, } } } impl IProvider for SingletonProvider where InjectableType: Injectable, { fn provide( &self, _di_container: &DIContainerType, _dependency_history: DependencyHistory, ) -> Result, InjectableError> { Ok(Providable::Singleton(self.singleton.clone())) } } pub struct FunctionProvider { function: Rc, providable_func_kind: ProvidableFunctionKind, } impl FunctionProvider { pub fn new( function: Rc, providable_func_kind: ProvidableFunctionKind, ) -> Self { Self { function, providable_func_kind, } } } impl IProvider for FunctionProvider { fn provide( &self, _di_container: &DIContainerType, _dependency_history: DependencyHistory, ) -> Result, InjectableError> { Ok(Providable::Function( self.function.clone(), self.providable_func_kind, )) } } #[cfg(test)] mod tests { use super::*; use crate::dependency_history::MockDependencyHistory; use crate::di_container::blocking::MockDIContainer; use crate::test_utils::subjects; #[test] fn transient_type_provider_works() { let transient_type_provider = TransientTypeProvider::::new(); let di_container = MockDIContainer::new(); let dependency_history_mock = MockDependencyHistory::new(); assert!( matches!( transient_type_provider.provide(&di_container, dependency_history_mock), Ok(Providable::Transient(_)) ), "The provided type is not transient" ); } #[test] fn singleton_provider_works() { let singleton_provider = SingletonProvider::::new( SingletonPtr::new(subjects::UserManager {}), ); let di_container = MockDIContainer::new(); assert!( matches!( singleton_provider .provide(&di_container, MockDependencyHistory::new()) .unwrap(), Providable::Singleton(_) ), "The provided type is not a singleton" ); } #[test] fn function_provider_works() { use std::any::Any; use std::rc::Rc; use crate::castable_function::AnyCastableFunction; #[derive(Debug)] struct FooFactory; impl AnyCastableFunction for FooFactory { fn as_any(&self) -> &dyn Any { self } } let instant_func_provider = FunctionProvider::new(Rc::new(FooFactory), ProvidableFunctionKind::Instant); let di_container = MockDIContainer::new(); assert!( matches!( instant_func_provider .provide(&di_container, MockDependencyHistory::new()), Ok(Providable::Function(_, ProvidableFunctionKind::Instant)) ), concat!( "The provided type is not a Providable::Function of kind ", "ProvidableFunctionKind::Instant" ) ); } }