From b31422d48a600ccccb682567f5eb11fc0bca547c Mon Sep 17 00:00:00 2001
From: HampusM <hampus@hampusmat.com>
Date: Sat, 20 Aug 2022 17:08:58 +0200
Subject: feat: allow bind interface to default factory

This commit will allow interface traits to be bound to default factories. Default factories being factories that doesn't take any arguments
---
 src/di_container.rs | 44 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)

(limited to 'src/di_container.rs')

diff --git a/src/di_container.rs b/src/di_container.rs
index eaa0366..84fb9e0 100644
--- a/src/di_container.rs
+++ b/src/di_container.rs
@@ -181,6 +181,39 @@ where
 
         Ok(())
     }
+
+    /// Creates a binding of type `Interface` to a factory that takes no arguments
+    /// inside of the associated [`DIContainer`].
+    ///
+    /// *This function is only available if Syrette is built with the "factory" feature.*
+    ///
+    /// # Errors
+    /// Will return Err if the associated [`DIContainer`] already have a binding for
+    /// the interface.
+    #[cfg(feature = "factory")]
+    pub fn to_default_factory<Return>(
+        &mut self,
+        factory_func: &'static dyn Fn<(), Output = TransientPtr<Return>>,
+    ) -> error_stack::Result<(), BindingBuilderError>
+    where
+        Return: 'static + ?Sized,
+    {
+        let factory_impl = CastableFactory::new(factory_func);
+
+        self.di_container
+            .bindings
+            .set::<Interface>(Box::new(crate::provider::FactoryProvider::new(
+                crate::ptr::FactoryPtr::new(factory_impl),
+            )))
+            .ok_or_else(|| {
+                report!(BindingBuilderError).attach_printable(format!(
+                    "Binding already exists for interface '{}'",
+                    type_name::<Interface>()
+                ))
+            })?;
+
+        Ok(())
+    }
 }
 
 /// Dependency injection container.
@@ -290,6 +323,17 @@ impl DIContainer
                 .map_err(unable_to_cast_binding::<Interface>);
         }
 
+        #[cfg(feature = "factory")]
+        if let Providable::Factory(binding_factory) = binding_providable {
+            use crate::interfaces::factory::IFactory;
+
+            let factory = binding_factory
+                .cast::<dyn IFactory<(), Interface>>()
+                .map_err(unable_to_cast_binding::<Interface>)?;
+
+            return Ok(factory());
+        }
+
         Err(report!(DIContainerError).attach_printable(format!(
             "Binding for interface '{}' is not transient",
             type_name::<Interface>()
-- 
cgit v1.2.3-18-g5258