#pragma once

#include "Types.h"

#include <vector>
#include <memory>
#include <algorithm>

namespace td {
namespace game {

class Player;

enum class TeamColor : std::int8_t {
    None = -1,
    Red,
    Blue
};

struct Spawn {
    Direction direction;
    std::int32_t x, y;
};

struct TeamCastle {
    std::int32_t x, y;
    std::uint16_t life = 1000;
};

class Team {
private:
    std::vector<Player*> m_Players;
    TeamColor m_Color;
    Spawn m_Spawn;
    TeamCastle m_TeamCastle;
public:
    Team(TeamColor color);

    void addPlayer(Player* newPlayer);
    void removePlayer(const Player* player);

    TeamColor getColor() const;

    const Spawn& getSpawn() const { return m_Spawn; }
    Spawn& getSpawn() { return m_Spawn; }

    const TeamCastle& getCastle() const { return m_TeamCastle; }
    TeamCastle& getCastle() { return m_TeamCastle; }

    std::uint8_t getPlayerCount() const;
};

} // namespace game
} // namespace td