#include "LearnableAttacks.hpp"

using namespace CreatureLib::Library;

void LearnableAttacks::AddLevelAttack(uint8_t level, const AttackData* attack) {

    List<const AttackData*> levelData;
    if (_learnedByLevel.TryGet(level, levelData)) {
        levelData.Append(attack);
    } else {
        levelData = {attack};
        _learnedByLevel.Insert(level, levelData);
    }
    _distinctAttacks.insert(attack);
}

const List<const AttackData*>& LearnableAttacks::GetAttacksForLevel(uint8_t level) const {
    return _learnedByLevel.Get(level);
}