CreatureLib/src/Battling/Models/Battle.cpp

#include "Battle.hpp"
#include <Arbutils/Assert.hpp>
#include "../../Library/Exceptions/NotImplementedException.hpp"
#include "../Flow/TurnHandler.hpp"
#include "../Flow/TurnOrdering.hpp"

using namespace CreatureLib;
using namespace CreatureLib::Battling;

const ArbUt::BorrowedPtr<const BattleLibrary>& Battle::GetLibrary() const noexcept { return _library; }

bool Battle::CanUse(const ArbUt::BorrowedPtr<BaseTurnChoice>& choice) {
    AssertNotNull(choice)
    if (choice->GetKind() == TurnChoiceKind::Attack) {
        // HOOK: change number of uses needed.
        return choice.ForceAs<AttackTurnChoice>()->GetAttack()->GetRemainingUses() >= 1;
    }
    return true;
}

bool Battle::TrySetChoice(BaseTurnChoice* choice) {
    AssertNotNull(choice)
    if (!CanUse(choice))
        return false;
    choice->GetUser()->GetBattleSide()->SetChoice(choice);
    CheckChoicesSetAndRun();
    return true;
}

void Battle::CheckChoicesSetAndRun() {
    for (auto side : _sides) {
        if (!side->AllChoicesSet()) {
            return;
        }
    }
    for (auto side : _sides) {
        if (!side->AllPossibleSlotsFilled()) {
            return;
        }
    }

    auto choices = std::vector<std::shared_ptr<BaseTurnChoice>>(_numberOfSides * _creaturesPerSide);
    auto i = 0;
    for (auto side : _sides) {
        for (const auto& choice : side->GetChoices()) {
            AssertNotNull(choice)
            if (choice->GetKind() == TurnChoiceKind::Attack) {
                auto attack = ((AttackTurnChoice*)choice.get())->GetAttack();
                uint8_t uses = 1;
                // HOOK: change number of uses needed.
                if (attack->GetRemainingUses() < uses) {
                    // TODO: Implement default move
                    throw NotImplementedException("Not enough remaining uses, change to default move.");
                }
                // HOOK: Check if we need to change the move
            }
            choice->__SetRandomValue(_random.Get());
            choices[i] = choice;
            i++;
        }
        side->ResetChoices();
    }
    _currentTurn++;
    TurnOrdering::OrderChoices(choices, _random.GetRNG());
    this->_currentTurnQueue.reset(new ChoiceQueue(choices));
    TurnHandler::RunTurn(this->_currentTurnQueue);
    if (this->_currentTurnQueue->HasCompletedQueue) {
        this->_currentTurnQueue = nullptr;
    }
}

ArbUt::BorrowedPtr<ChoiceQueue> Battle::GetCurrentTurnQueue() const noexcept { return _currentTurnQueue; }

BattleRandom* Battle::GetRandom() noexcept { return &_random; }

bool Battle::CreatureInField(ArbUt::BorrowedPtr<Creature> creature) const {
    AssertNotNull(creature)
    return std::any_of(_sides.begin(), _sides.end(), [creature](auto c) { return c->CreatureOnSide(creature); });
}

void Battle::ForceRecall(uint8_t side, uint8_t index) { _sides[side]->SetCreature(nullptr, index); }

size_t Battle::ScriptCount() const { return 1; }
void Battle::GetActiveScripts(ArbUt::List<ScriptWrapper>& scripts) {
    scripts.Append(ScriptWrapper::FromSet(&_volatile));
}

void Battle::SwitchCreature(uint8_t sideIndex, uint8_t index, Creature* c) {
    auto side = this->_sides[sideIndex];
    side->SetCreature(c, index);
}

bool Battle::CanSlotBeFilled(uint8_t side, uint8_t index) const {
    for (const auto& party : _parties) {
        if (party->IsResponsibleForIndex(side, index)) {
            if (party->HasCreaturesNotInField())
                return true;
        }
    }
    return false;
}

void Battle::ValidateBattleState() {
    if (_hasEnded) {
        return;
    }
    bool survivingSideExists = false;
    uint8_t winningSide = 255;
    for (uint8_t i = 0; i < _sides.Count(); i++) {
        auto side = _sides[i];
        if (side->HasFled()) {
            this->_battleResult = BattleResult::Inconclusive();
            this->_hasEnded = true;
            return;
        }
        if (!side->IsDefeated()) {
            if (survivingSideExists) {
                return;
            }
            survivingSideExists = true;
            winningSide = i;
        }
    }
    if (!survivingSideExists) {
        this->_battleResult = BattleResult::Inconclusive();
        this->_hasEnded = true;
        return;
    }
    this->_battleResult = BattleResult::Conclusive(winningSide);
    this->_hasEnded = true;
}
void Battle::AddVolatileScript(const ArbUt::BasicStringView& key) {
    auto script = _volatile.Get(key);
    if (script != nullptr) {
        script->Stack();
        return;
    }
    script = _library->LoadScript(ScriptCategory::Battle, key);
    if (script == nullptr) {
        std::stringstream ss;
        ss << "Invalid volatile script requested for battle: '" << key.c_str() << "'.";
        throw CreatureException(ss.str());
    }
    return _volatile.Add(script.GetRaw());
}
void Battle::AddVolatileScript(Script* script) { return _volatile.Add(script); }
void Battle::RemoveVolatileScript(Script* script) { _volatile.Remove(script->GetName()); }
void Battle::DisplayText(const ArbUt::StringView& text) const { TriggerEventListener(new DisplayTextEvent(text)); }