#pragma once

#include "DI/interfaces/wrapper.hpp"
#include "DI/object_type.hpp"
#include "DI/type_traits.hpp"

#include <memory>
#include <type_traits>
#include <unordered_map>

class Container;

template <typename Interface>
class BindingBuilder
{
public:
	explicit BindingBuilder(Container *container) noexcept;

	template <class ObjectImpl,
			  class = std::enable_if_t<std::is_base_of_v<Interface, ObjectImpl>>>
	void to() noexcept;

	void to_factory(Interface func) noexcept;

private:
	Container *_container;
};

class Container
{
public:
	Container() noexcept = default;

	template <class Interface>
	auto bind() noexcept -> BindingBuilder<Interface>;

	template <class Interface, class = std::enable_if_t<std::is_abstract_v<Interface>>>
	auto get() const noexcept -> std::shared_ptr<Interface>;

	template <typename Interface, typename = std::enable_if_t<is_func_v<Interface>>>
	auto get() const noexcept -> Interface;

	std::unordered_map<BaseObjectType, std::shared_ptr<IGenericWrapper>, ObjectTypeHasher>
		bindings;
};

#include "container.tpp"