#pragma once

#include <algorithm>
#include <cassert>

namespace sp {

template <typename MessageBase>
void MessageDispatcher<MessageBase>::RegisterHandler(const std::shared_ptr<MessageHandler>& a_Handler) {
	assert(a_Handler);
	m_Handlers.push_back(a_Handler);
}

template <typename MessageBase>
void MessageDispatcher<MessageBase>::UnregisterHandler(const std::shared_ptr<MessageHandler>& a_Handler) {
	auto found = std::find(m_Handlers.begin(), m_Handlers.end(), a_Handler);
	if (found != m_Handlers.end())
		m_Handlers.erase(found);
}

template <typename MessageBase>
void MessageDispatcher<MessageBase>::Dispatch(const MessageBase& a_Message) {
	for (auto& handler : m_Handlers) {
		a_Message.Dispatch(*handler.lock());
	}
}

}  // namespace sp