1 files changed, 53 insertions, 0 deletions
diff --git a/src/DI/container.tpp b/src/DI/container.tpp
new file mode 100644
index 0000000..21bf81a
--- /dev/null
+++ b/src/DI/container.tpp
@@ -0,0 +1,53 @@
+#pragma once
+
+#include "container.hpp"
+
+#include "function_wrapper.hpp"
+#include "object_wrapper.hpp"
+
+template <class Interface>
+BindingBuilder<Interface>::BindingBuilder(Container *container) : _container(container)
+{
+}
+
+template <class Interface>
+template <class ObjectImpl, class>
+void BindingBuilder<Interface>::to()
+{
+	_container->bindings.emplace(
+		ObjectType<Interface>(),
+		std::dynamic_pointer_cast<IGenericWrapper>(
+			std::make_shared<ObjectWrapper<Interface, ObjectImpl>>(*_container)));
+}
+
+template <class Interface>
+void BindingBuilder<Interface>::to_factory(Interface func)
+{
+	_container->bindings.emplace(ObjectType<Interface>(),
+								 std::dynamic_pointer_cast<IGenericWrapper>(
+									 std::make_shared<FunctionWrapper<Interface>>(func)));
+}
+
+template <class Interface>
+BindingBuilder<Interface> Container::bind()
+{
+	return BindingBuilder<Interface>(this);
+}
+
+template <class Interface, class>
+std::shared_ptr<Interface> Container::get() const
+{
+	auto wrapper = std::dynamic_pointer_cast<IWrapper<std::shared_ptr<Interface>>>(
+		bindings.at(ObjectType<Interface>()));
+
+	return wrapper->get();
+}
+
+template <typename Interface, typename>
+Interface Container::get() const
+{
+	auto wrapper = std::dynamic_pointer_cast<IWrapper<Interface>>(
+		bindings.at(ObjectType<Interface>()));
+
+	return wrapper->get();
+}