#pragma once

#include <random>
#include <vector>

#include <entt/entt.hpp>

#include <kurator/sim/State.h>


namespace kurator
{
namespace sim
{


class TeamManager
{
public:
	using Team = std::vector<entt::entity>;
	TeamManager();
	void extend(State& ctx);
	void add(int team, entt::entity entity);
	entt::entity random(int exclude);
	void update(State& ctx);
private:
	std::vector<Team> teams;
	std::mt19937 generator;
};


}  // namespace sim
}  // namespace kurator