#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;
    }

    RoguelikeRecruitImageAnalyzer analyzer(m_ctrler->get_image());

    if (!analyzer.analyze()) {
        return false;
    }

    bool recruited = false;

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

    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);
    //}

    return recruited;
}

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(m_ctrler->get_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("Roguelike-" + oper.name, oper.elite);
}