use crate::{Activator, ServiceDependency, ServiceDescriptor, ServiceLifetime, Type};
use std::sync::OnceLock;
pub struct InjectBuilder {
activator: Activator,
lifetime: ServiceLifetime,
key_type: Option<Type>,
dependencies: Vec<ServiceDependency>,
}
impl InjectBuilder {
pub fn new(activator: Activator, lifetime: ServiceLifetime) -> Self {
Self {
activator,
lifetime,
key_type: None,
dependencies: Vec::default(),
}
}
pub fn depends_on(mut self, dependency: ServiceDependency) -> Self {
if !self.dependencies.contains(&dependency) {
self.dependencies.push(dependency);
}
self
}
pub fn with_key<TKey>(mut self) -> Self {
self.key_type = Some(Type::of::<TKey>());
self
}
pub fn as_mut(mut self) -> Self {
self.activator.as_mut();
self
}
pub fn build(mut self) -> ServiceDescriptor {
ServiceDescriptor::new(
self.lifetime,
if let Some(key) = self.key_type {
self.activator.service_type().with_key(&key)
} else {
self.activator.service_type().clone()
},
self.activator.implementation_type().clone(),
if self.dependencies.is_empty() {
Vec::new()
} else {
self.dependencies.shrink_to_fit();
self.dependencies
},
OnceLock::new(),
self.activator.factory(),
)
}
}
impl From<InjectBuilder> for ServiceDescriptor {
fn from(value: InjectBuilder) -> Self {
value.build()
}
}