#pragma once

#include <array>
#include <functional>
#include <iostream>
#include <memory>

namespace sp {

namespace details {

template <typename TBase>
using ArrayType = std::vector<std::function<std::unique_ptr<TBase>(void)>>;



template <typename TBase, typename... TMessages>
struct ArrayFiller {};

template <typename TBase, typename... TMessages>
struct ArrayFiller<TBase, std::tuple<TMessages...>> {
	static ArrayType<TBase> ArrayCreate() {
		ArrayType<TBase> array;
		array.reserve(sizeof...(TMessages));
		ArrayFiller<TBase, TMessages...>::ArrayAppend(array);
		return array;
	}
};

template <typename TBase, typename TMessage, typename... TMessages>
struct ArrayFiller<TBase, TMessage, TMessages...> {
	static void ArrayAppend(details::ArrayType<TBase>& array) {
		ArrayFiller<TBase, TMessage>::ArrayAppend(array);
		ArrayFiller<TBase, TMessages...>::ArrayAppend(array);
	}
};

template <typename TBase, typename TMessage>
struct ArrayFiller<TBase, TMessage> {
	static void ArrayAppend(details::ArrayType<TBase>& array) {
		array.push_back([]() -> std::unique_ptr<TBase> { return std::make_unique<TMessage>(); });
	}
};

}  // namespace details

template <typename TBase, typename TTMessages>
class MessageFactory {
  public:
	using IdType = typename TBase::MsgIdType;

	MessageFactory() : m_Factory(details::ArrayFiller<TBase, TTMessages>::ArrayCreate()) {}

	std::unique_ptr<TBase> CreateMessage(IdType id) {
		if (id >= m_Factory.size())
			return nullptr;
		return m_Factory.at(id)();
	}

  private:
	details::ArrayType<TBase> m_Factory;
};



}  // namespace sp