PokemonSwSh_MaxLair_BossFinder.cpp

/*  Max Lair (Boss Finder)
 *
 *  From: https://github.com/PokemonAutomation/
 *
 */

#include "Common/Compiler.h"
#include "Common/Cpp/Exceptions.h"
#include "CommonFramework/Notifications/ProgramNotifications.h"
#include "CommonTools/StartupChecks/VideoResolutionCheck.h"
#include "NintendoSwitch/NintendoSwitch_Settings.h"
#include "NintendoSwitch/Commands/NintendoSwitch_Commands_PushButtons.h"
#include "Pokemon/Pokemon_Strings.h"
#include "PokemonSwSh/Programs/PokemonSwSh_GameEntry.h"
#include "Program/PokemonSwSh_MaxLair_Run_Adventure.h"
#include "PokemonSwSh_MaxLair_BossFinder.h"

namespace PokemonAutomation{
namespace NintendoSwitch{
namespace PokemonSwSh{
using namespace MaxLairInternal;



MaxLairBossFinder_Descriptor::MaxLairBossFinder_Descriptor()
    : MultiSwitchProgramDescriptor(
        "PokemonSwSh:MaxLair-BossFinder",
        STRING_POKEMON + " SwSh", "Max Lair - Boss Finder",
        "Programs/PokemonSwSh/MaxLair-BossFinder.html",
        "Auto Max Lair 2.0 - Run adventures until you find the boss you want. Once you find your boss, switch to the other programs to shiny hunt it.",
        ProgramControllerClass::StandardController_NoRestrictions,
        FeedbackType::REQUIRED,
        AllowCommandsWhenRunning::DISABLE_COMMANDS,
        1, 4, 1
    )
{}
std::unique_ptr<StatsTracker> MaxLairBossFinder_Descriptor::make_stats() const{
    return std::unique_ptr<StatsTracker>(new Stats());
}



class MaxLairBossFinder_ConsoleOptions : public ConsoleSpecificOptions{
public:
    MaxLairBossFinder_ConsoleOptions(std::string label, const LanguageSet& languages, bool host)
        : ConsoleSpecificOptions(std::move(label), languages, host)
        , normal_ball(
            "<b>Normal Ball:</b> Ball for catching non-boss " + STRING_POKEMON + ".",
            LockMode::LOCK_WHILE_RUNNING,
            "poke-ball"
        )
    {
        PA_ADD_OPTION(normal_ball);
    }

    PokemonBallSelectOption normal_ball;
};

class MaxLairBossFinder_ConsoleFactory : public ConsoleSpecificOptionsFactory{
public:
    virtual std::unique_ptr<ConsoleSpecificOptions> make(
        std::string label,
        const LanguageSet& languages,
        bool host
    ) const override{
        return std::unique_ptr<ConsoleSpecificOptions>(new MaxLairBossFinder_ConsoleOptions(std::move(label), languages, host));
    }
};


MaxLairBossFinder::MaxLairBossFinder()
    : MultiSwitchProgramInstance({"Notifs", "LiveHost"})
    , GO_HOME_WHEN_DONE(false)
    , CONSOLES(MaxLairBossFinder_ConsoleFactory())
    , NOTIFICATION_STATUS("Status Update", true, false)
    , NOTIFICATION_SHINY("Shiny Catch", true, true, ImageAttachmentMode::JPG, {"Notifs", "Showcase"})
    , NOTIFICATIONS({
        &HOSTING.NOTIFICATIONS.NOTIFICATION,
        &NOTIFICATION_STATUS,
        &NOTIFICATION_SHINY,
        &NOTIFICATION_PROGRAM_FINISH,
        &NOTIFICATION_ERROR_FATAL,
    })
{
    PA_ADD_OPTION(START_LOCATION);
    PA_ADD_OPTION(GO_HOME_WHEN_DONE);

    PA_ADD_OPTION(CONSOLES);
    PA_ADD_OPTION(BOSS_LIST);
    PA_ADD_OPTION(HOSTING);

    PA_ADD_OPTION(TOUCH_DATE_INTERVAL);
    PA_ADD_OPTION(NOTIFICATIONS);
}

std::string MaxLairBossFinder::check_validity() const{
    std::string error = MultiSwitchProgramInstance::check_validity();
    if (!error.empty()){
        return error;
    }

    size_t active_consoles = CONSOLES.active_consoles();
    error = CONSOLES.HOST.check_validity(active_consoles);
    if (!error.empty()){
        return error;
    }
    error = HOSTING.check_validity(active_consoles);
    if (!error.empty()){
        return error;
    }
    return std::string();
}
void MaxLairBossFinder::update_active_consoles(size_t switch_count){
    CONSOLES.set_active_consoles(switch_count);
}




class EndBattleDecider_BossFinder : public EndBattleDecider{
public:
    EndBattleDecider_BossFinder(const Consoles& consoles, const BossActionTable& boss_list)
        : m_consoles(consoles)
        , m_boss_list(boss_list)
    {}
    virtual const std::string& normal_ball(
        size_t console_index
    ) const override{
        return console(console_index).normal_ball.slug();
    }
    virtual const std::string& boss_ball(
        size_t console_index, const std::string& boss_slug
    ) const override{
        return get_filter(boss_slug).ball.slug();
    }
    virtual CaughtScreenAction end_adventure_action(
        size_t console_index, const std::string& boss_slug,
        const PathStats& path_stats,
        bool any_shiny, bool boss_is_shiny
    ) const override{
        if (boss_slug.empty()){
            return CaughtScreenAction::TAKE_NON_BOSS_SHINY_AND_CONTINUE;
        }
        switch (get_filter(boss_slug).action){
        case BossAction::CATCH_AND_STOP_PROGRAM:
            return CaughtScreenAction::STOP_PROGRAM;
        case BossAction::CATCH_AND_STOP_IF_SHINY:
            return boss_is_shiny
                ? CaughtScreenAction::STOP_PROGRAM
                : CaughtScreenAction::TAKE_NON_BOSS_SHINY_AND_CONTINUE;
        }
        throw InternalProgramError(nullptr, PA_CURRENT_FUNCTION, "Invalid enum.");
    }
    virtual bool is_in_save_list(const std::string& boss_slug) const override {
            return get_filter(boss_slug).save_on_the_go;
    }


private:
    const BossActionRow& get_filter(const std::string& boss_slug) const{
        for (const StaticTableRow* row : m_boss_list.table()){
            const BossActionRow& filter = static_cast<const BossActionRow&>(*row);
            if (boss_slug == filter.slug()){
                return filter;
            }
        }
        throw InternalProgramError(nullptr, PA_CURRENT_FUNCTION, "Unrecognized boss slug: " + boss_slug);
    }

private:
    const MaxLairBossFinder_ConsoleOptions& console(size_t index) const{
        return static_cast<const MaxLairBossFinder_ConsoleOptions&>(m_consoles[index]);
    }

    const Consoles& m_consoles;
    const BossActionTable& m_boss_list;

};





void MaxLairBossFinder::program(MultiSwitchProgramEnvironment& env, CancellableScope& scope){
    size_t host_index = CONSOLES.HOST.current_value();
    if (host_index >= env.consoles.size()){
        throw UserSetupError(env.logger(), "Invalid Host Switch");
    }

    env.run_in_parallel(scope, [&](ConsoleHandle& console, ProControllerContext& context){
        assert_16_9_720p_min(console, console);
        if (START_LOCATION.start_in_grip_menu()){
            grip_menu_connect_go_home(context);
            resume_game_no_interact(console, context, ConsoleSettings::instance().TOLERATE_SYSTEM_UPDATE_MENU_FAST);
        }else{
            pbf_press_button(context, BUTTON_B, 40ms, 40ms);
        }
    });

    EndBattleDecider_BossFinder decider(CONSOLES, BOSS_LIST);

    loop_adventures(
        env, scope, CONSOLES,
        host_index, 0,
        decider,
        GO_HOME_WHEN_DONE,
        HOSTING,
        TOUCH_DATE_INTERVAL,
        NOTIFICATION_STATUS,
        NOTIFICATION_SHINY
    );

    env.update_stats();
    send_program_finished_notification(env, NOTIFICATION_PROGRAM_FINISH);
}






}
}
}