refactor: add factory class & make DI container return unique ptrs

1 files changed, 32 insertions, 0 deletions

diff --git a/src/DI/allocation.hpp b/src/DI/allocation.hpp
new file mode 100644
index 0000000..ac0fa38
--- /dev/null
+++ b/src/DI/allocation.hpp
@@ -0,0 +1,32 @@
+#pragma once
+
+#include <memory>
+
+template <class Traits, class Type>
+struct RebindAllocHelper
+{
+	using type = typename Traits::template rebind_alloc<Type>;
+};
+
+template <class Allocator>
+class AllocDestructor
+{
+	using _alloc_traits = std::allocator_traits<Allocator>;
+
+public:
+	using Pointer = typename _alloc_traits::pointer;
+	using Size = typename _alloc_traits::size_type;
+
+	using pointer = Pointer;
+	using size = Size;
+
+	AllocDestructor(Allocator &allocator, Size alloc_size) noexcept;
+
+	void operator()(Pointer ptr) noexcept;
+
+private:
+	Allocator &_allocator;
+	Size _size;
+};
+
+#include "allocation.tpp"