MaaAssistantArknights/src/MeoAssistant/RoguelikeRecruitTaskPlugin.cpp

#include "RoguelikeRecruitTaskPlugin.h"

#include "RoguelikeRecruitImageAnalyzer.h"
#include "Controller.h"
#include "TaskData.h"
#include "Resource.h"
#include "Logger.hpp"
#include "RuntimeStatus.h"
#include "ProcessTask.h"

bool asst::RoguelikeRecruitTaskPlugin::verify(AsstMsg msg, const json::value& details) const
{
    if (msg != AsstMsg::SubTaskCompleted
        || details.get("subtask", std::string()) != "ProcessTask") {
        return false;
    }

    if (details.at("details").at("task").as_string() == "Roguelike1ChooseOper") {
        return true;
    }
    else {
        return false;
    }
}

bool asst::RoguelikeRecruitTaskPlugin::_run()
{
    LogTraceFunction;

    if (check_core_char()) {
        return true;
    }

    bool recruited = false;

    auto recruit_oper = [&](const BattleRecruitOperInfo& info) {
        Log.info("Choose：", info.name, info.elite, info.level);
        select_oper(info);
        recruited = true;
    };

    constexpr size_t SwipeTimes = 3;
    for (size_t i = 0; i != SwipeTimes; ++i) {
        auto image = m_ctrler->get_image();
        RoguelikeRecruitImageAnalyzer analyzer(image);
        if (!analyzer.analyze()) {
            return false;
        }

        const auto& oper_list = analyzer.get_result();
        for (const auto& info : oper_list) {
            // 先看看有没有精二的
            if (info.elite != 2) {
                continue;
            }
            recruit_oper(info);
            break;
        }

        if (!recruited) {
            for (const auto& info : oper_list) {
                if (!info.required) {
                    continue;
                }
                // 拿个精一 50 以上的
                if (info.elite == 0 ||
                    (info.elite == 1 && info.level < 50)) {
                    continue;
                }
                recruit_oper(info);
                break;
            }
        }

        if (!recruited) {
            Log.info("All are lower");
            // 随便招个精一的
            for (const auto& info : oper_list) {
                if (!info.required) {
                    continue;
                }
                if (info.elite == 0) {
                    continue;
                }
                recruit_oper(info);
                break;
            }
        }

        // 这玩意选了也没啥用，不如省点理智
        //if (!recruited) {
        //    // 随便招个
        //    Log.info("All are very lower");
        //    auto info = oper_list.front();
        //    recruit_oper(info);
        //}

        // 没有合适的干员，尝试向右翻页
        if (!recruited) {
            ProcessTask(*this, { "SlowlySwipeToTheRight" }).run();
            sleep(Task.get("Roguelike1Custom-HijackSquad")->rear_delay);
            continue;
        }

        // 已经招募到合适的干员
        return true;
    }

    ProcessTask(*this, { "SwipeToTheLeft" }).run();
    return false;
}

bool asst::RoguelikeRecruitTaskPlugin::check_core_char()
{
    auto core_opt = m_status->get_str("RoguelikeCoreChar");
    if (!core_opt || core_opt->empty()) {
        return false;
    }
    constexpr size_t SwipeTimes = 2;
    for (size_t i = 0; i != SwipeTimes; ++i) {
        auto image = m_ctrler->get_image();
        RoguelikeRecruitImageAnalyzer analyzer(image);
        if (!analyzer.analyze()) {
            return false;
        }
        const auto& chars = analyzer.get_result();
        auto it = std::find_if(chars.cbegin(), chars.cend(),
            [&](const BattleRecruitOperInfo& oper) -> bool {
                return oper.name == core_opt.value();
            });

        if (it == chars.cend()) {
            ProcessTask(*this, { "SlowlySwipeToTheRight" }).run();
            sleep(Task.get("Roguelike1Custom-HijackSquad")->rear_delay);
            continue;
        }
        select_oper(*it);
        m_status->set_str("RoguelikeCoreChar", "");
        return true;
    }
    ProcessTask(*this, { "SwipeToTheLeft" }).run();
    return false;
}

void asst::RoguelikeRecruitTaskPlugin::select_oper(const BattleRecruitOperInfo& oper)
{
    m_ctrler->click(oper.rect);

    m_status->set_number(RuntimeStatus::RoguelikeCharElitePrefix + oper.name, oper.elite);
    m_status->set_number(RuntimeStatus::RoguelikeCharLevelPrefix + oper.name, oper.level);

    std::string overview_str = m_status->get_str(RuntimeStatus::RoguelikeCharOverview).value_or(json::value().to_string());
    json::value overview = json::parse(overview_str).value_or(json::value());
    overview[oper.name] = json::object{
        { "elite", oper.elite },
        { "level", oper.level },
    };
    m_status->set_str(RuntimeStatus::RoguelikeCharOverview, overview.to_string());
}